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

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

@@ -1,5 +1,20 @@
 # Changelog
 
+## [integrations/langfuse-v4.0.1] - 2026-01-27
+
+### 🐛 Bug Fixes
+
+- *(langfuse)* Apply trace_id from tracing context to Langfuse traces (#2781)
+
+
+## [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.1.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: langfuse-haystack
-Version: 4.0.0
+Version: 4.1.0
 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.1.0}/example/basic_rag.py

@@ -62,5 +62,5 @@ if __name__ == "__main__":
     question = "What does Rhodes Statue look like?"
     response = pipeline.run({"text_embedder": {"text": question}, "prompt_builder": {"question": question}})
 
-    print(response["llm"]["replies"][0])  # noqa: T201
-    print(response["tracer"]["trace_url"])  # noqa: T201
+    print(response["llm"]["replies"][0])
+    print(response["tracer"]["trace_url"])

{langfuse_haystack-4.0.0 → langfuse_haystack-4.1.0}/example/chat.py

@@ -59,5 +59,5 @@ if __name__ == "__main__":
             },
         }
     )
-    print(response["llm"]["replies"][0])  # noqa: T201
-    print(response["tracer"]["trace_url"])  # noqa: T201
+    print(response["llm"]["replies"][0])
+    print(response["tracer"]["trace_url"])

langfuse_haystack-4.1.0/pydoc/config_docusaurus.yml

@@ -0,0 +1,14 @@
+loaders:
+  - modules:
+      - haystack_integrations.components.connectors.langfuse.langfuse_connector
+      - haystack_integrations.tracing.langfuse.tracer
+    search_path: [../src]
+processors:
+  - type: filter
+    documented_only: true
+    skip_empty_modules: true
+renderer:
+  description: Langfuse integration for Haystack
+  id: integrations-langfuse
+  filename: langfuse.md
+  title: langfuse

{langfuse_haystack-4.0.0 → langfuse_haystack-4.1.0}/pyproject.toml

@@ -45,7 +45,7 @@ installer = "uv"
 dependencies = ["haystack-pydoc-tools", "ruff"]
 
 [tool.hatch.envs.default.scripts]
-docs = ["pydoc-markdown pydoc/config_docusaurus.yml"]
+docs = ["haystack-pydoc pydoc/config_docusaurus.yml"]
 fmt = "ruff check --fix {args}; ruff format {args}"
 fmt-check = "ruff check {args} && ruff format --check {args}"
 
@@ -76,8 +76,6 @@ non_interactive = true
 check_untyped_defs = true
 disallow_incomplete_defs = true
 
-[tool.hatch.metadata]
-allow-direct-references = true
 
 
 [tool.ruff]
