add: usage_details instead usage (#2481)

Author: Hansehart

integrations/langfuse/src/haystack_integrations/tracing/langfuse/tracer.py

@@ -328,7 +328,7 @@ def handle(self, span: LangfuseSpan, component_type: Optional[str]) -> None:
         if component_type in _SUPPORTED_GENERATORS:
             meta = span.get_data().get(_COMPONENT_OUTPUT_KEY, {}).get("meta")
             if meta:
-                span.raw_span().update(usage=meta[0].get("usage") or None, model=meta[0].get("model"))
+                span.raw_span().update(usage_details=meta[0].get("usage") or None, model=meta[0].get("model"))
 
         if component_type in _SUPPORTED_CHAT_GENERATORS:
             replies = span.get_data().get(_COMPONENT_OUTPUT_KEY, {}).get("replies")
@@ -342,7 +342,7 @@ def handle(self, span: LangfuseSpan, component_type: Optional[str]) -> None:
                         logger.error(f"Failed to parse completion_start_time: {completion_start_time}")
                         completion_start_time = None
                 span.raw_span().update(
-                    usage=meta.get("usage") or None,
+                    usage_details=meta.get("usage") or None,
                     model=meta.get("model"),
                     completion_start_time=completion_start_time,
                 )

integrations/langfuse/tests/test_tracer.py

@@ -203,7 +203,7 @@ def test_handle_generator(self):
         handler.handle(mock_span, component_type="OpenAIGenerator")
 
         assert mock_span.update.call_count == 1
-        assert mock_span.update.call_args_list[0][1] == {"usage": None, "model": "test_model"}
+        assert mock_span.update.call_args_list[0][1] == {"usage_details": None, "model": "test_model"}
 
     def test_handle_chat_generator(self):
         mock_span = Mock()
@@ -225,7 +225,7 @@ def test_handle_chat_generator(self):
 
         assert mock_span.update.call_count == 1
         assert mock_span.update.call_args_list[0][1] == {
-            "usage": None,
+            "usage_details": None,
             "model": "test_model",
             "completion_start_time": datetime.datetime(  # noqa: DTZ001
                 2021, 7, 27, 16, 2, 8, 12345
@@ -254,7 +254,7 @@ def test_handle_bad_completion_start_time(self, caplog):
 
         assert mock_span.update.call_count == 1
         assert mock_span.update.call_args_list[0][1] == {
-            "usage": None,
+            "usage_details": None,
             "model": "test_model",
             "completion_start_time": None,
         }
@@ -348,7 +348,7 @@ def test_trace_generation(self):
             }
             with tracer.trace(operation_name="operation_name", tags=tags) as span:
                 ...
-            assert span.raw_span()._data["usage"] is None
+            assert span.raw_span()._data["usage_details"] is None
             assert span.raw_span()._data["model"] == "test_model"
             assert span.raw_span()._data["completion_start_time"] == datetime.datetime(2021, 7, 27, 16, 2, 8, 12345)  # noqa: DTZ001
 
@@ -415,7 +415,7 @@ def test_trace_generation_invalid_start_time(self):
             }
             with tracer.trace(operation_name="operation_name", tags=tags) as span:
                 ...
-            assert span.raw_span()._data["usage"] is None
+            assert span.raw_span()._data["usage_details"] is None
             assert span.raw_span()._data["model"] == "test_model"
             assert span.raw_span()._data["completion_start_time"] is None