ڎj5dZddlZddlmZmZddlmZddlmZddlm Z m Z m Z m Z ddl mZd d lmZd d lmZd d lmZmZmZmZmZd d lmZGddeZeGddeZeGddZdS)zThe model response module.N) dataclassfield)datetime)StrEnum)AnyLiteralSelfList) ChatUsage) _generate_id) DictMixin) TextBlock ToolCallBlock ThinkingBlock DataBlock Base64Source)JSONSerializableObjectceZdZdZdZ dZdS)FinishedReasonz*The finished reason of the model response. interrupted completedN)__name__ __module__ __qualname____doc__ INTERRUPTED COMPLETEDNC:\Projects\DSCons\.venv\Lib\site-packages\agentscope\model\_model_response.pyrrs#44KJI**r!rc eZdZUdZeeezezeze d< e e d< e e Z ee d< e dZee d< e dZed e d < e d Zed ze d < e dZee d< e dZeeefe d< d deded zdefdZ d deded zdedefdZdededededef dZ d dedededed zdef dZd edefdZ d S)! ChatResponsezThe response of chat models.contentis_lastdefault_factoryidcLtjSNrnow isoformatr r!r"zChatResponse./HLNN4L4L4N4Nr! created_atcdS)N chat_responser r r!r"r/zChatResponse.3sr!r3typecdSr+r r r!r"r/zChatResponse.7Dr!NusagectjSr+)rrr r!r"r/zChatResponse.;s  8r!finished_reasonciSr+r r r!r"r/zChatResponse.Ar!metadatatextblock_idreturnc,|jD]8}t|tr!| ||jkr|xj|z c_|cS9t|jt sJ|jt||p t|S)z$Append text to the current response.N)r=r))r% isinstancerr)r=listappendr)selfr=r>blocks r" append_textzChatResponse.append_textEs\  E%++  H$8$8 d"  $,-----  4H$>  ? ? ?    r!thinking extra_fieldsc |jD]e}t|trN| ||jkrA|xj|z c_|D]\}}|t ||||cSft|jtsJt||p t}|D]\}}|t ||||j ||S)aAppend thinking to the current response. Args: thinking (`str`): The thinking content to append. block_id (`str | None`, defaults to `None`): The id of the ``ThinkingBlock`` to accumulate into. If no matching block exists in ``content``, a new one is appended. **extra_fields (`Any`): Additional provider-specific fields to attach to the ``ThinkingBlock`` (e.g. Anthropic's ``signature``, OpenAI Responses API's ``reasoning_item_id``). Only non-``None`` values are applied. N)rGr)) r%rArr)rGitemssetattrrBrrC)rDrGr>rHrEkeyvalues r"append_thinkingzChatResponse.append_thinkingUs*\  E%//  H$8$8(*"."4"4"6"633JC(sE222 $,-----xH4N OOO&,,.. + +JC sE*** E""" r!nameinputc |jD]c}t|trL|j|krA|xj|z c_|D]\}}|t ||||cSdt|||}|D]\}}|t |||t|jtsJ|j||S)aAppend tool call to the current response by tool call block ID. Args: block_id (`str`): The id of the ``ToolCallBlock`` to accumulate into. If no matching block exists in ``content``, a new one is appended. name (`str`): The name of the tool being called. input (`str`): The incremental JSON string arguments to append. **extra_fields (`Any`): Additional provider-specific fields to attach to the ``ToolCallBlock`` (e.g. OpenAI Responses API's ``call_id``). Only non-``None`` values are applied. N)r)rOrP) r%rArr)rPrJrKrBrC)rDr>rOrPrHrErLrMs r"append_tool_callzChatResponse.append_tool_call|s.\  E%// EH4H4H u$ "."4"4"6"633JC(sE222     ',,.. + +JC sE***$,----- E""" r!data media_typec 2|jD]}t|tr|j|krt|jt rt|jj|krd|jjrtj |jjnd}tj ||z d|j_|cS|j t|t tj | d|||S)uAppend raw media bytes to the ``DataBlock`` with the given id. The accumulated bytes are stored base64-encoded in ``DataBlock.source.data`` (via a :class:`Base64Source`). Callers must pass the *incremental* raw media bytes only — the method takes care of base64 (de)coding internally, so consumers never have to worry about base64 padding (``=``) corrupting concatenation. .. note:: Byte-level delta accumulation only has a well-defined semantics for streaming media where chunks can be safely concatenated (currently ``audio/*``). For non-streamable media types (e.g. ``image/*``, ``video/*``) each ``DataBlock`` should be treated as a complete file and passed atomically rather than accumulated through this method. Args: block_id (`str`): The id of the ``DataBlock`` to accumulate into. If no matching block exists in ``content``, a new one is appended. data (`bytes`): The incremental raw media bytes to append. media_type (`str`): The IANA media type of the raw bytes (e.g. ``audio/pcm``, ``audio/wav``). Used both to tag a newly-created block and to guard against accidentally mixing bytes from different media types under the same id. name (`str | None`, defaults to `None`): The optional ``name`` field used when a new ``DataBlock`` needs to be created. Ignored when an existing block with ``block_id`` is found. Returns: `Self`: The current ``ChatResponse`` instance for chaining. r!ascii)rSrT)r)sourcerO) r%rArr)rWrrTrSbase64 b64decode b64encodedecoderC)rDr>rSrTrOrE old_bytess r"append_data_blockzChatResponse.append_data_blocks,Z\  E5),, H((u|\::)L+z99|(F$U\%6777 %+$4$%%&// !   #)$//66w??)      r!c\d|jD}|jD]F}|j|vr9||j}t|trJ|xj|jz c_|jpiD]\}}|t|||t|tr|xj |j z c_ t|trK|xj |j z c_ |jpiD]\}}|t|||t|trt|jtr4t|jtr|jj|jjks|j|_|jjdr|jjrt'j|jjnd}|jjrt'j|jjnd}t'j||zd|j_0|jj|j_H|r6|jd|D|jr |j|_|S)z-Append chat response to the current response.ci|] }|j| Sr )r)).0_s r" z5ChatResponse.append_chat_response..sAAAa!$AAAr!Nzaudio/r!rVc3BK|]}|dVdS)T)deepN) model_copy)r`rEs r" z4ChatResponse.append_chat_response..5sG    d ++      r!)r%r)poprArrG model_extrarJrKrr=rrPrrWrrT startswithrSrXrYrZr[extendvaluesr7) rDr3new_block_dictrE delta_blockrLrMr\ delta_bytess r"append_chat_responsez!ChatResponse.append_chat_responsesBA=+@AAA\< D< DEx>)),00:: e]338DNNk&::NN (3'>'D"&K&K&M&M77 U ,#E36667 y11-DJJ+"22JJJ}55*DKK;#44KK(3'>'D"&K&K&M&M77 U ,#E36667 y11!D#5<>>D&{'9<HHD"L3&-899 (3'9 0;;HEED %|0%F,U\->???!$" +16%F,[-?-DEEE!$$ -3,<% 3-- &// )) -8,>,C )   L    +2244         -&,DJ r!r+)!rrrrr rrrr__annotations__boolrrr)strr1r4rr7r r9rr<dictrr rFrrNrRbytesr]ror r!r"r$r$ s&& )m+m;iG HHHH,MMMMeL111B111 e,N,NOOOJOOO'%*U//&&&D'/ "E#eLLAAAE9t AAAC&+e88'''O^38%" 333Hd3../,sTzT& $%%%*% %  %%%%N))) )  )  ))))` HHHH H Dj H  HHHHTM$M4MMMMMMr!r$c*eZdZUdZeed< eeZe ed< edZ e ed< edZ e ded < ed Z ed zed < ed Zee efed< eejZeed<d S)StructuredResponsez'The structured response of chat models.r%r'r)cLtjSr+r,r r!r"r/zStructuredResponse.Kr0r!r1cdS)Nstructured_responser r r!r"r/zStructuredResponse.Os 5r!ryr4cdSr+r r r!r"r/zStructuredResponse.Sr6r!Nr7ciSr+r r r!r"r/zStructuredResponse.Wr;r!r<)defaultr9)rrrrrsrprrr)rrr1r4rr7r r<rrrr9r r!r"rvrvAs311 MMM-eL111B111 e,N,NOOOJOOO'+0555,,,D'' (K#eLLAAAE9t AAAC27%" 333Hd3../,&+e('''O^:9r!rv)rrX dataclassesrrrenumrtypingrrr r _model_usager _utils._commonr _utils._mixinrmessagerrrrrtypesrrr$rvr r!r"rs ((((((((++++++++++++######))))))%%%%%%+*****+++++W+++ ]]]]]9]] ]@  ::::::: :::r!