diff --git a/src/agents/run.py b/src/agents/run.py index dfff7e38..bb386640 100644 --- a/src/agents/run.py +++ b/src/agents/run.py @@ -651,8 +651,8 @@ async def _run_single_turn_streamed( usage = ( Usage( requests=1, - input_tokens=event.response.usage.input_tokens, - output_tokens=event.response.usage.output_tokens, + input_tokens=event.response.usage.prompt_tokens, + output_tokens=event.response.usage.completion_tokens, total_tokens=event.response.usage.total_tokens, ) if event.response.usage