From 71e6565b2def3f4cd4cf51f06ef7be4fa1453440 Mon Sep 17 00:00:00 2001 From: anakin87 Date: Fri, 22 Dec 2023 11:39:21 +0100 Subject: [PATCH] bring back filters --- integrations/pinecone/src/pinecone_haystack/document_store.py | 1 + integrations/pinecone/tests/conftest.py | 2 +- 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/integrations/pinecone/src/pinecone_haystack/document_store.py b/integrations/pinecone/src/pinecone_haystack/document_store.py index 0cb32a7df..a02968c53 100644 --- a/integrations/pinecone/src/pinecone_haystack/document_store.py +++ b/integrations/pinecone/src/pinecone_haystack/document_store.py @@ -234,6 +234,7 @@ def _embedding_retrieval( vector=query_embedding, top_k=top_k, namespace=namespace or self.namespace, + filter=filters, include_values=True, include_metadata=True, ) diff --git a/integrations/pinecone/tests/conftest.py b/integrations/pinecone/tests/conftest.py index 95ad5fdf3..ea0fc0167 100644 --- a/integrations/pinecone/tests/conftest.py +++ b/integrations/pinecone/tests/conftest.py @@ -6,7 +6,7 @@ from pinecone_haystack.document_store import PineconeDocumentStore # This is the approximate time it takes for the documents to be available -SLEEP_TIME = 25 +SLEEP_TIME = 20 @pytest.fixture()