haystack-velesdb 3.7.0__tar.gz → 3.8.1__tar.gz

{haystack_velesdb-3.7.0/src/haystack_velesdb.egg-info → haystack_velesdb-3.8.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: haystack-velesdb
-Version: 3.7.0
+Version: 3.8.1
 Summary: Haystack 2.x DocumentStore for VelesDB: The Local AI Memory Database.
 Author-email: VelesDB Team <contact@wiscale.fr>
 License: MIT
@@ -20,8 +20,8 @@ Requires-Python: >=3.10
 Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: haystack-ai>=2.0.0
-Requires-Dist: velesdb>=1.14.0
-Requires-Dist: velesdb-common>=1.14.0
+Requires-Dist: velesdb>=3.8.0
+Requires-Dist: velesdb-common>=3.8.0
 Provides-Extra: dev
 Requires-Dist: pytest<9.0,>=7.0; extra == "dev"
 Dynamic: license-file

{haystack_velesdb-3.7.0 → haystack_velesdb-3.8.1}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "haystack-velesdb"
-version = "3.7.0"
+version = "3.8.1"
 description = "Haystack 2.x DocumentStore for VelesDB: The Local AI Memory Database."
 readme = "README.md"
 license = {text = "MIT"}
@@ -31,8 +31,8 @@ classifiers = [
 requires-python = ">=3.10"
 dependencies = [
     "haystack-ai>=2.0.0",
-    "velesdb>=1.14.0",
-    "velesdb-common>=1.14.0",
+    "velesdb>=3.8.0",
+    "velesdb-common>=3.8.0",
 ]
 
 [project.optional-dependencies]

haystack_velesdb-3.8.1/src/haystack_velesdb/__init__.py

@@ -0,0 +1,7 @@
+"""Haystack 2.x DocumentStore integration for VelesDB."""
+
+from haystack_velesdb.document_store import VelesDBDocumentStore
+from haystack_velesdb.retriever import VelesDBEmbeddingRetriever
+
+__all__ = ["VelesDBDocumentStore", "VelesDBEmbeddingRetriever"]
+__version__ = "3.8.1"

haystack_velesdb-3.8.1/src/haystack_velesdb/retriever.py

@@ -0,0 +1,65 @@
+"""Haystack retriever component backed by a :class:`VelesDBDocumentStore`."""
+
+from typing import Any, Dict, List, Optional
+
+from haystack import component, default_from_dict, default_to_dict
+from haystack.dataclasses import Document
+
+from haystack_velesdb.document_store import VelesDBDocumentStore
+
+
+@component
+class VelesDBEmbeddingRetriever:
+    """Retrieves documents from a :class:`VelesDBDocumentStore` by dense
+    embedding similarity, for use inside a Haystack ``Pipeline``.
+
+    Ships the component the ecosystem expects (like ``QdrantEmbeddingRetriever``)
+    so callers no longer have to hand-roll a ``@component`` wrapper: connect an
+    embedder's ``embedding`` output to this component's ``query_embedding`` input.
+    ``top_k`` / ``filters`` set at construction can be overridden per ``run``.
+    """
+
+    def __init__(
+        self,
+        document_store: VelesDBDocumentStore,
+        top_k: int = 10,
+        filters: Optional[Dict[str, Any]] = None,
+        scale_score: bool = True,
+    ) -> None:
+        self._document_store = document_store
+        self._top_k = top_k
+        self._filters = filters
+        self._scale_score = scale_score
+
+    @component.output_types(documents=List[Document])
+    def run(
+        self,
+        query_embedding: List[float],
+        top_k: Optional[int] = None,
+        filters: Optional[Dict[str, Any]] = None,
+    ) -> Dict[str, List[Document]]:
+        """Run dense retrieval for ``query_embedding``."""
+        documents = self._document_store.embedding_retrieval(
+            query_embedding,
+            top_k=self._top_k if top_k is None else top_k,
+            filters=self._filters if filters is None else filters,
+            scale_score=self._scale_score,
+        )
+        return {"documents": documents}
+
+    def to_dict(self) -> Dict[str, Any]:
+        """Serialize the component (and its store) for pipeline persistence."""
+        return default_to_dict(
+            self,
+            document_store=self._document_store.to_dict(),
+            top_k=self._top_k,
+            filters=self._filters,
+            scale_score=self._scale_score,
+        )
+
+    @classmethod
+    def from_dict(cls, data: Dict[str, Any]) -> "VelesDBEmbeddingRetriever":
+        """Rebuild the component (and its store) from a serialized pipeline."""
+        store_data = data["init_parameters"]["document_store"]
+        data["init_parameters"]["document_store"] = VelesDBDocumentStore.from_dict(store_data)
+        return default_from_dict(cls, data)

{haystack_velesdb-3.7.0 → haystack_velesdb-3.8.1/src/haystack_velesdb.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: haystack-velesdb
-Version: 3.7.0
+Version: 3.8.1
 Summary: Haystack 2.x DocumentStore for VelesDB: The Local AI Memory Database.
 Author-email: VelesDB Team <contact@wiscale.fr>
 License: MIT
@@ -20,8 +20,8 @@ Requires-Python: >=3.10
 Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: haystack-ai>=2.0.0
-Requires-Dist: velesdb>=1.14.0
-Requires-Dist: velesdb-common>=1.14.0
+Requires-Dist: velesdb>=3.8.0
+Requires-Dist: velesdb-common>=3.8.0
 Provides-Extra: dev
 Requires-Dist: pytest<9.0,>=7.0; extra == "dev"
 Dynamic: license-file

{haystack_velesdb-3.7.0 → haystack_velesdb-3.8.1}/src/haystack_velesdb.egg-info/SOURCES.txt

@@ -4,6 +4,7 @@ pyproject.toml
 src/haystack_velesdb/__init__.py
 src/haystack_velesdb/document_store.py
 src/haystack_velesdb/py.typed
+src/haystack_velesdb/retriever.py
 src/haystack_velesdb.egg-info/PKG-INFO
 src/haystack_velesdb.egg-info/SOURCES.txt
 src/haystack_velesdb.egg-info/dependency_links.txt

{haystack_velesdb-3.7.0 → haystack_velesdb-3.8.1}/src/haystack_velesdb.egg-info/requires.txt

@@ -1,6 +1,6 @@
 haystack-ai>=2.0.0
-velesdb>=1.14.0
-velesdb-common>=1.14.0
+velesdb>=3.8.0
+velesdb-common>=3.8.0
 
 [dev]
 pytest<9.0,>=7.0

{haystack_velesdb-3.7.0 → haystack_velesdb-3.8.1}/tests/test_real_haystack_integration.py

@@ -286,3 +286,34 @@ def test_pipeline_with_decorated_retriever(tmp_path) -> None:
     assert len(docs) == 2
     # Top result should be the closest English doc to the query vector.
     assert docs[0].id == "doc-en-1"
+
+
+def test_shipped_embedding_retriever_component_runs(tmp_path) -> None:
+    """The package now ships VelesDBEmbeddingRetriever, so callers no longer
+    hand-roll a @component wrapper. It runs standalone and inside a Pipeline."""
+    from haystack_velesdb import VelesDBEmbeddingRetriever
+
+    store = _store(tmp_path)
+    store.write_documents(_docs())
+    retriever = VelesDBEmbeddingRetriever(document_store=store, top_k=2)
+
+    out = retriever.run(query_embedding=[1.0, 0.0, 0.0, 0.0])
+    assert [d.id for d in out["documents"]][0] == "doc-en-1"
+
+    pipeline = Pipeline()
+    pipeline.add_component("retriever", retriever)
+    result = pipeline.run({"retriever": {"query_embedding": [1.0, 0.0, 0.0, 0.0]}})
+    assert len(result["retriever"]["documents"]) == 2
+
+
+def test_shipped_embedding_retriever_serialization_roundtrip(tmp_path) -> None:
+    """to_dict/from_dict rebuilds the component and its store for pipeline YAML."""
+    from haystack_velesdb import VelesDBEmbeddingRetriever
+
+    retriever = VelesDBEmbeddingRetriever(
+        document_store=_store(tmp_path), top_k=3, scale_score=False
+    )
+    restored = VelesDBEmbeddingRetriever.from_dict(retriever.to_dict())
+    assert restored._top_k == 3
+    assert restored._scale_score is False
+    assert isinstance(restored._document_store, VelesDBDocumentStore)

haystack_velesdb-3.7.0/src/haystack_velesdb/__init__.py

@@ -1,6 +0,0 @@
-"""Haystack 2.x DocumentStore integration for VelesDB."""
-
-from haystack_velesdb.document_store import VelesDBDocumentStore
-
-__all__ = ["VelesDBDocumentStore"]
-__version__ = "3.7.0"