From e493ce99817fdd5ebfc9be36170431e5fd6de579 Mon Sep 17 00:00:00 2001
From: Jyong <76649700+JohnJyong@users.noreply.github.com>
Date: Thu, 18 Jul 2024 20:25:28 +0800
Subject: [PATCH] update clean embedding cache logic (#6434)

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

diff --git a/api/schedule/clean_embedding_cache_task.py b/api/schedule/clean_embedding_cache_task.py
index 3d49b487c6..f68c54600a 100644
--- a/api/schedule/clean_embedding_cache_task.py
+++ b/api/schedule/clean_embedding_cache_task.py
@@ -16,16 +16,14 @@ def clean_embedding_cache_task():
     clean_days = int(dify_config.CLEAN_DAY_SETTING)
     start_at = time.perf_counter()
     thirty_days_ago = datetime.datetime.now() - datetime.timedelta(days=clean_days)
-    page = 1
     while True:
         try:
             embeddings = db.session.query(Embedding).filter(Embedding.created_at < thirty_days_ago) \
-                .order_by(Embedding.created_at.desc()).paginate(page=page, per_page=100)
+                .order_by(Embedding.created_at.desc()).limit(100).all()
         except NotFound:
             break
         for embedding in embeddings:
             db.session.delete(embedding)
         db.session.commit()
-        page += 1
     end_at = time.perf_counter()
     click.echo(click.style('Cleaned embedding cache from db success latency: {}'.format(end_at - start_at), fg='green'))