o
    i                     @   sH   d dl Z d dlmZ ddlmZmZ G dd deZG dd deZdS )	    N)	Tokenizer   )	AwsClientAwsClientV2c                   @   sn   e Zd Z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
d
d
d
d
ddddZdS )BedrockClientNaws_access_keyaws_secret_keyaws_session_token
aws_regiontimeoutr   r	   r
   r   r   c             	   C      t j| d|||||d d S Nbedrock)servicer   r	   r
   r   r   )r   __init__selfr   r	   r
   r   r    r   [/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/cohere/bedrock_client.pyr   	      	
zBedrockClient.__init__.)modeltop_nrank_fieldsreturn_documentsmax_chunks_per_docrequest_optionsc          	      C   s   t d)Nz{Please use cohere.BedrockClientV2 instead: Rerank API on Bedrock is not supported with cohere.BedrockClient for this model.)NotImplementedError)	r   query	documentsr   r   r   r   r   r   r   r   r   rerank   s   zBedrockClient.rerank)	__name__
__module____qualname__typingOptionalstrfloatr   r    r   r   r   r   r      s$    
r   c                   @   sV   e Zd Z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 )
BedrockClientV2Nr   r   r	   r
   r   r   c             	   C   r   r   )r   r   r   r   r   r   r       r   zBedrockClientV2.__init__)r!   r"   r#   r$   r%   r&   r'   r   r   r   r   r   r(      s"    r(   )r$   
tokenizersr   
aws_clientr   r   r   r(   r   r   r   r   <module>   s
    