diff --git a/litellm/types/router.py b/litellm/types/router.py index d545f7ae639..14c0be1fb07 100644 --- a/litellm/types/router.py +++ b/litellm/types/router.py @@ -637,8 +637,7 @@ class LiteLLMParamsTypedDict(TypedDict, total=False): class DeploymentTypedDict(TypedDict, total=False): model_name: Required[str] litellm_params: Required[LiteLLMParamsTypedDict] - model_info: dict - + model_info: dict | ModelInfo SPECIAL_MODEL_INFO_PARAMS = tuple(MirroredPricingParams.model_fields)