ڎj&BUdZddlZddlmZmZmZddlmZddlm Z m Z ddl m Z dd l mZdd lmZmZmZmZmZmZGd d eZGd deZGddeZGddeZGddeZeezezezZeed<GddZdS)z(Internal utilities for the model module.N)AnySelf TypeAlias)Field) ChatResponseFinishedReason) ChatUsage)logger) Base64Source DataBlock TextBlock ThinkingBlock ToolCallBlock URLSourcec^eZdZUdZeeZeeed<de ddfdZ de fdZ dS) _AccTextBlockzThe accumulating counterpart of ``TextBlock``, whose ``text`` field holds the incremental fragments until they are joined in ``build``.default_factorytextblockreturnNcD|j|jdS)zCollect one delta block in constant time. Args: block (`TextBlock`): The delta block to collect. N)rappendselfrs EC:\Projects\DSCons\.venv\Lib\site-packages\agentscope\model\_utils.pyrz_AccTextBlock.appends" $$$$$c ztdii|dd|jiS)z.Join the fragments into a plain ``TextBlock``.r)r model_dumpjoinrrs rbuildz_AccTextBlock.build&s>MMLdoo//L9K9KLLMMMr) __name__ __module__ __qualname____doc__rlistrstr__annotations__rrr&r"rrrrsKKeD$s)%I%$%%%%NyNNNNNNrrc^eZdZUdZeeZeeed<de ddfdZ de fdZ dS) _AccThinkingBlockzThe accumulating counterpart of ``ThinkingBlock``, whose ``thinking`` field holds the incremental fragments until they are joined in ``build``.rthinkingrrNc|j|j|jpiD]\}}|t |||dS)zCollect one delta block in constant time. Args: block (`ThinkingBlock`): The delta block to collect. N)r0r model_extraitemssetattr)rrkeyvalues rrz_AccThinkingBlock.append4sh U^,,,!,299;; * *JC c5))) * *rc ztdii|dd|jiS)z2Join the fragments into a plain ``ThinkingBlock``.r0r!r")rr#r$r0r%s rr&z_AccThinkingBlock.buildBsH  G""GJ 0F0FGG   r) r'r(r)r*rr+r0r,r-rrr&r"rrr/r/+s %Hd3i *M *d * * * * }      rr/c^eZdZUdZeeZeeed<de ddfdZ de fdZ dS) _AccToolCallBlockzThe accumulating counterpart of ``ToolCallBlock``, whose ``input`` field holds the incremental JSON fragments until they are joined in ``build``.rinputrrNc|j|j|js|jr|j|_dSdSdS)zCollect one delta block in constant time. Args: block (`ToolCallBlock`): The delta block to collect. N)r:rnamers rrz_AccToolCallBlock.appendRsR %+&&&y #UZ # DIII # # # #rc ztdii|dd|jiS)z2Join the fragments into a plain ``ToolCallBlock``.r:r!r")rr#r$r:r%s rr&z_AccToolCallBlock.build_sH  A""AGRWWTZ-@-@AA   r) r'r(r)r*rr+r:r,r-rrr&r"rrr9r9IsuE49 #M #d # # # # }      rr9c^eZdZUdZeeZeeed<de ddfdZ de fdZ dS) _AccBase64SourceaThe accumulating counterpart of ``Base64Source`` for audio streams, whose ``data`` field holds the raw decoded bytes of each delta. Base64 strings cannot be concatenated directly: a delta whose byte length is not a multiple of three is padded with ``=``, and everything after that padding is silently dropped on decoding. So each delta is decoded on arrival and the bytes are encoded once in ``build``. rdatasourcerNcz|jr3|jtj|jdSdS)zDecode one delta source and collect its raw bytes. Args: source (`Base64Source`): The delta source to collect. N)r@rbase64 b64decoderrAs rrz_AccBase64Source.appendtsA ; < I  V-fk:: ; ; ; ; ; < z,_StreamAccumulator.build..s >>>1QWWYY>>>rT)r_is_lastrYr[r")rXrrWvaluesrYr[)rkwargss rr&z_StreamAccumulator.build sm!# 7 7F4L >> (;(;(=(=>>>* 0       r)rN) r'r(r)r*r\rrrer&r"rrrUrUsp : : : :*,*4****X  |       rrU)r*rCtypingrrrpydanticr_model_responserr _model_usager _loggingr messager rrrrrrr/r9r?rNrSr-rUr"rrrts... ''''''''''99999999######NNNNNINNN,        <        :     |   @9 9 9 9 9 I9 9 9 z%%(99MI 9 Q Q Q Q Q Q Q Q Q Q r