o i[! @svddlmZddlZddlZddlmZddlmZddl m Z m Z m Z ddl mZddlmZddlmZdd lmZdd lmZdd lmZdd lmZdd lmZddlmZz$ddlZddl m!Z!ddl"m#Z#ddl$m%Z%ddl&m'Z'ddl(m)Z)Wne*yZ+ze*de+dZ+[+wweddGdddeZ,d%ddZ-d%dd Z.Gd!d"d"ee!Z/Gd#d$d$e'Z0dS)&) annotationsN)Callable) dataclass)AnyLiteraloverload) ModelProfile) UserError)amazon_model_profile)anthropic_model_profile)cohere_model_profile)deepseek_model_profile)meta_model_profile)mistral_model_profile)Provider) BaseClient)Config) NoRegionError)Session)FrozenAuthTokenuPlease install the `boto3` package to use the Bedrock provider, you can use the `bedrock` optional group — `pip install "pydantic-ai-slim[bedrock]"`T)kw_onlyc@s6eZdZUdZdZded<dZded<dZded<d S) BedrockModelProfilezProfile for models used with BedrockModel. ALL FIELDS MUST BE `bedrock_` PREFIXED SO YOU CAN MERGE THEM WITH OTHER MODELS. Fboolbedrock_supports_tool_choicetextzLiteral['text', 'json']bedrock_tool_result_format bedrock_send_back_thinking_partsN)__name__ __module__ __qualname____doc__r__annotations__rrr"r"c/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/pydantic_ai/providers/bedrock.pyr!s   r model_namestrreturnModelProfile | NonecC$t|}d|vrtdd|S|S)z;Get the model profile for an Amazon model used via Bedrock.novaT)r)r rupdater$profiler"r"r#bedrock_amazon_model_profile-r-cCr()zNs z/BedrockProvider.model_profile..cSstddt|S)Njson)r)rr*rr<r"r"r#r=Qs ) anthropicmistralcohereamazonr8deepseek.rz(.+)-v\d+(?::\d+)?$) r r-rr0splitlenrematchgroup)r4r$provider_to_profilepartsprovidermodel_name_with_version version_matchr"r"r# model_profileLs*      zBedrockProvider.model_profilebedrock_clientNonecCdSr6r")r4rRr"r"r#__init__szBedrockProvider.__init__N)r: region_name profile_nameaws_read_timeoutaws_connect_timeoutapi_keyr: str | NonerWrXrY float | NonerZcCrTr6r")r4r[r:rWrXrYrZr"r"r#rUvs )aws_access_key_idaws_secret_access_keyaws_session_tokenr:rWrXrYrZr^r_r`c CrTr6r") r4r^r_r`r:rWrXrYrZr"r"r#rUs ) rRr^r_r`r:rWrXr[rYrZBaseClient | Nonec  Cs|dur ||_dS| pttdd} | pttdd} | | d} z.|dur6tjt|||d}d| d <n tj|||||d }|jd tdi| |d |_WdSt ya}zt d |d}~ww)aInitialize the Bedrock provider. Args: bedrock_client: A boto3 client for Bedrock Runtime. If provided, other arguments are ignored. aws_access_key_id: The AWS access key ID. If not set, the `AWS_ACCESS_KEY_ID` environment variable will be used if available. aws_secret_access_key: The AWS secret access key. If not set, the `AWS_SECRET_ACCESS_KEY` environment variable will be used if available. aws_session_token: The AWS session token. If not set, the `AWS_SESSION_TOKEN` environment variable will be used if available. api_key: The API key for Bedrock client. Can be used instead of `aws_access_key_id`, `aws_secret_access_key`, and `aws_session_token`. If not set, the `AWS_BEARER_TOKEN_BEDROCK` environment variable will be used if available. base_url: The base URL for the Bedrock client. region_name: The AWS region name. If not set, the `AWS_DEFAULT_REGION` environment variable will be used if available. profile_name: The AWS profile name. aws_read_timeout: The read timeout for Bedrock client. aws_connect_timeout: The connect timeout for Bedrock client. NAWS_READ_TIMEOUTi,AWS_CONNECT_TIMEOUT<) read_timeoutconnect_timeout)botocore_sessionrWrXbearersignature_version)r^r_r`rWrXzbedrock-runtime)configr9zGYou must provide a `region_name` or a boto3 client for Bedrock Runtime.r") r7floatosgetenvboto3r_BearerTokenSessionr;rrr )r4rRr^r_r`r:rWrXr[rYrZrerfrjsessionexcr"r"r#rUs>    )r&r%)r&rr$r%r&r')rRrr&rS)r[r%r:r\rWr\rXr\rYr]rZr]r&rS)r^r\r_r\r`r\r:r\rWr\rXr\rYr]rZr]r&rS)rRrar^r\r_r\r`r\r:r\rWr\rXr\r[r\rYr]rZr]r&rS) rrrr propertyr5r:r;rQrrUr"r"r"r#r1=sN    '  r1cs2eZdZdfdd Zdd d Zdd d ZZS)rotokenr%cst||_dSr6)superrUrt)r4rt __class__r"r#rUs  z_BearerTokenSession.__init___kwargsrr&rcKs t|jSr6)rrt)r4rxr"r"r#get_auth_tokens z"_BearerTokenSession.get_auth_tokenrScCrTr6r"r3r"r"r#get_credentialsrVz#_BearerTokenSession.get_credentials)rtr%)rxrr&r)r&rS)rrrrUryrz __classcell__r"r"rvr#ros rorr)1 __future__r _annotationsrlrIcollections.abcr dataclassesrtypingrrr pydantic_airpydantic_ai.exceptionsr pydantic_ai.profiles.amazonr pydantic_ai.profiles.anthropicr pydantic_ai.profiles.coherer pydantic_ai.profiles.deepseekr pydantic_ai.profiles.metarpydantic_ai.profiles.mistralrpydantic_ai.providersrrnbotocore.clientrbotocore.configrbotocore.exceptionsrbotocore.sessionrbotocore.tokensr ImportError _import_errorrr-r0r1ror"r"r"r#sJ