diff --git a/enterprise/__init__.py b/enterprise/__init__.py new file mode 100644 index 000000000..b6e690fd5 --- /dev/null +++ b/enterprise/__init__.py @@ -0,0 +1 @@ +from . import * diff --git a/litellm/proxy/proxy_server.py b/litellm/proxy/proxy_server.py index 36254f426..718033c84 100644 --- a/litellm/proxy/proxy_server.py +++ b/litellm/proxy/proxy_server.py @@ -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 = (