cognee-community-vector-adapter-qdrant 0.2.0__tar.gz → 0.2.1__tar.gz

{cognee_community_vector_adapter_qdrant-0.2.0 → cognee_community_vector_adapter_qdrant-0.2.1}/PKG-INFO

@@ -1,13 +1,13 @@
 Metadata-Version: 2.4
 Name: cognee-community-vector-adapter-qdrant
-Version: 0.2.0
+Version: 0.2.1
 Summary: Qdrant vector database adapter for cognee
 Requires-Python: >=3.11,<=3.13
 Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
-Requires-Dist: cognee (==0.5.1)
+Requires-Dist: cognee (==0.5.2)
 Requires-Dist: instructor (>=1.11)
 Requires-Dist: qdrant-client (>=1.16.0)
 Requires-Dist: starlette (>=0.48.0)

{cognee_community_vector_adapter_qdrant-0.2.0 → cognee_community_vector_adapter_qdrant-0.2.1}/cognee_community_vector_adapter_qdrant/qdrant_adapter.py

@@ -180,6 +180,7 @@ class QDrantAdapter(VectorDBInterface):
         query_vector: list[float] | None = None,
         limit: int | None = 15,
         with_vector: bool = False,
+        include_payload: bool = False,
     ) -> list[ScoredResult]:
         if query_text is None and query_vector is None:
             raise MissingQueryParameterError()
@@ -218,6 +219,7 @@ class QDrantAdapter(VectorDBInterface):
                 using="text",
                 limit=limit,
                 with_vectors=with_vector,
+                with_payload=include_payload,
             )
 
             await client.close()
@@ -228,7 +230,9 @@ class QDrantAdapter(VectorDBInterface):
             return [
                 ScoredResult(
                     id=parse_id(str(result.id)),
-                    payload={
+                    payload=None
+                    if not result.payload
+                    else {
                         **result.payload,
                         "id": parse_id(str(result.id)),
                     },
@@ -248,6 +252,7 @@ class QDrantAdapter(VectorDBInterface):
         query_texts: list[str],
         limit: int | None = None,
         with_vectors: bool = False,
+        include_payload: bool = False,
     ):
         """
         Perform batch search in a Qdrant collection with dynamic search requests.
@@ -258,6 +263,7 @@ class QDrantAdapter(VectorDBInterface):
         - limit (int): List of result limits for search requests.
         - with_vectors (bool, optional): Bool indicating whether to return
                                          vectors for search requests.
+        - include_payload (bool, optional): Bool indicating whether to return payload in results.
 
         Returns:
         - results: The search results from Qdrant.
@@ -291,6 +297,7 @@ class QDrantAdapter(VectorDBInterface):
                 ),
                 limit=limit,
                 with_vectors=with_vectors,
+                with_payload=include_payload,
             )
 
             await client.close()

{cognee_community_vector_adapter_qdrant-0.2.0 → cognee_community_vector_adapter_qdrant-0.2.1}/pyproject.toml

@@ -1,12 +1,12 @@
 [project]
 name = "cognee-community-vector-adapter-qdrant"
-version = "0.2.0"
+version = "0.2.1"
 description = "Qdrant vector database adapter for cognee"
 readme = "README.md"
 requires-python = ">=3.11,<=3.13"
 dependencies = [
     "qdrant-client>=1.16.0",
-    "cognee==0.5.1",
+    "cognee==0.5.2",
     "starlette>=0.48.0",
     "instructor>=1.11"
 ]