َj)-dZddlmZddlmZddlmZddlmZddl m Z m Z m Z m Z mZmZmZGd d eZGd d eZd S)zThe DeepSeek formatter module.)Any)Field) FormatterBase)logger)Msg TextBlock DataBlock ThinkingBlock HintBlock ToolCallBlockToolResultBlockc~eZdZUdZeddZeeed<dee dee ee ffdZ d S) DeepSeekChatFormatterzThe DeepSeek formatter class for chatbot scenario, where only a user and an agent are involved. We use the `role` field to identify different entities in the conversation. cdgSNz text/plainrVC:\Projects\DSCons\.venv\Lib\site-packages\agentscope\formatter\_deepseek_formatter.pyzDeepSeekChatFormatter.rgThe supported input types. Defaults to ``["text/plain"]`` (DeepSeek does not support multimodal input).default_factory description input_typesmsgsreturnc K||g}|D]}g}g}g}|D]}t|tr|d|jd6t|t r||jft|trZ|s|s|rxd d|D}|j |p|rdndd} |j dkr|rd |nd| d <|r|| d <|| g}g}g}t|j tr|d |j d3g} |j D]j} t| tr| | j2t| tr#| d | jjd k| r+|d d | dt|t r-||jd|j|jddt|t(r|s|s|rxd d|D}|j |p|rdndd} |j dkr|rd |nd| d <|r|| d <|| g}g}g}||j\} }|d|j| |jdt/jdt3|d d|D}|j |p|rdndd}|j dkr|rd |nd|d <|r||d <|ds|d s|r|||S)aFormat message objects into DeepSeek API format. Args: msgs (`list[Msg]`): The list of message objects to format. Returns: `list[dict[str, Any]]`: The formatted messages as a list of dictionaries. text)typer! c3BK|]}|ddVdSr!Nget.0bs r z/DeepSeekChatFormatter.format..EA1123AEE&"--111111rNr&rolecontent assistantreasoning_content tool_callsuser[z* attached, not supported by this provider]function)name arguments)idr"r6c3BK|]}|ddVdSr%r'r)s rr,z/DeepSeekChatFormatter.format..r-rtool)r/ tool_call_idr0r7z2Unsupported block type %s in the message, skipped.c3BK|]}|ddVdSr%r'r)s rr,z/DeepSeekChatFormatter.format..s0#N#N!AEE&"$5$5#N#N#N#N#N#Nrr0)assert_list_of_msgsget_content_blocks isinstancer appendr!r thinkingr joinr/hintstrr source media_typerr9r7inputrconvert_tool_result_to_stringoutputrwarningr"r()selfrmessagesmsgcontent_blocksreasoning_content_blocksr3block content_textmsg_flush_hinthint_text_partssub msg_flushtextual_output_ content_msg msg_deepseeks rformatzDeepSeekChatFormatter.format"s`   &&&!Q .Q .C#%N-/ $J//11o o eY//n"))65:*N*NOOOO}55k,33ENCCCCy11h&(%(4( (,yy117E111(( %(H'3(<(2 ::: 8{22$<!( *B C C C%'++>? &F;EN<8 777)+350%' !%*c22 %+ CC68#(:""C)#y99" / 6 6sx @ @ @ @!+C!;!;" / 6 6%F (=%F%F%F!"!"!"+$OO,2/3yy/I/I!"!" }557%%"'($.(- -2[))     77+&(%(4( (,yy117E111(( %(H'3(<(2 :55 8{22$<!( *B C C C%'&&9: &A6@Il3  222)+350%' (,(J(J ))%NAOO$*,1H'5$)J NLU  ))#N#N~#N#N#NNNK&F:+E442,,Lx;&&0DII677701  8-7 \*Y' .##L11 ., .  ---rN) __name__ __module__ __qualname____doc__rrlistrE__annotations__r dictrr[rrrrrs #U.. <Kcd3id d38n ddddddrrc JeZdZUdZeddZeed<eddZe eed <d e e d e e ee ffd Z d e e d e e ee ffd Z dd e e ded e e ee ffdZede d e ee ffdZdS)DeepSeekMultiAgentFormatterzq DeepSeek 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.)defaultrconversation_history_promptcdgSrrrrrrz$DeepSeekMultiAgentFormatter.rrrrrrrc"K||g}d}t|dkrG|djdkr6|||dd{Vd}d}|||d23d{V\}}|xdkr0|||d{VAdkr2||||d{Vd}xy6|S) zlFormat input messages into the structure required by the DeepSeek API for multi-agent conversations.rsystemNrT tool_sequence agent_messageF) r>lenr/rA_format_system_message_group_messagesextend_format_tool_sequence_format_agent_message)rLrformatted_msgs start_indexis_first_agent_messagetypgroups rr[z"DeepSeekMultiAgentFormatter.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**% !IsDcbKt|j|d{VS)zqGiven a sequence of tool call/result messages, format them into the required format for the DeepSeek API.)rN)rrr[)rLrs rrpz1DeepSeekMultiAgentFormatter._format_tool_sequencesN +(   &,, rTis_firstcLK|r|j}nd}g}g}|D]R}|D];}t|tr$||jd|j r#z r4r.)rfr?r@r rAr7r!rC) rLrrxrfrraccumulated_textrNrQconversation_blocks_texts rrqz1DeepSeekMultiAgentFormatter._format_agent_message s  -*.*J ' '*, '%' I IC//11 I IeY//I$++sx,G,G5:,G,GHHH I$&  + )),--.!! % $   ! !"7   rrNc4Kd|dS)z'Format system message for DeepSeek API.rir.)get_text_content)rNs rrmz2DeepSeekMultiAgentFormatter._format_system_message1s) ++--   rN)T)r\r]r^r_rrfrErarr`r rbrr[rpboolrq staticmethodrmrrrrdrds{ (-u *N (((#U.. <KcctDcN/C>3i d38n &&3i&& d38n  &&&&P  c3h   \   rrdN)r_typingrpydanticr_formatter_baser_loggingrmessager r r r r rrrrdrrrrs$$******rrrrrMrrrjp p p p p -p p p p p r