ڎj(>dZddlZddlmZddlZddlmZddlmZddl m Z dd l m Z m Z mZmZmZmZmZmZmZd e ezd eeeffd ZGd deZGddeZdS)z)The Moonshot AI formatter for agentscope.N)Any)Field)_OpenAIFormatterBase)logger) Msg URLSource Base64Source TextBlock DataBlock ThinkingBlock HintBlock ToolCallBlockToolResultBlocksourcereturnct|trd|jd|j}nFt|trt |j}|dr|d}t|d5}tj |  d}dddn #1swxYwYd|jd|}ntj|d}|tj |j d}d|jd|}nt%d t'|d d |id S) uhConvert an image source to Moonshot ``image_url`` format. Moonshot's vision API only accepts base64 data URIs or file IDs — raw remote URLs are rejected. This helper downloads remote ``http(s)://`` URLs and converts them to base64 data URIs, while ``file://`` URLs and ``Base64Source`` go through the same conversion as the OpenAI base. zdata:z;base64,zfile://rbzutf-8N)timeoutzUnsupported image source type: image_urlurl)typer) isinstancer media_typedatar strr startswith removeprefixopenbase64 b64encodereaddecoderequestsgetraise_for_statuscontent ValueErrorr)rrurl_str local_pathfencodedresponses VC:\Projects\DSCons\.venv\Lib\site-packages\agentscope\formatter\_moonshot_formatter.py_moonshot_format_image_sourcer1s&,''K>f'>>>> FI & &Kfj//   i ( ( ? --i88Jj$'' E1 *1668844;;GDD E E E E E E E E E E E E E E E>&+>>W>>CC|GR888H  % % ' ' '&x'788??HHG>&+>>W>>CCI4<<IIJJJS\  s:CCCceZdZUdZeddZeeed<de e zde ee ffdZ d eedee ee ffd Zd S) MoonshotChatFormatteraThe Moonshot AI formatter for chatbot scenario. Moonshot's API is OpenAI-compatible, but thinking models (``kimi-k2.6``, ``kimi-k2-thinking``) return a ``reasoning_content`` field alongside ``content`` in assistant messages. This formatter preserves that field when re-sending assistant messages back to the API so that the *Preserved Thinking* feature works correctly in multi-turn conversations. c gdSN)z text/plainzimage/*zaudio/*r6r0zMoonshotChatFormatter.F D D Dr7PThe supported input types. Defaults to ``["text/plain", "image/*", "audio/*"]``.default_factory description input_typesrrc t|SNr1selfrs r0_format_image_sourcez*MoonshotChatFormatter._format_image_sourceM-V444r7msgsc" K||g}d}|t|kr_||}g}g}g}|D]}t|tr||j3t|tr|d|jdft|tr-| |} | || t|tr:|s|s|r[|j |j |pdd} |j dkr|rd|nd| d <|r|| d <|| g}g}g}t|jt r#|d d|jdgd \g} |jD]v} t| tr| d| jd5t| tr,| | } | | | w| r|d | d t|t"r-||jd |j |jdd:t|t(r<|s|s|r[|j |j |pdd}|j dkr|rd|nd|d <|r||d <||g}g}g}||j\}}|d|j||j d|rg}|D]v}t|tr|d|jd5t|tr,| |}|||w|r|d d|dt/jdt3||j |j |pdd} |j dkr|rd|nd| d <|r|| d <| ds| d s|r|| |dz }|t|k_|S)aFormat messages into the Moonshot / OpenAI-compatible API format. Behaves identically to :class:`OpenAIChatFormatter` except that :class:`ThinkingBlock` content is placed into the ``reasoning_content`` field of the assistant message dict (required for Preserved Thinking). Args: msgs (`list[Msg]`): The list of message objects to format. Returns: `list[dict[str, Any]]`: The formatted messages as a list of dictionaries. rtextrrHN)rolenamer) assistant reasoning_content tool_callsuserrJr)function)rK arguments)idrrStool)rJ tool_call_idr)rKzsystem-reminderz2Unsupported block type %s in the message, skipped.r)r)assert_list_of_msgslenget_content_blocksrrappendthinkingr rHr _format_openai_data_blockrrJrKjoinhintrrrUinputrconvert_tool_result_to_stringoutputrwarningrr')rCrFmessagesimsgcontent_blocksreasoning_partsrPblock formatted msg_moonshot hint_partssub formatted_sub msg_flushtextual_outputmultimodal_data promo_contentitemfmt_items r0formatzMoonshotChatFormatter.formatTs$   &&&! #d))mmq'C)+N)+O%'J//11E E e]33D$**5>::::y11"))65:*N*NOOOOy11| $ > >u E EI ,&--i888y11w%((($'H$'H'5'=(( 8{22$3!( / : : :%'))<= &D9CL6  555)+*,%' !%*c22 (.-3UZ$H$H,"24 #(: E EC)#y99 E * 1 1-3SX$F$F!"!"!"!"",C!;!;E040N0N$'1"1" $1#<$.$5$5m$D$D$D%$OO)/J G G }55H%%"'($.(- -2[))     77<%((($'H$'H'5'=%% 8{22$3!( / : : :%'&&9: &A6@Il3  222)+*,%' ::5<HH&'OO$*,1H'5$)J '(* $3 C CD)$ :: C - 4 4-3TY$G$G!"!"!"!"",D)!listr__annotations__r r dictrrDr rur6r7r0r3r3;s#UDD DKc5L(5 c3h5555~3i~ d38n ~~~~~~r7r3c peZdZUdZeddZeed<eddZe eed <d e e zd e ee ffd Zd e ed e e ee ffdZd e ed e e ee ffdZ dd e eded e e ee ffdZeded e ee ffdZdS)MoonshotMultiAgentFormattera5Formatter for the Moonshot AI API in multi-agent conversations. Moonshot's API is OpenAI-compatible, so the multi-agent history collapsing strategy is the same as :class:`OpenAIMultiAgentFormatter`. Tool sequences are delegated to :class:`MoonshotChatFormatter` so that ``reasoning_content`` is preserved correctly for multi-turn *Preserved Thinking* conversations. .. note:: Telling the assistant's name in the system prompt is important in multi-agent conversations so that the model knows which role it is playing. zg# Conversation History The content between tags contains your conversation history z7The prompt to use for the conversation history section.)defaultr=conversation_history_promptc gdSr5r6r6r7r0r8z$MoonshotMultiAgentFormatter.-r9r7r:r;r>rrc t|Sr@rArBs r0rDz0MoonshotMultiAgentFormatter._format_image_source4rEr7rFc"K||g}d}|rG|djdkr6|||dd{Vd}d}|||d23d{V\}}|dkrB|t |j|d{VR|dkr1|| ||d{Vd }6|S) aFormat input messages into the Moonshot AI API format for multi-agent conversations. Non-tool messages from all agents are collapsed into a single user message with ```` tags. Tool call / result sequences are delegated to :class:`MoonshotChatFormatter`. Args: msgs (`list[Msg]`): The list of message objects to format. Returns: `list[dict[str, Any]]`: The formatted messages as a list of dictionaries. rsystemNrT tool_sequencer> agent_messageF) rXrJr[_format_system_message_group_messagesextendr3r>ru_format_agent_message)rCrFformatted_msgs start_indexis_first_agent_messagetypgroups r0ruz"MoonshotMultiAgentFormatter.format:s   &&&%'  DGLH,,  ! !11$q'::::::::   K!% $ 4 4T+,,5G H H / / / / / / /*#uo%%%%/$($4fUmm$$$$$$ ''%%44. */&!I sDcbKt|j|d{VS)zOFormat a sequence of tool-related messages using MoonshotChatFormatter.rN)r3r>ru)rCrFs r0_format_tool_sequencez1MoonshotMultiAgentFormatter._format_tool_sequencegsN +(   &,, r7Tis_firstcK|r|j}nd}g}g}|D]}|D]}}t|tr%||jd|j`` user message.rNz: NrMz z rHrIrQrR) rrZrr r[rKrHr r]r^r) rCrFrraccumulated_text media_blocksrfrirj history_text content_lists r0rz1MoonshotMultiAgentFormatter._format_agent_messageqsz  -*.*J ' '*, '&(#%  7 7C//11 7 7eY//7$++sx,G,G5:,G,GHHHHy117 $ > >u E EI ,$++I666  7    Iyy!122  + !! .0  H    F F G G GL)))L99::r7rfc4Kd|dS)z0Format a system message for the Moonshot AI API.rrR)get_text_content)rfs r0rz2MoonshotMultiAgentFormatter._format_system_messages)++--   r7N)T)rvrwrxryrrrr{r>rzr r r|rrDr rurboolr staticmethodrr6r7r0r~r~s  (-u *N (((#UDD DKc5L(5 c3h5555 +c+tDcN/C++++Z3i d38n (;(;3i(;(; d38n  (;(;(;(;T # $sCx.   \   r7r~)ryr"typingrr&pydanticr_openai_formatterr_loggingrmessager r r r r rrrrr|rr1r3r~r6r7r0rsx// 333333                         $  #s(^    FWWWWW0WWWtL L L L L "6L L L L L r7