uPjtddlmZmZddlmZddlmZmZmZddl Z ddl m Z m Z ddl mZddlmZer dd lmZeZneZGd d eZy) )ABCabstractmethod)Mapping) TYPE_CHECKINGAnyFinalN)RerankBilledUnitsRerankResponse) ModelInfoBaseLLMException)Loggingc>eZdZe d%dedededzdedzdeeefdzdef dZe d&ded edededzdef d Z ediiifded e jd e d e dedzdededede fdZe d&dedzdededzdefdZededefdZe d'dededededeeeeefzdedzdedzdeedzdedzdedzdedzdedzdefdZded edee j,zdefd!Z d%dededzd"edzd#edzdeeeff d$Zy)(BaseRerankConfigNheadersmodelapi_keyoptional_paramslitellm_paramsreturncyN)selfrrrrrs p/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/llms/base_llm/rerank/transformation.pyvalidate_environmentz%BaseRerankConfig.validate_environments optional_rerank_paramsciSrr)rrrrrs rtransform_rerank_requestz)BaseRerankConfig.transform_rerank_request s  r raw_responsemodel_response logging_obj request_datac |Srr) rrr"r#r$rr%rrs rtransform_rerank_responsez*BaseRerankConfig.transform_rerank_response*s rapi_basec|xsdS)z{ OPTIONAL Get the complete url for the request Some providers need `model` in `api_base` r)rr(rrs rget_complete_urlz!BaseRerankConfig.get_complete_url8s~2rcyrr)rrs r"get_supported_cohere_rerank_paramsz3BaseRerankConfig.get_supported_cohere_rerank_paramsHs rnon_default_params drop_paramsquery documentscustom_llm_providertop_n rank_fieldsreturn_documentsmax_chunks_per_docmax_tokens_per_doc instructionc yrr) rr.rr/r0r1r2r3r4r5r6r7r8s rmap_cohere_rerank_paramsz)BaseRerankConfig.map_cohere_rerank_paramsLs r error_message status_codect|||)N)r<messagerr )rr;r<rs rget_error_classz BaseRerankConfig.get_error_class^s#!  r billed_units model_infoc^| d|vs|d|y|jd}|y|d|z}|dfS)a% Calculates the cost per query for a given rerank model. Input: - model: str, the model name without provider prefix - custom_llm_provider: str, the provider used for the model. If provided, used to check if the litellm model info is for that provider. - num_queries: int, the number of queries to calculate the cost for - model_info: ModelInfo, the model info for the given model Returns: Tuple[float, float] - prompt_cost_in_usd, completion_cost_in_usd input_cost_per_query)rD search_unitsrD)get)rrr2r@rArE prompt_costs rcalculate_rerank_costz&BaseRerankConfig.calculate_rerank_costes[*  %Z7019#*..~>  '(>?,N Cr)NNNr)NNNTNNN)__name__ __module__ __qualname__rdictstrrobjectrr!httpxResponser LiteLLMLoggingObjr'r+listr-boolrintr:Headersrr?r r tuplefloatrHrrrrrs  #'+6:    t     V ,t3      '+ !%  t   # "!  nn '  '  t        (, *         +/ (,(,)-)-"&        d38n,-  !4Z Tz #Y% +  $J  $J 4Z    " S s TTYTaTaMa fv +/15'+ # # !4Z# ($. # $ # ue|  # rr)abcrrcollections.abcrtypingrrrrOlitellm.types.rerankr r litellm.types.utilsr chat.transformationr*litellm.litellm_core_utils.litellm_loggingr_LiteLLMLoggingObjrQrrrrr`s=##,, B)2X*t st r