feat: Add force_disable_check_same_thread init param for Qdrant local…

… client (#779) * fix: added missing init param which is use by qdrant local client * fix: Corrected linting
deepset-ai · Jun 6, 2024 · 78157a9 · 78157a9
1 parent 3f3fa2b
commit 78157a9
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 0 deletions.
diff --git a/integrations/qdrant/src/haystack_integrations/document_stores/qdrant/document_store.py b/integrations/qdrant/src/haystack_integrations/document_stores/qdrant/document_store.py
@@ -68,6 +68,7 @@ def __init__(
         timeout: Optional[int] = None,
         host: Optional[str] = None,
         path: Optional[str] = None,
+        force_disable_check_same_thread: bool = False,
         index: str = "Document",
         embedding_dim: int = 768,
         on_disk: bool = False,
@@ -109,6 +110,7 @@ def __init__(
         self.timeout = timeout
         self.host = host
         self.path = path
+        self.force_disable_check_same_thread = force_disable_check_same_thread
         self.metadata = metadata or {}
         self.api_key = api_key
 
@@ -155,6 +157,7 @@ def client(self):
                 host=self.host,
                 path=self.path,
                 metadata=self.metadata,
+                force_disable_check_same_thread=self.force_disable_check_same_thread,
             )
             # Make sure the collection is properly set up
             self._set_up_collection(

diff --git a/integrations/qdrant/tests/test_dict_converters.py b/integrations/qdrant/tests/test_dict_converters.py
@@ -25,6 +25,7 @@ def test_to_dict():
             "content_field": "content",
             "name_field": "name",
             "embedding_field": "embedding",
+            "force_disable_check_same_thread": False,
             "use_sparse_embeddings": False,
             "similarity": "cosine",
             "return_embedding": False,
@@ -64,6 +65,7 @@ def test_from_dict():
                 "content_field": "content",
                 "name_field": "name",
                 "embedding_field": "embedding",
+                "force_disable_check_same_thread": False,
                 "use_sparse_embeddings": True,
                 "similarity": "cosine",
                 "return_embedding": False,
@@ -88,6 +90,7 @@ def test_from_dict():
             document_store.content_field == "content",
             document_store.name_field == "name",
             document_store.embedding_field == "embedding",
+            document_store.force_disable_check_same_thread is False,
             document_store.use_sparse_embeddings is True,
             document_store.on_disk is False,
             document_store.similarity == "cosine",

diff --git a/integrations/qdrant/tests/test_retriever.py b/integrations/qdrant/tests/test_retriever.py
@@ -49,6 +49,7 @@ def test_to_dict(self):
                         "on_disk": False,
                         "content_field": "content",
                         "name_field": "name",
+                        "force_disable_check_same_thread": False,
                         "embedding_field": "embedding",
                         "use_sparse_embeddings": False,
                         "similarity": "cosine",
@@ -172,6 +173,7 @@ def test_to_dict(self):
                         "content_field": "content",
                         "name_field": "name",
                         "embedding_field": "embedding",
+                        "force_disable_check_same_thread": False,
                         "use_sparse_embeddings": False,
                         "similarity": "cosine",
                         "return_embedding": False,
@@ -281,6 +283,7 @@ def test_to_dict(self):
                         "content_field": "content",
                         "name_field": "name",
                         "embedding_field": "embedding",
+                        "force_disable_check_same_thread": False,
                         "use_sparse_embeddings": False,
                         "similarity": "cosine",
                         "return_embedding": False,