feat: update

Author: earayu

aperag/llm/litellm_logging.py

@@ -25,7 +25,12 @@ def setup_litellm_logging():
 
     litellm.suppress_debug_info = True
     litellm.disable_streaming_logging = True
-    logging.info("LiteLLM debug info suppression is enabled.")
+
+    # Disable callbacks to prevent LoggingWorker event loop issues in Celery tasks
+    # See: https://github.com/BerriAI/litellm/issues/14521
+    litellm.callbacks = []
+
+    logging.info("LiteLLM debug info suppression is enabled and callbacks are disabled.")
 
     # Filter Pydantic serialization warnings globally
     import warnings

aperag/llm/litellm_track.py

@@ -15,7 +15,6 @@
 import logging
 from datetime import datetime
 
-import litellm
 from litellm.integrations.custom_logger import CustomLogger
 
 logger = logging.getLogger(__name__)
@@ -76,4 +75,7 @@ def log_post_api_call_for_completion(self, kwargs, response_obj, start_time, end
 
 
 def register_custom_llm_track():
-    litellm.callbacks = [MyCustomHandler()]
+    # Disabled to prevent LoggingWorker event loop issues in Celery tasks
+    # See: https://github.com/BerriAI/litellm/issues/14521
+    # litellm.callbacks = [MyCustomHandler()]
+    pass