diff --git a/litellm/llms/azure_ai/common_utils.py b/litellm/llms/azure_ai/common_utils.py index 281e0579c77..2c711c36a0f 100644 --- a/litellm/llms/azure_ai/common_utils.py +++ b/litellm/llms/azure_ai/common_utils.py @@ -59,6 +59,17 @@ class AzureFoundryModelInfo(BaseLLMModelInfo): return AzureAIAnthropicTokenCounter() return None + def get_models( + self, api_key: Optional[str] = None, api_base: Optional[str] = None + ) -> List[str]: + """ + Returns a list of models supported by Azure AI. + + Azure AI doesn't have a standard model listing endpoint, + so this returns an empty list. + """ + return [] + ######################################################### # Not implemented methods ######################################################### diff --git a/litellm/utils.py b/litellm/utils.py index 3e88c6fe9e3..948d6e97f4d 100644 --- a/litellm/utils.py +++ b/litellm/utils.py @@ -8244,6 +8244,10 @@ class ProviderConfigManager: return litellm.ClarifaiConfig() elif LlmProviders.BEDROCK == provider: return litellm.llms.bedrock.common_utils.BedrockModelInfo() + elif LlmProviders.AZURE_AI == provider: + from litellm.llms.azure_ai.common_utils import AzureFoundryModelInfo + + return AzureFoundryModelInfo(model=model) return None @staticmethod