uPj# UddlmZddlmZmZddlZddlmZmZddl m Z ddl m Z ddl mZedZdd eeefdzd edzfd Z dd edzd eeefdzd eded eeeff dZdZeed<GddeZy))Mapping)FinalLiteralN)BaseLLMModelInfoBaseTokenCounter)get_secret_str)AllMessageValues)GenericLiteLLMParams) Authorizationzapi-keyzApi-KeyzOcp-Apim-Subscription-Keylitellm_paramsreturnc`ddlm}|rtj|n t}||S)aY Resolve an Entra ID / OAuth access token for an Azure AI Foundry deployment. Accepts the same credential set as the `azure` provider: service principal (`tenant_id` / `client_id` / `client_secret`), a pre-fetched `azure_ad_token`, an OIDC federated token, username/password, or `DefaultAzureCredential` / managed identity. r)get_azure_ad_token)litellm.llms.azure.common_utilsrr model_validate)r rparamss g/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/llms/azure_ai/common_utils.pyget_azure_ai_entra_tokenr s,CDR ! 0 0 @XlXnF f %%api_keyapi_key_headerapi_key_env_varcl|r||dk(rd|iS|iSt|}|rdd|iStd|d)z Build the auth headers for an Azure AI Foundry route. Prefers the API key when one is configured, and otherwise falls back to Entra ID / OAuth, sending the access token as a bearer token. r zBearer )r z*:!;<<  FFWXH H r!azure_model_router_selected_model%AZURE_MODEL_ROUTER_SELECTED_MODEL_KEYceZdZdZddedzfdZedededfdZede ee fdzdedzfd Z e ddedzde ee fdzde fd Z edd edzdedzfd Zedd edzdedzfdZeddedzdedzfdZdedzfdZdd edzd edzdeefdZededefdZededefdZedefdZ ddededeededed edzd edzdefdZy)AzureFoundryModelInfoz/Model info for Azure AI / Azure Foundry models.Nmodelc||_yN)_model)selfr!s r__init__zAzureFoundryModelInfo.__init__<s  rr )agents model_routerdefaultcPd|vry|j}d|vs d|vsd|vsd|vryy)aY 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-routerr(r))lower)r! model_lowers rget_azure_ai_routez(AzureFoundryModelInfo.get_azure_ai_route?s@  "[[] { *+-,,!r hidden_paramsc\|sy|jt}t|tr|r|Sy)aThe model Azure Model Router actually served, stamped by ``AzureModelRouterConfig``. Reading this beats re-deriving the route from a model string: the stamp is set on the code path that was actually taken, so it holds no matter what the caller named the model. N)getr isinstancestr)r0selecteds rget_model_router_selected_modelz5AzureFoundryModelInfo.get_model_router_selected_modelXs/'++,QR h $Orctj|y|$|jdxs|jdnd}td||fDS)a7Whether a request went down the Azure Model Router route. Prefers the response stamp, then the deployment's litellm model path, and only then the caller-supplied name. The last two go through ``get_azure_ai_route`` so the model-router name heuristic lives in exactly one place. NTlitellm_model_namer!c3pK|].}t|txrtj|dk(0yw)r(N)r3r4r r/).0 candidates r z=AzureFoundryModelInfo.is_model_router_call..vs:  y# & p+@+S+ST]+^bp+p p s46)r r6r2any)r!r0deployment_models ris_model_router_callz*AzureFoundryModelInfo.is_model_router_callfse ! @ @ O [UbUnM  2 3 Q}7H7H7Qtx  .6   rapi_basecD|xstjxs tdS)NAZURE_AI_API_BASE)litellmr@r)r@s r get_api_basez"AzureFoundryModelInfo.get_api_base{sR7++R~>Q/RRrrcD|xstjxs tdS)NAZURE_AI_API_KEY)rCrr)rs r get_api_keyz!AzureFoundryModelInfo.get_api_keysO'//O^ "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 r+r,)splitr!s rstrip_model_router_prefixz/AzureFoundryModelInfo.strip_model_router_prefixsE e #;;215 5 e #;;215 5 rc0tj|}|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 rWrVs rget_base_modelz$AzureFoundryModelInfo.get_base_models&??F rctj|}|dk(r ddlm}|Sd|j 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 r(r)AzureModelRouterConfigrL)AzureAnthropicConfig)AzureAIStudioConfig) r r/7litellm.llms.azure_ai.azure_model_router.transformationr[r-.litellm.llms.azure_ai.anthropic.transformationr\)litellm.llms.azure_ai.chat.transformationr])r!azure_ai_router[r\r]s rget_azure_ai_config_for_modelz3AzureFoundryModelInfo.get_azure_ai_config_for_modelsV !6 H H O ^ + *+ +  & () ) U&( (rheadersmessagesoptional_paramsr ctd)z+Azure Foundry sends api key in query paramsz5Azure Foundry does not support environment validation)NotImplementedError)r%rcr!rdrer rr@s rvalidate_environmentz*AzureFoundryModelInfo.validate_environments""YZZrr#)NN)__name__ __module__ __qualname____doc__r4r& staticmethodrr/robjectr6boolr?rDrGpropertyrHrrOlistrRrWrYrbdictr rhrQrrr r 9sM9cDj#'2U*V0 wsF{7Kd7R WZ]aWa   59 Tz sF{+d2    (SsTzSS4ZSSPS4ZP3:PPsTzS4Z#3d#: #*sTzUYZ]U^* c c   )S ) )R## [ [ ['( [  [  [t [* [  [rr r#)Nr rF)collections.abcrtypingrrrC litellm.llms.base_llm.base_utilsrrlitellm.secret_managers.mainrlitellm.types.llms.openair litellm.types.routerr AzureAIApiKeyHeaderr4rnrrr__annotations__r rQrrr{s#!O765`a &WS&[-AD-H &TWZ^T^ &"37*9-  4ZCK(4/(   S#X 40S%uR[,[r