Ë
    ÷¾™h<  ã                   óR   — d dl mZmZmZmZmZ d dlmZ d dlm	Z	m
Z
  G d„ de«      Zy)é    )ÚAnyÚDictÚListÚOptionalÚUnion)ÚCohereRerankConfig)ÚOptionalRerankParamsÚRerankRequestc                   óê   — e Zd ZdZdd„Zdee   dedefd„Zdedefd„Z		 	 	 	 	 	 dd	ee
   ded
ededeeeeeef   f      dee   dee   deee      dee   dee   dee   defd„Zdedede
de
fd„Zy)ÚCohereRerankV2Configz@
    Reference: https://docs.cohere.com/v2/reference/rerank
    ÚreturnNc                  ó   — y )N© )Úselfs    úh/var/www/Befach/backend/env/lib/python3.12/site-packages/litellm/llms/cohere/rerank_v2/transformation.pyÚ__init__zCohereRerankV2Config.__init__   s   € Øó    Úapi_baseÚmodelc                 ó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   s      r   Úget_complete_urlz%CohereRerankV2Config.get_complete_url   s5   € Ùà—‘ 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   c                 ó$   — t        |||||	|¬«      S )zf
        Map Cohere rerank params

        No mapping required - returns all supported params
        )r   r   r   r    r!   r   )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ô
ð 	
r   Úoptional_rerank_paramsÚheadersc                 ó  — 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*   Úrerank_requests        r   Útransform_rerank_requestz-CohereRerankV2Config.transform_rerank_request>   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   N)NNNTNN)Ú__name__Ú
__module__Ú__qualname__Ú__doc__r   r   Ústrr   Úlistr"   ÚdictÚboolr   r   r   r   Úintr	   r(   r1   r   r   r   r   r      s/  „ ñóð1¨°#©ð 1¸sð 1Àsó 1ð
¸ð 
Àó 
ð" .2Ø#Ø+/Ø+/Ø,0Ø,0ñ
à$ T™Nð
ð ð
ð ð	
ð
 ð
ð ˜˜c 4¨¨S¨¡>Ð1Ñ2Ñ3ð
ð & c™]ð
ð ˜‰}ð
ð ˜d 3™iÑ(ð
ð # 4™.ð
ð % S™Mð
ð % S™Mð
ð 
ó
ð8<àð<ð !5ð<ð ð	<ð
 
ô<r   r   N)Útypingr   r   r   r   r   Ú)litellm.llms.cohere.rerank.transformationr   Úlitellm.types.rerankr	   r
   r   r   r   r   ú<module>r>      s    ðß 3Õ 3å Hß DôJ<Ð-õ J<r   