瀏覽代碼

Add the loop on index worker

jherve 1 年之前
父節點
當前提交
19e949fe18
共有 1 個文件被更改,包括 8 次插入7 次删除
  1. 8 7
      src/media_observer/test.py

+ 8 - 7
src/media_observer/test.py

@@ -384,15 +384,16 @@ class SimilarityIndexWorker(Worker):
     new_embeddings_event: asyncio.Event
 
     async def run(self):
-        await self.new_embeddings_event.wait()
+        while True:
+            await self.new_embeddings_event.wait()
 
-        sim_index = SimilaritySearch.create(self.storage)
-        logger.info("Starting index..")
-        await sim_index.add_embeddings()
-        await sim_index.save()
-        logger.info("Similarity index ready")
+            sim_index = SimilaritySearch.create(self.storage)
+            logger.info("Starting index..")
+            await sim_index.add_embeddings()
+            await sim_index.save()
+            logger.info("Similarity index ready")
 
-        self.new_embeddings_event.clear()
+            self.new_embeddings_event.clear()
 
 
 @frozen