explicit projection when reading from Astra DB (#996)

superagent-ai · Jun 5, 2024 · 3d4ca4b · 3d4ca4b
1 parent 1a91451
commit 3d4ca4b
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