explicit projection when reading from Astra DB (deepset-ai#733)

Rusteam · May 15, 2024 · c4f1cc4 · c4f1cc4
1 parent 6f298ce
commit c4f1cc4
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/integrations/astra/src/haystack_integrations/document_stores/astra/astra_client.py b/integrations/astra/src/haystack_integrations/document_stores/astra/astra_client.py
@@ -208,6 +208,7 @@ def find_documents(self, find_query):
  filter=find_query.get("filter"),
  sort=find_query.get("sort"),
  options=find_query.get("options"),
+ projection={"*": 1},
  )
 
  if "data" in response_dict and "documents" in response_dict["data"]:
@@ -273,6 +274,7 @@ def update_document(self, document: Dict, id_key: str):
  filter={id_key: document_id},
  update={"$set": document},
  options={"returnDocument": "after"},
+ projection={"*": 1},
  )
 
  document[id_key] = document_id