Parcourir la source

update delete embeddings by id (#6489)

Jyong il y a 9 mois
Parent
commit
f73a3a58ae
1 fichiers modifiés avec 4 ajouts et 3 suppressions
  1. 4 3
      api/schedule/clean_embedding_cache_task.py

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