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

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

@@ -1,5 +1,12 @@
 # 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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: langfuse-haystack
-Version: 4.0.1
+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.1 → 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.1 → 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.1 → 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.1 → 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.1 → langfuse_haystack-4.1.0}/src/haystack_integrations/tracing/langfuse/tracer.py

@@ -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. "
@@ -343,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:
@@ -456,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)
@@ -543,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:
@@ -558,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 ""
@@ -565,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.1 → langfuse_haystack-4.1.0}/tests/test_tracer.py

@@ -149,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
@@ -157,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")]})
@@ -568,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"}
 
@@ -586,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.1/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