add debugging utils to print when connecting to prisma

litellm/proxy/common_utils/debug_utils.py

@@ -1,9 +1,11 @@
 # Start tracing memory allocations
+import json
 import os
 import tracemalloc
 
 from fastapi import APIRouter
 
+import litellm
 from litellm._logging import verbose_proxy_logger
 
 router = APIRouter()
@@ -130,3 +132,70 @@ async def get_otel_spans():
         "spans_grouped_by_parent": spans_grouped_by_parent,
         "most_recent_parent": most_recent_parent,
     }
+
+
+# Helper functions for debugging
+def init_verbose_loggers():
+    worker_config = litellm.get_secret("WORKER_CONFIG")
+    if os.path.isfile(worker_config):
+        return
+    # if not, assume it's a json string
+    _settings = json.loads(os.getenv("WORKER_CONFIG"))
+    if not isinstance(_settings, dict):
+        return
+
+    debug = _settings.get("debug", None)
+    detailed_debug = _settings.get("detailed_debug", None)
+    if debug is True:  # this needs to be first, so users can see Router init debugg
+        import logging
+
+        from litellm._logging import (
+            verbose_logger,
+            verbose_proxy_logger,
+            verbose_router_logger,
+        )
+
+        # this must ALWAYS remain logging.INFO, DO NOT MODIFY THIS
+        verbose_logger.setLevel(level=logging.INFO)  # sets package logs to info
+        verbose_router_logger.setLevel(level=logging.INFO)  # set router logs to info
+        verbose_proxy_logger.setLevel(level=logging.INFO)  # set proxy logs to info
+    if detailed_debug == True:
+        import logging
+
+        from litellm._logging import (
+            verbose_logger,
+            verbose_proxy_logger,
+            verbose_router_logger,
+        )
+
+        verbose_logger.setLevel(level=logging.DEBUG)  # set package log to debug
+        verbose_router_logger.setLevel(level=logging.DEBUG)  # set router logs to debug
+        verbose_proxy_logger.setLevel(level=logging.DEBUG)  # set proxy logs to debug
+    elif debug == False and detailed_debug == False:
+        # users can control proxy debugging using env variable = 'LITELLM_LOG'
+        litellm_log_setting = os.environ.get("LITELLM_LOG", "")
+        if litellm_log_setting != None:
+            if litellm_log_setting.upper() == "INFO":
+                import logging
+
+                from litellm._logging import verbose_proxy_logger, verbose_router_logger
+
+                # this must ALWAYS remain logging.INFO, DO NOT MODIFY THIS
+
+                verbose_router_logger.setLevel(
+                    level=logging.INFO
+                )  # set router logs to info
+                verbose_proxy_logger.setLevel(
+                    level=logging.INFO
+                )  # set proxy logs to info
+            elif litellm_log_setting.upper() == "DEBUG":
+                import logging
+
+                from litellm._logging import verbose_proxy_logger, verbose_router_logger
+
+                verbose_router_logger.setLevel(
+                    level=logging.DEBUG
+                )  # set router logs to info
+                verbose_proxy_logger.setLevel(
+                    level=logging.DEBUG
+                )  # set proxy logs to debug