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

fix(proxy_server.py): don't reconnect prisma if already connected

Browse source
This commit is contained in:
Krrish Dholakia 2024-01-09 11:38:42 +05:30
parent e97eff4243
commit cd350ab8d8
3 changed files with 8 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

5
litellm/tests/test_proxy_startup.py
View file

@ -19,7 +19,8 @@ from litellm.proxy.proxy_server import (
save_worker_config,
initialize,
startup_event,
llm_model_list
llm_model_list,
shutdown_event
)
def test_proxy_gunicorn_startup_direct_config():
@ -36,6 +37,7 @@ def test_proxy_gunicorn_startup_direct_config():
config_fp = f"{filepath}/test_configs/test_config_no_auth.yaml"
os.environ["WORKER_CONFIG"] = config_fp
asyncio.run(startup_event())
asyncio.run(shutdown_event())
except Exception as e:
if "Already connected to the query engine" in str(e):
pass
@ -51,6 +53,7 @@ def test_proxy_gunicorn_startup_config_dict():
worker_config = {"config": config_fp}
os.environ["WORKER_CONFIG"] = json.dumps(worker_config)
asyncio.run(startup_event())
asyncio.run(shutdown_event())
except Exception as e:
if "Already connected to the query engine" in str(e):
pass