usage

Author: antonpirker

tests/integrations/huggingface_hub/test_huggingface_hub.py

@@ -791,13 +791,15 @@ def test_chat_completion_streaming_with_tools(
         "gen_ai.response.finish_reasons": "tool_calls",
         "gen_ai.response.model": "test-model-123",
         "gen_ai.response.streaming": True,
-        "gen_ai.usage.input_tokens": 183,
-        "gen_ai.usage.output_tokens": 14,
-        "gen_ai.usage.total_tokens": 197,
         "thread.id": mock.ANY,
         "thread.name": mock.ANY,
     }
 
+    if HF_VERSION and HF_VERSION >= (0, 26, 0):
+        expected_data["gen_ai.usage.input_tokens"] = 183
+        expected_data["gen_ai.usage.output_tokens"] = 14
+        expected_data["gen_ai.usage.total_tokens"] = 197
+
     if send_default_pii and include_prompts:
         expected_data["gen_ai.request.messages"] = (
             '[{"role": "user", "content": "What is the weather in Paris?"}]'