ڎj8 dZddlZddlZddlZddlmZddlmZddlmZm Z m Z ddl Z ddl m Z mZddlmZmZmZmZmZGd d e ZdS) zThe formatter module.N)abstractmethod)fnmatch)AnyListAsyncGenerator) BaseModelField)Msg DataBlock TextBlock URLSource Base64Sourcec ReZdZUdZeddZeeed< e deefdZ e de d e dee ee ffd Zed eedd fd ZdeeeezzdeeeeezffdZed eedefdZd S) FormatterBasezThe base class for formatters.cdgS)N text/plainrRC:\Projects\DSCons\.venv\Lib\site-packages\agentscope\formatter\_formatter_base.pyzFormatterBase.sraThe supported input types, aligned with the model card's ``input_types`` field. Entries other than ``text/plain`` and ``application/x-thinking`` are treated as media-type patterns (glob-style, e.g. ``image/*``, ``audio/mp3``) that control which ``DataBlock``\s are forwarded to the API.)default_factory description input_typesreturnc$d|jDS)zDerive the accepted media-type patterns from :attr:`input_types` by excluding ``text/plain`` and ``application/x-thinking``.cg|]}|dv| S))rzapplication/x-thinkingr).0ts r z=FormatterBase.supported_input_media_types..*s.   @@@ @@@r)r)selfs rsupported_input_media_typesz)FormatterBase.supported_input_media_types&s&  %    rargskwargsc KdS)z[Format the Msg objects to a list of dictionaries that satisfy the API requirements.Nr)r!r#r$s rformatzFormatterBase.format0s rmsgsNct|tstd|D]7}t|ts tdt |d8dS)zAssert that the input is a list of Msg objects. Args: msgs (`list[Msg]`): A list of Msg objects to be validated. z$Input must be a list of Msg objects.zExpected Msg object, got z instead.N) isinstancelist TypeErrorr type)r'msgs rassert_list_of_msgsz!FormatterBase.assert_list_of_msgs5s|$%% DBCC C  Cc3'' DS DDD   routputc  t|tr|gfSg}g}|D] t tr| j3t t r jjdd}t fd|j Dr\tj }|d|d|d| td|d|d   gt jtr(|d|d jjd ,t jt rt#j jj}t'j|d 5}t+j jj}|||d|d|jd dddn #1swxYwY|r#td g|td }d||fS)aTurn the tool result list into a textual output to be compatible with the LLM API that doesn't support multimodal data in the tool result. For URL-based images, the URL is included in the list. For base64-encoded images, the local file path where the image is saved is included in the returned list. Args: output (`str | List[TextBlock | DataBlock]`): The output of the tool response, including text and multimodal data like images and audio. Returns: `tuple[str, list[TextBlock | DataBlock]]`: A tuple containing the textual representation of the tool result and a list of blocks to be promoted as a user message. /rc3LK|]}tjj|VdS)N)rsource media_type)r_blocks r z>FormatterBase.convert_tool_result_to_string..jsFEL3Q77rzA(n) zD file is returned and will be presented to you with the identifier [z].z- z (z file): )textz2 file is returned and can be accessed at the URL: z.F)suffixdeletez( file is returned and saved locally at: NzQThe multimodal data and their identifiers are listed as follows:z )r)strr appendr8r r3r4splitanyr" shortuuiduuidextendrurlr mimetypesguess_extensiontempfileNamedTemporaryFilebase64 b64decodedatawritenamejoin) r!r/textual_outputmultimodal_data main_type identifier extension temp_file decoded_datar6s @rconvert_tool_result_to_stringz+FormatterBase.convert_tool_result_to_stringFs6. fc " " 2:  "; ; E%++: %%ej1111E9--7 !L399#>>qA !=4"+!1!1J"))===&=== $**%%K*%K%K %K%K%K"   i88"))...;@<;K...   l;; ) 9 /!!I"4($ #'-'7 8I'J'J ! 555&--2Y22>Gn222                 9  !  - Oyy((/99sAG<<H H c Kd}g}|D]}|E|ds|drd}nd}||I|dkrM|ds|dr||||fWV|g}d}|dkrL|ds|dr ||fWV|g}d}|||r ||fWVdSdS)zGroup messages into tool sequences and agent messages. Args: msgs (`list[Msg]`): A list of Msg objects to be grouped. N tool_call tool_result tool_sequence agent_message)get_content_blocksr=has_content_blocks)r' group_typegroupr-s r_group_messageszFormatterBase._group_messagess  & &C!))1++M::1"1JJ!0J S!!!_,,))1++M::1LL%%%%$e+++++ EE!0JJ..))&++M::&%e+++++ EE!0JJLL%%%  $e# # # # # # # $ $r)__name__ __module__ __qualname____doc__r rr*r<__annotations__propertyr"rrdictr& staticmethodr r.rr r tuplerUrr_rrrrrsx(("U.. 9   Kc   % T#Y   X #d38n9M^$s)\ g:d9y011g: sDY.// 0g:g:g:g:R)$DI)$.)$)$)$\)$)$)$rr)rcrHrDrFabcrrtypingrrrr@pydanticrr messager r r rrrrrrrms ,,,,,,,,,,%%%%%%%%C$C$C$C$C$IC$C$C$C$C$r