fix: 保留 Code Agent 最终回复全文

This commit is contained in:
lyon
2026-06-16 01:45:34 +08:00
parent 3607724e49
commit fac75d208e
2 changed files with 56 additions and 4 deletions
+47 -1
View File
@@ -77,6 +77,14 @@ interface CompletedAssistantMessage {
text: string;
}
interface FinalAssistantMessage {
itemId: string | null;
text: string;
messageIndex: number | null;
messageCount: number | null;
source: string;
}
interface AssistantDeltaProgressItem {
itemId: string | null;
text: string;
@@ -604,7 +612,9 @@ async function runCodexStdioTurnWithSession(options: CodexStdioTurnOptions, sess
if (pendingInterrupt) await pendingInterrupt.catch(() => undefined);
if (terminal.status !== "completed") emitEvents(await session.close());
emitEvents(flushAssistantDeltaProgress(assistantDeltaProgress));
if (completedAssistantMessages.length === 0) emitEvents(assistantMessageEventsForTurn(assistantText, terminal.status === "completed"));
const finalAssistant = terminal.status === "completed" ? finalAssistantMessageForTurn(completedAssistantMessages, assistantText) : null;
if (finalAssistant) emitEvent(assistantFinalResponseEvent(finalAssistant));
else if (completedAssistantMessages.length === 0) emitEvents(assistantMessageEventsForTurn(assistantText, false));
emitEvents(suppressedNotificationEvents(suppressedNotifications));
emitEvent({ type: "terminal_status", payload: { terminalStatus: terminal.status, failureKind: terminal.failureKind, message: terminal.message } });
await liveEventWrite;
@@ -883,6 +893,42 @@ function assistantMessageEventsForTurn(assistantDeltaText: string, completed: bo
}];
}
function finalAssistantMessageForTurn(completedMessages: CompletedAssistantMessage[], assistantDeltaText: string): FinalAssistantMessage | null {
const latestCompleted = completedMessages.at(-1) ?? null;
if (latestCompleted && latestCompleted.text.trim().length > 0) {
return {
itemId: latestCompleted.itemId,
text: latestCompleted.text,
messageIndex: completedMessages.length,
messageCount: completedMessages.length,
source: "completed-agent-message-final",
};
}
if (assistantDeltaText.trim().length === 0) return null;
return {
itemId: null,
text: assistantDeltaText,
messageIndex: 1,
messageCount: 1,
source: "agent-message-delta-final",
};
}
function assistantFinalResponseEvent(message: FinalAssistantMessage): BackendEvent {
return {
type: "assistant_message",
payload: {
text: message.text,
itemId: message.itemId,
source: message.source,
messageIndex: message.messageIndex,
messageCount: message.messageCount,
replyAuthority: true,
final: true,
},
};
}
function createAssistantDeltaProgressState(): AssistantDeltaProgressState {
return new Map();
}