Ë
    uPêjÇ  ã                   óF   — d dl mZmZ d dlmZ d dlmZmZ  G d„ de«      Zy)é    )ÚAnyÚFinal)ÚCohereRerankConfig)ÚOptionalRerankParamsÚRerankRequestc                   ó  — e Zd ZdZdd„Z	 ddedz  dededz  defd„Zdedefd	„Z		 	 	 	 	 	 	 dd
edz  dede
dedeeeeef   z     dedz  dedz  dee   dz  de
dz  dedz  dedz  dedz  defd„Z	 ddededededz  def
d„Zy)ÚCohereRerankV2Configz@
    Reference: https://docs.cohere.com/v2/reference/rerank
    ÚreturnNc                  ó   — y ©N© )Úselfs    úq/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/llms/cohere/rerank_v2/transformation.pyÚ__init__zCohereRerankV2Config.__init__   s   € Øó    Úapi_baseÚmodelÚoptional_paramsc                 óZ   — |r)|j                  d«      }|j                  d«      s|› d�}|S y)Nú/z
/v2/rerankzhttps://api.cohere.ai/v2/rerank)ÚrstripÚendswith)r   r   r   r   s       r   Úget_complete_urlz%CohereRerankV2Config.get_complete_url   s7   € ñ à—‘ sÓ+ˆHØ×$Ñ$ \Ô2Ø&˜Z zÐ2�ØˆOØ0r   c                 ó
   — g d¢S )N)ÚqueryÚ	documentsÚtop_nÚmax_tokens_per_docÚrank_fieldsÚreturn_documentsr   )r   r   s     r   Ú"get_supported_cohere_rerank_paramsz7CohereRerankV2Config.get_supported_cohere_rerank_params   s   € ò
ð 	
r   Únon_default_paramsÚdrop_paramsr   r   Úcustom_llm_providerr   r   r    Úmax_chunks_per_docr   Úinstructionc           
      ó6   — t        t        |||||	|¬«      «      S )zf
        Map Cohere rerank params

        No mapping required - returns all supported params
        )r   r   r   r   r    r   )Údictr   )r   r"   r   r#   r   r   r$   r   r   r    r%   r   r&   s                r   Úmap_cohere_rerank_paramsz-CohereRerankV2Config.map_cohere_rerank_params'   s+   € ô( Ü ØØ#ØØ'Ø!1Ø#5ôó	
ð 		
r   Úoptional_rerank_paramsÚheadersÚlitellm_paramsc                 ó  — d|vrt        d«      ‚d|vrt        d«      ‚t        ||d   |d   |j                  dd «      |j                  dd «      |j                  dd «      |j                  dd «      ¬	«      }|j                  d
¬«      S )Nr   z#query is required for Cohere rerankr   z'documents is required for Cohere rerankr   r   r    r   )r   r   r   r   r   r    r   T)Úexclude_none)Ú
ValueErrorr   ÚgetÚ
model_dump)r   r   r*   r+   r,   Úrerank_requests         r   Útransform_rerank_requestz-CohereRerankV2Config.transform_rerank_requestF   s¤   € ð Ð0Ñ0ÜÐBÓCÐCØÐ4Ñ4ÜÐFÓGÐGÜ -ØØ(¨Ñ1Ø,¨[Ñ9Ø(×,Ñ,¨W°dÓ;Ø.×2Ñ2°=À$ÓGØ3×7Ñ7Ð8JÈDÓQØ5×9Ñ9Ð:NÐPTÓUô!
ˆð ×(Ñ(°dÐ(Ó;Ð;r   )r
   Nr   )NNNTNNN)Ú__name__Ú
__module__Ú__qualname__Ú__doc__r   Ústrr(   r   Úlistr!   Úboolr   Úintr)   r3   r   r   r   r	   r	      si  „ ñóð (,ñ	1à˜‘*ð1ð ð1ð  ™ð	1ð
 
ó1ð
¸ð 
Àó 
ð" +/Ø Ø(,Ø(,Ø)-Ø)-Ø"&ñ
à  4™Kð
ð ð
ð ð	
ð
 ð
ð ˜˜d 3¨ 8™nÑ,Ñ-ð
ð ! 4™Zð
ð �T‰zð
ð ˜#‘Y Ñ%ð
ð  ™+ð
ð   $™Jð
ð   $™Jð
ð ˜4‘Zð
ð 
ó
ðH '+ñ<àð<ð !%ð<ð ð	<ð
 ˜t™ð<ð 
ô<r   r	   N)	Útypingr   r   Ú)litellm.llms.cohere.rerank.transformationr   Úlitellm.types.rerankr   r   r	   r   r   r   ú<module>r?      s   ðß å Hß DôS<Ð-õ S<r   