o
    i                     @   sT   d dl Z ddlmZmZ ddlmZ ddlmZ G dd deZG dd	 d	eZ	dS )
    N   )	AwsClientAwsClientV2)Client)Modec                   @   `   e Zd ZU eed< dddddddeje deje deje deje deje f
d	d
Z	dS )SagemakerClientsagemaker_finetuningNaws_access_keyaws_secret_keyaws_session_token
aws_regiontimeoutr   r   r   r   r   c             	   C   @   t j| d|||||d z	t|d| _W d S  ty   Y d S w N	sagemaker)servicer   r   r   r   r   )r   )r   __init__r   r	   	Exceptionselfr   r   r   r   r    r   ]/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/cohere/sagemaker_client.pyr         		zSagemakerClient.__init__
__name__
__module____qualname__r   __annotations__typingOptionalstrfloatr   r   r   r   r   r      $   
 r   c                   @   r   )SagemakerClientV2r	   Nr
   r   r   r   r   r   c             	   C   r   r   )r   r   r   r	   r   r   r   r   r   r   &   r   zSagemakerClientV2.__init__r   r   r   r   r   r%   #   r$   r%   )
r    
aws_clientr   r   %manually_maintained.cohere_aws.clientr   #manually_maintained.cohere_aws.moder   r   r%   r   r   r   r   <module>   s    