fix: 保留 Code Agent 最终回复全文
This commit is contained in:
@@ -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();
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user