Ver Fonte

update delete embeddings by id (#6489)

Jyong há 9 meses atrás
pai
commit
f73a3a58ae
1 ficheiros alterados com 4 adições e 3 exclusões
  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: