Browse Source

Add the loop on index worker

jherve 1 year ago
parent
commit
19e949fe18
1 changed files with 8 additions and 7 deletions
  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