tPj@ xUddlZddlmZddlZddlmZmZmZmZm Z m Z ddl m Z ddl mZddlddlmZddlmZeZeed <ej-d eege d g ej-d eege d g ej-deege d g eefdede defdZy)N)Final) APIRouterDepends HTTPExceptionRequestResponsestatus)ORJSONResponse)verbose_proxy_logger)*)user_api_key_auth)ProxyBaseLLMRequestProcessingrouterz /v2/rerankrerank) dependenciesresponse_classtagsz /v1/rerankz/rerankrequestfastapi_responseuser_api_key_dictcJKddlm}m}m}m}m}m}m} m} i} |jd{} tj| } || |||| |d{} |j|| dd{} || d|| d{} | d{}tj|j| j!dd d t#|d ixsi}|j!d dxsd }|j!ddxsd }|j!ddxsd }|j!ddxsi}|j$j't)j*d||j!ddxs| j!dd|||| |j!ddt#|dd | |d ||S77[7B717*#t,$r}|j/||| d{7t1j2d|t5|t6rRt9t#|dt;|t#|ddt#|ddt#|dt<j>|}t9t#|d|t#|ddt#|ddt#|ddd}~wwxYww)Nr)add_litellm_data_to_requestgeneral_settings llm_router proxy_configproxy_logging_obj route_request user_modelversion)datarrrrrr)rr call_typearerank)r route_typerrlitellm_call_idsuccess)r$r _hidden_paramsmodel_id cache_keyapi_baseadditional_headers response_costallowed_model_region) rcall_idr(r)r*rr, model_region request_data hidden_params)roriginal_exceptionr0z;litellm.proxy.proxy_server.rerank(): Exception occured - %smessagetypeNoneparam status_code)r3r4r6codei) litellm.proxy.proxy_serverrrrrrrrrbodyorjsonloads pre_call_hookasyncio create_taskupdate_request_statusgetgetattrheadersupdaterget_custom_headers Exceptionpost_call_failure_hookr error isinstancerProxyExceptionstrr HTTP_400_BAD_REQUEST)rrrrrrrrrrrr r;llm_callresponser1r(r)r*r+e error_msgs m/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/proxy/rerank_endpoints/endpoints.pyrrs.    DI#LLN*||D!1-/%   '44GX_cow4xx!. !! !   !).   3 3DHHM^`bDclu 3 v 'x1A2FL" '++J=C(,,[$?E2 '++J=C$1$5$56JD$Q$WUW  '' ) < < "3%))*;TBgdhhO`bfFg!#!+//F$%68NPRS!+ %  i+ y )< 66/ATX7    ""#`bcd a ' 9c!f5Q/a&1Q v/J/JK  #$I 9i8Q/a&1Q s3  sJ#F?F0*F?F3F?4F65F?F9 F?F<DF?/J#0F?3F?6F?9F?<F?? J JG" B;JJ  J#)r?typingrr<fastapirrrrrr fastapi.responsesr litellm._loggingr litellm.proxy._types$litellm.proxy.auth.user_api_key_authr 'litellm.proxy.common_request_processingrr__annotations__postUserAPIKeyAuthrr9rRr^s PP,1"BQ +,-!    +,-!     +,-!   )00A(BZ ZZ&Z   $Zr]