langfuse-haystack 4.0.0__tar.gz → 4.0.1__tar.gz

{langfuse_haystack-4.0.0 → langfuse_haystack-4.0.1}/CHANGELOG.md

@@ -1,5 +1,13 @@
 # Changelog
 
+## [integrations/langfuse-v4.0.0] - 2026-01-12
+
+### 🧹 Chores
+
+- Make fmt command more forgiving (#2671)
+- [**breaking**] Langfuse - drop Python 3.9 and use X|Y typing (#2709)
+
+
 ## [integrations/langfuse-v3.3.1] - 2025-12-10
 
 ### 🚀 Features

{langfuse_haystack-4.0.0 → langfuse_haystack-4.0.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: langfuse-haystack
-Version: 4.0.0
+Version: 4.0.1
 Summary: Langfuse integration for Haystack
 Project-URL: Documentation, https://github.com/deepset-ai/haystack-core-integrations/tree/main/integrations/langfuse#readme
 Project-URL: Issues, https://github.com/deepset-ai/haystack-core-integrations/issues

{langfuse_haystack-4.0.0 → langfuse_haystack-4.0.1}/src/haystack_integrations/tracing/langfuse/tracer.py

@@ -22,7 +22,7 @@ from haystack.tracing import utils as tracing_utils
 
 import langfuse
 from langfuse import LangfuseSpan as LangfuseClientSpan
-from langfuse.types import TraceMetadata
+from langfuse.types import TraceContext, TraceMetadata
 
 logger = logging.getLogger(__name__)
 
@@ -287,8 +287,12 @@ class DefaultSpanHandler(SpanHandler):
             root_span_type: Literal["agent", "span"] = (
                 "agent" if context.operation_name == "haystack.agent.run" else "span"
             )
+            trace_context: TraceContext | None = cast(
+                TraceContext, {"trace_id": tracing_ctx.get("trace_id")} if tracing_ctx.get("trace_id") else None
+            )
             # Create a new trace when there's no parent span
             span_context_manager = self.tracer.start_as_current_observation(
+                trace_context=trace_context,
                 name=context.trace_name,
                 version=tracing_ctx.get("version"),
                 as_type=root_span_type,

{langfuse_haystack-4.0.0 → langfuse_haystack-4.0.1}/tests/test_tracer.py

@@ -18,6 +18,7 @@ from haystack_integrations.tracing.langfuse.tracer import (
     LangfuseTracer,
     SpanContext,
     _sanitize_usage_data,
+    tracing_context_var,
 )
 
 
@@ -222,6 +223,44 @@ class TestSanitizeUsageData:
 
 
 class TestDefaultSpanHandler:
+    def test_create_span_root_trace(self):
+        """Test creating a root span (no parent)."""
+        mock_client = Mock()
+        mock_client.start_as_current_span = Mock(return_value=MockContextManager())
+        mock_client.start_as_current_observation = Mock(return_value=MockContextManager())
+
+        handler = DefaultSpanHandler()
+        handler.init_tracer(mock_client)
+
+        context = MagicMock(spec=SpanContext)
+        context.parent_span = None
+        context.trace_name = "test_trace"
+        context.operation_name = "haystack.pipeline.run"
+        context.public = False
+
+        # Set up tracing context
+        token = tracing_context_var.set(
+            {"user_id": "test_user", "session_id": "test_session", "trace_id": "test_trace_id"}
+        )
+        try:
+            result = handler.create_span(context)
+
+            # Assert tracer was called with correct parameters
+            handler.tracer.start_as_current_observation.assert_called_once_with(
+                trace_context={"trace_id": "test_trace_id"},
+                name="test_trace",
+                version=None,
+                as_type="span",
+            )
+            call_kwargs = handler.tracer.start_as_current_observation.call_args[1]
+            assert "test_trace" == call_kwargs["name"]
+            assert "span" == call_kwargs["as_type"]
+
+            # Assert result is a LangfuseSpan
+            assert isinstance(result, LangfuseSpan)
+        finally:
+            tracing_context_var.reset(token)
+
     def test_handle_generator(self):
         mock_span = Mock()
         mock_span.raw_span.return_value = mock_span