
    ^h              	          S SK r S SKJrJrJrJrJrJr  S SKJ	r	  S SK
Jr  S SK7  S SKJr  \" 5       rS SKr\R#                  S\" \5      /\	S/S	9\R#                  S
\" \5      /\	S/S	9\" \5      4S\S\S\4S jj5       5       rg)    N)	APIRouterDependsHTTPExceptionRequestResponsestatus)ORJSONResponse)verbose_proxy_logger)*)user_api_key_authz
/v1/rerankrerank)dependenciesresponse_classtagsz/rerankrequestfastapi_responseuser_api_key_dictc                   #    SSK JnJnJnJnJnJnJn	Jn
J	n  0 n U R                  5       I S h  vN n[        R                  " U5      nU" UU UUUUS9I S h  vN nUR                  X,SS9I S h  vN nU	" USUU
S9I S h  vN nUI S h  vN n[        R                  " UR!                  UR#                  SS	5      S
S95        [%        US0 5      =(       d    0 nUR#                  SS 5      =(       d    S	nUR#                  SS 5      =(       d    S	nUR#                  SS 5      =(       d    S	nUR#                  SS 5      =(       d    0 nUR&                  R)                  U" SUUUUU[%        USS	5      US.UD65        U$  GNM GN' GN GN N! [*         a  nUR-                  UUUS9I S h  vN    [.        R0                  " SR3                  [5        U5      5      5        [7        U[8        5      (       aO  [;        [%        US[5        U5      5      [%        USS5      [%        USS5      [%        US[<        R>                  5      S9e[5        U5       n[;        [%        USU5      [%        USS5      [%        USS5      [%        USS5      S9eS nAff = f7f)Nr   )	add_litellm_data_to_requestgeneral_settingsget_custom_headers
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   r   bodyorjsonloadspre_call_hookasynciocreate_taskupdate_request_statusgetgetattrheadersupdate	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   r   r6   llm_callresponsehidden_paramsr&   r'   r(   r)   e	error_msgs                          b/home/james-whalen/.local/lib/python3.13/site-packages/litellm/proxy/rerank_endpoints/endpoints.pyr   r      s    "
 
 
 DL\\^#||D! 1-/%
 
 '44/h 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
 E<(F
 E?F
 0F1F
 F	F
 FC/F
 ;J<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   routerr:   postUserAPIKeyAuthr   r4       rN   <module>rX      s     P P , 1 " B	  +,-!
	   +,-!
	   )00A(B^^^ &^^rW   