test fix - import loc

litellm/proxy/client/cli/interface.py

@@ -101,7 +101,7 @@ def show_commands():
 
 def setup_shell(ctx: click.Context):
     """Set up the interactive shell with banner and initial info."""
-    from .banner import show_banner
+    from litellm.proxy.common_utils.banner import show_banner
     
     show_banner()
     
@@ -123,7 +123,7 @@ def handle_special_commands(user_input: str) -> bool:
         return True
     elif user_input.lower() == "clear":
         click.clear()
-        from .banner import show_banner
+        from litellm.proxy.common_utils.banner import show_banner
         show_banner()
         show_commands()
         return True

litellm/proxy/common_utils/banner.py

@@ -1,5 +1,3 @@
-# third party imports
-import click
 
 # LiteLLM ASCII banner
 LITELLM_BANNER = """   ██╗     ██╗████████╗███████╗██╗     ██╗     ███╗   ███╗
@@ -12,4 +10,8 @@ LITELLM_BANNER = """   ██╗     ██╗████████╗██
 
 def show_banner():
     """Display the LiteLLM CLI banner."""
-    click.echo(f"\n{LITELLM_BANNER}\n") 
+    try:
+      import click
+      click.echo(f"\n{LITELLM_BANNER}\n") 
+    except ImportError:
+      print("\n") # noqa: T201

litellm/proxy/common_utils/html_forms/cli_sso_success.py

@@ -1,5 +1,4 @@
-
-from litellm.proxy.client.cli.banner import LITELLM_BANNER
+from litellm.proxy.common_utils.banner import LITELLM_BANNER
 
 
 def render_cli_sso_success_page() -> str:

litellm/proxy/proxy_server.py

@@ -2977,7 +2977,7 @@ async def initialize(  # noqa: PLR0915
     config=None,
 ):
     global user_model, user_api_base, user_debug, user_detailed_debug, user_user_max_tokens, user_request_timeout, user_temperature, user_telemetry, user_headers, experimental, llm_model_list, llm_router, general_settings, master_key, user_custom_auth, prisma_client
-    from litellm.proxy.client.cli.banner import show_banner
+    from litellm.proxy.common_utils.banner import show_banner
     show_banner()
     if os.getenv("LITELLM_DONT_SHOW_FEEDBACK_BOX", "").lower() != "true":
         generate_feedback_box()