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

fix(proxy_server.py): use tomli instead of tomllib

Browse source
This commit is contained in:
Krrish Dholakia 2023-10-11 10:43:42 -07:00
parent 6e2e020428
commit 5e5466dfaf
1 changed files with 5 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

7
litellm/proxy/proxy_server.py
View file

@ -1,5 +1,4 @@
import sys, os, platform, appdirs import sys, os, platform, appdirs
import tomllib
import shutil, random, traceback import shutil, random, traceback
sys.path.insert( sys.path.insert(
0, os.path.abspath("../..") 0, os.path.abspath("../..")
@ -8,11 +7,15 @@ sys.path.insert(
try: try:
import uvicorn import uvicorn
import fastapi import fastapi
import tomli as tomllib
except ImportError: except ImportError:
import subprocess import subprocess
import sys import sys
subprocess.check_call([sys.executable, "-m", "pip", "install", "uvicorn", "fastapi"]) subprocess.check_call([sys.executable, "-m", "pip", "install", "uvicorn", "fastapi", "tomli"])
import uvicorn
import fastapi
import tomli as tomllib
print() print()