weave-haystack 2.0.0__tar.gz → 3.1.0__tar.gz

weave_haystack-3.1.0/CHANGELOG.md

@@ -0,0 +1,9 @@
+# Changelog
+
+## [integrations/weave-v3.0.0] - 2026-01-13
+
+### 🧹 Chores
+
+- Standardize Weave integration (#2738)
+
+<!-- generated by git-cliff -->

{weave_haystack-2.0.0 → weave_haystack-3.1.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-2.0.0 → weave_haystack-3.1.0}/PKG-INFO

@@ -1,9 +1,9 @@
 Metadata-Version: 2.4
 Name: weave-haystack
-Version: 2.0.0
+Version: 3.1.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/weights_and_biases_weave/README.md
+Project-URL: Documentation, https://github.com/deepset-ai/haystack-core-integrations/blob/main/integrations/weave/README.md
 Project-URL: Issues, https://github.com/deepset-ai/haystack-core-integrations/issues
 Author-email: deepset GmbH <info@deepset.ai>
 License-Expression: Apache-2.0
@@ -11,15 +11,14 @@ License-File: LICENSE.txt
 Classifier: Development Status :: 4 - Beta
 Classifier: License :: OSI Approved :: Apache Software License
 Classifier: Programming Language :: Python
-Classifier: Programming Language :: Python :: 3.9
 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 :: Implementation :: CPython
 Classifier: Programming Language :: Python :: Implementation :: PyPy
-Requires-Python: >=3.9
-Requires-Dist: haystack-ai
+Requires-Python: >=3.10
+Requires-Dist: haystack-ai>=2.22.0
 Requires-Dist: weave>=0.52.7
 Description-Content-Type: text/markdown
 
@@ -29,7 +28,7 @@ Description-Content-Type: text/markdown
 [![PyPI - Python Version](https://img.shields.io/pypi/pyversions/weave-haystack)](https://pypi.org/project/weave-haystack)
 
 - [Integration page](https://haystack.deepset.ai/integrations/weights-and-bias-tracer)
-- [Changelog](https://github.com/deepset-ai/haystack-core-integrations/blob/main/integrations/weights_and_biases_weave/CHANGELOG.md)
+- [Changelog](https://github.com/deepset-ai/haystack-core-integrations/blob/main/integrations/weave/CHANGELOG.md)
 
 ---
 

{weave_haystack-2.0.0 → weave_haystack-3.1.0}/README.md

@@ -4,7 +4,7 @@
 [![PyPI - Python Version](https://img.shields.io/pypi/pyversions/weave-haystack)](https://pypi.org/project/weave-haystack)
 
 - [Integration page](https://haystack.deepset.ai/integrations/weights-and-bias-tracer)
-- [Changelog](https://github.com/deepset-ai/haystack-core-integrations/blob/main/integrations/weights_and_biases_weave/CHANGELOG.md)
+- [Changelog](https://github.com/deepset-ai/haystack-core-integrations/blob/main/integrations/weave/CHANGELOG.md)
 
 ---
 

weave_haystack-3.1.0/pydoc/config_docusaurus.yml

@@ -0,0 +1,29 @@
+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-weave
+  markdown:
+    add_member_class_prefix: false
+    add_method_class_prefix: true
+    classdef_code_block: false
+    descriptive_class_title: false
+    descriptive_module_title: true
+    filename: weave.md
+  title: Weave
+  type: haystack_pydoc_tools.renderers.DocusaurusRenderer

{weave_haystack-2.0.0 → weave_haystack-3.1.0}/pyproject.toml

@@ -7,7 +7,7 @@ name = "weave-haystack"
 dynamic = ["version"]
 description = "An integration of Weights & Biases Weave tracer with Haystack"
 readme = "README.md"
-requires-python = ">=3.9"
+requires-python = ">=3.10"
 license = "Apache-2.0"
 keywords = []
 authors = [{ name = "deepset GmbH", email = "info@deepset.ai" }]
@@ -15,7 +15,6 @@ classifiers = [
   "License :: OSI Approved :: Apache Software License",
   "Development Status :: 4 - Beta",
   "Programming Language :: Python",
-  "Programming Language :: Python :: 3.9",
   "Programming Language :: Python :: 3.10",
   "Programming Language :: Python :: 3.11",
   "Programming Language :: Python :: 3.12",
@@ -23,11 +22,11 @@ classifiers = [
   "Programming Language :: Python :: Implementation :: CPython",
   "Programming Language :: Python :: Implementation :: PyPy",
 ]
-dependencies = ["haystack-ai", "weave>=0.52.7"]
+dependencies = ["haystack-ai>=2.22.0", "weave>=0.52.7"]
 
 [project.urls]
 Source = "https://github.com/deepset-ai/haystack-core-integrations"
-Documentation = "https://github.com/deepset-ai/haystack-core-integrations/blob/main/integrations/weights_and_biases_weave/README.md"
+Documentation = "https://github.com/deepset-ai/haystack-core-integrations/blob/main/integrations/weave/README.md"
 Issues = "https://github.com/deepset-ai/haystack-core-integrations/issues"
 
 [tool.hatch.build.targets.wheel]
@@ -35,18 +34,18 @@ packages = ["src/haystack_integrations"]
 
 [tool.hatch.version]
 source = "vcs"
-tag-pattern = 'integrations\/weights_and_biases_weave-v(?P<version>.*)'
+tag-pattern = 'integrations\/weave-v(?P<version>.*)'
 
 [tool.hatch.version.raw-options]
 root = "../.."
-git_describe_command = 'git describe --tags --match="integrations/weights_and_biases_weave-v[0-9]*"'
+git_describe_command = 'git describe --tags --match="integrations/weave-v[0-9]*"'
 
 [tool.hatch.envs.default]
 installer = "uv"
 dependencies = ["haystack-pydoc-tools", "ruff"]
 [tool.hatch.envs.default.scripts]
-docs = ["pydoc-markdown pydoc/config.yml"]
-fmt = "ruff check --fix {args} && ruff format {args}"
+docs = ["pydoc-markdown pydoc/config_docusaurus.yml"]
+fmt = "ruff check --fix {args}; ruff format {args}"
 fmt-check = "ruff check {args} && ruff format --check {args}"
 
 [tool.hatch.envs.test]
@@ -63,7 +62,7 @@ dependencies = [
 unit = 'pytest -m "not integration" {args:tests}'
 integration = 'pytest -m "integration" {args:tests}'
 all = 'pytest {args:tests}'
-cov-retry = 'all --cov=haystack_integrations --reruns 3 --reruns-delay 30 -x'
+cov-retry = 'pytest --cov=haystack_integrations --reruns 3 --reruns-delay 30 -x {args:tests}'
 types = """mypy -p haystack_integrations.components.connectors.weave \
 -p haystack_integrations.tracing.weave {args}"""
 
@@ -74,7 +73,6 @@ check_untyped_defs = true
 disallow_incomplete_defs = true
 
 [tool.ruff]
-target-version = "py39"
 line-length = 120
 
 [tool.ruff.lint]
@@ -121,10 +119,6 @@ ignore = [
   "PLR0913",
   "PLR0915",
 ]
-unfixable = [
-  # Don't touch unused imports
-  "F401",
-]
 
 [tool.ruff.lint.isort]
 known-first-party = ["haystack_integrations"]

{weave_haystack-2.0.0 → weave_haystack-3.1.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-2.0.0 → weave_haystack-3.1.0}/src/haystack_integrations/components/connectors/weave/weave_connector.py

@@ -1,4 +1,8 @@
-from typing import Any, Optional
+# SPDX-FileCopyrightText: 2025-present deepset GmbH <info@deepset.ai>
+#
+# SPDX-License-Identifier: Apache-2.0
+
+from typing import Any
 
 from haystack import component, default_from_dict, default_to_dict, logging, tracing
 
@@ -71,7 +75,7 @@ class WeaveConnector:
 
     """
 
-    def __init__(self, pipeline_name: str, weave_init_kwargs: Optional[dict[str, Any]] = None) -> None:
+    def __init__(self, pipeline_name: str, weave_init_kwargs: dict[str, Any] | None = None) -> None:
         """
         Initialize WeaveConnector.
 
@@ -80,7 +84,7 @@ class WeaveConnector:
         """
         self.pipeline_name = pipeline_name
         self.weave_init_kwargs = weave_init_kwargs or {}
-        self.tracer: Optional[WeaveTracer] = None
+        self.tracer: WeaveTracer | None = None
 
     def warm_up(self) -> None:
         """Initialize the WeaveTracer."""
@@ -90,6 +94,8 @@ class WeaveConnector:
 
     @component.output_types(pipeline_name=str)
     def run(self) -> dict[str, str]:
+        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}
 

{weave_haystack-2.0.0 → weave_haystack-3.1.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-2.0.0 → weave_haystack-3.1.0}/src/haystack_integrations/tracing/weave/tracer.py

@@ -1,7 +1,11 @@
+# SPDX-FileCopyrightText: 2025-present deepset GmbH <info@deepset.ai>
+#
+# SPDX-License-Identifier: Apache-2.0
+
 import contextlib
 import os
 from collections.abc import Iterator
-from typing import Any, Optional, Union
+from typing import Any
 
 from haystack import logging
 from haystack.tracing import Span, Tracer
@@ -22,9 +26,7 @@ class WeaveSpan(Span):
     that describe the operation.
     """
 
-    def __init__(
-        self, call: Optional[Call] = None, parent: Optional[Call] = None, operation: Optional[str] = None
-    ) -> None:
+    def __init__(self, call: Call | None = None, parent: Call | None = None, operation: str | None = None) -> None:
         self._call = call
         self._parent = parent
         self._operation = operation
@@ -86,12 +88,10 @@ class WeaveTracer(Tracer):
             )
 
         self._client = weave.init(project_name, **weave_init_kwargs)
-        self._current_span: Optional[WeaveSpan] = None
+        self._current_span: WeaveSpan | None = None
 
     @staticmethod
-    def create_call(
-        attributes: dict, client: WeaveClient, parent_span: Union[WeaveSpan, None], operation_name: str
-    ) -> Call:
+    def create_call(attributes: dict, client: WeaveClient, parent_span: WeaveSpan | None, operation_name: str) -> Call:
         comp_name = attributes.pop("haystack.component.name", "")
         comp_type = attributes.pop("haystack.component.type", "")
         comp_input = attributes.pop("haystack.component.input", {})
@@ -104,18 +104,16 @@ class WeaveTracer(Tracer):
         )
         return call
 
-    def current_span(self) -> Optional[Span]:
+    def current_span(self) -> Span | None:
         """Get the current active span."""
         return self._current_span
 
     @staticmethod
-    def _create_component_span(parent_span: Optional[WeaveSpan], operation_name: str) -> WeaveSpan:
+    def _create_component_span(parent_span: WeaveSpan | None, operation_name: str) -> WeaveSpan:
         """Create a span for component runs with deferred call creation."""
         return WeaveSpan(parent=parent_span.raw_span() if parent_span else None, operation=operation_name)
 
-    def _create_regular_span(
-        self, operation_name: str, tags: Optional[dict], parent_span: Optional[WeaveSpan]
-    ) -> WeaveSpan:
+    def _create_regular_span(self, operation_name: str, tags: dict | None, parent_span: WeaveSpan | None) -> WeaveSpan:
         """Create a span for regular operations with immediate call creation."""
         call = self._client.create_call(
             op=operation_name,
@@ -124,7 +122,7 @@ class WeaveTracer(Tracer):
         )
         return WeaveSpan(call=call)
 
-    def _finish_component_call(self, span: WeaveSpan, parent_span: Optional[WeaveSpan], operation_name: str) -> None:
+    def _finish_component_call(self, span: WeaveSpan, parent_span: WeaveSpan | None, operation_name: str) -> None:
         """Create and finish call for component runs."""
         attributes = span.get_attributes()
         call = self.create_call(
@@ -149,14 +147,14 @@ class WeaveTracer(Tracer):
             output={key: coerce_tag_value(value) for key, value in pipeline_output.items()},
         )
 
+    # the current implementation violates the Liskov Substitution Principle by using WeaveSpan instead of Span
+    # unfortunately, it seems hard to fix without rewriting the Tracer
     @contextlib.contextmanager
-    def trace(
+    def trace(  # type: ignore[override]
         self,
         operation_name: str,
-        tags: Optional[dict[str, Any]] = None,
-        # the current implementation violates the Liskov Substitution Principle by using WeaveSpan instead of Span
-        # unfortunately, it seems hard to fix without rewriting the Tracer
-        parent_span: Optional[WeaveSpan] = None,  # type: ignore[override]
+        tags: dict[str, Any] | None = None,
+        parent_span: WeaveSpan | None = None,
     ) -> Iterator[WeaveSpan]:
         """
         A context manager that creates and manages spans for tracking operations in Weights & Biases Weave.

{weave_haystack-2.0.0 → weave_haystack-3.1.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-2.0.0 → weave_haystack-3.1.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
@@ -146,6 +150,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-2.0.0/CHANGELOG.md

@@ -1,21 +0,0 @@
-# Changelog
-
-## [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-2.0.0/pydoc/config.yml

@@ -1,30 +0,0 @@
-loaders:
-  - type: haystack_pydoc_tools.loaders.CustomPythonLoader
-    search_path: [../src]
-    modules: [
-      "haystack_integrations.components.connectors.weave.weave_connector",
-      "haystack_integrations.tracing.weave.tracer",
-    ]
-    ignore_when_discovered: ["__init__"]
-processors:
-  - type: filter
-    expression:
-    documented_only: true
-    do_not_filter_modules: false
-    skip_empty_modules: true
-  - type: smart
-  - type: crossref
-renderer:
-  type: haystack_pydoc_tools.renderers.ReadmeIntegrationRenderer
-  excerpt: Weights & Bias integration for Haystack
-  category_slug: integrations-api
-  title: weights and bias
-  slug: integrations-weights-bias
-  order: 260
-  markdown:
-    descriptive_class_title: false
-    classdef_code_block: false
-    descriptive_module_title: true
-    add_method_class_prefix: true
-    add_member_class_prefix: false
-    filename: _readme_weights_and_bias.md