o ưiw @s<dZddlmZddlmZddlmZGdddeZdS)z, Deprecated. Only PostgresSQL is supported. )verbose_proxy_logger) DynamoDBArgs)CustomDBc@s<eZdZUddlmZmZeed<defddZddZ d S) DynamoDBWrapperr) CredentialsStaticCredentials credentialsdatabase_argumentscCsddlm}m}d|_|jdkr||_n,|jdkrA|jdur:t|jtr:|jdur:t|jtr:||j|jd|_nt d|||_ |j |_ dS)Nr) PayPerRequest ThroughputZPAY_PER_REQUESTZPROVISIONED_THROUGHPUT)readwritezhInvalid args passed in. Need to set both read_capacity_units and write_capacity_units. Args passed in - ) Zaiodynamo.modelsr r Zthroughput_typeZ billing_modeZread_capacity_units isinstanceintZwrite_capacity_units Exceptionr Z region_name)selfr r r rQ/home/app/Keep/.python/lib/python3.10/site-packages/litellm/proxy/db/dynamo_db.py__init__s"        zDynamoDBWrapper.__init__cCs|jjdurdStd|jjddl}ddl}|d}|j|jj|jj|jj d|j |jj |jj d}|dd}|dd}|dd }td |||j d <||j d <||j d <dS)Nz(DynamoDB: setting env vars based on arn=rsts)RoleArnRoleSessionNameZWebIdentityToken)rrrZ AccessKeyIdZSecretAccessKeyZ SessionTokenz(Got STS assumed Role, aws_access_key_id=ZAWS_ACCESS_KEY_IDZAWS_SECRET_ACCESS_KEYZAWS_SESSION_TOKEN)r Z aws_role_namerdebugosboto3clientZassume_role_with_web_identityZaws_session_nameZaws_web_identity_tokenZ assume_roleZassume_role_aws_role_nameZassume_role_aws_session_nameenviron)rrrZ sts_clientZ assumed_roleZaws_access_key_idZaws_secret_access_keyZaws_session_tokenrrrset_env_vars_based_on_arn$s4        z)DynamoDBWrapper.set_env_vars_based_on_arnN) __name__ __module__ __qualname__Zaiodynamo.credentialsrr__annotations__rrrrrrrr s  rN)__doc__Zlitellm._loggingrZlitellm.proxy._typesrZlitellm.proxy.db.base_clientrrrrrrs