Ensure deletion queries

Author: ejscribner

api/core/rag/datasource/vdb/couchbase/couchbase_vector.py

@@ -240,16 +240,20 @@ def delete_by_ids(self, ids: list[str]) -> None:
             WHERE META().id IN $doc_ids;
             """
         try:
-            self._cluster.query(query, named_parameters={'doc_ids': ids})
+            result = self._cluster.query(query, named_parameters={'doc_ids': ids})
+            # force evaluation of the query to ensure deletion occurs
+            list(result)
         except Exception as e:
-            print(e)
+            logger.error(e)
 
     def delete_by_document_id(self, document_id: str):
         query = f"""
                 DELETE FROM `{self._client_config.bucket_name}`.{self._client_config.scope_name}.{self._collection_name}
                 WHERE META().id = $doc_id;
                 """
-        self._cluster.query(query,named_parameters={'doc_id':document_id})
+        result = self._cluster.query(query,named_parameters={'doc_id':document_id})
+        # force evaluation of the query to ensure deletion occurs
+        list(result)
 
     # def get_ids_by_metadata_field(self, key: str, value: str):
     #     query = f"""
@@ -265,7 +269,9 @@ def delete_by_metadata_field(self, key: str, value: str) -> None:
             DELETE FROM `{self._client_config.bucket_name}`.{self._client_config.scope_name}.{self._collection_name}
             WHERE metadata.{key} = $value;
             """
-        self._cluster.query(query, named_parameters={'value':value})
+        result = self._cluster.query(query, named_parameters={'value':value})
+        # force evaluation of the query to ensure deletion occurs
+        list(result)
 
     def search_by_vector(
             self,
@@ -387,4 +393,4 @@ def init_vector(self, dataset: Dataset, attributes: list, embeddings: Embeddings
                 bucket_name=config.get('COUCHBASE_BUCKET_NAME'),
                 scope_name=config.get('COUCHBASE_SCOPE_NAME'),
             )
-        )
+        )