o i @sddlmZddlZddlmZddlZddlmZddl m Z ddl m Z ddl mZddlmZdd lmZdd lmZdd lmZdd lmZdd lmZmZmZddlmZzddlmZWne ysZ!ze de!dZ![!wwGdddeeZ"dS)) annotationsN)overload) AsyncOpenAI) ModelProfile) UserError)cached_async_http_client)cohere_model_profile)deepseek_model_profile)grok_model_profile)meta_model_profile)mistral_model_profile)OpenAIJsonSchemaTransformerOpenAIModelProfileopenai_model_profile)Provider)AsyncAzureOpenAIuPlease install the `openai` package to use the Azure provider, you can use the `openai` optional group — `pip install "pydantic-ai-slim[openai]"`c@seZdZdZed ddZed ddZed!d d Zd"d dZe d#ddZ e dddddd$ddZ ddddddd%ddZ dS)& AzureProviderzzProvider for Azure OpenAI API. See for more information. returnstrcCsdS)Nazureselfrra/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/pydantic_ai/providers/azure.pyname#szAzureProvider.namecCs|jdusJ|jSN) _base_urlrrrrbase_url'szAzureProvider.base_urlrcCs|jSr)_clientrrrrclient,szAzureProvider.client model_nameModelProfile | NonecCsv|}tttttttd}|D]$\}}||r6|dr(|t |d}||}t t d |Sqt |S)N)llamazmeta-deepseekz mistralai-mistralzcohere-grok-)json_schema_transformer)lowerr r r rr items startswithendswithlenrr updater)rr prefix_to_profileprefix profile_funcprofilerrr model_profile0s"   zAzureProvider.model_profile openai_clientrNonecCdSrr)rr3rrr__init__KszAzureProvider.__init__N)azure_endpoint api_versionapi_key http_clientr7 str | Noner8r9r:httpx.AsyncClient | NonecCr5rr)rr7r8r9r:rrrr6Ns)r7r8r9r3r:AsyncAzureOpenAI | NonecCs|dur'|dus Jd|dusJd|dusJdt|j|_||_dS|p-td}|s4td|s?dtjvr?td|sJd tjvrJtd |pPtd d }t ||||d |_t|jj|_dS)aCreate a new Azure provider. Args: azure_endpoint: The Azure endpoint to use for authentication, if not provided, the `AZURE_OPENAI_ENDPOINT` environment variable will be used if available. api_version: The API version to use for authentication, if not provided, the `OPENAI_API_VERSION` environment variable will be used if available. api_key: The API key to use for authentication, if not provided, the `AZURE_OPENAI_API_KEY` environment variable will be used if available. openai_client: An existing [`AsyncAzureOpenAI`](https://github.com/openai/openai-python#microsoft-azure-openai) client to use. If provided, `base_url`, `api_key`, and `http_client` must be `None`. http_client: An existing `httpx.AsyncClient` to use for making HTTP requests. Nz8Cannot provide both `openai_client` and `azure_endpoint`z5Cannot provide both `openai_client` and `http_client`z1Cannot provide both `openai_client` and `api_key`AZURE_OPENAI_ENDPOINTzeMust provide one of the `azure_endpoint` argument or the `AZURE_OPENAI_ENDPOINT` environment variableAZURE_OPENAI_API_KEYz]Must provide one of the `api_key` argument or the `AZURE_OPENAI_API_KEY` environment variableOPENAI_API_VERSIONz_Must provide one of the `api_version` argument or the `OPENAI_API_VERSION` environment variabler)provider)r7r9r8r:) rrrrosgetenvrenvironrr)rr7r8r9r3r:rrrr6Xs6  )rr)rr)r rrr!)r3rrr4) r7r;r8r;r9r;r:r<rr4) r7r;r8r;r9r;r3r=r:r<rr4) __name__ __module__ __qualname____doc__propertyrrrr2rr6rrrrrs.      r)# __future__r _annotationsrBtypingrhttpxopenair pydantic_airpydantic_ai.exceptionsrpydantic_ai.modelsrpydantic_ai.profiles.cohererpydantic_ai.profiles.deepseekr pydantic_ai.profiles.grokr pydantic_ai.profiles.metar pydantic_ai.profiles.mistralr pydantic_ai.profiles.openair rrpydantic_ai.providersrr ImportError _import_errorrrrrrs4