Add usage to context in streaming

rm-openai · rm-openai · commit e507637ebc2b · 2025-04-24T15:54:02.000-04:00
diff --git a/src/agents/run.py b/src/agents/run.py
@@ -696,6 +696,7 @@ async def _run_single_turn_streamed(
                     usage=usage,
                     response_id=event.response.id,
                 )
+                context_wrapper.usage.add(usage)
 
             streamed_result._event_queue.put_nowait(RawResponsesStreamEvent(data=event))
 

Original file line number	Diff line number	Diff line change
`@@ -696,6 +696,7 @@ async def _run_single_turn_streamed(`
`696`	`696`	`usage=usage,`
`697`	`697`	`response_id=event.response.id,`
`698`	`698`	`)`
	`699`	`+ context_wrapper.usage.add(usage)`
`699`	`700`
`700`	`701`	`streamed_result._event_queue.put_nowait(RawResponsesStreamEvent(data=event))`
`701`	`702`