uPjX"UdZddlmZddlmZmZddlZddlmZddl m Z ddl m Z ddlmZdd lmZdd lmZmZmZmZmZd d lmZd Zeed<GddeZy)uH Transformation logic for DashScope's OpenAI-compatible /v1/reranks API. Supports - qwen3-rerank (Other DashScope rerankers — gte-rerank-v2 / qwen3-vl-rerank — share the same endpoint but have not been validated against this transformer. Behavior with those models is undefined.) Endpoint - https://dashscope.aliyuncs.com/compatible-api/v1/reranks Note: chat/embed live under `/compatible-mode/v1/`, but DashScope's rerank route is exposed under `/compatible-api/v1/reranks` per the docs. Override with `DASHSCOPE_API_BASE_RERANK` to point at a different host or path. Empirically, qwen3-rerank accepts `return_documents=true` and echoes `results[].document.text` back, even though the public docs list the flag as supported only for gte-rerank-v2 / qwen3-vl-rerank. Docs - https://help.aliyun.com/zh/model-studio/text-rerank-api )Mapping)AnyFinalN)uuid)Logging)BaseLLMException)BaseRerankConfig)get_secret_str)OptionalRerankParamsRerankBilledUnitsRerankResponseRerankResponseMeta RerankTokens)DashScopeErrorz8https://dashscope.aliyuncs.com/compatible-api/v1/reranksDEFAULT_RERANK_URLceZdZdZd$dZ d%dedzdededzdefdZ d&d eded edzdedzd eee fdzdef d Z dede fd Z d'dedzdede dede eeeefzdedzdedzde edzde dzdedzdedzdedzdefdZ d%deded ed edzdef dZ d(dedej&deded edzdedzdedzd edzdefd Zd!ed"ed eej.zdefd#Zy))DashScopeRerankConfigak Reference: https://help.aliyun.com/zh/model-studio/text-rerank-api Targets DashScope's qwen3-rerank model. Request fields: model, query, documents, top_n, return_documents. Response: results[].index, results[].relevance_score, optionally results[].document.text (when return_documents=true), plus a top-level usage.total_tokens counter. returnNcyN)selfs q/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/llms/dashscope/rerank/transformation.py__init__zDashScopeRerankConfig.__init__:s api_basemodeloptional_paramsc|tdxst}|tk(rtS|jd}|jds|jdr|S|jdr|dS|dS)NDASHSCOPE_API_BASE_RERANK/z/reranksz/rerankz/v1)r rrstripendswith)rrrrcleaneds rget_complete_urlz&DashScopeRerankConfig.get_complete_url=s  %&ABXFXH ) )% %!-   J '7+;+;I+FN   E "Yh' '(##rheadersapi_keylitellm_paramscT| td}| tdd|ddd}i||S)NDASHSCOPE_API_KEYzZDashScope API key is required. Set 'DASHSCOPE_API_KEY' env var or pass api_key explicitly.zBearer zapplication/json) Authorizationacceptz content-type)r ValueError)rr'rr(rr)default_headerss rvalidate_environmentz*DashScopeRerankConfig.validate_environmentSsT ?$%89G ?l  'wi0(."  ./-W--rc gdS)N)query documentstop_nreturn_documentsr)rrs r"get_supported_cohere_rerank_paramsz8DashScopeRerankConfig.get_supported_cohere_rerank_paramsisBBrnon_default_params drop_paramsr2r3custom_llm_providerr4 rank_fieldsr5max_chunks_per_docmax_tokens_per_doc instructionc Nt||} ||| d<| | | d<t| S)N)r2r3r4r5)r dict)rr7rr8r2r3r9r4r:r5r;r<r=paramss rmap_cohere_rerank_paramsz.DashScopeRerankConfig.map_cohere_rerank_paramslsA"/C/   #F7O  ')9F% &F|roptional_rerank_paramscd|vr tdd|vr td||d|dd}|jd|d|d<|jd|d|d<|S)Nr2z&query is required for DashScope rerankr3z*documents is required for DashScope rerank)rr2r3r4r5)r.get)rrrBr'r)requests rtransform_rerank_requestz.DashScopeRerankConfig.transform_rerank_requests 0 0EF F 4 4IJ J+G4/ <*  " % %g . :5g>GG  ! % %&8 9 E*@AS*TG& 'r raw_responsemodel_response logging_obj request_datac |xsi}|xsi}|xsi} |j} |j |j d|d|i| d| vr4d| vr0t|j| j dt| | j d} | t|jd| g} | D]\} | d | d d } | j d }t|tr|| d <nt|trd |i| d <| j| ^| j dxsi}|j d}t|}t|}t||}t| j dxsttj | |S#t$r"t|j|jwxYw)N) status_codemessager2complete_input_dict)inputr(additional_argsoriginal_responsecoderesultsrMz)No results in DashScope rerank response: indexrelevance_score)rTrUdocumenttextusage total_tokens)rY) input_tokens) billed_unitstokensid)r]rSmeta)json ExceptionrrLrW post_callrDstr isinstancer?appendr rrr ruuid4)rrrGrHrIr(rJrr) response_jsonrStransformed_resultsritemdocrXrYr[r\r^s rtransform_rerank_responsez/DashScopeRerankConfig.transform_rerank_responses$)r )/R'-2 #/#4#4#6M ""7+2LA+   ] "y 'E (44%)))S5GH  '**95 ? (44CM?S 24 -A7#$%6#7$D%% #C#t$#&Z C%$*C=Z  & &t , -%((17R#ii7 /\J $,?(l6R  &;#djjl*;'  i  (44$))  s F+G error_messagerLcht|trtj|}t |||S)N)rLrMr')rcr?httpxHeadersr)rrlrLr's rget_error_classz%DashScopeRerankConfig.get_error_classs2 gt $mmG,G#!  r)rNr)NNN)NNNTNNN)NNNN)__name__ __module__ __qualname____doc__rrbr?r&robjectr0listr6boolrintrArFrnResponser LiteLLMLoggingObjrkrorrprrrrr0s (, $*$$ $ $4#'+6: ...t .  .  V ,t3 . .,CCC+/ (,(,)-)-"& 4K    d38n,- !4ZTz#Y%+ $J $J4Z @'+ !%  t   :#$('+&*H H nnH ' H ' H t H TkH H t H  H T       %    rr)rtcollections.abcrtypingrrrn litellm._uuidr*litellm.litellm_core_utils.litellm_loggingrrz)litellm.llms.base_llm.chat.transformationr+litellm.llms.base_llm.rerank.transformationr litellm.secret_managers.mainr litellm.types.rerankr r r rr common_utilsrr__annotations__rrrrrsM0$ SFH7*VEVD ,D r