ڎjMdZddlZddlmZmZddlmZddlmZddl m Z dd l m Z m Z mZmZmZmZmZmZmZGd d eZGd d eZdS)adThe xAI formatter module. This formatter converts AgentScope ``Msg`` objects into the protobuf ``Message`` objects expected by the ``xai_sdk`` gRPC client. Unlike every other formatter, the ``format()`` method returns a list of ``chat_pb2.Message`` proto objects rather than plain dicts, because the ``xai_sdk`` chat API accepts proto messages directly. N)AnyList)Field) FormatterBase)logger) Msg TextBlock ThinkingBlock ToolCallBlockToolResultBlock DataBlock URLSource Base64Source HintBlockceZdZUdZegddZeeed<dee de de e fd Z d ed e defd Z d e defdZdS)XAIChatFormatteraFormatter for the xAI chat model. Converts ``Msg`` objects into ``xai_sdk`` protobuf ``Message`` objects that can be appended directly to a ``xai_sdk`` chat session. Unlike other formatters whose ``format()`` returns ``list[dict]``, this formatter returns ``list[chat_pb2.Message]``. The type annotation is intentionally widened to ``list[Any]`` to accommodate this difference. z text/plainz image/jpegz image/pngUThe supported input types. Defaults to ``["text/plain", "image/jpeg", "image/png"]``.default description input_typesmsgskwargsreturnc Kddlm}m}m}m}m}m}||g} |D]} | } d| D} | j dkr>d d| D} | || l| j dkrg}| D]}t|trt|tr|r| ||g}t|jt r$| ||j||j|}|r| ||t|t$r| |jt|t(r|jjdrbt|jt0rt!|jj}|d r|d }t7|d 5}t9j|d }d d d n #1swxYwY| |d |jjd|| ||2t|jt@r8| |d |jjd|jj!tEj#d|jjtEj#dtI|j%|r| ||h| j dkrg}g}| D]}t|tLr}|r|'}|j()d|_ |r?|j*+}d d|D|_g}|D]h}|j,+}|j-|_-|j.)d|_$|j/|j0_/|j1|j0_2i| |g}nC|rAd d|D} | r| || g}|3|j4}| |||j-t|tjr| |t|t$r| |t|trt|tr|s|r<|r|'}|j()d|_ |r?|j*+}d d|D|_g}|D]h}|j,+}|j-|_-|j.)d|_$|j/|j0_/|j1|j0_2i| |g}nC|rAd d|D} | r| || g}t|jt r%| ||j||j|}|r| |||r|'}|j()d|_ |r?|j*+}d d|D|_g}|D]h}|j,+}|j-|_-|j.)d|_$|j/|j0_/|j1|j0_2i| |D|r?d d|D} | r| || tEj#d| j | S)aConvert a list of ``Msg`` objects to ``xai_sdk`` proto messages. Args: msgs (`list[Msg]`): A list of ``Msg`` objects representing the conversation. **kwargs (`Any`): Unused; retained for interface compatibility. Returns: `list[Any]`: A list of ``chat_pb2.Message`` proto objects, ready to be appended to a ``xai_sdk`` chat session via ``chat.append()``. r) assistantimagesystem tool_resultuserchat_pb2c<g|]}t|t|S) isinstancer .0bs QC:\Projects\DSCons\.venv\Lib\site-packages\agentscope\formatter\_xai_formatter.py z+XAIChatFormatter.format..Us'III 1i0H0HI1IIIr! c3$K|] }|jV dSNtextr(s r+ z*XAIChatFormatter.format..Xs$ = =A = = = = = =r-r#image/file://rbutf-8Ndata:;base64,zoUnsupported media type %s for xAI API. Only image/jpeg and image/png are supported. This block will be skipped.z3Unsupported block type %s in user message, skipped.rROLE_ASSISTANTc3$K|] }|jV dSr0r1r(s r+r3z*XAIChatFormatter.format..s53"3"/0AF3"3"3"3"3"3"r-TOOL_CALL_TYPE_CLIENT_SIDE_TOOLc3$K|] }|jV dSr0r1r(s r+r3z*XAIChatFormatter.format..s$,J,JQV,J,J,J,J,J,Jr-) tool_call_idc3$K|] }|jV dSr0r1r(s r+r3z*XAIChatFormatter.format..s57&7&347&7&7&7&7&7&r-c3$K|] }|jV dSr0r1r(s r+r3z*XAIChatFormatter.format..s$0N0NA0N0N0N0N0N0Nr-c3$K|] }|jV dSr0r1r(s r+r3z*XAIChatFormatter.format..s$*H*Ha16*H*H*H*H*H*Hr-c3$K|] }|jV dSr0r1r(s r+r3z*XAIChatFormatter.format..s$$B$BQV$B$B$B$B$B$Br-z'Unsupported message role '%s', skipped.)6 xai_sdk.chatrr r!r"r#r$assert_list_of_msgsget_content_blocksrolejoinappendr'r rhintstr_xai_user_args_from_blocksr r2rsource media_type startswithrurl removeprefixopenbase64 b64encodereaddecoderdatar warningtype__name__rMessage MessageRoleValuecontentadd tool_callsid ToolCallTypenamefunctioninput arguments_extract_result_textoutputr )selfrrrr r!r"r#r$ xai_messagesmsgblocks text_blocksr2 content_argsblock hint_argsurl_str local_pathfencoded pending_textpending_tool_calls msg_protoctcproto_tc output_texts r+formatzXAIChatFormatter.format2sV &                   &&&"$ E E C++--FIIfIIIKx8##yy = = = = ===##FF4LL1111V##%' #::E!%779#E9557'.(//l0CDDD+-L%ej#66F(//UZ0@0@AAAA(,(G(G % %))I )F , 3 3DD)4D E E E#E955*$++EJ7777#E955( <2==hGG!)%, BB"*-el.>*?*?#*#5#5i#@#@!H181E1E(12&2&J*.j$)?)?%:1282B,-FFHH3*3**0&//)0%:%:%:%:%:%:%:%:%:%:%:%:%:%:%:%1$7$7(--@EL4K-@-@6=-@-@)*)*%&%&%&%&%1$7$7g$G$G$G$G!+EL,!G!G" , 3 3$)E)F 0G)F)F27,2C)F)F%&%&!"!"!"#N!>!& 7 ' KK0  = ''l(;<<<[((02 :<"#XFXFE!%99WF..(0(8(8(:(:I-5-A-G-G 0..IN ,2$-$5$9$9$;$;)-3"3"4@3"3"3"*"*"02 &8GG+4+?+C+C+E+E.0e 080E0K0K$E1"1" :< 1 6>@h 1 ; ;(// :::13..).#'99,J,J\,J,J,J#J#JD#E , 3 3IIdOO D D D+-L&*&?&?!L'' %++'K +-2X$E=99/F*11%8888#E955,F$++E2222#E=99)F#E955&F-2212,4,<,<,>,> 191E1K1K$42"2" $0!6(1(9(=(=(?(?A-1YY7&7&8D7&7&7&.&.&AF46L*< !K !KB/8/C/G/G/I/IH24%HK(0(=(C(C,M)*)*%-M >@WH$5$:BD(H$5$?$? , 3 3I > > >57 2 2!-2'+yy0N0N0N0N0N'N'N#'!I$0$7$7 $$H$H$H/1 %ej#66F(//UZ0@0@AAAA(,(G(G % %))I )F , 3 3DD)4D E E E%= ( 0 0 2 2I%-%9%?%?(&&IN$*%-1133!%*H*H<*H*H*H!H!H') 0??#,#7#;#;#=#=&(e (0(=(C(C=)) 24).68h)33 '' 2222!=99$B$B\$B$B$BBBD=$++IIdOO<<<=H s9:I??J J rkr c g}|D]}t|tr||j3t|tr|jjds tj d|jjt|jtrt|jj }|dr| d}t|d5}tj|d}dddn #1swxYwY||d|jjd|v|||t|jt&r8||d|jjd|jj|S) aConvert a list of ``TextBlock | DataBlock`` into the positional args expected by ``xai_sdk.chat.user(*args)``. DataBlocks that are not images (or use an unsupported media type) are dropped with a warning, matching the existing user-role DataBlock handling above. Args: blocks (`list`): The ``hint`` list from a :class:`HintBlock`. image (`Any`): The ``xai_sdk.chat.image`` constructor, passed in to keep the local import contract identical to ``format``. Returns: `list`: Positional args for ``user(*args)``; empty if nothing survived. r4zxUnsupported media type %s for xAI API. Only image/jpeg and image/png are supported. This hint sub-block will be skipped.r5r6r7Nr8r9)r'r rHr2rrLrMrNr rWrrJrOrPrQrRrSrTrUrrV) rhrkr argssubrprqrrrss r+rKz+XAIChatFormatter._xai_user_args_from_blockssG0" " C#y))!  CH%%%%C++ z,77AAN? -  cj)44!#*.11G)))44 4%,%9%9)%D%D !*d33q&,&6qvvxx&@&@&G&G '''G !E!4 (=!4!4*1!4!4 EE'NN3333 L99KK8CJ$988&)jo88  s::EE E rgc|dSt|tr|St|trg}|D]}t|tr||j2t|tr||]|t|d|St|S)aUExtract a plain-text string from a ``ToolResultBlock`` output. Args: output (`Any`): The raw output of a ``ToolResultBlock``, which may be a string, a list of blocks, or another type. Returns: `str`: A plain-text representation of the output. Nr.)r'rJlistr rHr2rG)rhrgpartsitems r+rfz%XAIChatFormatter._extract_result_textYs >2 fc " " M fd # # $E , ,dI..,LL++++c**,LL&&&&LLT++++99U## #6{{r-N)rY __module__ __qualname____doc__rrrrJ__annotations__r rrr{rKrfr&r-r+rrs#U999 IKcg3igg c ggggR<<<  <<<<|33r-rceZdZUdZeddZeed<egddZe eed<d e e d e d e e fd Z d e e d ed efdZdS)XAIMultiAgentFormatteraFormatter for the xAI chat model in multi-agent conversations. Produces ``xai_sdk`` protobuf ``Message`` objects (same as :class:`XAIChatFormatter`). Prior agent-to-agent messages are collapsed into a single ``user`` message with ```` tags; tool call / result sequences are delegated to :class:`XAIChatFormatter`. .. note:: ``format()`` returns ``list[Any]`` (protobuf messages), not ``list[dict]``. zg# Conversation History The content between tags contains your conversation history z7The prompt to use for the conversation history section.rconversation_history_promptrrrrrrcJKddlm}m}||g}d}|rK|djdkr:|d}|||d}d}|||d23d{V\} } | dkrB|t|j  | d{VR| d kr9| | | } | r||| d }6|S) aaConvert a list of ``Msg`` objects to ``xai_sdk`` proto messages. Conversation history (non-tool messages) is collapsed into a single ``user`` protobuf message containing ```` tags. Tool sequences are formatted by :class:`XAIChatFormatter`. Args: msgs (`list[Msg]`): A list of ``Msg`` objects representing the conversation. **kwargs (`Any`): Unused; retained for interface compatibility. Returns: `list[Any]`: A list of ``chat_pb2.Message`` proto objects. r)r!r#r!rTN tool_sequence)r agent_message)is_firstF) rCr!r#rDrFget_text_contentrH_group_messagesextendrrr{_build_history_text) rhrrr!r#ri start_indexr2is_first_agent_messagetypgroup history_texts r+r{zXAIMultiAgentFormatter.formats* .-------   &&&"$   DGLH,,7++--D   t - - -K!% $ 4 4T+,,5G H H / / / / / / /*#uo%%##*$($4fUmm$$$$$$ ''#773 8    < ''\(:(:;;;).&!I sD"rcg}|D]}g}|jr||jd|D]1}t|tr||j2|r(|d||sdS|r|jnd}|dzd|zdzS)uBuild a ``…`` text block from agent messages. Args: msgs (`list[Msg]`): Non-tool messages to collapse into history. is_first (`bool`): When ``True``, prepend :attr:`conversation_history_prompt` before the tag. Returns: `str`: The formatted history string, or an empty string when there is no text content. : rz r.z )rbrHrEr'r r2rGr)rhrrlinesrjrrnprefixs r+rz*XAIMultiAgentFormatter._build_history_texts . .C!Ex - ^^^,,,//11 - -eY//-LL,,, . SXXe__--- 25=E112 % %(8(88>IIr-N)rYrrrrrrJrrrr rrr{boolrr&r-r+rrvs  (-u *N (((#U999 IKc23i22 c 2222hJS JJJJJJJJr-r)rrRtypingrrpydanticr_formatter_baser_loggingr messager r r r rrrrrrrr&r-r+rsQ ******                      UUUUU}UUUp oJoJoJoJoJ]oJoJoJoJoJr-