File tree Expand file tree Collapse file tree 2 files changed +2
-3
lines changed
Expand file tree Collapse file tree 2 files changed +2
-3
lines changed Original file line number Diff line number Diff 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 }
Original file line number Diff line number Diff 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 },
You can’t perform that action at this time.
0 commit comments