o
    ưi                     @   s   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dZdS )    )AnyDictListOptionalUnion)BaseRerankConfigNT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                 C   s   |pi }|d ur||d< |d ur||d< |d ur||d< |d ur$||d< |	d ur,|	|d< |
d ur4|
|d< | j |||||||||	|
|dS )Nr   r   r   r   r   r   )r	   r
   r   r   r   r   r   r   r   r   r   )Zmap_cohere_rerank_params)r   r	   r
   r   r   r   r   r   r   r   r   r   Zall_non_default_params r   V/home/app/Keep/.python/lib/python3.10/site-packages/litellm/rerank_api/rerank_utils.pyget_optional_rerank_params   s4   r   )NNNTNNN)typingr   r   r   r   r   Z+litellm.llms.base_llm.rerank.transformationr   strboolintdictr   r   r   r   r   <module>   sH    	
	
