Merge 2c980cd43c into da71eaebd2
This commit is contained in:
commit
5774320666
@ -163,6 +163,7 @@ export function createAgentEventHandler({
|
||||
seq: number,
|
||||
jobState: "done" | "error",
|
||||
error?: unknown,
|
||||
usage?: Record<string, unknown>,
|
||||
) => {
|
||||
const text = chatRunState.buffers.get(clientRunId)?.trim() ?? "";
|
||||
chatRunState.buffers.delete(clientRunId);
|
||||
@ -180,6 +181,7 @@ export function createAgentEventHandler({
|
||||
timestamp: Date.now(),
|
||||
}
|
||||
: undefined,
|
||||
usage: usage,
|
||||
};
|
||||
// Suppress webchat broadcast for heartbeat runs when showOk is false
|
||||
if (!shouldSuppressHeartbeatBroadcast(clientRunId)) {
|
||||
@ -264,6 +266,7 @@ export function createAgentEventHandler({
|
||||
evt.seq,
|
||||
lifecyclePhase === "error" ? "error" : "done",
|
||||
evt.data?.error,
|
||||
evt.data?.usage as Record<string, unknown> | undefined,
|
||||
);
|
||||
} else {
|
||||
emitChatFinal(
|
||||
@ -272,6 +275,7 @@ export function createAgentEventHandler({
|
||||
evt.seq,
|
||||
lifecyclePhase === "error" ? "error" : "done",
|
||||
evt.data?.error,
|
||||
evt.data?.usage as Record<string, unknown> | undefined,
|
||||
);
|
||||
}
|
||||
} else if (isAborted && (lifecyclePhase === "end" || lifecyclePhase === "error")) {
|
||||
|
||||
@ -149,6 +149,7 @@ function broadcastChatFinal(params: {
|
||||
runId: string;
|
||||
sessionKey: string;
|
||||
message?: Record<string, unknown>;
|
||||
usage?: Record<string, unknown>;
|
||||
}) {
|
||||
const seq = nextChatSeq({ agentRunSeq: params.context.agentRunSeq }, params.runId);
|
||||
const payload = {
|
||||
@ -157,6 +158,7 @@ function broadcastChatFinal(params: {
|
||||
seq,
|
||||
state: "final" as const,
|
||||
message: params.message,
|
||||
usage: params.usage,
|
||||
};
|
||||
params.context.broadcast("chat", payload);
|
||||
params.context.nodeSendToSession(params.sessionKey, "chat", payload);
|
||||
@ -545,6 +547,7 @@ export const chatHandlers: GatewayRequestHandlers = {
|
||||
runId: clientRunId,
|
||||
sessionKey: p.sessionKey,
|
||||
message,
|
||||
usage: message?.usage as Record<string, unknown> | undefined,
|
||||
});
|
||||
}
|
||||
context.dedupe.set(`chat:${clientRunId}`, {
|
||||
|
||||
Loading…
Reference in New Issue
Block a user