Skip to content

Commit 3cefce9

Browse files
authored
fix: strip #2518 total_tokens from OpenAI tracing usage payloads (#2521)
1 parent f2f0b8c commit 3cefce9

File tree

2 files changed

+2
-3
lines changed

2 files changed

+2
-3
lines changed

src/agents/tracing/processors.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,6 @@ class BackendSpanExporter(TracingExporter):
3333
{
3434
"input_tokens",
3535
"output_tokens",
36-
"total_tokens",
3736
"input_tokens_details",
3837
"output_tokens_details",
3938
}

tests/test_trace_processor.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -316,14 +316,15 @@ def export(self):
316316
sent_payload = mock_client.return_value.post.call_args.kwargs["json"]["data"][0]
317317
sent_usage = sent_payload["span_data"]["usage"]
318318
assert "requests" not in sent_usage
319+
assert "total_tokens" not in sent_usage
319320
assert sent_usage["input_tokens"] == 10
320321
assert sent_usage["output_tokens"] == 5
321-
assert sent_usage["total_tokens"] == 15
322322
assert sent_usage["input_tokens_details"] == {"cached_tokens": 1}
323323
assert sent_usage["output_tokens_details"] == {"reasoning_tokens": 2}
324324

325325
# Ensure the original exported object has not been mutated.
326326
assert "requests" in item.exported_payload["span_data"]["usage"]
327+
assert item.exported_payload["span_data"]["usage"]["total_tokens"] == 15
327328
exporter.close()
328329

329330

@@ -400,7 +401,6 @@ def test_sanitize_for_openai_tracing_api_keeps_allowed_generation_usage():
400401
"usage": {
401402
"input_tokens": 1,
402403
"output_tokens": 2,
403-
"total_tokens": 3,
404404
"input_tokens_details": {"cached_tokens": 0},
405405
"output_tokens_details": {"reasoning_tokens": 0},
406406
},

0 commit comments

Comments
 (0)