ڎjz(dZddlZddlZddlmZmZmZddlmZm Z m Z ddl m Z m Z ddlmZd d lmZmZd d lmZmZmZmZd ZGd de ZdS)z"The PowerShell tool in agentscope.N)AsyncGeneratorAnyList) BackendBase LocalBackend_normalize_newlines)ToolBaseToolMiddlewareBase) ToolChunk) TextBlockToolResultState)PermissionBehaviorPermissionContextPermissionDecisionPermissionRule)pwshpowershell.exec eZdZUdZdZeed< dZeed< dddddd dd d d d ddddgdZe ee fed<dZ e ed<dZ e ed<dZe ed<dZe ed<dZe ed< d(deejezdzdeedzdedzddffd Zdefd Zd!e ee fd"edefd#Zd!e ee fdeefd$Z d)deded&edeedffd'Z xZ!S)* PowerShellz8Execute PowerShell commands through a workspace backend.namea<Executes a PowerShell command and returns its output. Each command starts in the configured working directory, but PowerShell session state does not persist between commands. Commands run without loading the user's PowerShell profile. IMPORTANT: Avoid using this tool for filesystem operations when a dedicated tool can accomplish the task. Prefer the dedicated tools because their calls are easier for the user to review and authorize: - File search: Use Glob (NOT Get-ChildItem) - Content search: Use Grep (NOT Select-String) - Read files: Use Read (NOT Get-Content) - Edit files: Use Edit - Write files: Use Write (NOT Set-Content or Out-File) - Communication: Output text directly (NOT Write-Output) # Instructions - Verify the parent directory before creating new directories or files. - Quote paths containing spaces, for example: Get-Item "path with spaces". - Prefer absolute paths and avoid Set-Location so the working directory is clear in every command. - You may specify an optional timeout in milliseconds (up to 600000ms / 10 minutes). The default timeout is 120000ms (2 minutes). - Write a concise description of what the command does. Include more context for pipelines, uncommon parameters, or commands with side effects. - Run independent commands in parallel tool calls. Keep dependent commands together and use PowerShell-native error handling when later work depends on earlier work succeeding. - Prefer new git commits over amending existing commits. Avoid destructive git operations and never bypass hooks unless the user explicitly asks. descriptionobjectstringz"The PowerShell command to execute.)typerz5Clear, concise description of what this command does.integerz?Optional timeout in milliseconds (default: 120000, max: 600000)' r)rrdefaultmaximumminimum)commandrtimeoutr$)r propertiesrequired input_schemaFis_mcp is_read_onlyis_concurrency_safeis_external_toolis_state_injectedNcwd middlewaresbackendreturnct||tj|nd|_|p t |_d|_dS)aInitialize the PowerShell tool. Args: cwd (`str | os.PathLike[str] | None`, optional): Working directory used when executing commands. middlewares (`List[ToolMiddlewareBase] | None`, optional): Tool middlewares wrapping command execution. backend (`BackendBase | None`, optional): Backend used for subprocess execution. Defaults to the host-local backend. )r/N)super__init__osfspath_cwdr_backend _executable)selfr.r/r0 __class__s RC:\Projects\DSCons\.venv\Lib\site-packages\agentscope\tool\_builtin\_powershell.pyr4zPowerShell.__init___sU" [111&)oBIcNNN4 1<>> '+c K|jMtD]>}|j|dddddgdd{V}|jd kr ||_n?d |_|jS) z>Prefer PowerShell 6+ and cache the first available executable.N-NoLogo -NoProfile-NonInteractivez-Commandzexit 0g$@)r%r)r9_SHELL_CANDIDATESr8 exec_shell exit_code)r: candidateprobes r<_resolve_executablezPowerShell._resolve_executableus   #. 4 4 "m66!!$)" !7        ?c))'0D$E*$4 r= tool_inputcontextc>KttjddS)aFAsk the user to confirm every PowerShell command. PowerShell-specific command validation is intentionally outside this implementation. Since no command is classified as safe, every invocation prompts the user. This is a regular ASK that allow rules and BYPASS mode may still override. zExecute PowerShell commandz,PowerShell command validation is not enabled)behaviormessagedecision_reason)rrASK)r:rIrJs r<check_permissionszPowerShell.check_permissionss+"'+0J    r=c KgS)zReturn no automatic allow-rule suggestions. A broad rule would weaken the conservative permission boundary before PowerShell-specific command validation is available. )r:rIs r<generate_suggestionszPowerShell.generate_suggestionss  r=r%c DKt|d}tj|dd}d|d}tj|dd} |d{V}|j|ddd d |g|j|d z d{V} nL#t$r?} ttd |d| gtj dWVYd} ~ dSd} ~ wwxYwt| jdd} t| jdd} | jdkrA| jdkr6ttd|d|gtj dWVdS| sjd |d} | r| d| z } | r| d| z } t'| dkr | dddz} tt| gtj dWVdS| }| r |r|dz }|| z }t'|dkr |dddz}tt|gtjdWVdS)aExecute a PowerShell command through the configured backend. Args: command (`str`): PowerShell source text to execute. description (`str`, optional): Human-readable description of the command. timeout (`int`, optional): Timeout in milliseconds, capped at 600000. Yields: `ToolChunk`: A final chunk containing the command output. r z utf-16-leasciia$ProgressPreference = [System.Management.Automation.ActionPreference]::SilentlyContinue $OutputEncoding = [Console]::OutputEncoding = [System.Text.UTF8Encoding]::new($false) $AgentScopeCommand = [System.Text.Encoding]::Unicode.GetString([System.Convert]::FromBase64String('z1')) & ([ScriptBlock]::Create($AgentScopeCommand))Nr?r@rAz-EncodedCommandg@@)r.r%zCommand failed: z Error: )textT)contentstateis_lastzutf-8replace)errorss timed outzCommand timed out after zms:  z Stdout: z Stderr: i0uz ... (output truncated))minbase64 b64encodeencodedecoderHr8rDr7 Exceptionr rrERRORr stdoutstderrrEoklenRUNNING)r:r$rr% timeout_msencoded_user_commandpowershell_scriptencoded_command executableresultexcrfrg error_resultoutputs r<callzPowerShell.calls(&)) %/ NN; ' '  &//  <% < < < !*  $ $[ 1 1  &//  #7799999999J=33 %%# I"V+4        FF   GGG#GG &+     FFFFF % M  ; ;  % M  ; ;    r ! !fm|&C&C)z))&))&+        Fyy{{ 9g999L 7 6f 6 66  7 6f 6 66 <  5(( %(+EE" 5556%+     F   $ f F v;;  FUF^&@@FF+++,!)         sA C D#$4DD#)NNN)rTr)"__name__ __module__ __qualname____doc__rstr__annotations__rr(dictrr)boolr*r+r,r-r5PathLikerr rr4rHrrrPrrSintrr rt __classcell__)r;s@r<rrsBBD#/MKMMM>2!C !K"5"!    ,K1$$L$sCx.6FDL$ %%%%"d"""#t###.27;&* ,, 2;s# #d *,,-4,t# ,  ,,,,,,, 3    , cN #       $ cN  n      q q q q  q  4 ( q q q q q q q q r=r)rxr`r5typingrrrr8rrr _baser r _responser rMrr permissionrrrrrCrrRr=r<rs(( ,,,,,,,,,,DDDDDDDDDD00000000!!!!!!11111111/B B B B B B B B B B r=