Forráskód Böngészése

fix: error in migrate_annotation_vector_database when exec vdb-migrate (#15937)

Co-authored-by: crazywoola <427733928@qq.com>
Naoki KOBAYASHI 1 hónapja
szülő
commit
e91531fc23
1 módosított fájl, 8 hozzáadás és 2 törlés
  1. 8 2
      api/commands.py

+ 8 - 2
api/commands.py

@@ -160,11 +160,17 @@ def migrate_annotation_vector_database():
     while True:
         try:
             # get apps info
+            per_page = 50
             apps = (
-                App.query.filter(App.status == "normal")
+                db.session.query(App)
+                .filter(App.status == "normal")
                 .order_by(App.created_at.desc())
-                .paginate(page=page, per_page=50)
+                .limit(per_page)
+                .offset((page - 1) * per_page)
+                .all()
             )
+            if not apps:
+                break
         except NotFound:
             break