o
    i                     @   sf   d dl mZmZmZmZmZmZ ddlmZ edde	fgZ
de
_G dd deZG d	d
 d
eZdS )    )AnyDictIteratorList
NamedTupleOptional   )CohereObjectDocumenttextz_
Returned by co.rerank,
dict which always contains text but can also contain aribitrary fields
c                	       sL   e Zd Z			d
deeef dededdf fddZdefdd	Z	  Z
S )RerankResultNdocumentindexrelevance_scorereturnc                    s(   t  j|i | || _|| _|| _d S N)super__init__r   r   r   )selfr   r   r   argskwargs	__class__ r/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/cohere/manually_maintained/cohere_aws/rerank.pyr      s   
zRerankResult.__init__c                 C   sn   | j }| j}| jd u rd| d| dS d| jv r*| jd }d| d| d| dS d| j d| d| dS )NzRerankResult<index: z, relevance_score: >r   zRerankResult<text: z	, index: zRerankResult<document: )r   r   r   )r   scorer   r   r   r   r   __repr__   s   


zRerankResult.__repr__)NNN)__name__
__module____qualname__r   strr   intfloatr   r   __classcell__r   r   r   r   r      s    

r   c                       s   e Zd Z	ddeeeef  ddf fddZdeeef dee	 fddZ
defdd	Zdefd
dZdefddZde	fddZ  ZS )	RerankingNresponser   c                    s.   t  jdi | |d usJ | || _d S )Nr   )r   r   _resultsresults)r   r&   r   r   r   r   r   &   s   zReranking.__init__c                 C   s\   g }|d D ]%}d|  v r|t|d |d |d  q|t|d |d d q|S )Nr(   r   r   r   )r   r   )keysappendr   )r   r&   r(   resr   r   r   r'   -   s   zReranking._resultsc                 C   
   t | jS r   )r!   r(   r   r   r   r   __str__8      
zReranking.__str__c                 C   s
   | j  S r   )r(   r   r-   r   r   r   r   ;   r/   zReranking.__repr__c                 C   r,   r   )iterr(   r-   r   r   r   __iter__>   r/   zReranking.__iter__c                 C   s
   | j | S r   )r(   )r   r   r   r   r   __getitem__A   r/   zReranking.__getitem__r   )r   r   r    r   r   r!   r   r   r   r   r'   r.   r   r   r1   r2   r$   r   r   r   r   r%   $   s    r%   N)typingr   r   r   r   r   r   r&   r	   r!   RerankDocument__doc__r   r%   r   r   r   r   <module>   s     