test: improve vector store tests (#3855)

2024-04-26 19:18:42 +08:00
parent 4d66a86579
commit 045827043d
6 changed files with 58 additions and 74 deletions
--- a/api/core/rag/datasource/vdb/relyt/relyt_vector.py
+++ b/api/core/rag/datasource/vdb/relyt/relyt_vector.py
@@ -199,10 +199,10 @@ class RelytVector(BaseVector):
        if ids:
            self.delete_by_uuids(ids)

-    def delete_by_ids(self, doc_ids: list[str]) -> None:
+    def delete_by_ids(self, ids: list[str]) -> None:

        with Session(self.client) as session:
-            ids_str = ','.join(f"'{doc_id}'" for doc_id in doc_ids)
+            ids_str = ','.join(f"'{doc_id}'" for doc_id in ids)
            select_statement = sql_text(
                f"""SELECT id FROM "{self._collection_name}" WHERE metadata->>'doc_id' in ({ids_str}); """
            )