From f73a3a58ae61b31e32984a902560a93b909aa639 Mon Sep 17 00:00:00 2001
From: Jyong <76649700+JohnJyong@users.noreply.github.com>
Date: Sat, 20 Jul 2024 09:04:21 +0800
Subject: [PATCH] update delete embeddings by id (#6489)

---
 api/schedule/clean_embedding_cache_task.py | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/api/schedule/clean_embedding_cache_task.py b/api/schedule/clean_embedding_cache_task.py
index 49b1be3a3..ccc106226 100644
--- 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: