update delete embeddings by id (#6489)

2 years ago · f73a3a58ae
parent 1e0e573165
commit f73a3a58ae
1 changed files with 4 additions and 3 deletions
--- a/api/schedule/clean_embedding_cache_task.py
+++ b/api/schedule/clean_embedding_cache_task.py
@ -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: