phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

(fix) proxy - self.connect() for get_data()

Browse source
This commit is contained in:
ishaan-jaff 2024-01-05 20:47:50 +05:30
parent b4901e6dea
commit 898c072103
1 changed files with 10 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

10
litellm/proxy/utils.py
View file

@ -345,15 +345,21 @@ class PrismaClient:
user_id: Optional[str] = None, user_id: Optional[str] = None,
): ):
try: try:
print_verbose("PrismaClient: get_data")
# incase prisma is not connected
if self.db.is_connected() == False:
await self.connect()
response = None response = None
if token is not None: if token is not None:
# check if plain text or hash # check if plain text or hash
hashed_token = token hashed_token = token
if token.startswith("sk-"): if token.startswith("sk-"):
hashed_token = self.hash_token(token=token) hashed_token = self.hash_token(token=token)
print_verbose("PrismaClient: find_unique")
response = await self.db.litellm_verificationtoken.find_unique( response = await self.db.litellm_verificationtoken.find_unique(
where={"token": hashed_token} where={"token": hashed_token}
) )
print_verbose(f"PrismaClient: response={response}")
if response: if response:
# Token exists, now check expiration. # Token exists, now check expiration.
if response.expires is not None and expires is not None: if response.expires is not None and expires is not None:
@ -381,6 +387,10 @@ class PrismaClient:
) )
return response return response
except Exception as e: except Exception as e:
print_verbose(f"LiteLLM Prisma Client Exception: {e}")
import traceback
traceback.print_exc()
asyncio.create_task( asyncio.create_task(
self.proxy_logging_obj.failure_handler(original_exception=e) self.proxy_logging_obj.failure_handler(original_exception=e)
) )