
    h]              
       `   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 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
      ej'                  d ee      ge	d	g
      ej'                  d ee      ge	d	g
       ee      fdededefd                     Zy)    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                   K   ddl m}m}m}m}m}m}m}	m}
 i }	 | j                          d {   }t        j                  |      } ||| |||
|       d {   }|j                  ||d       d {   } ||d||	       d {   }| d {   }t        j                  |j                  |j!                  dd	      d
             t#        |di       xs i }|j!                  dd       xs d	}|j!                  dd       xs d	}|j!                  dd       xs d	}|j!                  dd       xs i }|j$                  j'                  t)        j*                  d|||||
t#        |dd	      |d|       |S 7 L7 $7 7 7 # t,        $ r}|j/                  |||       d {  7   t1        j2                  dj5                  t7        |                   t9        |t:              rRt=        t#        |dt7        |            t#        |dd      t#        |dd      t#        |dt>        j@                              t7        |       }t=        t#        |d|      t#        |dd      t#        |dd      t#        |dd            d }~ww xY 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   	call_typearerank)r   
route_typer   r   litellm_call_id success)r"   r   _hidden_paramsmodel_id	cache_keyapi_baseadditional_headersallowed_model_region)r   r&   r'   r(   r   model_regionrequest_data)r   original_exceptionr,   z;litellm.proxy.proxy_server.rerank(): Exception occured - {}messagetypeNoneparamstatus_code)r.   r/   r1   codei   )!litellm.proxy.proxy_serverr   r   r   r   r   r   r   r   bodyorjsonloadspre_call_hookasynciocreate_taskupdate_request_statusgetgetattrheadersupdater   get_custom_headers	Exceptionpost_call_failure_hookr
   errorformatstr
isinstancer   ProxyExceptionr   HTTP_400_BAD_REQUEST)r   r   r   r   r   r   r   r   r   r   r   r   r6   llm_callresponsehidden_paramsr&   r'   r(   r)   e	error_msgs                         d/var/www/Befach/backend/env/lib/python3.12/site-packages/litellm/proxy/rerank_endpoints/endpoints.pyr   r      s    .	 	 	 DL\\^#||D! 1-/%
 
 '44/dh 5 
 

 ' !!	
 
 "> 	33 $):B ?	 4 	
  *:B?E2 $$Z6<"!%%k48>B	 $$Z6<"*../CTJPb  '')<< 	"3!#!$%68NPRS!	 %		
 k $



 ":  66/ATX 7 
 	
 	
 	""IPPQTUVQWX	
 a' 9c!f5Q/a&1Qv/J/JK	  q6(I 9i8Q/a&1Qs3	 !s   J
F E9*F E<F 4E?5F F		F FC%F 8J
9F <F ?F F F 	JJ&F)'CJJJ
)r7   fastapir   r   r   r   r   r   fastapi.responsesr	   litellm._loggingr
   litellm.proxy._types$litellm.proxy.auth.user_api_key_authr   'litellm.proxy.common_request_processingr   routerr:   postUserAPIKeyAuthr   r4       rO   <module>rZ      s     P P , 1 " B Q	  +,-!
	   +,-!
	   +,-!
	   )00A(B]]] &]$]rY   