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

(fix) prisma always installed on deploys

Browse source
This commit is contained in:
ishaan-jaff 2023-11-21 17:27:07 -08:00
parent dcc60ae31a
commit 410c059d52
1 changed files with 11 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

5
litellm/proxy/proxy_server.py
View file

@ -207,15 +207,16 @@ async def user_api_key_auth(request: Request):
def prisma_setup(database_url: Optional[str]): def prisma_setup(database_url: Optional[str]):
global prisma_client global prisma_client
if database_url: if database_url:
try:
import os import os
print("LiteLLM: DATABASE_URL Set in config, trying to 'pip install prisma'") print("LiteLLM: DATABASE_URL Set in config, trying to 'pip install prisma'")
os.environ["DATABASE_URL"] = database_url os.environ["DATABASE_URL"] = database_url
subprocess.run(['pip', 'install', 'prisma'])
subprocess.run(['python3', '-m', 'pip', 'install', 'prisma'])
subprocess.run(['prisma', 'db', 'push']) subprocess.run(['prisma', 'db', 'push'])
# Now you can import the Prisma Client # Now you can import the Prisma Client
from prisma import Client from prisma import Client
prisma_client = Client() prisma_client = Client()
except Exception as e:
print("Error when initializing prisma, Ensure you run pip install prisma", e)
def rq_setup(use_queue: bool): def rq_setup(use_queue: bool):
global request_queue, redis_connection, redis_job global request_queue, redis_connection, redis_job