Merge e507637 into sapling-pr-archive-rm-openai

rm-openai · web-flow · commit 8d428ebe0275 · 2025-04-24T15:54:14.000-04:00
diff --git a/src/agents/run.py b/src/agents/run.py
@@ -701,6 +701,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
`@@ -701,6 +701,7 @@ async def _run_single_turn_streamed(`
`701`	`701`	`usage=usage,`
`702`	`702`	`response_id=event.response.id,`
`703`	`703`	`)`
	`704`	`+ context_wrapper.usage.add(usage)`
`704`	`705`
`705`	`706`	`streamed_result._event_queue.put_nowait(RawResponsesStreamEvent(data=event))`
`706`	`707`