uPj/UdZddlmZddlZddlmZddlmZddl m Z ddl m Z ddl mZmZdd lmZmZd d lmZd Zeed <Gdde Zy)a( Transformation logic from OpenAI /v1/embeddings format to DashScope's /v1/embeddings format. Supports - text-embedding-v4 - text-embedding-v3 Endpoint - https://dashscope.aliyuncs.com/compatible-mode/v1/embeddings Docs - https://help.aliyun.com/zh/model-studio/text-embedding-synchronous-api )FinalN)Logging)BaseLLMException)BaseEmbeddingConfig)get_secret_str)AllEmbeddingInputValuesAllMessageValues)EmbeddingResponseUsage)DashScopeErrorz1https://dashscope.aliyuncs.com/compatible-mode/v1DEFAULT_API_BASEc\eZdZdZddZdedeefdZ ddededed e def d Z dd eded ee ded ededzdedzdefdZ d dedzdedzdeded ede dzdefdZ dededed edef dZdedej"dedededzdeded edefdZdeded eej,zdefdZy)!DashScopeEmbeddingConfigz Reference: https://help.aliyun.com/zh/model-studio/text-embedding-synchronous-api DashScope exposes an OpenAI-compatible /v1/embeddings endpoint, so the request and response shapes are nearly identical to OpenAI's. returnNcyN)selfs p/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/llms/dashscope/embed/transformation.py__init__z!DashScopeEmbeddingConfig.__init__&s modelc gdS)N dimensionsencoding_formatuserr)rrs rget_supported_openai_paramsz4DashScopeEmbeddingConfig.get_supported_openai_params)s 98rnon_default_paramsoptional_params drop_paramscr|j|}|jD]\}}| ||vs|||<|Sr)ritems)rr r!rr" supportedkvs rmap_openai_paramsz*DashScopeEmbeddingConfig.map_openai_params/sQ ;;EB &,,. 'DAqyI~%&"  'rheadersmessageslitellm_paramsapi_keyapi_basecR| td}| tddd|d}i||S)NDASHSCOPE_API_KEYzZDashScope API key is required. Set 'DASHSCOPE_API_KEY' env var or pass api_key explicitly.zapplication/jsonzBearer )z Content-Type Authorization)r ValueError) rr)rr*r!r+r,r-default_headerss rvalidate_environmentz-DashScopeEmbeddingConfig.validate_environmentAsQ ?$%89G ?l /&wi0" ./-W--rstreamc|xstdxst}|jd}|jdr|S|dS)NDASHSCOPE_API_BASE/z /embeddings)rrrstripendswith)rr-r,rr!r+r4bases rget_complete_urlz)DashScopeEmbeddingConfig.get_complete_urlWsFS>*>?SCS{{3 == 'K{##rinputcP||d}dD]}|j|}||||<|S)N)rr<r)get)rrr<r!r)datakeyvalues rtransform_embedding_requestz4DashScopeEmbeddingConfig.transform_embedding_requestfsH = "C#'',E !S  " r raw_responsemodel_response logging_obj request_datac  |j} |j |j d|d|i| d| vrR| d} t | tr| j dt| n t| } t|j| d|_ | j d g|_ | j d ||_ | j d xsi} | j d d }| j d|}t|d t|d |d| vrt|d| d|S#t$r} t|jd| d} ~ wwxYw)Nz,Failed to parse DashScope response as JSON: ) status_codemessager<complete_input_dict)r<r,additional_argsoriginal_responseerrorrIlistr?rusage prompt_tokensr total_tokens)rPcompletion_tokensrQid)json Exceptionr rH post_callr> isinstancedictstrobjectr?rsetattrr )rrrCrDrEr,rFr!r+ response_jsonerMrIrOrPrQs rtransform_embedding_responsez5DashScopeEmbeddingConfig.transform_embedding_responsewsv #/#4#4#6M ""7+2LA+   m #(1EAKESWAXUYYy#e*=^abg^hG (44  !'+//;,00%@$((17R$yy!< #ii F    +"#)   = ND-*= >S  (44FqcJ  sD++ E4EE error_messagerHcht|trtj|}t |||S)N)rHrIr))rWrXhttpxHeadersr )rr_rHr)s rget_error_classz(DashScopeEmbeddingConfig.get_error_classs2 gt $mmG,G#!  r)rN)F)NNr)__name__ __module__ __qualname____doc__rrYrNrrXboolr(r r3r;rrBraResponser LiteLLMLoggingObjr^intrbrrcrrrrrs 99c9"       2##...'( .  .  .t.*. .:# $* $t $ $  $  $t  $  $'     "66nn6* 6 ' 6 t 6666 6p       %    rr)rgtypingrra*litellm.litellm_core_utils.litellm_loggingrrj)litellm.llms.base_llm.chat.transformationr.litellm.llms.base_llm.embedding.transformationrlitellm.secret_managers.mainrlitellm.types.llms.openairr litellm.types.utilsr r common_utilsr r__annotations__rrrrrusA  SFN7O8)M%M] 2] r