ڎj4dZddlmZmZddlmZmZmZmZddl Z ddl m Z ddl mZdd lmZmZmZmZmZmZmZdd lmZGd d eZGd deZGddeZGddeZGddeZdS)zThe agent state class.)AnyType) BaseModelFieldfield_serializermodel_validatorN) _generate_id)Task) TextBlock DataBlockMsg ToolCallBlock ToolCallStateToolResultBlock HintBlock)PermissionContextcHeZdZUdZeeed<eed<eed<eed<dS)ReadCacheEntryzThe read file cache.lines updated_atbytes file_pathN)__name__ __module__ __qualname____doc__liststr__annotations__floatEC:\Projects\DSCons\.venv\Lib\site-packages\agentscope\state\_state.pyrrsD 9 LLLNNNNNr$rceZdZUdZeddZeed< eddZe ed< ee Z e e ed < ee Z e eed < d ed e dzfdZd ede ed dfdZ ddeedzd dfdZdS) ToolContextz!The tool context, e.g. tool cachedr )defaultgtmax_cache_filesiai'max_cache_bytesdefault_factoryread_file_cacheactivated_groupsrreturnNc:K|jD]}|j|kr tjj|d{V}||jkr|cS|j|dS#t$r|j|YdSwxYwdS)zGet cached file content if still valid. Args: file_path: The absolute path of the file. Returns: The cached entry if valid, otherwise None. N) r/raiofilesospathgetmtimerremove Exception)selfrentryrs r% get_cachezToolContext.get_cache.s)  E)++  '/{'7'@'@'K'K!K!K!K!K!K!KJ!U%555$ ,33E:::#tt    (//666444 ,ts6A.A..$BBrcK tjjd{V}n#t$rYdSwxYwt d|Ddz }fd|jD|_t|j|jkr7|j dt|j|jk7t d|jD}|jrG||z|j kr9|j d}||j z}|jr||z|j k9|j t|||dS)zCache file content with LRU eviction. Args: file_path: The absolute path of the file. lines: The lines of the file content. Nc3ZK|]&}t|dV'dS)zutf-8N)lenencode).0lines r% z)ToolContext.cache_file..Ys6<z*ToolContext.cache_file..]s0   )++ +++r$rc3$K|] }|jV dSN)r)r@r:s r%rBz)ToolContext.cache_file..hs$II55;IIIIIIr$)rrrr)r3r4r5r6r8sumr/r>r+popr,rappendr)r9rrrnew_entry_bytes current_sizeremoveds ` r% cache_filezToolContext.cache_fileJs '{/88CCCCCCCCJJ    FF  <>reserved_file_pathsc\Kp tfd|jD|_dS)a Drop read caches whose paths are not in ``reserved_file_paths``. Args: reserved_file_paths: File paths from Read calls that remain in the context. Caches for these files are kept; all others are evicted. c&g|] }|jv |Sr#rD)r@r:rOs r%rEz0ToolContext.clean_file_cache..s1   "555 555r$N)setr/)r9rOs `r%clean_file_cachezToolContext.clean_file_cachezsL2:SUU    -   r$rG)rrrrrr+intr!r,r"rr/rr0r r;rNrRrSr#r$r%r'r' s9++ 5333OS333-"U5U;;;OU;;;>,1E$,G,G,GOT.)GGG3"'%"="="=d3i===$1F8. #. d3i. D. . . . d04   X_        r$r'cDeZdZUdZedZeeed<dS) TaskContextzThe task context.cgSrGr#r#r$r%zTaskContext.sbr$r-tasksN) rrrrrrYrr r!r#r$r%rVrVs;jj999E4:999r$rVceZdZUdZeeZeed< dZ e ed< dZ e e ezdzed< edde e ezdzd edzfd ZdZedzed <dS) ReplyContextz'The context of the current agent reply.r-reply_idrcur_iterNstructured_schemavaluer1cXt|tr|n|S)z3Serialize a schema class into its JSON schema dict.) isinstancetypemodel_json_schemar9r_s r%_serialize_structured_schemaz)ReplyContext._serialize_structured_schemas* -7ud,C,CNu&&(((Nr$structured_output)rrrrrr r\r r!r]rTr^rrdictrrerfr#r$r%r[r[s11E,777Hc777#Hc8<tI-4;;;7)**OI%,O OOO+*O&*td{)))<z5AgentState.get_awaiting_tool_calls..KOOOqadOOOr$ tool_resultczg|]7}|jtjks|jtjk,|jv5|8Sr#)staterASKING SUBMITTEDrr@tc result_idss r%rEz6AgentState.get_awaiting_tool_calls..LsR   x=///M333Z8O8O 9P8O8Or$ tool_call)rmrrget_content_blocksr9rlast_msgrs @r%rz"AgentState.get_awaiting_tool_calls8s| I<# =K ' '8=D+@+@IOOH$?$? $N$NOOO     11+>>    r$c|jsgS|jd}|jdks|j|ks|j|jkrgSd|dDfd|dDS)aBGet the current reply's tool calls with no matching tool result yet, whatever their state. A reply accumulates into a single message, so the calls of the finished reasoning-acting rounds are excluded while the ones of the ongoing round are returned. Args: name (`str`): Only messages authored by this agent name are inspected; observed messages from other agents are ignored. Returns: `list[ToolCallBlock]`: The unfinished tool call blocks, empty if none. rrch|] }|j Sr#rrs r%rz7AgentState.get_unfinished_tool_calls..mrr$rc&g|] }|jv |Sr#rrs r%rEz8AgentState.get_unfinished_tool_calls..ns0   uJ&& &&&r$r)rmrrrr\rrs @r%get_unfinished_tool_callsz$AgentState.get_unfinished_tool_callsUs| I<# M[ ( (}$${dm++IOOH$?$? $N$NOOO     11+>>    r$)*rrrrrr rjr r!rlrr rrmrr classmethodrrvpropertyr\setterrTr]r[rrrr~r'rrVrrgrrrrrrrrrr#r$r%ririsGGeL999J999/24GS4 I-. .333&t444GT#Y444O _(###     [$# +#+++X+_,c,d,,,_,+#+++X+ _,c,d,,,_,#(% "E"E"EM<EEE$ -2E)---) !&k B B BL+BBB "'{!C!C!CM;CCC8 &+U4%@%@%@NDcN@@@  !I - = O   <8C8D8888" C D4G    : c d=6I      r$ri)rtypingrrpydanticrrrr aiofiles.osr3_utils._commonr _taskr messager rrrrrr permissionrrr'rVr[rir#r$r%rsHHHHHHHHHHHH))))))+*****Yk k k k k )k k k \)=====9===6B B B B B B B B B B r$