clone/codex
1
0
Fork 0
mirror of https://github.com/openai/codex.git synced 2026-04-24 14:45:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(cli): stream delta response items

Browse Source
This commit is contained in:
aibrahim-oai
2025-07-12 19:05:27 -07:00
parent c46bb67d77
commit afcb2f4f82
2 changed files with 58 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
codex-cli/src/components/chat/terminal-chat.tsx
View File

@@ -255,7 +255,18 @@ export default function TerminalChat({
onItem: (item) => {
log(`onItem: ${JSON.stringify(item)}`);
setItems((prev) => {
const updated = uniqueById([...prev, item as ResponseItem]);
let updated = prev;
if (item.id) {
const idx = prev.findIndex((i) => i.id === item.id);
if (idx !== -1) {
updated = [...prev];
updated[idx] = item as ResponseItem;
} else {
updated = uniqueById([...prev, item as ResponseItem]);
}
} else {
updated = uniqueById([...prev, item as ResponseItem]);
}
saveRollout(sessionId, updated);
return updated;
});

49
codex-cli/src/utils/agent/agent-loop.ts
View File

@@ -669,10 +669,16 @@ export class AgentLoop {
}
// Skip items we've already processed to avoid staging duplicates
if (item.id && alreadyStagedItemIds.has(item.id)) {
if (
item.id &&
alreadyStagedItemIds.has(item.id) &&
item.status !== "in_progress"
) {
return;
}
alreadyStagedItemIds.add(item.id);
if (item.id && item.status !== "in_progress") {
alreadyStagedItemIds.add(item.id);
}
// Store the item so the final flush can still operate on a complete list.
// We'll nil out entries once they're delivered.
@@ -1035,11 +1041,42 @@ export class AgentLoop {
try {
let newTurnInput: Array<ResponseInputItem> = [];
const partials = new Map<string, string>();
// eslint-disable-next-line no-await-in-loop
for await (const event of stream as AsyncIterable<ResponseEvent>) {
log(`AgentLoop.run(): response event ${event.type}`);
// process and surface each item (no-op until we can depend on streaming events)
if (event.type === "response.output_text.delta") {
const id = event.item_id;
const soFar = partials.get(id) ?? "";
const text = soFar + event.delta;
partials.set(id, text);
stageItem({
id,
type: "message",
role: "assistant",
status: "in_progress",
content: [{ type: "output_text", text }],
} as ResponseItem);
continue;
}
if (event.type === "response.output_text.done") {
const id = event.item_id;
const text = event.text;
partials.set(id, text);
stageItem({
id,
type: "message",
role: "assistant",
status: "completed",
content: [{ type: "output_text", text }],
} as ResponseItem);
continue;
}
// process and surface each item when completed
if (event.type === "response.output_item.done") {
const item = event.item;
// 1) if it's a reasoning item, annotate it
@@ -1062,6 +1099,12 @@ export class AgentLoop {
if (callId) {
this.pendingAborts.add(callId);
}
} else if (
item.type === "message" &&
(item as { role?: string }).role === "assistant"
) {
// Final message already emitted via output_text.done
continue;
} else {
stageItem(item as ResponseItem);
}