phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-27 11:43:54 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

(fix) re-connect prisma if not connected

Browse source
This commit is contained in:
ishaan-jaff 2024-01-05 17:58:23 +05:30
parent 06ef89f65a
commit 9d4ce57801
1 changed files with 11 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

11
litellm/proxy/utils.py
View file

@ -395,6 +395,10 @@ class PrismaClient:
Add a key to the database. If it already exists, do nothing. Add a key to the database. If it already exists, do nothing.
""" """
try: try:
# incase prisma is not connected
if self.db.is_connected() == False:
await self.connect()
if table_name == "user+key": if table_name == "user+key":
token = data["token"] token = data["token"]
hashed_token = self.hash_token(token=token) hashed_token = self.hash_token(token=token)
@ -473,6 +477,10 @@ class PrismaClient:
Update existing data Update existing data
""" """
try: try:
# incase prisma is not connected
if self.db.is_connected() == False:
await self.connect()
db_data = self.jsonify_object(data=data) db_data = self.jsonify_object(data=data)
if token is not None: if token is not None:
print_verbose(f"token: {token}") print_verbose(f"token: {token}")
@ -515,6 +523,9 @@ class PrismaClient:
Allow user to delete a key(s) Allow user to delete a key(s)
""" """
try: try:
# incase prisma is not connected
if self.db.is_connected() == False:
await self.connect()
hashed_tokens = [self.hash_token(token=token) for token in tokens] hashed_tokens = [self.hash_token(token=token) for token in tokens]
await self.db.litellm_verificationtoken.delete_many( await self.db.litellm_verificationtoken.delete_many(
where={"token": {"in": hashed_tokens}} where={"token": {"in": hashed_tokens}}