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

(fix) error cli users see when importing enterprise folder

Browse source
This commit is contained in:
ishaan-jaff 2024-03-15 08:10:11 -07:00
parent f07a652148
commit fa1676b253
2 changed files with 7 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

1
enterprise/__init__.py Normal file
View file

@ -0,0 +1 @@
from . import *

8
litellm/proxy/proxy_server.py
View file

@ -144,9 +144,13 @@ from typing import Union
try:
# when using litellm cli
import litellm.proxy.enterprise as enterprise
except:
except Exception as e:
# when using litellm docker image
import enterprise # type: ignore
try:
import enterprise # type: ignore
except Exception as e:
print("Error importing enterprise: " + str(e)) # noqa
pass
ui_link = f"/ui/"
ui_message = (