o ưi@sXddlmZmZmZddlZddlmZmZddlm Z ddl m Z GdddeZ dS))ListLiteralOptionalN)BaseLLMModelInfoBaseTokenCounter)get_secret_str)AllMessageValuesc@sPeZdZdZd#deefddZedededfdd Z ed#d eedeefd d Z ed#d eedeefddZ e d#deedeefddZ deefddZ d$d eed eedeefddZededefddZededefddZedefddZ  d$dededeeded ed eed eedefd!d"ZdS)%AzureFoundryModelInfoz/Model info for Azure AI / Azure Foundry models.NmodelcCs ||_dSN)_model)selfr rY/home/app/Keep/.python/lib/python3.10/site-packages/litellm/llms/azure_ai/common_utils.py__init__ s zAzureFoundryModelInfo.__init__return)agents model_routerdefaultcCs<d|vrdS|}d|vsd|vsd|vsd|vrdSdS)aa Get the Azure AI route for the given model. Similar to BedrockModelInfo.get_bedrock_route(). Supported routes: - agents: azure_ai/agents/ - model_router: azure_ai/model_router/ or models with "model-router"/"model_router" in name - default: standard models zagents/r model_router/ model-router/z model-routerrr)lower)r Z model_lowerrrrget_azure_ai_routes z(AzureFoundryModelInfo.get_azure_ai_routeapi_basecCs|ptjptdS)NZAZURE_AI_API_BASE)litellmrr)rrrr get_api_base(sz"AzureFoundryModelInfo.get_api_baseapi_keycCs|p tjp tjp tdS)NZAZURE_AI_API_KEY)rrZ openai_keyr)rrrr get_api_key,sz!AzureFoundryModelInfo.get_api_key api_versioncCs|ptjptd}|S)NZAZURE_API_VERSION)rrr)r rrrrr5sz!AzureFoundryModelInfo.api_versioncCs*|jrd|jvrddlm}|SdS)z Factory method to create a token counter for Azure AI. Returns: AzureAIAnthropicTokenCounter for Claude models, None otherwise. clauder)AzureAIAnthropicTokenCounterN)r rZ:litellm.llms.azure_ai.anthropic.count_tokens.token_counterr )r r rrrget_token_counter<s z'AzureFoundryModelInfo.get_token_countercCsgS)z 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. r)r rrrrr get_modelsLs z AzureFoundryModelInfo.get_modelscCs4d|vr |dddSd|vr|dddS|S)at Strip the model_router prefix from model name. Examples: - "model_router/gpt-4o" -> "gpt-4o" - "model-router/gpt-4o" -> "gpt-4o" - "gpt-4o" -> "gpt-4o" Args: model: Model name potentially with model_router prefix Returns: Model name without the prefix rr)splitr rrrstrip_model_router_prefix[s z/AzureFoundryModelInfo.strip_model_router_prefixcCst|}|S)z Get the base model name, stripping any Azure AI routing prefixes. Args: model: Model name potentially with routing prefixes Returns: Base model name )r r&r%rrrget_base_modelqs z$AzureFoundryModelInfo.get_base_modelcCsTt|}|dkrddlm}|Sd|vr!ddlm}|Sddlm}|S)a Get the appropriate Azure AI config class for the given model. Routes to specialized configs based on model type: - Model Router: AzureModelRouterConfig - Claude models: AzureAnthropicConfig - Default: AzureAIStudioConfig Args: model: The model name Returns: The appropriate config instance rr)AzureModelRouterConfigr)AzureAnthropicConfig)AzureAIStudioConfig) r rZ7litellm.llms.azure_ai.azure_model_router.transformationr(rZ.litellm.llms.azure_ai.anthropic.transformationr)Z)litellm.llms.azure_ai.chat.transformationr*)r Zazure_ai_router(r)r*rrrget_azure_ai_config_for_models     z3AzureFoundryModelInfo.get_azure_ai_config_for_modelheadersmessagesoptional_paramslitellm_paramscCstd)z+Azure Foundry sends api key in query paramsz5Azure Foundry does not support environment validation)NotImplementedError)r r,r r-r.r/rrrrrvalidate_environments z*AzureFoundryModelInfo.validate_environmentr )NN)__name__ __module__ __qualname____doc__rstrr staticmethodrrrrpropertyrrr!rr"r&r'r+dictrr1rrrrr sZ & r ) typingrrrrZ litellm.llms.base_llm.base_utilsrrZlitellm.secret_managers.mainrZlitellm.types.llms.openairr rrrrs