
    i                         d dl mZmZmZmZmZ d dlmZ 	 	 	 	 	 	 	 dd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e   defdZy)    )AnyDictListOptionalUnion)BaseRerankConfigNrerank_provider_configmodeldrop_paramsquery	documentscustom_llm_providertop_nrank_fieldsreturn_documentsmax_chunks_per_docmax_tokens_per_docnon_default_paramsreturnc                     |xs i }|||d<   |||d<   |||d<   |||d<   |	|	|d<   |
|
|d<   | j                  |||||||||	|
|      S )Nr   r   r   r   r   r   )r
   r   r   r   r   r   r   r   r   r   r   )map_cohere_rerank_params)r	   r
   r   r   r   r   r   r   r   r   r   r   all_non_default_paramss                p/Users/manta/Documents/Projects/TheRoad-I1/.venv/lib/python3.12/site-packages/litellm/rerank_api/rerank_utils.pyget_optional_rerank_paramsr      s     052*/w'*/w'.7{+#5E12%7I34%7I34!::/)--1 ;      )NNNTNNN)typingr   r   r   r   r   +litellm.llms.base_llm.rerank.transformationr   strboolintdictr    r   r   <module>r#      s    3 3 H *.'+'+(,(,)-','' ' 	'
 E#tCH~-./' "#' C=' $s)$' tn' !' !' !' 
'r   