Skip to content

Commit

Permalink
fix(proxy/utils.py): fix isoformat to string logic
Browse files Browse the repository at this point in the history
  • Loading branch information
krrishdholakia committed Jan 18, 2024
1 parent 7db04af commit 96122a4
Show file tree
Hide file tree
Showing 2 changed files with 13 additions and 10 deletions.
7 changes: 4 additions & 3 deletions litellm/proxy/proxy_server.py
Original file line number Diff line number Diff line change
Expand Up @@ -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 (
Expand Down
16 changes: 9 additions & 7 deletions litellm/proxy/utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -370,19 +370,21 @@ async def get_data(
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(
Expand Down

0 comments on commit 96122a4

Please sign in to comment.