From d62da29cbeb70f243f5b95a8a84ad2863d5b0b14 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Fri, 24 Nov 2023 15:46:25 -0800 Subject: [PATCH] fix: fix linting issues --- litellm/proxy/proxy_server.py | 2 ++ litellm/router.py | 2 +- 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/litellm/proxy/proxy_server.py b/litellm/proxy/proxy_server.py index 0a9b8d4d15..cb73687fed 100644 --- a/litellm/proxy/proxy_server.py +++ b/litellm/proxy/proxy_server.py @@ -790,6 +790,8 @@ async def delete_key_fn(request: Request): async def info_key_fn(key: str = fastapi.Query(..., description="Key in the request parameters")): global prisma_client try: + if prisma_client is None: + raise Exception(f"Database not connected. Connect a database to your proxy - https://docs.litellm.ai/docs/simple_proxy#managing-auth---virtual-keys") key_info = await prisma_client.litellm_verificationtoken.find_unique( where={ "token": key diff --git a/litellm/router.py b/litellm/router.py index 82199ee725..6f277cb5f5 100644 --- a/litellm/router.py +++ b/litellm/router.py @@ -81,7 +81,7 @@ class Router: self.allowed_fails = allowed_fails or litellm.allowed_fails self.failed_calls = InMemoryCache() # cache to track failed call per deployment, if num failed calls within 1 minute > allowed fails, then add it to cooldown - self.num_retries = num_retries or litellm.num_retries + self.num_retries = num_retries or litellm.num_retries or 0 self.set_verbose = set_verbose self.timeout = timeout or litellm.request_timeout self.routing_strategy = routing_strategy