修复刷新页面索引时的逻辑错误

api/model/embedding_search/title_collection.py

@@ -14,7 +14,7 @@ class TitleCollectionModel(BaseModel):
     page_id: Mapped[int] = mapped_column(sqlalchemy.Integer, index=True, nullable=True)
 
 class TitleCollectionHelper(BaseHelper):
-    async def add(self, title: str, page_id: Optional[int] = None) -> Union[int, bool]:
+    async def add(self, title: str, page_id: Optional[int] = None) -> TitleCollectionModel | None:
         stmt = select(TitleCollectionModel.id).where(TitleCollectionModel.title == title)
         result = await self.session.scalar(stmt)
 

maintenance/update_title_index.py

@@ -1,4 +1,5 @@
 import asyncio
+import sys
 import base as _
 import local
 from service.database import DatabaseService
@@ -10,9 +11,12 @@ async def main():
     dbs = await DatabaseService.create()
     mw_api = MediaWikiApi.create()
 
+    start_title = sys.argv[1] if len(sys.argv) > 1 else None
+
     continue_key = None
     while True:
-        page_res = await mw_api.get_all_pages(continue_key)
+        page_res = await mw_api.get_all_pages(continue_key, start_title)
+        start_title = None
 
         title_list = page_res["title_list"]
 

service/embedding_search.py

@@ -160,7 +160,7 @@ class EmbeddingSearchService:
             # Collection changed, remove old index
             if self.collection_id != self.title_index.collection_id:
                 async with PageIndexHelper(self.dbs, self.title_index.collection_id) as old_page_index:
-                    await self.page_index.init_table()
+                    await old_page_index.init_table()
                     old_page_index.remove_by_page_id(self.page_id)
                 self.title_index.collection_id = self.collection_id