AI/dify
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git synced 2025-06-04 11:14:10 +08:00
Code Issues Packages Projects Releases Wiki Activity

update delete embeddings by id (#6489)

Browse Source
This commit is contained in:
Jyong 2024-07-20 09:04:21 +08:00 committed by GitHub
parent 1e0e573165
commit f73a3a58ae
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
api/schedule/clean_embedding_cache_task.py
View File

@ -25,9 +25,10 @@ def clean_embedding_cache_task():
except NotFound:
break
if embedding_ids:
db.session.execute(text(
"DELETE FROM embeddings WHERE id in :embedding_ids"
), {'embedding_ids': tuple(embedding_ids)})
for embedding_id in embedding_ids:
db.session.execute(text(
"DELETE FROM embeddings WHERE id = :embedding_id"
), {'embedding_id': embedding_id})
db.session.commit()
else: