ڎj=dZddlZddlZddlmZmZddlmZmZddl m Z m Z ddl m Z mZmZmZmZdd lmZdd lmZmZdd lmZd d lmZmZGddeZdS)zThe edit tool in agentscope.N)AnyList)ToolBaseToolMiddlewareBase)DEFAULT_DANGEROUS_FILESDEFAULT_DANGEROUS_DIRECTORIES)PermissionContextPermissionDecisionPermissionBehaviorPermissionModePermissionRule) ToolChunk) TextBlockToolResultState) AgentState) BackendBase_normalize_newlinesc eZdZUdZdZeed< dZeed< dddddd ddd dd d d ddgddZe ee fed<d Z e ed<d Z e ed<d Ze ed<d Ze ed<dZe ed<eeddfdeedeedeedzdedzddf fd Zd e ee fd!edefd"Zd#edzd e ee fde fd$Zd e ee fdeefd%Z d,d&ed'ed(ed)e d*edzdef d+Z xZ!S)-Editz@The edit tool for performing exact string replacements in files.namea:Performs exact string replacements in files. Usage: - You must use your `Read` tool at least once in the conversation before editing. This tool will error if you attempt an edit without reading the file. - When editing text from Read tool output, ensure you preserve the exact indentation (tabs/spaces) as it appears AFTER the line number prefix. The line number prefix format is: line number + tab. Everything after that is the actual file content to match. Never include any part of the line number prefix in the old_string or new_string. - ALWAYS prefer editing existing files in the codebase. NEVER write new files unless explicitly required. - Only use emojis if the user explicitly requests it. Avoid adding emojis to files unless asked. - The edit will FAIL if `old_string` is not unique in the file. descriptionobjectstringz&The absolute path to the file to edit.)typerzUThe exact string to replace. Must match exactly including whitespace and indentation.z&The string to replace old_string with.booleanzfIf true, replace all occurrences. If false (default), only replace if there is exactly one occurrence.F)rrdefault) file_path old_string new_string replace_all)r r!r")r propertiesrequired input_schemais_mcp is_read_onlyis_concurrency_safeis_external_toolTis_state_injectedNdangerous_filesdangerous_directories middlewaresbackendreturncddlm}t|t ||_t ||_|p ||_dS)aInitialize the edit tool. Args: dangerous_files (`list[str]`, optional): Sensitive files that require explicit user confirmation, even in BYPASS mode. Matched by basename (case-insensitive). Defaults to `DEFAULT_DANGEROUS_FILES`. Pass a custom list to fully replace the defaults, or `[]` to disable the filename check. dangerous_directories (`list[str]`, optional): Sensitive directories that require explicit user confirmation. Matched when any path segment equals an entry (case-insensitive). Defaults to `DEFAULT_DANGEROUS_DIRECTORIES`. Pass a custom list to fully replace the defaults, or `[]` to disable the directory check. middlewares (`List[ToolMiddlewareBase] | None`, optional): Tool middlewares wrapping the tool execution. backend (`BackendBase | None`, optional): The sandbox backend to use for file I/O. When ``None``, a :class:`LocalBackend` is created. r) LocalBackend)r.N)_backendr2super__init__listr,r-)selfr,r-r.r/r2 __class__s LC:\Projects\DSCons\.venv\Lib\site-packages\agentscope\tool\_builtin\_edit.pyr5z Edit.__init__Wsi: +***** [111#O44%)*?%@%@"1<<>> tool_inputcontextcK|d}|sttjdS||r ttjd|ddS|jtjtj fvr6| ||r ttj d|d d Sttjd S) a|Check permissions for file editing. This method implements Edit-specific permission checks: 1. Dangerous path check (safety check, bypass-immune) 2. ACCEPT_EDITS mode check for files in working directories Args: tool_input (`dict[str, Any]`): The tool input containing "file_path" key context (`PermissionContext`): The permission context with mode and rules Returns: `PermissionDecision`: ASK for dangerous paths, ALLOW for safe operations in ACCEPT_EDITS mode, PASSTHROUGH otherwise r zNo file path provided)behaviormessagez6Permission required: Edit operation on sensitive file z)Safety check: dangerous file or directoryT)r>r?decision_reason bypass_immunezPermission granted for editing z (in working directory)z5File is in working directory and not a dangerous path)r>r?r@) getr r PASSTHROUGH_is_dangerous_pathASKmoder ACCEPT_EDITSDONT_ASK_path_in_allowed_working_pathALLOW)r7r;r<r s r9check_permissionszEdit.check_permissions{s$.NN;//  %+7/   " "9 - - %+/."+.. K"   <  '  #   11)WEE )/5.i...%' "'3    r: rule_contentclK|dS|dd}|sdStj||S)aCheck if a permission rule matches the file path. Matches rule_content as a glob pattern against the "file_path" parameter using fnmatch. If rule_content is None, matches all invocations (tool-name-level rule). Args: rule_content (`str | None`): Glob pattern to match against the file path (e.g., "src/**"), or None to match all invocations tool_input (`dict[str, Any]`): The tool input data containing "file_path" key Returns: `bool`: True if the glob pattern matches the file path, False otherwise NTr rBF)rCfnmatch)r7rMr;r s r9 match_rulezEdit.match_rulesE,  4NN;33  5y,777r:cK|dd}|sgS|j|}|r|ddznd}t |j|t jdgS)a Generate suggested permission rules for the file path. Suggests a glob pattern covering the parent directory of the file, allowing the user to grant permission for the entire directory at once. Args: tool_input (`dict[str, Any]`): The tool input data containing "file_path" key Returns: `List[PermissionRule]`: A single suggested rule covering the parent directory (e.g., file "/src/main.py" -> rule "src/**") r rBz/\z/**z** suggested) tool_namerMr>source)rCr3dirnamerstriprrr rK)r7r;r parentpatterns r9generate_suggestionszEdit.generate_suggestionss$NN;33  I&&y115;D6==''%// )$+1"      r:r r!r"r# _agent_statec NK|j|s.ttd|gtjdS|j|d{Vs.ttd|gtjdS||kr+ttdgtjdSd}|h|j|d{V}|+ttdgtjdSd |j }n |j |d{V}t| d d }nR#t$rE} ttd t| gtjdcYd} ~ Sd} ~ wwxYw||} | dkr.ttd|gtjdS| dkr4|s2ttd| d|dgtjdS|r|||} n|||d} |j|| d d{VnR#t$rE} ttdt| gtjdcYd} ~ Sd} ~ wwxYw|rd| dnd} d t)j|d| dd|d|d} ttd| d|gtjd| ||r| ndd S)!z'Execute the edit and return the result.z0Error: file_path must be an absolute path, got: )textT)contentstateis_lastNzError: File not found: zCError: old_string and new_string are identical. No changes to make.zBError: To edit a file, you must first read it using the Read tool.rBzutf-8replace)errorszError reading file: rzError: old_string not found in rzError: old_string appears z times in zT. Set replace_all=true to replace all occurrences, or make old_string more specific.zError writing file: zall z occurrencesz 1 occurrence)keependsza/zb/r )fromfiletofilenzSuccessfully replaced z in )diffr occurrences)r]r^r_metadata)r3isabsrrrERROR file_exists tool_context get_cachejoinlines read_filerdecode Exceptionstrcountr` write_fileencodedifflib unified_diff splitlinesRUNNING)r7r r!r"r#rZr]cacherawergupdated_contentreplacement_msg diff_texts r9callz Edit.callsU}""9-- 6*366&+    ]..y99999999 #HY#H#HIII&+    # #=&+      #&3==iHHHHHHHHE} !"6 */     ggek**GG  M33I>>>>>>>>.JJwyJ99    &,K3q66,K,KLLLM)/  mmJ//  !  JyJJ &+  ??;?999(1999 &+      %ooj*EEOO%ooO -**&&w//           "(Gs1vv(G(GHHHI%+       1< O ,; , , , , GG  ""D"11**D*99)i))'I''      &/&&#&& ")!&.9@{{q    s=7AE<< G :GG G 4J77 L:L;LL)FN)"__name__ __module__ __qualname____doc__rrs__annotations__rr&dictrr'boolr(r)r*r+rr r6rrrr5r r rLrPrrYrrr __classcell__)r8s@r9rrsJJD#/CKCCC"2!G !<!G ""!!  4>==9$$L$sCx.>FDL$ %%%%"d""""t"""&=+H7;&* "2"2c"2 $Cy"2,-4 "2 t# "2  "2"2"2"2"2"2H> cN> #>   > > > > @8Dj8cN8  8888<" cN"  n " " " " R"*. i i i i  i  i !4' i  i i i i i i i i r:r)rrwrOtypingrr_baserr _constantsrr permissionr r r rr _responserr?rrr^rr3rrrr:r9rsO""00000000"!!!!!1111111166666666M M M M M 8M M M M M r: