diff --git a/litellm/proxy/proxy_server.py b/litellm/proxy/proxy_server.py index 27576bf51a..d3667892bb 100644 --- a/litellm/proxy/proxy_server.py +++ b/litellm/proxy/proxy_server.py @@ -331,9 +331,10 @@ async def user_api_key_auth( f"LLM Model List pre access group check: {llm_model_list}" ) access_groups = [] - for m in llm_model_list: - for group in m.get("model_info", {}).get("access_groups", []): - access_groups.append((m["model_name"], group)) + if llm_model_list is not None: + for m in llm_model_list: + for group in m.get("model_info", {}).get("access_groups", []): + access_groups.append((m["model_name"], group)) allowed_models = valid_token.models if ( diff --git a/litellm/proxy/utils.py b/litellm/proxy/utils.py index e7191cafce..ab1fea463e 100644 --- a/litellm/proxy/utils.py +++ b/litellm/proxy/utils.py @@ -370,19 +370,21 @@ class PrismaClient: response = await self.db.litellm_verificationtoken.find_unique( where={"token": hashed_token} ) + if response is not None: + # for prisma we need to cast the expires time to str + if isinstance(response.expires, datetime): + response.expires = response.expires.isoformat() elif query_type == "find_all" and user_id is not None: response = await self.db.litellm_verificationtoken.find_many( where={"user_id": user_id} ) + if response is not None and len(response) > 0: + for r in response: + if isinstance(r.expires, datetime): + r.expires = r.expires.isoformat() print_verbose(f"PrismaClient: response={response}") if response is not None: - if isinstance(response, LiteLLM_VerificationToken): - # for prisma we need to cast the expires time to str - if isinstance(response.expires, datetime): - response.expires = response.expires.isoformat() - return response - else: - return response + return response else: # Token does not exist. raise HTTPException(