explicit projection when reading from Astra DB

superagent-ai · May 14, 2024 · cda4454 · cda4454
1 parent e383a64
commit cda4454
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/libs/superagent/app/vectorstores/astra_client.py b/libs/superagent/app/vectorstores/astra_client.py
@@ -160,6 +160,7 @@ def _query(self, vector, top_k, filters=None):
         query = {
             "sort": {"$vector": vector},
             "options": {"limit": top_k, "includeSimilarity": True},
+            "projection": {"*": 1},
         }
         if filters is not None:
             query["filter"] = filters