g`ddlZddlmZmZddlmZddlmZGddeZGdd eZ y) N) AwsClient AwsClientV2)Client)Modec eZdZUeed<dddddddej edej edej edej edej ef d Z y) SagemakerClientsagemaker_finetuningNaws_access_keyaws_secret_keyaws_session_token aws_regiontimeoutr r rrrc ~tj|d||||| t||_y#t$rYywxYwN sagemaker)servicer r rrr)r)r__init__rr Exceptionselfr r rrrs L/var/www/openai/venv/lib/python3.12/site-packages/cohere/sagemaker_client.pyrzSagemakerClient.__init__ sK  ))/!  (.*(ED %    0 << __name__ __module__ __qualname__r__annotations__typingOptionalstrfloatrrr r }  48376:/3.2#OOC0#OOC0  &s3  , __U+r%r c eZdZUeed<dddddddej edej edej edej edej ef d Z y) SagemakerClientV2r Nr r r rrrc ~tj|d||||| t||_y#t$rYywxYwr)rrrr rrs rrzSagemakerClientV2.__init__&sK  ))/!  (.*(ED %   rrr$r%rr(r(#r&r%r() r aws_clientrr%manually_maintained.cohere_aws.clientr#manually_maintained.cohere_aws.moderr r(r$r%rr-s) .95i6 r%