uPj9UdZddlmZmZmZddlZddlmZddlm Z ddl m Z m Z ddl mZmZer ddlZddlmZeZneZd Zeed <d Zeed <d ddddddZeeed<GddeZy)a DashScope Image Generation Configuration Handles transformation between OpenAI-compatible format and DashScope multimodal-generation API. API endpoint: POST https://dashscope-intl.aliyuncs.com/api/v1/services/aigc/multimodal-generation/generation Request format: { "model": "qwen-image-2.0-pro", "input": { "messages": [{"role": "user", "content": [{"text": ""}]}] }, "parameters": {"size": "1024*1024", "n": 1, ...} } Response format: { "output": { "choices": [{"message": {"content": [{"image": ""}]}}] }, "usage": {"output_width": 1024, "output_height": 1024, "output_image_count": 1} } ) TYPE_CHECKINGAnyFinalN)BaseImageGenerationConfig)get_secret_str)AllMessageValues#OpenAIImageGenerationOptionalParams) ImageObject ImageResponse)LoggingzYhttps://dashscope-intl.aliyuncs.com/api/v1/services/aigc/multimodal-generation/generationDEFAULT_API_BASEz/compatible-mode/v1CHAT_COMPATIBLE_MODE_PATHz256*256z512*512z 1024*1024z 1792*1024z 1024*1792z 2048*2048)256x256512x512 1024x1024 1792x1024 1024x1792 2048x2048OPENAI_TO_DASHSCOPE_SIZEc6eZdZdZdedeefdZdededede def dZ dd ed zd ed zdeded ed e d zdefdZ ddededee ded ed ed zd ed zdefdZ dededed ededef dZ ddedej dedededed eddd ed zde d zdefdZy )DashScopeImageGenerationConfigz Configuration for DashScope image generation (qwen-image-2.0, qwen-image-2.0-pro, qwen-image-3.0, qwen-image-3.0-pro). modelreturnc ddgS)Nnsize)selfrs {/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/llms/dashscope/image_generation/transformation.pyget_supported_openai_paramsz:DashScopeImageGenerationConfig.get_supported_openai_paramsFs V}non_default_paramsoptional_params drop_paramsc|j|}i}|jD]C\}}||vr ||vr|dk(r*tj||j dd|d<?|||<E|S)Nrx*)r itemsrgetreplace) rr"r#rr$supported_paramsmappedkvs rmap_openai_paramsz0DashScopeImageGenerationConfig.map_openai_paramsIs#'"B"B5"I &,,. DAqO#((F{!9!=!=a3PSAT!Uvq   r!Napi_baseapi_keylitellm_paramsstreamc|r&|jdjts|nd}|xstdxstS)N/DASHSCOPE_API_BASE_IMAGE)rstripendswithrrr )rr0r1rr#r2r3image_api_bases rget_complete_urlz/DashScopeImageGenerationConfig.get_complete_url^sA!)=)F)FG`)aHgk _0J!K_O__r!headersmessagescX|xs td}|s tdd||d<d|d<|S)NDASHSCOPE_API_KEYzDASHSCOPE_API_KEY is not setzBearer Authorizationzapplication/jsonz Content-Type)r ValueError) rr;rr<r#r2r1r0 final_api_keys rvalidate_environmentz3DashScopeImageGenerationConfig.validate_environmentlsC 'M.9L*M ;< <%,]O#< "4r!promptc^i}|jD] \}}|||< |ddd|igdgi|dS)zl Transform OpenAI-style image generation request to DashScope multimodal-generation format. r<usertext)rolecontent)rinput parameters)r() rrrCr#r2r;rJr-r.s r"transform_image_generation_requestzADashScopeImageGenerationConfig.transform_image_generation_request}sb#% #))+ DAqJqM  &%+V$4#5%  r! raw_responsemodel_response logging_obj request_dataencodingztiktoken.Encoding | None json_modec |jdk7r2|j|j|j|j |j } d| vrEd| vrA|jt | jd| |j|j|jsg|_| jdijdg} | D]d}|jdijd g}|D];}|jd }|s|jjt| =f|S#t $r0} |jd| |j|jd} ~ wwxYw) z Transform DashScope response to litellm ImageResponse. DashScope response: output.choices[0].message.content[0].image OpenAI response: data[0].url ) error_message status_coder;z5Failed to parse DashScope image generation response: NcodeoutputmessagechoicesrHimage)url) rUget_error_classrFr;json Exceptionstrr)dataappendr )rrrLrMrNrOr#r2rPr1rQ response_dataerYchoice content_list content_item image_urls r#transform_image_generation_responsezBDashScopeImageGenerationConfig.transform_image_generation_responses&  # #s *&&*//(44$,,'   #/#4#4#6M ] "x}'D&&!-"3"3I}"MN(44$,,'  """$N &**8R8<a9b      8# `* `t ` `  `  `t  `  `*##'(     t* "           L#!%77nn7& 7 ' 7  777-7t7$;7 7r!r)rltypingrrrrp5litellm.llms.base_llm.image_generation.transformationrlitellm.secret_managers.mainrlitellm.types.llms.openairr litellm.types.utilsr r tiktoken*litellm.litellm_core_utils.litellm_loggingr _LiteLLMLoggingObjrrr __annotations__rrrnrrr!rr|s2-, 8;X*u%u#858 )%+P%>Pr!