fcaa4a9f30
* Add user mgmt functionality to client & CLI * Add user mgmt client & CLI tests * Add user mgmt client & CLI info to README.md's * lint fixes: litellm/proxy/client/users.py * Fix mypy errors
54 lines
1.2 KiB
Python
54 lines
1.2 KiB
Python
# stdlib imports
|
|
from typing import Optional
|
|
|
|
# third party imports
|
|
import click
|
|
|
|
# local imports
|
|
from .commands.models import models
|
|
from .commands.credentials import credentials
|
|
from .commands.chat import chat
|
|
from .commands.http import http
|
|
from .commands.keys import keys
|
|
from .commands.users import users
|
|
|
|
|
|
@click.group()
|
|
@click.option(
|
|
"--base-url",
|
|
envvar="LITELLM_PROXY_URL",
|
|
show_envvar=True,
|
|
default="http://localhost:4000",
|
|
help="Base URL of the LiteLLM proxy server",
|
|
)
|
|
@click.option(
|
|
"--api-key",
|
|
envvar="LITELLM_PROXY_API_KEY",
|
|
show_envvar=True,
|
|
help="API key for authentication",
|
|
)
|
|
@click.pass_context
|
|
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)
|
|
ctx.obj["base_url"] = base_url
|
|
ctx.obj["api_key"] = api_key
|
|
|
|
|
|
# Add the models command group
|
|
cli.add_command(models)
|
|
# Add the credentials command group
|
|
cli.add_command(credentials)
|
|
# Add the chat command group
|
|
cli.add_command(chat)
|
|
# Add the http command group
|
|
cli.add_command(http)
|
|
# Add the keys command group
|
|
cli.add_command(keys)
|
|
# Add the users command group
|
|
cli.add_command(users)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
cli()
|