[codex] Add incremental chat completion traces (#708)

Co-authored-by: James Grugett <jahooma@gmail.com>

Author: brandonkachen

agents/e2e/base2-free-summary-format.e2e.test.ts

@@ -324,6 +324,7 @@ describe('Base2-Free Summary Format Compliance', () => {
         })
         const runStateWithMessages = withMessageHistory({
           runState: {
+            traceSessionId: 'test-trace-session',
             sessionState,
             output: { type: 'error', message: '' },
           },
@@ -461,6 +462,7 @@ describe('Base2-Free Summary Format Compliance', () => {
         })
         const runStateWithMessages = withMessageHistory({
           runState: {
+            traceSessionId: 'test-trace-session',
             sessionState,
             output: { type: 'error', message: '' },
           },

agents/e2e/context-pruner.e2e.test.ts

@@ -151,7 +151,11 @@ Do not do anything else. Just spawn context-pruner and then report the result.`,
       // Create initial session state with the large message history
       const sessionState = await initialSessionState({})
       const runStateWithMessages = withMessageHistory({
-        runState: { sessionState, output: { type: 'error', message: '' } },
+        runState: {
+          traceSessionId: 'test-trace-session',
+          sessionState,
+          output: { type: 'error', message: '' },
+        },
         messages: initialMessages,
       })
 
@@ -277,7 +281,11 @@ Do not do anything else. Just spawn context-pruner and then report the result.`,
 
       const sessionState = await initialSessionState({})
       const runStateWithMessages = withMessageHistory({
-        runState: { sessionState, output: { type: 'error', message: '' } },
+        runState: {
+          traceSessionId: 'test-trace-session',
+          sessionState,
+          output: { type: 'error', message: '' },
+        },
         messages: initialMessages,
       })
 

agents/e2e/context-pruning-threshold.e2e.test.ts

@@ -324,7 +324,11 @@ describe('Context Pruning Threshold E2E', () => {
 
       const sessionState = await initialSessionState({})
       const runStateWithMessages = withMessageHistory({
-        runState: { sessionState, output: { type: 'error', message: '' } },
+        runState: {
+          traceSessionId: 'test-trace-session',
+          sessionState,
+          output: { type: 'error', message: '' },
+        },
         messages,
       })
 
@@ -398,7 +402,11 @@ describe('Context Pruning Threshold E2E', () => {
 
       const sessionState = await initialSessionState({})
       const runStateWithMessages = withMessageHistory({
-        runState: { sessionState, output: { type: 'error', message: '' } },
+        runState: {
+          traceSessionId: 'test-trace-session',
+          sessionState,
+          output: { type: 'error', message: '' },
+        },
         messages,
       })
 
@@ -494,6 +502,7 @@ describe('Context Pruning Threshold E2E', () => {
       const sessionStateCal = await initialSessionState({})
       const runStateCal = withMessageHistory({
         runState: {
+          traceSessionId: 'test-trace-session',
           sessionState: sessionStateCal,
           output: { type: 'error', message: '' },
         },
@@ -544,7 +553,11 @@ describe('Context Pruning Threshold E2E', () => {
       // =========================================================================
       const sessionState = await initialSessionState({})
       const runStateWithMessages = withMessageHistory({
-        runState: { sessionState, output: { type: 'error', message: '' } },
+        runState: {
+          traceSessionId: 'test-trace-session',
+          sessionState,
+          output: { type: 'error', message: '' },
+        },
         messages,
       })