َjTdZddlZddlZddlmZddlmZddlZddlm Z ddl m Z dd l m Z dd lmZdd lmZmZmZmZmZmZmZmZmZGd d e eZGddeZGddeZdS)zThe Anthropic formatter module.N)ABC)Any)Field) FormatterBase)logger)_json_loads_with_repair) Msg TextBlock ThinkingBlock HintBlock DataBlock ToolCallBlockToolResultBlock URLSource Base64SourceceZdZdZdeedeeeeffdZ de deeefdzfdZ e de ezdeeeffd ZdS) _AnthropicFormatterBasezMixin for formatting Anthropic formatters to avoid duplication between AnthropicChatFormatter and AnthropicMultiAgentFormatter.msgsreturnc K||g}|D]}g}d}|D]l}|r3|r1t|ts|d|dg}d}t|t r%|jr|d|jdrt|trst|dd}||d|d t|d d}|r|d |j |d tj d t|tr|r'|rdn|j } || |dg}d}t|jtr#|dd|jdgdvg} |jD]v} t| t r| d| jd5t| t r,|| } | r| | w| r|d| dt|t r.||} | r|| Ut|t$r:|d|j|jt+|jpddt|tr|r'|s%|rdn|j } || |dg}g}|j}t|tr|r|d|dn|D]}t|t r%|jr|d|jd`_. Fuserrolecontenttexttyperredacted_thinking_dataNredacted_thinking)r data signaturethinking)r r%r$zODropping ThinkingBlock without signature; Anthropic requires a valid signature.tool_usez{})r idnameinput/r[z file returned, URL: ]z file returned, type: z(empty tool output) tool_result)r tool_use_idrTz2Unsupported block type %s in the message, skipped.)assert_list_of_msgsget_content_blocks isinstancerappendr rr getattrr%r debugrrhintstrr_format_anthropic_data_blockrr'r(r r)outputsource media_typesplitrurlwarningr )selfrmessagesmsgcontent_blockshas_tool_resultblock redacted_datar$r hint_partssub formatted_subformatted_blocktool_result_contentr8 out_block fmt_blockr9 main_typefallbacks WC:\Projects\DSCons\.venv\Lib\site-packages\agentscope\formatter\_anthropic_formatter.py_format_messagesz(_AnthropicFormatterBase._format_messages s(   &&&!Z Z C#%N#O//11K K # ,& ,'' ,OO!'NCC&(N&+OeY//| z&--%+UZ@@ }55r$+0%%M %0&--(;(5%,!' %% % *11,6051:!"!"#L!5  y11J%0)8Fvvch %)nEE*,*/!%*c22 (.-3UZ$H$H,"24 #(: E EC)#y99 E * 1 1-3SX$F$F!"!"!"!"",C!;!;E$($E$Ec$J$J!.$1!E$.$5$5m$D$D$D%$OO)/J G G y11f&*&G&G&N&NO&?&--o>>>}55a"))$."'($)J &= % 3t&&    77P&,o,)8Fvvch %)nEE*,68'"\F!&#..&&!/66)/ @ @*0 & &I))Y??&$->!&$7$>$>4:4=N)*)*%&%&%& ",Iy!A!A&,0,M,M$--"-" $-!&$7$>$>y$I$I$I$I-6-=F060A0G0G0L0LQ0OI'1&)'D'D %*-B -B-B4:J-B-B-B)1 -J -J-J5;5F-J-J-J)1%8$>$>17(J(J%&%&%&/+22%+5JKK#))$1+08':'+OONLU   "1>vvch $#1rCNc|j}|jtfd|jDst jddSdst jddS||S)a-Format a DataBlock into Anthropic API format. Args: block (`DataBlock`): The data block to format. Returns: `dict[str, Any] | None`: The formatted data block, or None if the media type is not supported. c3BK|]}tj|VdS)N)fnmatch).0patternr:s rN zG_AnthropicFormatterBase._format_anthropic_data_block..(sC   OJ 0 0      rPz(Media type %s is not supported, skipped.Nzimage/z4Anthropic only supports image data, got %s, skipped.)r9r:anysupported_input_media_typesr r= startswith_format_image_source)r>rCr9r:s @rNr7z4_AnthropicFormatterBase._format_anthropic_data_blocks&     ;       N:   4$$X..  NF   4((000rPr9ct|trdd|j|jddSt|trt |j}|dr|d}t|d5}tj |  d}dddn #1swxYwYdd|j|ddStj|d }|tj |j d}dd|j|ddSt%d t'|) aFormat an image source into Anthropic API format. Args: source (`URLSource | Base64Source`): The image source to format. Returns: `dict[str, Any]`: The formatted image source. imagebase64)r r:r#)r r9zfile://rbzutf-8N)timeoutzUnsupported source type: )r1rr:r#rr6r<rY removeprefixopenr] b64encodereaddecoderequestsgetraise_for_statusr ValueErrorr )r9r< file_pathfr#responses rNrZz,_AnthropicFormatterBase._format_image_source<s fl + +& I$"("3"K  * * Ifj//C~~i(( ,,Y77 )T**Fa!+AFFHH55<zAnthropicChatFormatter.z y 9rPEThe supported input types. Defaults to ``["text/plain", "image/*"]``.default_factory description input_typesrrc<K||d{VS)rN)rO)r>rs rNformatzAnthropicChatFormatter.formats.(**4000000000rPN) rmrnrorprrrqr6__annotations__r rrrrrtrPrNrvrvss #U99 9Kc13i1 d38n 111111rPrvc eZdZUdZeddZeed<eddZe eed <d e e d e e ee ffd Z d e e d ed e e ee ffdZede d e ee ffdZdS)AnthropicMultiAgentFormatterzmAnthropic formatter for multi-agent conversations, where more than a user and an agent are involved. zg# Conversation History The content between tags contains your conversation history z7The prompt to use for the conversation history section.)defaultr~conversation_history_promptc ddgSrxrtrtrPrNryz%AnthropicMultiAgentFormatter.rzrPr{r|rrrc"K||g}d}t|dkrG|djdkr6|||dd{Vd}d}|||d23d{V\}}|xdkr0|||d{VAdkr2||||d{Vd}xy6|S) zmFormat input messages into the structure required by the Anthropic API for multi-agent conversations.rsystemNrT tool_sequence agent_messageF) r/lenrr2_format_system_message_group_messagesextendrO_format_agent_message)r>rformatted_msgs start_indexis_first_agent_messagetypgroups rNrz#AnthropicMultiAgentFormatter.formats   &&& t99q==T!W\X55  ! !11$q'::::::::   K!% $ 4 4T+,,5G H H 3 3 3 3 3 3 3*#u$____"))"33E::::::::%__"))"88!2 .3**% !IsDis_firstcKg}g}|D]}|jpd}g}|D]}t|tr||j2t|t r[||} | rD|r-|dd|dg}|| |r/|dd|} || |r+|dd|d|r|r|d dr#|j dz|ddz|dd<n!| dd|j dzd|d  dr|d dxxd z cc<n|dd d|rd |d gSgS)z0Format agent messages into conversation history.Agentr rz:  rz z z rr) r(r0r1r r2rrr7joinrgrinsert) r>rrconversation_blocksaccumulated_textr@ agent_nameagent_text_partsrCrHrs rNrz2AnthropicMultiAgentFormatter._format_agent_messages ! 7 7C,WJ! //11 D DeY// D$++EJ7777y11 D&*&G&G&N&NO& D+2/66,2,0II6F,G,G!"!" 02,+22?CCC 7#- M M:J1K1K M M  '' 666    & &" II&677     8 "1%))&11 4#$)!,V45$A&v.. $** & $ @'!(#2&**622 #B'///>A////#**#\::  #2  rPr@cKg}|D]1}t|tr||j2ddd|dgdS)zFormat a system message.rrrrr)r0r1r r2rr)r@ text_partsrCs rNrz3AnthropicMultiAgentFormatter._format_system_messages ++-- . .E%++ .!!%*---# IIj11   rPN)rmrnrorprrr6rrrqr rrrrboolrrsrrtrPrNrrs?(-u *N (((#U99 9KcctDcN/C>I3iII d38n  IIIIV # $sCx.   \   rPr)rpr]rSabcrtypingrrfpydanticr_formatter_baser_loggingr _utils._commonr messager r r rrrrrrrrvrrtrPrNrs%% ******444444                      UIUIUIUIUImSUIUIUIp "1"1"1"1"14"1"1"1JP P P P P #:P P P P P rP