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(proxy_cli.py): ensure proxy always retries if db push fails to connect to db

Browse source
This commit is contained in:
Krrish Dholakia 2024-01-17 17:37:59 -08:00
parent 7a80df83b2
commit 3bdfb2bc6d
1 changed files with 15 additions and 13 deletions
Download patch file Download diff file Expand all files Collapse all files

8
litellm/proxy/proxy_cli.py
View file

@ -405,6 +405,7 @@ def run_server(
is_prisma_runnable = False is_prisma_runnable = False
if is_prisma_runnable: if is_prisma_runnable:
for _ in range(4):
# run prisma db push, before starting server # run prisma db push, before starting server
# Save the current working directory # Save the current working directory
original_dir = os.getcwd() original_dir = os.getcwd()
@ -413,9 +414,10 @@ def run_server(
dname = os.path.dirname(abspath) dname = os.path.dirname(abspath)
os.chdir(dname) os.chdir(dname)
try: try:
subprocess.run( subprocess.run(["prisma", "db", "push", "--accept-data-loss"])
["prisma", "db", "push", "--accept-data-loss"] break # Exit the loop if the subprocess succeeds
) # this looks like a weird edge case when prisma just wont start on render. we need to have the --accept-data-loss except subprocess.CalledProcessError as e:
print(f"Error: {e}")
finally: finally:
os.chdir(original_dir) os.chdir(original_dir)
else: else: