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

fix(proxy/utils.py): don't keep connecting to db if connection already established

Browse source
This commit is contained in:
Krrish Dholakia 2024-01-03 17:43:44 +05:30
parent f2da345173
commit 469ae0a378
2 changed files with 7 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

2
litellm/proxy/proxy_server.py
View file

@ -1014,7 +1014,7 @@ async def startup_event():
) # start the background health check coroutine. ) # start the background health check coroutine.
print_verbose(f"prisma client - {prisma_client}") print_verbose(f"prisma client - {prisma_client}")
if prisma_client: if prisma_client is not None:
await prisma_client.connect() await prisma_client.connect()
if prisma_client is not None and master_key is not None: if prisma_client is not None and master_key is not None:

8
litellm/proxy/utils.py
View file

@ -255,7 +255,7 @@ class PrismaClient:
) )
## init logging object ## init logging object
self.proxy_logging_obj = proxy_logging_obj self.proxy_logging_obj = proxy_logging_obj
self.connected = False
os.environ["DATABASE_URL"] = database_url os.environ["DATABASE_URL"] = database_url
# Save the current working directory # Save the current working directory
original_dir = os.getcwd() original_dir = os.getcwd()
@ -505,7 +505,11 @@ class PrismaClient:
) )
async def connect(self): async def connect(self):
try: try:
await self.db.connect() if self.connected == False:
await self.db.connect()
self.connected = True
else:
return
except Exception as e: except Exception as e:
asyncio.create_task( asyncio.create_task(
self.proxy_logging_obj.failure_handler(original_exception=e) self.proxy_logging_obj.failure_handler(original_exception=e)