jyapayne/dify
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

test: improve vector store tests (#3855)

Browse source
This commit is contained in:
Bowen Liang 2024-04-26 19:18:42 +08:00 • committed by GitHub
commit 045827043d
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
6 changed files with 58 additions and 74 deletions
Download patch file Download diff file Expand all files Collapse all files

4
api/core/rag/datasource/vdb/relyt/relyt_vector.py
View file

@ -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}); """
)