From fa1676b253fa8f861e57cc6b031581c270cf9497 Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Fri, 15 Mar 2024 08:10:11 -0700 Subject: [PATCH] (fix) error cli users see when importing enterprise folder --- enterprise/__init__.py | 1 + litellm/proxy/proxy_server.py | 8 ++++++-- 2 files changed, 7 insertions(+), 2 deletions(-) create mode 100644 enterprise/__init__.py 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 = (