weave-haystack 3.0.0__tar.gz → 3.2.0__tar.gz

weave_haystack-3.2.0/CHANGELOG.md

@@ -0,0 +1,20 @@
+# Changelog
+
+## [integrations/weave-v3.1.0] - 2026-01-14
+
+### 🚀 Features
+
+- Update weave components to auto call run `warm_up` (#2750)
+
+### 🧹 Chores
+
+- Standardize Weave integration - pt.2 (#2739)
+
+
+## [integrations/weave-v3.0.0] - 2026-01-13
+
+### 🧹 Chores
+
+- Standardize Weave integration (#2738)
+
+<!-- generated by git-cliff -->

{weave_haystack-3.0.0 → weave_haystack-3.2.0}/LICENSE.txt

@@ -58,7 +58,7 @@ APPENDIX: How to apply the Apache License to your work.
 
 To apply the Apache License to your work, attach the following boilerplate notice, with the fields enclosed by brackets "[]" replaced with your own identifying information. (Don't include the brackets!)  The text should be enclosed in the appropriate comment syntax for the file format. We also recommend that a file or class name and description of purpose be included on the same "printed page" as the copyright notice for easier identification within third-party archives.
 
-Copyright [yyyy] [name of copyright owner]
+Copyright 2025 deepset GmbH
 
 Licensed under the Apache License, Version 2.0 (the "License");
 you may not use this file except in compliance with the License.

{weave_haystack-3.0.0 → weave_haystack-3.2.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: weave-haystack
-Version: 3.0.0
+Version: 3.2.0
 Summary: An integration of Weights & Biases Weave tracer with Haystack
 Project-URL: Source, https://github.com/deepset-ai/haystack-core-integrations
 Project-URL: Documentation, https://github.com/deepset-ai/haystack-core-integrations/blob/main/integrations/weave/README.md
@@ -15,6 +15,7 @@ Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
+Classifier: Programming Language :: Python :: 3.14
 Classifier: Programming Language :: Python :: Implementation :: CPython
 Classifier: Programming Language :: Python :: Implementation :: PyPy
 Requires-Python: >=3.10

{weave_haystack-3.0.0 → weave_haystack-3.2.0}/example/hybrid_retrieval.py

@@ -20,7 +20,7 @@ def index():
     ]
 
     embedder = SentenceTransformersDocumentEmbedder()
-    embedder.warm_up()
+    # Components warm up automatically on first run.
     embed_docs = embedder.run(documents)
 
     document_store.write_documents(embed_docs["documents"])

weave_haystack-3.2.0/pydoc/config_docusaurus.yml

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

{weave_haystack-3.0.0 → weave_haystack-3.2.0}/pyproject.toml

@@ -19,6 +19,7 @@ classifiers = [
   "Programming Language :: Python :: 3.11",
   "Programming Language :: Python :: 3.12",
   "Programming Language :: Python :: 3.13",
+  "Programming Language :: Python :: 3.14",
   "Programming Language :: Python :: Implementation :: CPython",
   "Programming Language :: Python :: Implementation :: PyPy",
 ]
@@ -44,7 +45,7 @@ git_describe_command = 'git describe --tags --match="integrations/weave-v[0-9]*"
 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}"
 
@@ -62,7 +63,8 @@ dependencies = [
 unit = 'pytest -m "not integration" {args:tests}'
 integration = 'pytest -m "integration" {args:tests}'
 all = 'pytest {args:tests}'
-cov-retry = 'pytest --cov=haystack_integrations --reruns 3 --reruns-delay 30 -x {args:tests}'
+unit-cov-retry = 'pytest --cov=haystack_integrations --reruns 3 --reruns-delay 30 -x -m "not integration" {args:tests}'
+integration-cov-append-retry = 'pytest --cov=haystack_integrations --cov-append --reruns 3 --reruns-delay 30 -x -m "integration" {args:tests}'
 types = """mypy -p haystack_integrations.components.connectors.weave \
 -p haystack_integrations.tracing.weave {args}"""
 
@@ -78,9 +80,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",
@@ -105,6 +115,7 @@ select = [
 ]
 ignore = [
   # Allow non-abstract empty methods in abstract base classes
+  "ANN401",
   "B027",
   # Allow boolean positional values in function calls, like `dict.get(... True)`
   "FBT003",
@@ -128,11 +139,13 @@ ban-relative-imports = "parents"
 
 [tool.ruff.lint.per-file-ignores]
 # Tests can use magic values, assertions, and relative imports
-"tests/**/*" = ["PLR2004", "S101", "TID252"]
+"example/**/*" = ["D", "ANN"]
+"tests/**/*" = ["D", "ANN", "PLR2004", "S101", "TID252"]
 
 [tool.coverage.run]
 source = ["haystack_integrations"]
 branch = true
+relative_files = true
 parallel = false
 
 

{weave_haystack-3.0.0 → weave_haystack-3.2.0}/src/haystack_integrations/components/connectors/weave/__init__.py

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

{weave_haystack-3.0.0 → weave_haystack-3.2.0}/src/haystack_integrations/components/connectors/weave/weave_connector.py

@@ -1,6 +1,12 @@
+# SPDX-FileCopyrightText: 2025-present deepset GmbH <info@deepset.ai>
+#
+# SPDX-License-Identifier: Apache-2.0
+
+from dataclasses import asdict, is_dataclass
 from typing import Any
 
 from haystack import component, default_from_dict, default_to_dict, logging, tracing
+from pydantic import BaseModel
 
 from haystack_integrations.tracing.weave import WeaveTracer
 from weave.trace.settings import UserSettings
@@ -90,6 +96,9 @@ class WeaveConnector:
 
     @component.output_types(pipeline_name=str)
     def run(self) -> dict[str, str]:
+        """Run the WeaveConnector, initializing the tracer if needed."""
+        if self.tracer is None:
+            self.warm_up()
         # NOTE: this is a no-op component - it simply triggers the Tracer to sends traces to Weights & Biases
         return {"pipeline_name": self.pipeline_name}
 
@@ -103,7 +112,10 @@ class WeaveConnector:
         weave_init_kwargs = self.weave_init_kwargs.copy()
         settings = weave_init_kwargs.get("settings", None)
         if isinstance(settings, UserSettings):
-            weave_init_kwargs["settings"] = settings.model_dump(mode="json", exclude_defaults=True)
+            if isinstance(settings, BaseModel):
+                weave_init_kwargs["settings"] = settings.model_dump(mode="json", exclude_defaults=True)
+            elif is_dataclass(settings):
+                weave_init_kwargs["settings"] = asdict(settings)
 
         return default_to_dict(self, pipeline_name=self.pipeline_name, weave_init_kwargs=weave_init_kwargs)
 

{weave_haystack-3.0.0 → weave_haystack-3.2.0}/src/haystack_integrations/tracing/weave/__init__.py

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

{weave_haystack-3.0.0 → weave_haystack-3.2.0}/src/haystack_integrations/tracing/weave/tracer.py

@@ -1,3 +1,7 @@
+# SPDX-FileCopyrightText: 2025-present deepset GmbH <info@deepset.ai>
+#
+# SPDX-License-Identifier: Apache-2.0
+
 import contextlib
 import os
 from collections.abc import Iterator
@@ -38,6 +42,7 @@ class WeaveSpan(Span):
         self._attributes[key] = value
 
     def set_tags(self, tags: dict[str, Any]) -> None:
+        """Set multiple tags at once by iterating over the provided dictionary."""
         for k, v in tags.items():
             self.set_tag(k, v)
 
@@ -52,9 +57,11 @@ class WeaveSpan(Span):
         return {"weave.call_id": self._call.id, "weave.run_id": getattr(self._call, "run_id", "")}
 
     def set_call(self, call: Call) -> None:
+        """Set the underlying Weave Call object for this span."""
         self._call = call
 
     def get_attributes(self) -> dict[str, Any]:
+        """Return the accumulated attributes dictionary for this span."""
         return self._attributes
 
 
@@ -88,6 +95,7 @@ class WeaveTracer(Tracer):
 
     @staticmethod
     def create_call(attributes: dict, client: WeaveClient, parent_span: WeaveSpan | None, operation_name: str) -> Call:
+        """Create and return a Weave Call from the given span attributes and client."""
         comp_name = attributes.pop("haystack.component.name", "")
         comp_type = attributes.pop("haystack.component.type", "")
         comp_input = attributes.pop("haystack.component.input", {})

{weave_haystack-3.0.0 → weave_haystack-3.2.0}/tests/test_tracer.py

@@ -1,3 +1,7 @@
+# SPDX-FileCopyrightText: 2025-present deepset GmbH <info@deepset.ai>
+#
+# SPDX-License-Identifier: Apache-2.0
+
 import logging
 from unittest.mock import patch
 

{weave_haystack-3.0.0 → weave_haystack-3.2.0}/tests/test_weave_connector.py

@@ -1,3 +1,7 @@
+# SPDX-FileCopyrightText: 2025-present deepset GmbH <info@deepset.ai>
+#
+# SPDX-License-Identifier: Apache-2.0
+
 import os
 from collections.abc import Generator
 from typing import Any
@@ -21,7 +25,7 @@ class FailingComponent:
     def run(self) -> dict[str, Any]:
         """Execute the component's logic - always raises an exception."""
         msg = "Test error"
-        raise PipelineRuntimeError(msg)
+        raise PipelineRuntimeError(component_name="FailingComponent", component_type=FailingComponent, message=msg)
 
 
 @pytest.fixture
@@ -93,9 +97,7 @@ class TestWeaveConnector:
         serialized: dict[str, Any] = connector.to_dict()
 
         assert serialized["init_parameters"]["pipeline_name"] == "test_pipeline"
-        assert serialized["init_parameters"]["weave_init_kwargs"] == {
-            "settings": {"implicitly_patch_integrations": False}
-        }
+        assert serialized["init_parameters"]["weave_init_kwargs"]["settings"]["implicitly_patch_integrations"] is False
         assert "type" in serialized
         assert serialized["type"] == "haystack_integrations.components.connectors.weave.weave_connector.WeaveConnector"
 
@@ -104,7 +106,7 @@ class TestWeaveConnector:
         assert isinstance(deserialized, WeaveConnector)
         assert deserialized.pipeline_name == "test_pipeline"
         assert deserialized.tracer is None  # tracer is only initialized with warm_up
-        assert deserialized.weave_init_kwargs == {"settings": {"implicitly_patch_integrations": False}}
+        assert deserialized.weave_init_kwargs["settings"]["implicitly_patch_integrations"] is False
 
     def test_pipeline_tracing(self, mock_weave_client: Mock, sample_pipeline: Pipeline) -> None:
         """Test that pipeline operations are correctly traced"""
@@ -146,6 +148,11 @@ class TestWeaveConnector:
         error_traces = [args for args in finish_call_args if args[1].get("exception") is not None]
         assert len(error_traces) > 0
 
+    @pytest.mark.skipif(
+        not os.environ.get("WANDB_API_KEY", None),
+        reason="Export an env var called WANDB_API_KEY containing the Weights & Bias API key to run this test.",
+    )
+    @pytest.mark.integration
     def test_run_method(self) -> None:
         """Test the basic run method of WeaveConnector"""
         connector = WeaveConnector(pipeline_name="test_pipeline")

weave_haystack-3.0.0/CHANGELOG.md

@@ -1,36 +0,0 @@
-# Changelog
-
-## [integrations/weights_and_biases_weave-v2.0.0] - 2025-09-16
-
-### 🐛 Bug Fixes
-
-- Weave - pin gql (#2198)
-
-### 🚜 Refactor
-
-- [**breaking**] Weave - update patching support and unpin gql (#2272)
-
-### 🧹 Chores
-
-- Standardize readmes - part 2 (#2205)
-
-
-## [integrations/weights_and_biases_weave-v1.0.0] - 2025-07-01
-
-### 🐛 Bug Fixes
-
-- [**breaking**] Weave - move Connector, fix types, add py.typed (#2038)
-
-### 🧹 Chores
-
-- Align core-integrations Hatch scripts (#1898)
-- Remove black (#1985)
-
-
-## [integrations/weights_and_biases_weave-v0.0.1] - 2025-03-28
-
-### 🚜 Refactor
-
-- Renaming the project/package to weights_biases_weave (#1565)
-
-<!-- generated by git-cliff -->

weave_haystack-3.0.0/pydoc/config_docusaurus.yml

@@ -1,29 +0,0 @@
-loaders:
-- ignore_when_discovered:
-  - __init__
-  modules:
-  - haystack_integrations.components.connectors.weave.weave_connector
-  - haystack_integrations.tracing.weave.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: Weights & Bias integration for Haystack
-  id: integrations-weights-bias
-  markdown:
-    add_member_class_prefix: false
-    add_method_class_prefix: true
-    classdef_code_block: false
-    descriptive_class_title: false
-    descriptive_module_title: true
-    filename: weights_and_bias.md
-  title: weights and bias
-  type: haystack_pydoc_tools.renderers.DocusaurusRenderer