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

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
commit f73a3a58ae
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 4 additions and 3 deletions
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: