revert

LoadBalancer.py

@@ -14,7 +14,7 @@ CACHE_DIR = os.getenv("CACHE_DIR")
 
 class LoadBalancer:
     def __init__(self, cache_dir, token, repo, polling_interval=4, max_retries=3, initial_delay=1):
-        self.version = "0.0.0.1 Alpha"
+        self.version = "0.0.0.2 Alpha"
         self.instances = []
         self.instances_health = {}
         self.polling_interval = polling_interval
@@ -34,8 +34,9 @@ class LoadBalancer:
         if not os.path.exists(self.CACHE_DIR):
             os.makedirs(self.CACHE_DIR)
 
-        # Initialize file structure
+        # Initialize file structure and prefetching
         self.file_structure = indexer()
+        self.start_prefetching()
 
         # Start polling and file checking in separate threads
         polling_thread = Thread(target=self.start_polling)
@@ -47,9 +48,9 @@ class LoadBalancer:
         content_check_thread.daemon = True
         content_check_thread.start()
 
-    async def start_prefetching(self):
-        """Start the metadata prefetching."""
-        await self.prefetch_metadata()
+    def start_prefetching(self):
+        """Start the metadata prefetching in the FastAPI event loop."""
+        asyncio.create_task(self.prefetch_metadata())
 
     async def prefetch_metadata(self):
         """Prefetch metadata for all items in the file structure."""
@@ -89,8 +90,7 @@ class LoadBalancer:
             time.sleep(120)  # Wait for 2 minutes
             current_file_structure = indexer()
             if current_file_structure != self.previous_file_structure:
-                # Schedule the prefetching task
-                asyncio.get_event_loop().create_task(self.start_prefetching())
+                self.start_prefetching()
                 self.previous_file_structure = current_file_structure
             else:
                 logging.info("No new content detected.")