From e03bc3ec7e889f1475368f2de8e499e4763a598a Mon Sep 17 00:00:00 2001 From: Krish Dholakia Date: Sat, 19 Jul 2025 12:33:57 -0700 Subject: [PATCH] feat(proxy_server.py): add model hub to the swagger (#12767) user request --- litellm/proxy/proxy_server.py | 12 ++++++++++-- 1 file changed, 10 insertions(+), 2 deletions(-) diff --git a/litellm/proxy/proxy_server.py b/litellm/proxy/proxy_server.py index fd8a32e9c6..d2424b9ed5 100644 --- a/litellm/proxy/proxy_server.py +++ b/litellm/proxy/proxy_server.py @@ -466,11 +466,14 @@ else: ) ui_link = f"{server_root_path}/ui/" +model_hub_link = f"{server_root_path}/ui/model_hub_table" ui_message = ( f"šŸ‘‰ [```LiteLLM Admin Panel on /ui```]({ui_link}). Create, Edit Keys with SSO" ) ui_message += "\n\nšŸ’ø [```LiteLLM Model Cost Map```](https://models.litellm.ai/)." +ui_message += f"\n\nšŸ”Ž [```LiteLLM Model Hub```]({model_hub_link}). See available models on the proxy. [**Docs**](https://docs.litellm.ai/docs/proxy/model_hub)" + custom_swagger_message = "[**Customize Swagger Docs**](https://docs.litellm.ai/docs/proxy/enterprise#swagger-docs---custom-routes--branding)" ### CUSTOM BRANDING [ENTERPRISE FEATURE] ### @@ -1858,8 +1861,12 @@ class ProxyConfig: PrometheusLogger = None if PrometheusLogger is not None: - verbose_proxy_logger.debug("mounting metrics endpoint") - PrometheusLogger._mount_metrics_endpoint(premium_user) + verbose_proxy_logger.debug( + "mounting metrics endpoint" + ) + PrometheusLogger._mount_metrics_endpoint( + premium_user + ) print( # noqa f"{blue_color_code} Initialized Success Callbacks - {litellm.success_callback} {reset_color_code}" ) # noqa @@ -3513,6 +3520,7 @@ class ProxyStartupEvent: if litellm.prometheus_initialize_budget_metrics is True: try: from litellm_enterprise.integrations.prometheus import PrometheusLogger + PrometheusLogger.initialize_budget_metrics_cron_job(scheduler=scheduler) except Exception: PrometheusLogger = None