langfuse-haystack 5.0.2__tar.gz → 5.1.0__tar.gz

{langfuse_haystack-5.0.2 → langfuse_haystack-5.1.0}/CHANGELOG.md

@@ -1,5 +1,12 @@
 # Changelog
 
+## [integrations/langfuse-v5.0.2] - 2026-06-01
+
+### 🐛 Bug Fixes
+
+- Update langfuse deserialization to not require `span_handler` key (#3378)
+
+
 ## [integrations/langfuse-v5.0.1] - 2026-05-18
 
 ### 🐛 Bug Fixes

{langfuse_haystack-5.0.2 → langfuse_haystack-5.1.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: langfuse-haystack
-Version: 5.0.2
+Version: 5.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-5.0.2 → langfuse_haystack-5.1.0}/example/basic_rag.py

@@ -6,12 +6,15 @@ os.environ["HAYSTACK_CONTENT_TRACING_ENABLED"] = "true"
 from datasets import load_dataset
 from haystack import Document, Pipeline
 from haystack.components.builders import PromptBuilder
-from haystack.components.embedders import SentenceTransformersDocumentEmbedder, SentenceTransformersTextEmbedder
 from haystack.components.generators import OpenAIGenerator
 from haystack.components.retrievers import InMemoryEmbeddingRetriever
 from haystack.document_stores.in_memory import InMemoryDocumentStore
 
 from haystack_integrations.components.connectors.langfuse import LangfuseConnector
+from haystack_integrations.components.embedders.sentence_transformers import (
+    SentenceTransformersDocumentEmbedder,
+    SentenceTransformersTextEmbedder,
+)
 
 
 def get_pipeline(document_store: InMemoryDocumentStore):
@@ -54,7 +57,6 @@ if __name__ == "__main__":
     document_store = InMemoryDocumentStore()
     dataset = load_dataset("bilgeyucel/seven-wonders", split="train")
     embedder = SentenceTransformersDocumentEmbedder("sentence-transformers/all-MiniLM-L6-v2")
-    embedder.warm_up()
     docs_with_embeddings = embedder.run([Document(**ds) for ds in dataset]).get("documents") or []  # type: ignore
     document_store.write_documents(docs_with_embeddings)
 

{langfuse_haystack-5.0.2 → langfuse_haystack-5.1.0}/example/chat.py

@@ -9,7 +9,7 @@ os.environ["HAYSTACK_CONTENT_TRACING_ENABLED"] = "true"
 
 from haystack import Pipeline
 from haystack.components.builders import ChatPromptBuilder
-from haystack.components.generators.chat import HuggingFaceAPIChatGenerator, OpenAIChatGenerator
+from haystack.components.generators.chat import OpenAIChatGenerator
 from haystack.dataclasses import ChatMessage
 from haystack.utils.auth import Secret
 from haystack.utils.hf import HFGenerationAPIType
@@ -17,6 +17,7 @@ from haystack.utils.hf import HFGenerationAPIType
 from haystack_integrations.components.connectors.langfuse import LangfuseConnector
 from haystack_integrations.components.generators.anthropic import AnthropicChatGenerator
 from haystack_integrations.components.generators.cohere import CohereChatGenerator
+from haystack_integrations.components.generators.huggingface_api import HuggingFaceAPIChatGenerator
 
 os.environ["HAYSTACK_CONTENT_TRACING_ENABLED"] = "true"
 

{langfuse_haystack-5.0.2 → langfuse_haystack-5.1.0}/src/haystack_integrations/components/connectors/langfuse/langfuse_connector.py

@@ -6,11 +6,11 @@ from typing import Any
 
 import httpx
 from haystack import component, default_from_dict, default_to_dict, logging, tracing
-from haystack.utils import Secret, deserialize_secrets_inplace
-from haystack.utils.base_serialization import deserialize_class_instance, serialize_class_instance
-from langfuse import Langfuse
+from haystack.core.serialization import component_to_dict
+from haystack.utils import Secret, deserialize_chatgenerator_inplace, deserialize_secrets_inplace
 
 from haystack_integrations.tracing.langfuse import LangfuseTracer, SpanHandler
+from langfuse import Langfuse
 
 logger = logging.getLogger(__name__)
 
@@ -198,7 +198,7 @@ class LangfuseConnector:
 
         :returns: The serialized component as a dictionary.
         """
-        span_handler = serialize_class_instance(self.span_handler) if self.span_handler else None
+        span_handler = component_to_dict(self.span_handler, "span_handler") if self.span_handler else None
         if self.langfuse_client_kwargs:
             # pop httpx_client and mask from self._langfuse_client_kwargs to prevent serialization issues
             langfuse_client_kwargs = {
@@ -228,6 +228,12 @@ class LangfuseConnector:
         """
         init_params = data["init_parameters"]
         deserialize_secrets_inplace(init_params, keys=["secret_key", "public_key"])
-        if init_params.get("span_handler") is not None:
-            init_params["span_handler"] = deserialize_class_instance(init_params["span_handler"])
+        span_handler_data = init_params.get("span_handler")
+        if span_handler_data is not None:
+            if "data" in span_handler_data:
+                # Pipelines serialized before this fix wrap the span handler as
+                # {"type": ..., "data": {"type": ..., "init_parameters": ...}}; unwrap it here so older
+                # serialized pipelines keep working.
+                init_params["span_handler"] = span_handler_data["data"]
+            deserialize_chatgenerator_inplace(init_params, key="span_handler")
         return default_from_dict(cls, data)

langfuse_haystack-5.1.0/tests/conftest.py

@@ -0,0 +1,11 @@
+import pytest
+
+
+@pytest.fixture(autouse=True)
+def allow_deserialization_of_test_modules(monkeypatch):
+    """
+    haystack-ai >= 3.0 refuses to deserialize classes and callables from modules outside its
+    trusted-module allowlist. Tools and callbacks defined in the test modules live outside that
+    allowlist, so trust them explicitly; haystack-ai 2.x ignores this environment variable.
+    """
+    monkeypatch.setenv("HAYSTACK_DESERIALIZATION_ALLOWLIST", "tests,test_*")

{langfuse_haystack-5.0.2 → langfuse_haystack-5.1.0}/tests/test_langfuse_connector.py

@@ -104,10 +104,7 @@ class TestLangfuseConnector:
                 },
                 "span_handler": {
                     "type": "tests.test_langfuse_connector.CustomSpanHandler",
-                    "data": {
-                        "type": "tests.test_langfuse_connector.CustomSpanHandler",
-                        "init_parameters": {},
-                    },
+                    "init_parameters": {},
                 },
                 "host": "https://example.com",
                 "langfuse_client_kwargs": {"timeout": 30.0},
