From 3ca45a412cb52176c8102e252a80e3b959e0457c Mon Sep 17 00:00:00 2001
From: mubashir1osmani <mubashir.osmani777@gmail.com>
Date: Mon, 27 Apr 2026 17:35:26 -0400
Subject: [PATCH] add include_model_info flag

---
 litellm/proxy/proxy_server.py | 10 ++++++++++
 litellm/proxy/utils.py        | 14 ++++++++++++++
 2 files changed, 24 insertions(+)

diff --git a/litellm/proxy/proxy_server.py b/litellm/proxy/proxy_server.py
index 3982d096e69..209e0b4a258 100644
--- a/litellm/proxy/proxy_server.py
+++ b/litellm/proxy/proxy_server.py
@@ -7025,6 +7025,7 @@ async def model_list(
     include_metadata: Optional[bool] = False,
     fallback_type: Optional[str] = None,
     scope: Optional[str] = None,
+    include_model_info: Optional[bool] = False,
 ):
     """
     Use `/model/info` - to get detailed model information, example - pricing, mode, etc.
@@ -7038,6 +7039,11 @@ async def model_list(
     - scope: Optional scope parameter. Currently only accepts "expand".
              When scope=expand is passed, proxy admins, team admins, and org admins
              will receive all proxy models as if they are a proxy admin.
+    - include_model_info: When true, embeds per-model-group info under a
+             "model_info" key on each entry (mode, max_input_tokens,
+             max_output_tokens, input/output cost per token, supports_*
+             flags, tpm/rpm). Sourced from the same router data as
+             /model_group/info.
     """
     global llm_model_list, general_settings, llm_router, prisma_client, user_api_key_cache, proxy_logging_obj
 
@@ -7107,6 +7113,7 @@ async def model_list(
                 include_metadata=include_metadata or False,
                 fallback_type=fallback_type,
                 llm_router=llm_router,
+                include_model_info=include_model_info or False,
             )
             model_data.append(model_info)
 
@@ -7140,6 +7147,7 @@ async def model_list(
             include_metadata=include_metadata or False,
             fallback_type=fallback_type,
             llm_router=llm_router,
+            include_model_info=include_model_info or False,
         )
         model_data.append(model_info)
 
@@ -7162,6 +7170,7 @@ async def model_list(
 async def model_info(
     model_id: str,
     user_api_key_dict: UserAPIKeyAuth = Depends(user_api_key_auth),
+    include_model_info: Optional[bool] = False,
 ):
     """
     Retrieve information about a specific model accessible to your API key.
@@ -7219,6 +7228,7 @@ async def model_info(
         include_metadata=False,
         fallback_type=None,
         llm_router=llm_router,
+        include_model_info=include_model_info or False,
     )
 
 
diff --git a/litellm/proxy/utils.py b/litellm/proxy/utils.py
index 712853a33c4..4c7fd29e878 100644
--- a/litellm/proxy/utils.py
+++ b/litellm/proxy/utils.py
@@ -5691,6 +5691,7 @@ def create_model_info_response(
     include_metadata: bool = False,
     fallback_type: Optional[str] = None,
     llm_router: Optional["Router"] = None,
+    include_model_info: bool = False,
 ) -> dict:
     """
     Create a standardized model info response.
@@ -5701,6 +5702,10 @@ def create_model_info_response(
         include_metadata: Whether to include metadata
         fallback_type: Type of fallbacks to include
         llm_router: LiteLLM router instance
+        include_model_info: When true, merges per-model-group info
+            (mode, max_input_tokens, max_output_tokens, input/output cost,
+            supports_* flags, tpm/rpm, etc.) under a top-level "model_info"
+            key. Pulled from llm_router.get_model_group_info().
 
     Returns:
         Dictionary containing model information
@@ -5714,6 +5719,15 @@ def create_model_info_response(
         "owned_by": provider,
     }
 
+    if include_model_info and llm_router is not None:
+        try:
+            group_info = llm_router.get_model_group_info(model_group=model_id)
+        except Exception:
+            group_info = None
+        model_info["model_info"] = (
+            group_info.model_dump(exclude_none=True) if group_info is not None else {}
+        )
+
     # Add metadata if requested
     if include_metadata:
         metadata = {}