َj@RdZddlZddlmZddlmZddlmZddlmZddl m Z d d l m Z d d l mZmZmZmZmZmZmZmZmZGd d e eZGddeZGddeZdS)z:The DashScope formatter module (OpenAI-compatible format).N)Any)fnmatch)ABC)Field) FormatterBase)logger) Msg TextBlock ThinkingBlockToolResultBlock URLSource DataBlock ToolCallBlock Base64Source HintBlockcReZdZUdZeddZeeed<e deefdZ e de fdZ d e deeefd zfd Zed eezdeeeffd Zed eezdeeeffdZed eezdeeeffdZd S)_DashScopeFormatterBasezqBase class for DashScope formatters (OpenAI-compatible format), providing shared data block formatting logic.c gdS)N) text/plainzimage/*zaudio/*zvideo/*rWC:\Projects\DSCons\.venv\Lib\site-packages\agentscope\formatter\_dashscope_formatter.pyz _DashScopeFormatterBase.s! ! ! ra The supported input types, aligned with the model card's ``input_types`` field. Media types (non ``text/plain`` / ``application/x-thinking`` entries) are used to filter ``DataBlock``\s; ``application/x-thinking`` enables passing ``reasoning_content`` back to the API.)default_factory description input_typesreturnc$d|jDS)zwDerive supported media types from :attr:`input_types`, excluding ``text/plain`` and ``application/x-thinking``.cg|]}|dv| S))rapplication/x-thinkingr).0ts r zG_DashScopeFormatterBase.supported_input_media_types..2s.   @@@ @@@rrselfs rsupported_input_media_typesz3_DashScopeFormatterBase.supported_input_media_types.s&  %    rcd|jvS)zReturn ``True`` if ``application/x-thinking`` is listed in :attr:`input_types`, meaning the model accepts ``reasoning_content`` in the conversation history.r"r&r's rsupports_thinking_inputz/_DashScopeFormatterBase.supports_thinking_input8s (4+;;;rblockNctfd|jDs:tjdjjd|jdSjjdd}|dkr|jS|dkr| jS|d kr| jStjd |dS) aMFormat a DataBlock into the OpenAI-compatible format for DashScope API. Supports: - Images: ``{"type": "image_url", "image_url": {"url": ...}}`` - Videos: ``{"type": "video_url", "video_url": {"url": ...}}`` - Audio: ``{"type": "input_audio", "input_audio": {...}}`` Args: block (`DataBlock`): The DataBlock to format. Returns: `dict[str, Any] | None`: A dictionary representing the formatted DataBlock, or ``None`` if the media type is unsupported. c3LK|]}tjj|VdS)N)rsource media_type)r#patternr,s r zG_DashScopeFormatterBase._format_dashscope_data_block..TsF   EL+W 5 5      rz]Unsupported media type %s for DashScope API. Supported types: %s. This block will be skipped.z, N/rimagevideoaudiozMUnsupported main media type %s for DashScope API. This block will be skipped.) anyr)r warningr/r0joinsplit_format_image_source_format_video_source_format_audio_source)r(r, main_types ` r_format_dashscope_data_blockz4_DashScopeFormatterBase._format_dashscope_data_block?s*    ;      N9 ' $:;;     4L+11#66q9   ,,U\:: :   ,,U\:: :   ,,U\:: : *    trr/c(t|trd|jd|j}nt|trt |j}|dr|d}t|d5}tj |  d}dddn #1swxYwYd|jd|}n"|}ntdt|dd |id S) zConvert an image source to OpenAI-compatible ``image_url`` format. Local ``file://`` URLs are read from disk and converted to base64 data URIs. Remote URLs are passed through unchanged. data:;base64,file://rbutf-8NzUnsupported image source type: image_urlurl)typerF isinstancerr0datarstrrG startswith removeprefixopenbase64 b64encodereaddecode ValueErrorrHr/rGurl_str local_pathfencodeds rr;z,_DashScopeFormatterBase._format_image_sourcersm fl + + OB&+BBV[BBCC  * * O&*ooG!!),, $11)<< *d++Iq$.qvvxx88??HHGIIIIIIIIIIIIIIIBf/BBBBMtF||MMNN N     :CCCc(t|trd|jd|j}nt|trt |j}|dr|d}t|d5}tj |  d}dddn #1swxYwYd|jd|}n"|}ntdt|dd |id S) zConvert a video source to DashScope's ``video_url`` format (OpenAI-compatible extension). Local ``file://`` URLs are read from disk and converted to base64 data URIs. Remote URLs are passed through unchanged. rArBrCrDrENzUnsupported video source type: video_urlrG)rHr\rIrUs rr<z,_DashScopeFormatterBase._format_video_sourcesm fl + + OB&+BBV[BBCC  * * O&*ooG!!),, $11)<< *d++Iq$.qvvxx88??HHGIIIIIIIIIIIIIIIBf/BBBBMtF||MMNN N    rZct|tr-|jdd}d|j|ddSt|t rt |j}|jdd}|dr~| d}t|d5}tj | d}d d d n #1swxYwYd||ddSd||ddStd t!|) aConvert an audio source to DashScope ``input_audio`` format. DashScope's compatible API accepts URLs directly in the ``data`` field (unlike standard OpenAI which requires base64). Local files are still read and base64-encoded. r3 input_audio)rKformat)rHr_rCrDrENzUnsupported audio source type: )rJrr0r:rKrrLrGrMrNrOrPrQrRrSrTrH)r/fmtrVrWrXrKs rr=z,_DashScopeFormatterBase._format_audio_sources fl + + #))#..r2C%"K!    fi ( ( &*ooG#))#..r2C!!),, $11)<< *d++Fq!+AFFHH55<>>y11v%,,~,$'H'5'=66 &B7AJ|4)>Bii .??J':;'--j999)+%' )+!%*c22&--(.-3UZ$H$H,"24 #(: E EC)#y99 E * 1 1-3SX$F$F!"!"!"!"",C!;!;E$($E$E(+%&%&!. $1!E$.$5$5m$D$D$D%*11)/J G G }55H%%"'($.(- -2[))     }55<3>&--en===778%,,~,$'H'5'=55 &A6@Il3)=AYY .>>I&9:'--i888)+%' )+ ::5<HH&'#))$*,1H'5$)J '(* $3 C CD)$ :: C - 4 4-3TY$G$G!"!"!"!"",D)! tags contains your conversation history )rdefaultconversation_history_promptrnrcKg}d}t|dkrG|djdkr6|||dd{Vd}d}|||d23d{V\}}|xdkr0|||d{VAdkr2||||d{Vd}xy6|S) aFormat input messages into the structure required by the DashScope OpenAI-compatible API. To support multi-agent conversations, this formatter processes messages as follows: - Prepends an instruction before the first conversation history section. - Combines conversation turns into a history section, where each entry is formatted as ``{name}: {content}``. - Wraps the conversation history with ```` and ```` tags. Returns: `list[dict[str, Any]]`: A list of dictionaries formatted for the DashScope API. rsystemNrT tool_sequence agent_messageF)rrsr_format_system_message_group_messagesextend_format_tool_sequence_format_agent_message)r(rnr start_indexis_first_agent_messagetypgroups rr`z#DashScopeMultiAgentFormatter.formats& t99q==T!W\X55  ! !11$q'::::::::   K!% $ 4 4T+,,5G H H 3 3 3 3 3 3 3*#u$____"))"88????????%__"))"88!2 .3**% !IsC9cbKt|j|d{VS)zrGiven a sequence of tool call/result messages, format them into the required format for the DashScope API.r&N)rmrr`)r(rns rrz2DashScopeMultiAgentFormatter._format_tool_sequencesN ,(   &,, rTis_firstcK|r|j}nd}g}g}g}g}|D]}|D]}} t| tr%||jd| j r^z z c3BK|]}|ddVdS)rprN)r)r#bs rr2zEDashScopeMultiAgentFormatter._format_agent_message..sA- - "#AEE&"  - - - - - - rrqrxrr) rrrJr rrzrprr?r9rinsertr) r(rnrrrconversation_blocksaccumulated_text media_blocksrr,rconversation_blocks_text content_lists rrz2DashScopeMultiAgentFormatter._format_agent_messages  -*.*J ' '*, '%'$&#%  = =C//11 = =eY//=$++sx,G,G5:,G,GHHHHy11=&*&G&G''O'2$++O<<< =    & &#3445     C"1%))&11 /#$)!,V45$A&v.. $**8=HI #2&**622 C#B'///>A////#**FL+ABBB#'99- - ':- - - $ $  .0 #    )ABB    L)))  M  ! !6l"K"K L L Lrrc4Kd|dS)z(Format system message for DashScope API.rrr)get_text_content)rs rrz3DashScopeMultiAgentFormatter._format_system_message+s) ++--   rN)T)rbrcrdrerrrLrgrfr rjr`rrrirrkrrrrrrs@ (-u6 *(((+c+tDz++++Z3i d38n BB3iBB d38n  BBBBH  c3h   \   rr)rerPtypingrrabcrpydanticr_formatter_baser_loggingr messager r r rrrrrrrrmrrrrrs@@ ******                      |K|K|K|K|KmS|K|K|K~zzzzz4zzzz] ] ] ] ] #:] ] ] ] ] r