@@ -86,9 +84,17 @@ line-length = 120
 [tool.ruff.lint]
 select = [
   "A",
+  "ANN",
   "ARG",
   "B",
   "C",
+  "D102",   # Missing docstring in public method
+  "D103",   # Missing docstring in public function
+  "D205",   # 1 blank line required between summary line and description
+  "D209",   # Closing triple quotes go to new line
+  "D213",   # summary lines must be positioned on the second physical line of the docstring
+  "D417",   # Missing argument descriptions in the docstring
+  "D419",   # Docstring is empty
   "DTZ",
   "E",
   "EM",
@@ -125,6 +131,8 @@ ignore = [
   "PLR0915",
   # Asserts
   "S101",
+  # Allow `Any` - used legitimately for dynamic types and SDK boundaries
+  "ANN401",
 ]
 
 [tool.ruff.lint.isort]
@@ -135,9 +143,10 @@ ban-relative-imports = "parents"
 
 [tool.ruff.lint.per-file-ignores]
 # Tests can use magic values, assertions, and relative imports
-"tests/**/*" = ["PLR2004", "S101", "TID252"]
+"tests/**/*" = ["D", "PLR2004", "S101", "TID252", "ANN"]
 # Examples can print their output
-"examples/**" = ["T201"]
+"examples/**" = ["T201", "ANN", "D"]
+"example/**" = ["T201", "ANN", "D"]
 "tests/**" = ["T201"]
 
 [tool.coverage.run]

{langfuse_haystack-4.0.0 → langfuse_haystack-4.1.0}/src/haystack_integrations/components/connectors/langfuse/langfuse_connector.py

@@ -19,6 +19,7 @@ logger = logging.getLogger(__name__)
 class LangfuseConnector:
     """
     LangfuseConnector connects Haystack LLM framework with [Langfuse](https://langfuse.com) in order to enable the
+
     tracing of operations and data flow within various components of a pipeline.
 
     To use LangfuseConnector, add it to your pipeline without connecting it to any other components.

{langfuse_haystack-4.0.0 → langfuse_haystack-4.1.0}/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__)
 
@@ -91,7 +91,10 @@ class LangfuseSpan(Span):
         if key.endswith(".input"):
             if "messages" in value:
                 messages = [m.to_openai_dict_format(require_tool_call_ids=False) for m in value["messages"]]
-                self._span.update(input=messages)
+                if isinstance(gen_kwargs := value.get("generation_kwargs"), dict):
+                    self._span.update(input={"messages": messages, "generation_kwargs": gen_kwargs})
+                else:
+                    self._span.update(input=messages)
             else:
                 coerced_value = tracing_utils.coerce_tag_value(value)
                 self._span.update(input=coerced_value)
@@ -125,6 +128,7 @@ class LangfuseSpan(Span):
         return self._data
 
     def get_correlation_data_for_logs(self) -> dict[str, Any]:
+        """Return correlation data for log enrichment."""
         return {}
 
 
@@ -234,9 +238,11 @@ class SpanHandler(ABC):
 
     @classmethod
     def from_dict(cls, data: dict[str, Any]) -> "SpanHandler":
+        """Deserialize a SpanHandler from a dictionary."""
         return default_from_dict(cls, data)
 
     def to_dict(self) -> dict[str, Any]:
+        """Serialize this SpanHandler to a dictionary."""
         return default_to_dict(self)
 
 
@@ -273,6 +279,7 @@ class DefaultSpanHandler(SpanHandler):
     """DefaultSpanHandler provides the default Langfuse tracing behavior for Haystack."""
 
     def create_span(self, context: SpanContext) -> LangfuseSpan:
+        """Create a Langfuse span based on the given context."""
         if self.tracer is None:
             message = (
                 "Tracer is not initialized. "
@@ -287,8 +294,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,
@@ -339,6 +350,7 @@ class DefaultSpanHandler(SpanHandler):
             return LangfuseSpan(self.tracer.start_as_current_span(name=context.name))
 
     def handle(self, span: LangfuseSpan, component_type: str | None) -> None:
+        """Process and enrich a span after component execution."""
         # If the span is at the pipeline level, we add input and output keys to the span
         at_pipeline_level = span.get_data().get(_PIPELINE_INPUT_KEY) is not None
         if at_pipeline_level:
@@ -452,6 +464,7 @@ class LangfuseTracer(Tracer):
     def trace(
         self, operation_name: str, tags: dict[str, Any] | None = None, parent_span: Span | None = None
     ) -> Iterator[Span]:
+        """Create and manage a tracing span as a context manager."""
         tags = tags or {}
         span_name = tags.get(_COMPONENT_NAME_KEY, operation_name)
         component_type = tags.get(_COMPONENT_TYPE_KEY)
@@ -539,6 +552,7 @@ class LangfuseTracer(Tracer):
                 self.flush()
 
     def flush(self) -> None:
+        """Flush all pending spans to Langfuse."""
         self._tracer.flush()
 
     def current_span(self) -> Span | None:
@@ -554,6 +568,7 @@ class LangfuseTracer(Tracer):
     def get_trace_url(self) -> str:
         """
         Return the URL to the tracing data.
+
         :return: The URL to the tracing data.
         """
         return self._tracer.get_trace_url() or ""
@@ -561,6 +576,7 @@ class LangfuseTracer(Tracer):
     def get_trace_id(self) -> str:
         """
         Return the trace ID.
+
         :return: The trace ID.
         """
         return self._tracer.get_current_trace_id() or ""

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

@@ -18,6 +18,7 @@ from haystack_integrations.tracing.langfuse.tracer import (
     LangfuseTracer,
     SpanContext,
     _sanitize_usage_data,
+    tracing_context_var,
 )
 
 
@@ -148,7 +149,6 @@ class TestLangfuseSpan:
     def test_set_content_tag_updates_input_and_output_with_messages(self):
         mock_context_manager = MockContextManager()
 
-        # test message input
         span = LangfuseSpan(mock_context_manager)
         span.set_content_tag("key.input", {"messages": [ChatMessage.from_user("message")]})
         assert mock_context_manager._span.update.call_count == 1
@@ -156,6 +156,20 @@ class TestLangfuseSpan:
         assert mock_context_manager._span.update.call_args_list[0][1] == {
             "input": [{"role": "user", "content": "message"}]
         }
+
+        # test generation_kwargs are captured
+        mock_context_manager._span.update.reset_mock()
+        span.set_content_tag(
+            "key.input",
+            {"messages": [ChatMessage.from_user("message")], "generation_kwargs": {"n": 2, "temperature": 0.7}},
+        )
+        assert mock_context_manager._span.update.call_count == 1
+        assert mock_context_manager._span.update.call_args_list[0][1] == {
+            "input": {
+                "messages": [{"role": "user", "content": "message"}],
+                "generation_kwargs": {"n": 2, "temperature": 0.7},
+            }
+        }
         # test replies ChatMessage list
         mock_context_manager._span.update.reset_mock()
         span.set_content_tag("key.output", {"replies": [ChatMessage.from_system("reply")]})
@@ -222,6 +236,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
@@ -529,7 +581,7 @@ class TestLangfuseTracer:
         assert tracer._public
 
     def test_create_new_span(self):
-        mock_raw_span = MagicMock()
+        mock_raw_span = Mock()
         mock_raw_span.operation_name = "operation_name"
         mock_raw_span.metadata = {"tag1": "value1", "tag2": "value2"}
 
@@ -547,7 +599,7 @@ class TestLangfuseTracer:
             mock_span_instance.get_data.return_value = {}
             mock_span_instance._context_manager = mock_context_manager
 
-            mock_tracer = MagicMock()
+            mock_tracer = Mock()
             mock_tracer.start_as_current_span.return_value = mock_context_manager
             mock_tracer.start_as_current_observation.return_value = mock_context_manager
 

langfuse_haystack-4.0.0/pydoc/config_docusaurus.yml

@@ -1,29 +0,0 @@
-loaders:
-- ignore_when_discovered:
-  - __init__
-  modules:
-  - haystack_integrations.components.connectors.langfuse.langfuse_connector
-  - haystack_integrations.tracing.langfuse.tracer
-  search_path:
-  - ../src
-  type: haystack_pydoc_tools.loaders.CustomPythonLoader
-processors:
-- do_not_filter_modules: false
-  documented_only: true
-  expression: null
-  skip_empty_modules: true
-  type: filter
-- type: smart
-- type: crossref
-renderer:
-  description: Langfuse integration for Haystack
-  id: integrations-langfuse
-  markdown:
-    add_member_class_prefix: false
-    add_method_class_prefix: true
-    classdef_code_block: false
-    descriptive_class_title: false
-    descriptive_module_title: true
-    filename: langfuse.md
-  title: langfuse
-  type: haystack_pydoc_tools.renderers.DocusaurusRenderer