fix: fix pyrightI

This commit is contained in:
Ishaan Jaffer 2026-01-20 12:44:29 -08:00
parent 2e690fce3f
commit f8ee2f1a03
2 changed files with 15 additions and 0 deletions

View file

@ -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
#########################################################

View file

@ -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