
    g              	          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  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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_authz
/v1/rerankrerank)dependenciesresponse_classtagsz/rerankrequestfastapi_responseuser_api_key_dictc                    K   ddl m}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)                   |d|||||t%        |dd	      |d|       |S 7 ?7 7 7 7 # t*        $ r}|j-                  |||       d {  7   t/        j0                  dj3                  t5        |                   t7        |t8              rRt;        t%        |dt5        |            t%        |dd      t%        |dd      t%        |dt<        j>                              t5        |       }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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                          ]/var/www/openai/venv/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   I>E: E.*E: E1E: 6E47E: 
E6	E: E8CE: -I>.E: 1E: 4E: 6E: 8E: :	I;I6FCI66I;;I>)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   