@@ -199,10 +196,7 @@ class TestLangfuseConnector:
                 },
                 "span_handler": {
                     "type": "tests.test_langfuse_connector.CustomSpanHandler",
-                    "data": {
-                        "type": "tests.test_langfuse_connector.CustomSpanHandler",
-                        "init_parameters": {},
-                    },
+                    "init_parameters": {},
                 },
                 "host": "https://example.com",
                 "langfuse_client_kwargs": {"timeout": 30.0},
@@ -218,6 +212,41 @@ class TestLangfuseConnector:
         assert langfuse_connector.host == "https://example.com"
         assert langfuse_connector.langfuse_client_kwargs == {"timeout": 30.0}
 
+    def test_from_dict_with_legacy_span_handler_format(self, monkeypatch):
+        # Pipelines serialized before this fix wrap span_handler as {"type": ..., "data": {...}}
+        monkeypatch.setenv("LANGFUSE_SECRET_KEY", "secret")
+        monkeypatch.setenv("LANGFUSE_PUBLIC_KEY", "public")
+
+        data = {
+            "type": "haystack_integrations.components.connectors.langfuse.langfuse_connector.LangfuseConnector",
+            "init_parameters": {
+                "name": "Chat example - OpenAI",
+                "public": True,
+                "secret_key": {
+                    "type": "env_var",
+                    "env_vars": ["LANGFUSE_SECRET_KEY"],
+                    "strict": True,
+                },
+                "public_key": {
+                    "type": "env_var",
+                    "env_vars": ["LANGFUSE_PUBLIC_KEY"],
+                    "strict": True,
+                },
+                "span_handler": {
+                    "type": "tests.test_langfuse_connector.CustomSpanHandler",
+                    "data": {
+                        "type": "tests.test_langfuse_connector.CustomSpanHandler",
+                        "init_parameters": {},
+                    },
+                },
+                "host": "https://example.com",
+                "langfuse_client_kwargs": {"timeout": 30.0},
+            },
+        }
+
+        langfuse_connector = LangfuseConnector.from_dict(data)
+        assert isinstance(langfuse_connector.span_handler, CustomSpanHandler)
+
     def test_pipeline_serialization(self, monkeypatch):
         # Set test env vars
         monkeypatch.setenv("LANGFUSE_SECRET_KEY", "secret")

langfuse_haystack-5.0.2/src/haystack_integrations/components/connectors/__init__.py

@@ -1,3 +0,0 @@
-# SPDX-FileCopyrightText: 2023-present deepset GmbH <info@deepset.ai>
-#
-# SPDX-License-Identifier: Apache-2.0