Make all commands show server URL (#10801)

* Make all commands show server URL

so users know which server they're accessing

* Write msg to stderr & only if it's a tty
This commit is contained in:
Marc Abramowitz 2025-06-09 17:37:40 -07:00 committed by GitHub
parent cd8ec4556f
commit ad9b79259f
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -1,4 +1,5 @@
# stdlib imports
import sys
from typing import Optional
# third party imports
@ -60,6 +61,8 @@ def print_version(base_url: str, api_key: Optional[str]):
def cli(ctx: click.Context, base_url: str, api_key: Optional[str]) -> None:
"""LiteLLM Proxy CLI - Manage your LiteLLM proxy server"""
ctx.ensure_object(dict)
if sys.stderr.isatty():
click.secho(f"Accessing LiteLLM server: {base_url} ...\n", fg="yellow", err=True)
ctx.obj["base_url"] = base_url
ctx.obj["api_key"] = api_key