weave-haystack 0.0.1__tar.gz → 1.0.0__tar.gz

weave_haystack-1.0.0/CHANGELOG.md

@@ -0,0 +1,9 @@
+# Changelog
+
+## [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-0.0.1 → weave_haystack-1.0.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: weave-haystack
-Version: 0.0.1
+Version: 1.0.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

{weave_haystack-0.0.1 → weave_haystack-1.0.0}/example/hybrid_retrieval.py

@@ -5,11 +5,10 @@ from haystack.components.rankers import TransformersSimilarityRanker
 from haystack.components.retrievers.in_memory import InMemoryBM25Retriever, InMemoryEmbeddingRetriever
 from haystack.document_stores.in_memory import InMemoryDocumentStore
 
-from haystack_integrations.components.connectors import WeaveConnector
+from haystack_integrations.components.connectors.weave import WeaveConnector
 
 
 def index():
-
     document_store = InMemoryDocumentStore()
     documents = [
         Document(content="The Eiffel Tower is located in Paris, France."),
@@ -30,7 +29,6 @@ def index():
 
 
 def hybrid_pipeline(document_store):
-
     text_embedder = SentenceTransformersTextEmbedder()
     embedding_retriever = InMemoryEmbeddingRetriever(document_store)
     bm25_retriever = InMemoryBM25Retriever(document_store)

{weave_haystack-0.0.1 → weave_haystack-1.0.0}/pydoc/config.yml

@@ -2,7 +2,7 @@ loaders:
   - type: haystack_pydoc_tools.loaders.CustomPythonLoader
     search_path: [../src]
     modules: [
-      "haystack_integrations.components.connectors.weave_connector",
+      "haystack_integrations.components.connectors.weave.weave_connector",
       "haystack_integrations.tracing.weave.tracer",
     ]
     ignore_when_discovered: ["__init__"]

{weave_haystack-0.0.1 → weave_haystack-1.0.0}/pyproject.toml

@@ -43,30 +43,35 @@ git_describe_command = 'git describe --tags --match="integrations/weights_and_bi
 
 [tool.hatch.envs.default]
 installer = "uv"
-dependencies = ["coverage[toml]>=6.5", "pytest", "pytest-rerunfailures", "haystack-pydoc-tools"]
+dependencies = ["haystack-pydoc-tools", "ruff"]
 [tool.hatch.envs.default.scripts]
-test = "pytest {args:tests}"
-test-cov = "coverage run -m pytest {args:tests}"
-test-cov-retry = "test-cov --reruns 3 --reruns-delay 30 -x"
-cov-report = ["- coverage combine", "coverage report"]
-cov = ["test-cov", "cov-report"]
-cov-retry = ["test-cov-retry", "cov-report"]
 docs = ["pydoc-markdown pydoc/config.yml"]
+fmt = "ruff check --fix {args} && ruff format {args}"
+fmt-check = "ruff check {args} && ruff format --check {args}"
+
+[tool.hatch.envs.test]
+dependencies = [
+    "pytest",
+    "pytest-asyncio",
+    "pytest-cov",
+    "pytest-rerunfailures",
+    "mypy",
+    "pip"
+]
 
-[tool.hatch.envs.lint]
-installer = "uv"
-detached = true
-dependencies = ["pip", "black>=23.1.0", "mypy>=1.0.0", "ruff>=0.0.243"]
-[tool.hatch.envs.lint.scripts]
-typing = "mypy --install-types --non-interactive --explicit-package-bases {args:src/ tests}"
-style = ["ruff check {args:}", "black --check --diff {args:.}"]
-fmt = ["black {args:.}", "ruff check --fix {args:}", "style"]
-all = ["style", "typing"]
-
-[tool.black]
-target-version = ["py39"]
-line-length = 120
-skip-string-normalization = true
+[tool.hatch.envs.test.scripts]
+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'
+types = """mypy -p haystack_integrations.components.connectors.weave \
+-p haystack_integrations.tracing.weave {args}"""
+
+[tool.mypy]
+install_types = true
+non_interactive = true
+check_untyped_defs = true
+disallow_incomplete_defs = true
 
 [tool.ruff]
 target-version = "py39"
@@ -146,14 +151,3 @@ exclude_lines = ["no cov", "if __name__ == .__main__.:", "if TYPE_CHECKING:"]
 markers = [
   "integration: integration tests"
 ]
-
-[[tool.mypy.overrides]]
-module = [
-  "grpc",
-  "haystack.*",
-  "haystack_integrations.*",
-  "numpy",
-  "pytest.*",
-  "weave.*"
-]
-ignore_missing_imports = true

{weave_haystack-0.0.1 → weave_haystack-1.0.0}/src/haystack_integrations/tracing/weave/tracer.py

@@ -53,7 +53,7 @@ class WeaveSpan(Span):
             return {}
         return {"weave.call_id": self._call.id, "weave.run_id": getattr(self._call, "run_id", "")}
 
-    def set_call(self, call: "weave.Call") -> None:
+    def set_call(self, call: Call) -> None:
         self._call = call
 
     def get_attributes(self) -> dict[str, Any]:
@@ -151,7 +151,12 @@ class WeaveTracer(Tracer):
 
     @contextlib.contextmanager
     def trace(
-        self, operation_name: str, tags: Optional[dict[str, Any]] = None, parent_span: Optional[WeaveSpan] = None
+        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]
     ) -> Iterator[WeaveSpan]:
         """
         A context manager that creates and manages spans for tracking operations in Weights & Biases Weave.

{weave_haystack-0.0.1 → weave_haystack-1.0.0}/tests/test_tracer.py

@@ -27,7 +27,6 @@ class MockWeaveClient:
 
 
 class TestWeaveTracer:
-
     def test_initialization(self):
         with patch("weave.init") as mock_init:
             mock_init.return_value = MockWeaveClient()

{weave_haystack-0.0.1 → weave_haystack-1.0.0}/tests/test_weave_connector.py

@@ -10,7 +10,7 @@ from haystack.core.errors import PipelineRuntimeError
 from haystack.tracing import tracer as haystack_configured_tracer
 from weave.trace.autopatch import AutopatchSettings
 
-from haystack_integrations.components.connectors import WeaveConnector
+from haystack_integrations.components.connectors.weave import WeaveConnector
 from haystack_integrations.tracing.weave import WeaveTracer
 
 
@@ -53,7 +53,7 @@ class TestWeaveConnector:
 
         assert serialized["init_parameters"]["pipeline_name"] == "test_pipeline"
         assert "type" in serialized
-        assert serialized["type"] == "haystack_integrations.components.connectors.weave_connector.WeaveConnector"
+        assert serialized["type"] == "haystack_integrations.components.connectors.weave.weave_connector.WeaveConnector"
 
         deserialized = WeaveConnector.from_dict(serialized)
 
@@ -73,7 +73,7 @@ class TestWeaveConnector:
         assert serialized["init_parameters"]["pipeline_name"] == "test_pipeline"
         assert serialized["init_parameters"]["weave_init_kwargs"] == {"autopatch_settings": {"disable_autopatch": True}}
         assert "type" in serialized
-        assert serialized["type"] == "haystack_integrations.components.connectors.weave_connector.WeaveConnector"
+        assert serialized["type"] == "haystack_integrations.components.connectors.weave.weave_connector.WeaveConnector"
 
         deserialized = WeaveConnector.from_dict(serialized)
 

weave_haystack-0.0.1/CHANGELOG.md

@@ -1,20 +0,0 @@
-# Changelog
-
-## [integrations/weights_bias-v0.0.2] - 2025-03-13
-
-### 🚀 Features
-
-- Add weave_init_kwargs to WeaveConnector (#1537)
-
-### 📚 Documentation
-
-- Update changelog for integrations/weights_bias (#1426)
-- Fixing WandB README (#1427)
-
-## [integrations/weights_bias-v0.0.1] - 2025-02-21
-
-### 🚀 Features
-
-- Integration with Weave the tracer from Weight and Bias  (#1412)
-
-<!-- generated by git-cliff -->

{weave_haystack-0.0.1/src/haystack_integrations/components/connectors → weave_haystack-1.0.0/src/haystack_integrations/components/connectors/weave}/weave_connector.py

@@ -1,9 +1,9 @@
 from typing import Any, Optional
 
 from haystack import component, default_from_dict, default_to_dict, logging, tracing
-from weave.trace.autopatch import AutopatchSettings
 
 from haystack_integrations.tracing.weave import WeaveTracer
+from weave.trace.autopatch import AutopatchSettings
 
 logger = logging.getLogger(__name__)