87 lines
2.5 KiB
TypeScript
87 lines
2.5 KiB
TypeScript
import { emitAgentEvent } from "../infra/agent-events.js";
|
|
import { createInlineCodeState } from "../markdown/code-spans.js";
|
|
import { formatAssistantErrorText } from "./pi-embedded-helpers.js";
|
|
import type { EmbeddedPiSubscribeContext } from "./pi-embedded-subscribe.handlers.types.js";
|
|
import { isAssistantMessage } from "./pi-embedded-utils.js";
|
|
|
|
export {
|
|
handleAutoCompactionEnd,
|
|
handleAutoCompactionStart,
|
|
} from "./pi-embedded-subscribe.handlers.compaction.js";
|
|
|
|
export function handleAgentStart(ctx: EmbeddedPiSubscribeContext) {
|
|
ctx.log.debug(`embedded run agent start: runId=${ctx.params.runId}`);
|
|
emitAgentEvent({
|
|
runId: ctx.params.runId,
|
|
stream: "lifecycle",
|
|
data: {
|
|
phase: "start",
|
|
startedAt: Date.now(),
|
|
},
|
|
});
|
|
void ctx.params.onAgentEvent?.({
|
|
stream: "lifecycle",
|
|
data: { phase: "start" },
|
|
});
|
|
}
|
|
|
|
export function handleAgentEnd(ctx: EmbeddedPiSubscribeContext) {
|
|
const lastAssistant = ctx.state.lastAssistant;
|
|
const isError = isAssistantMessage(lastAssistant) && lastAssistant.stopReason === "error";
|
|
|
|
if (isError && lastAssistant) {
|
|
const friendlyError = formatAssistantErrorText(lastAssistant, {
|
|
cfg: ctx.params.config,
|
|
sessionKey: ctx.params.sessionKey,
|
|
provider: lastAssistant.provider,
|
|
model: lastAssistant.model,
|
|
});
|
|
const errorText = (friendlyError || lastAssistant.errorMessage || "LLM request failed.").trim();
|
|
ctx.log.warn(
|
|
`embedded run agent end: runId=${ctx.params.runId} isError=true error=${errorText}`,
|
|
);
|
|
emitAgentEvent({
|
|
runId: ctx.params.runId,
|
|
stream: "lifecycle",
|
|
data: {
|
|
phase: "error",
|
|
error: errorText,
|
|
endedAt: Date.now(),
|
|
},
|
|
});
|
|
void ctx.params.onAgentEvent?.({
|
|
stream: "lifecycle",
|
|
data: {
|
|
phase: "error",
|
|
error: errorText,
|
|
},
|
|
});
|
|
} else {
|
|
ctx.log.debug(`embedded run agent end: runId=${ctx.params.runId} isError=${isError}`);
|
|
emitAgentEvent({
|
|
runId: ctx.params.runId,
|
|
stream: "lifecycle",
|
|
data: {
|
|
phase: "end",
|
|
endedAt: Date.now(),
|
|
},
|
|
});
|
|
void ctx.params.onAgentEvent?.({
|
|
stream: "lifecycle",
|
|
data: { phase: "end" },
|
|
});
|
|
}
|
|
|
|
ctx.flushBlockReplyBuffer();
|
|
|
|
ctx.state.blockState.thinking = false;
|
|
ctx.state.blockState.final = false;
|
|
ctx.state.blockState.inlineCode = createInlineCodeState();
|
|
|
|
if (ctx.state.pendingCompactionRetry > 0) {
|
|
ctx.resolveCompactionRetry();
|
|
} else {
|
|
ctx.maybeResolveCompactionWait();
|
|
}
|
|
}
|