Browse Source

update delete embeddings by id (#6489)

tags/0.6.15
Jyong 1 year ago
parent
commit
f73a3a58ae
No account linked to committer's email address
1 changed files with 4 additions and 3 deletions
  1. 4
    3
      api/schedule/clean_embedding_cache_task.py

+ 4
- 3
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:

Loading…
Cancel
Save