o
    ưi]                  
   @   s   d dl Z d dlmZmZmZmZmZmZ d dlm	Z	 d dl
mZ d dlT d dlmZ d dlmZ e Zd dlZejdeege	d	gd
ejdeege	d	gd
ejdeege	d	gd
eefdededefdd	ZdS )    N)	APIRouterDependsHTTPExceptionRequestResponsestatus)ORJSONResponse)verbose_proxy_logger)*)user_api_key_auth)ProxyBaseLLMRequestProcessingz
/v2/rerankrerank)dependenciesresponse_classtagsz
/v1/rerankz/rerankrequestfastapi_responseuser_api_key_dictc                    s  ddl m}m}m}m}m}m}m}	m}
 i }z| 	 I d H }t
|}||| |||
|dI d H }|j||ddI d H }||d||	dI d H }|I d H }t|j|dd	d
d t|di pai }|dd pid	}|dd pqd	}|dd pyd	}|dd pi }|jtjd|||||
t|dd	|d| |W S  ty } zR|j|||dI d H  tdt| t|trtt|dt|t|ddt|ddt|dtj dt| }tt|d|t|ddt|ddt|dddd }~ww )Nr   )add_litellm_data_to_requestgeneral_settings
llm_routerproxy_configproxy_logging_objroute_request
user_modelversion)datar   r   r   r   r   r   )r   r   Z	call_typeZarerank)r   Z
route_typer   r   litellm_call_id success)r   r   Z_hidden_paramsmodel_id	cache_keyapi_baseadditional_headersZallowed_model_region)r   r    r!   r"   r   Zmodel_regionrequest_data)r   Zoriginal_exceptionr$   z;litellm.proxy.proxy_server.rerank(): Exception occured - {}messagetypeNoneparamstatus_code)r%   r&   r(   codei   )!Zlitellm.proxy.proxy_serverr   r   r   r   r   r   r   r   bodyorjsonloadsZpre_call_hookasynciocreate_taskZupdate_request_statusgetgetattrheadersupdater   Zget_custom_headers	ExceptionZpost_call_failure_hookr	   errorformatstr
isinstancer   ZProxyExceptionr   ZHTTP_400_BAD_REQUEST)r   r   r   r   r   r   r   r   r   r   r   r   r,   Zllm_callresponseZhidden_paramsr    r!   r"   r#   e	error_msgr+   r+   _/home/app/Keep/.python/lib/python3.10/site-packages/litellm/proxy/rerank_endpoints/endpoints.pyr      s   (











)r-   Zfastapir   r   r   r   r   r   Zfastapi.responsesr   Zlitellm._loggingr	   Zlitellm.proxy._typesZ$litellm.proxy.auth.user_api_key_authr   Z'litellm.proxy.common_request_processingr   Zrouterr/   postZUserAPIKeyAuthr   r+   r+   r+   r=   <module>   sF    	