wcgw 4.1.0__tar.gz → 4.1.2__tar.gz

{wcgw-4.1.0 → wcgw-4.1.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: wcgw
-Version: 4.1.0
+Version: 4.1.2
 Summary: Shell and coding agent on claude and chatgpt
 Project-URL: Homepage, https://github.com/rusiaaman/wcgw
 Author-email: Aman Rusia <gapypi@arcfu.com>
@@ -11,6 +11,7 @@ Requires-Dist: fastapi>=0.115.0
 Requires-Dist: openai>=1.46.0
 Requires-Dist: petname>=2.6
 Requires-Dist: pexpect>=4.9.0
+Requires-Dist: psutil>=7.0.0
 Requires-Dist: pydantic>=2.9.2
 Requires-Dist: pygit2>=1.16.0
 Requires-Dist: pyte>=0.8.2
@@ -20,6 +21,8 @@ Requires-Dist: semantic-version>=2.10.0
 Requires-Dist: syntax-checker>=0.3.0
 Requires-Dist: tokenizers>=0.21.0
 Requires-Dist: toml>=0.10.2
+Requires-Dist: tree-sitter-bash>=0.23.3
+Requires-Dist: tree-sitter>=0.24.0
 Requires-Dist: typer>=0.12.5
 Requires-Dist: uvicorn>=0.31.0
 Requires-Dist: websockets>=13.1

{wcgw-4.1.0 → wcgw-4.1.2}/pyproject.toml

@@ -1,7 +1,7 @@
 [project]
 authors = [{ name = "Aman Rusia", email = "gapypi@arcfu.com" }]
 name = "wcgw"
-version = "4.1.0"
+version = "4.1.2"
 description = "Shell and coding agent on claude and chatgpt"
 readme = "README.md"
 requires-python = ">=3.11"
@@ -23,6 +23,9 @@ dependencies = [
     "tokenizers>=0.21.0",
     "pygit2>=1.16.0",
     "syntax-checker>=0.3.0",
+    "psutil>=7.0.0",
+    "tree-sitter>=0.24.0",
+    "tree-sitter-bash>=0.23.3",
 ]
 
 [project.urls]
@@ -58,6 +61,7 @@ dev-dependencies = [
     "line-profiler>=4.2.0",
     "pytest-asyncio>=0.25.3",
     "types-pexpect>=4.9.0.20241208",
+    "types-psutil>=7.0.0.20250218",
 ]
 
 [tool.pytest.ini_options]

{wcgw-4.1.0 → wcgw-4.1.2}/src/wcgw/client/bash_state/bash_state.py

@@ -16,6 +16,7 @@ from typing import (
 )
 
 import pexpect
+import psutil
 import pyte
 
 from ...types_ import (
@@ -30,6 +31,7 @@ from ...types_ import (
 )
 from ..encoder import EncoderDecoder
 from ..modes import BashCommandMode, FileEditMode, WriteIfEmptyMode
+from .parser.bash_statement_parser import BashStatementParser
 
 PROMPT_CONST = "wcgw→" + " "
 PROMPT_STATEMENT = "export GIT_PAGER=cat PAGER=cat PROMPT_COMMAND= PS1='wcgw→'' '"
@@ -87,6 +89,116 @@ def check_if_screen_command_available() -> bool:
         return False
 
 
+def get_wcgw_screen_sessions() -> list[str]:
+    """
+    Get a list of all WCGW screen session IDs.
+
+    Returns:
+        List of screen session IDs that match the wcgw pattern.
+    """
+    screen_sessions = []
+
+    try:
+        # Get list of all screen sessions
+        result = subprocess.run(
+            ["screen", "-ls"],
+            capture_output=True,
+            text=True,
+            check=False,  # Don't raise exception on non-zero exit code
+            timeout=0.5,
+        )
+        output = result.stdout or result.stderr or ""
+
+        # Parse screen output to get session IDs
+        for line in output.splitlines():
+            line = line.strip()
+            if not line or not line[0].isdigit():
+                continue
+
+            # Extract session info (e.g., "1234.wcgw.123456 (Detached)")
+            session_parts = line.split()
+            if not session_parts:
+                continue
+
+            session_id = session_parts[0].strip()
+
+            # Check if it's a WCGW session
+            if ".wcgw." in session_id:
+                screen_sessions.append(session_id)
+    except Exception:
+        # If anything goes wrong, just return empty list
+        pass
+
+    return screen_sessions
+
+
+def get_orphaned_wcgw_screens() -> list[str]:
+    """
+    Identify orphaned WCGW screen sessions where the parent process has PID 1
+    or doesn't exist.
+
+    Returns:
+        List of screen session IDs that are orphaned and match the wcgw pattern.
+    """
+    orphaned_screens = []
+
+    try:
+        # Get list of all WCGW screen sessions
+        screen_sessions = get_wcgw_screen_sessions()
+
+        for session_id in screen_sessions:
+            # Extract PID from session ID (first part before the dot)
+            try:
+                pid = int(session_id.split(".")[0])
+
+                # Check if process exists and if its parent is PID 1
+                try:
+                    process = psutil.Process(pid)
+                    parent_pid = process.ppid()
+
+                    if parent_pid == 1:
+                        # This is an orphaned process
+                        orphaned_screens.append(session_id)
+                except psutil.NoSuchProcess:
+                    # Process doesn't exist anymore, consider it orphaned
+                    orphaned_screens.append(session_id)
+            except (ValueError, IndexError):
+                # Couldn't parse PID, skip
+                continue
+    except Exception:
+        # If anything goes wrong, just return empty list
+        pass
+
+    return orphaned_screens
+
+
+def cleanup_orphaned_wcgw_screens(console: Console) -> None:
+    """
+    Clean up all orphaned WCGW screen sessions.
+
+    Args:
+        console: Console for logging.
+    """
+    orphaned_sessions = get_orphaned_wcgw_screens()
+
+    if not orphaned_sessions:
+        return
+
+    console.log(
+        f"Found {len(orphaned_sessions)} orphaned WCGW screen sessions to clean up"
+    )
+
+    for session in orphaned_sessions:
+        try:
+            subprocess.run(
+                ["screen", "-S", session, "-X", "quit"],
+                check=False,
+                timeout=CONFIG.timeout,
+            )
+        except Exception as e:
+            console.log(f"Failed to kill orphaned screen session: {session}\n{e}")
+
+
 def cleanup_all_screens_with_name(name: str, console: Console) -> None:
     """
     There could be in worst case multiple screens with same name, clear them if any.
@@ -269,6 +381,8 @@ class BashState:
         self.close_bg_expect_thread()
         if set_as_command is not None:
             self._last_command = set_as_command
+        # if s == "\n":
+        #     return self._shell.sendcontrol("m")
         output = self._shell.send(s)
         return output
 
@@ -387,6 +501,11 @@ class BashState:
         self._last_command = ""
         # Ensure self._cwd exists
         os.makedirs(self._cwd, exist_ok=True)
+
+        # Clean up orphaned WCGW screen sessions
+        if check_if_screen_command_available():
+            cleanup_orphaned_wcgw_screens(self.console)
+
         try:
             self._shell, self._shell_id = start_shell(
                 self._bash_command_mode.bash_mode == "restricted_mode",
@@ -817,10 +936,22 @@ def _execute_bash(
                 raise ValueError(WAITING_INPUT_MESSAGE)
 
             command = command_data.command.strip()
+
+            # Check for multiple statements using the bash statement parser
             if "\n" in command:
-                raise ValueError(
-                    "Command should not contain newline character in middle. Run only one command at a time."
-                )
+                try:
+                    parser = BashStatementParser()
+                    statements = parser.parse_string(command)
+                    if len(statements) > 1:
+                        return (
+                            "Error: Command contains multiple statements. Please run only one bash statement at a time.",
+                            0.0,
+                        )
+                except Exception:
+                    # Fall back to simple newline check if something goes wrong
+                    raise ValueError(
+                        "Command should not contain newline character in middle. Run only one command at a time."
+                    )
 
             for i in range(0, len(command), 128):
                 bash_state.send(command[i : i + 128], set_as_command=None)

wcgw-4.1.2/src/wcgw/client/bash_state/parser/__init__.py

@@ -0,0 +1,7 @@
+"""
+Parser for bash statements using tree-sitter.
+
+This module provides functionality to parse and identify individual bash statements.
+"""
+
+from .bash_statement_parser import BashStatementParser, Statement

wcgw-4.1.2/src/wcgw/client/bash_state/parser/bash_statement_parser.py

@@ -0,0 +1,181 @@
+#!/usr/bin/env python3
+"""
+Bash Statement Parser
+
+This script parses bash scripts and identifies individual statements using tree-sitter.
+It correctly handles multi-line strings, command chains with && and ||, and semicolon-separated statements.
+"""
+
+import sys
+from dataclasses import dataclass
+from typing import Any, List, Optional
+
+import tree_sitter_bash
+from tree_sitter import Language, Parser
+
+
+@dataclass
+class Statement:
+    """A bash statement with its source code and position information."""
+
+    text: str
+    start_line: int
+    end_line: int
+    start_byte: int
+    end_byte: int
+    node_type: str
+    parent_type: Optional[str] = None
+
+    def __str__(self) -> str:
+        return self.text.strip()
+
+
+class BashStatementParser:
+    def __init__(self) -> None:
+        # Use the precompiled bash language
+        self.language = Language(tree_sitter_bash.language())
+        self.parser = Parser(self.language)
+
+    def parse_file(self, file_path: str) -> List[Statement]:
+        """Parse a bash script file and return a list of statements."""
+        with open(file_path, "r", encoding="utf-8") as f:
+            content = f.read()
+        return self.parse_string(content)
+
+    def parse_string(self, content: str) -> List[Statement]:
+        """Parse a string containing bash script and return a list of statements."""
+        tree = self.parser.parse(bytes(content, "utf-8"))
+        root_node = tree.root_node
+
+        # For debugging: Uncomment to print the tree structure
+        # self._print_tree(root_node, content)
+
+        statements: List[Statement] = []
+        self._extract_statements(root_node, content, statements, None)
+
+        # Post-process statements to handle multi-line statements correctly
+        return self._post_process_statements(statements, content)
+
+    def _print_tree(self, node: Any, content: str, indent: str = "") -> None:
+        """Debug helper to print the entire syntax tree."""
+        node_text = content[node.start_byte : node.end_byte]
+        if len(node_text) > 40:
+            node_text = node_text[:37] + "..."
+        print(f"{indent}{node.type}: {repr(node_text)}")
+        for child in node.children:
+            self._print_tree(child, content, indent + "  ")
+
+    def _extract_statements(
+        self,
+        node: Any,
+        content: str,
+        statements: List[Statement],
+        parent_type: Optional[str],
+    ) -> None:
+        """Recursively extract statements from the syntax tree."""
+        # Node types that represent bash statements
+        statement_node_types = {
+            # Basic statements
+            "command",
+            "variable_assignment",
+            "declaration_command",
+            "unset_command",
+            # Control flow statements
+            "for_statement",
+            "c_style_for_statement",
+            "while_statement",
+            "if_statement",
+            "case_statement",
+            # Function definition
+            "function_definition",
+            # Command chains and groups
+            "pipeline",  # For command chains with | and |&
+            "list",  # For command chains with && and ||
+            "compound_statement",
+            "subshell",
+            "redirected_statement",
+        }
+
+        # Create a Statement object for this node if it's a recognized statement type
+        if node.type in statement_node_types:
+            # Get the text of this statement
+            start_byte = node.start_byte
+            end_byte = node.end_byte
+            statement_text = content[start_byte:end_byte]
+
+            # Get line numbers
+            start_line = (
+                node.start_point[0] + 1
+            )  # tree-sitter uses 0-indexed line numbers
+            end_line = node.end_point[0] + 1
+
+            statements.append(
+                Statement(
+                    text=statement_text,
+                    start_line=start_line,
+                    end_line=end_line,
+                    start_byte=start_byte,
+                    end_byte=end_byte,
+                    node_type=node.type,
+                    parent_type=parent_type,
+                )
+            )
+
+            # Update parent type for children
+            parent_type = node.type
+
+        # Recursively process all children
+        for child in node.children:
+            self._extract_statements(child, content, statements, parent_type)
+
+    def _post_process_statements(
+        self, statements: List[Statement], content: str
+    ) -> List[Statement]:
+        if not statements:
+            return []
+
+        # Filter out list statements that have been split
+        top_statements = []
+        for stmt in statements:
+            # Skip statements that are contained within others
+            is_contained = False
+            for other in statements:
+                if other is stmt:
+                    continue
+
+                # Check if completely contained (except for lists we've split)
+                if other.node_type != "list" or ";" not in other.text:
+                    if (
+                        other.start_line <= stmt.start_line
+                        and other.end_line >= stmt.end_line
+                        and len(other.text) > len(stmt.text)
+                        and stmt.text in other.text
+                    ):
+                        is_contained = True
+                        break
+
+            if not is_contained:
+                top_statements.append(stmt)
+
+        # Sort by position in file for consistent output
+        top_statements.sort(key=lambda s: (s.start_line, s.text))
+
+        return top_statements
+
+
+def main() -> None:
+    if len(sys.argv) < 2:
+        print("Usage: python bash_statement_parser.py <bash_script_file>")
+        sys.exit(1)
+
+    parser = BashStatementParser()
+    statements = parser.parse_file(sys.argv[1])
+
+    print(f"Found {len(statements)} statements:")
+    for i, stmt in enumerate(statements, 1):
+        print(f"\n--- Statement {i} (Lines {stmt.start_line}-{stmt.end_line}) ---")
+        print(stmt)
+
+
+if __name__ == "__main__":
+    main()

wcgw-4.1.2/tests/test_bash_parser.py

@@ -0,0 +1,80 @@
+"""
+Tests for the bash statement parser.
+"""
+
+import pytest
+from unittest.mock import patch
+
+from wcgw.client.bash_state.parser.bash_statement_parser import BashStatementParser
+
+
+def test_bash_statement_parser_basic():
+    """Test basic statement parsing."""
+    parser = BashStatementParser()
+    
+    # Test single statement
+    statements = parser.parse_string("echo hello")
+    assert len(statements) == 1
+    assert statements[0].text == "echo hello"
+    
+    # Test command with newlines inside string
+    statements = parser.parse_string('echo "hello\nworld"')
+    assert len(statements) == 1
+    
+    # Test command with && chain
+    statements = parser.parse_string("echo hello && echo world")
+    assert len(statements) == 1
+    
+    # Test command with || chain
+    statements = parser.parse_string("echo hello || echo world")
+    assert len(statements) == 1
+    
+    # Test command with pipe
+    statements = parser.parse_string("echo hello | grep hello")
+    assert len(statements) == 1
+
+
+def test_bash_statement_parser_multiple():
+    """Test multiple statement detection."""
+    parser = BashStatementParser()
+    
+    # Test multiple statements on separate lines
+    statements = parser.parse_string("echo hello\necho world")
+    assert len(statements) == 2
+    
+    # Test multiple statements with semicolons
+    statements = parser.parse_string("echo hello; echo world")
+    assert len(statements) == 2
+    
+    # Test more complex case
+    statements = parser.parse_string("echo hello; echo world && echo again")
+    assert len(statements) == 2
+    
+    # Test mixed separation
+    statements = parser.parse_string("echo a; echo b\necho c")
+    assert len(statements) == 3
+
+
+def test_bash_statement_parser_complex():
+    """Test complex statement handling."""
+    parser = BashStatementParser()
+    
+    # Test subshell
+    statements = parser.parse_string("(echo hello; echo world)")
+    assert len(statements) == 1
+    
+    # Test braces
+    statements = parser.parse_string("{ echo hello; echo world; }")
+    assert len(statements) == 1
+    
+    # Test semicolons in strings
+    statements = parser.parse_string('echo "hello;world"')
+    assert len(statements) == 1
+    
+    # Test escaped semicolons
+    statements = parser.parse_string('echo hello\\; echo world')
+    assert len(statements) == 1
+    
+    # Test quoted semicolons
+    statements = parser.parse_string("echo 'hello;world'")
+    assert len(statements) == 1

wcgw-4.1.2/tests/test_bash_parser_complex.py

@@ -0,0 +1,84 @@
+"""
+Tests specifically for complex bash parsing scenarios.
+"""
+
+from wcgw.client.bash_state.parser.bash_statement_parser import BashStatementParser
+
+
+def test_semicolon_lists():
+    """Test parsing of semicolon-separated commands."""
+    parser = BashStatementParser()
+
+    # Simple case: two commands separated by semicolon
+    statements = parser.parse_string("echo a; echo b")
+    assert len(statements) == 2
+    assert statements[0].text.strip() == "echo a"
+    assert statements[1].text.strip() == "echo b"
+
+    # Multiple semicolons
+    statements = parser.parse_string("echo a; echo b; echo c")
+    assert len(statements) == 3
+    assert statements[0].text.strip() == "echo a"
+    assert statements[1].text.strip() == "echo b"
+    assert statements[2].text.strip() == "echo c"
+
+    # Semicolons with whitespace
+    statements = parser.parse_string("echo a  ;  echo b")
+    assert len(statements) == 2
+    assert statements[0].text.strip() == "echo a"
+    assert statements[1].text.strip() == "echo b"
+
+
+def test_bash_command_with_semicolons_in_quotes():
+    """Test that semicolons inside quotes don't split statements."""
+    parser = BashStatementParser()
+
+    # Semicolon in single quotes
+    statements = parser.parse_string("echo 'a;b'")
+    assert len(statements) == 1
+
+    # Semicolon in double quotes
+    statements = parser.parse_string('echo "a;b"')
+    assert len(statements) == 1
+
+    # Mixed quotes
+    statements = parser.parse_string("echo \"a;b\" ; echo 'c;d'")
+    assert len(statements) == 2
+
+
+def test_complex_commands():
+    """Test complex command scenarios."""
+    parser = BashStatementParser()
+
+    # Command with redirection and semicolon
+    statements = parser.parse_string("cat > file.txt << EOF\ntest\nEOF\n; echo done")
+    assert len(statements) == 2
+
+    # Command with subshell and semicolon
+    statements = parser.parse_string("(cd /tmp && echo 'in tmp'); echo 'outside'")
+    assert len(statements) == 2
+
+    # Command with braces and semicolon
+    statements = parser.parse_string("{ echo a; echo b; }; echo c")
+    assert len(statements) == 2
+
+
+def test_command_chaining():
+    """Test command chains are treated as a single statement."""
+    parser = BashStatementParser()
+
+    # AND chaining
+    statements = parser.parse_string("echo a && echo b")
+    assert len(statements) == 1
+
+    # OR chaining
+    statements = parser.parse_string("echo a || echo b")
+    assert len(statements) == 1
+
+    # Pipe chaining
+    statements = parser.parse_string("echo a | grep a")
+    assert len(statements) == 1
+
+    # Mixed chaining
+    statements = parser.parse_string("echo a && echo b || echo c")
+    assert len(statements) == 1

{wcgw-4.1.0 → wcgw-4.1.2}/tests/test_tools.py

@@ -306,6 +306,24 @@ def test_bash_command(context: Context, temp_dir: str) -> None:
     assert isinstance(outputs[0], str)
     assert "hello world" in outputs[0]
 
+    # Test multiline
+    cmd = BashCommand(action_json=Command(command="echo 'hello \nworld'"))
+    outputs, _ = get_tool_output(
+        context, cmd, default_enc, 1.0, lambda x, y: ("", 0.0), None
+    )
+    assert len(outputs) == 1
+    assert isinstance(outputs[0], str)
+    assert "hello\nworld" in outputs[0]
+
+    # Multiple commands should raise exception
+    cmd = BashCommand(action_json=Command(command="echo 'hello'\necho world'"))
+    outputs, _ = get_tool_output(
+        context, cmd, default_enc, 1.0, lambda x, y: ("", 0.0), None
+    )
+    assert len(outputs) == 1
+    assert isinstance(outputs[0], str)
+    assert "Error: Command contains multiple statements" in outputs[0]
+
 
 def test_interaction_commands(context: Context, temp_dir: str) -> None:
     """Test the various interaction command types."""

{wcgw-4.1.0 → wcgw-4.1.2}/uv.lock

@@ -645,6 +645,21 @@ wheels = [
     { url = "https://files.pythonhosted.org/packages/e4/ea/d836f008d33151c7a1f62caf3d8dd782e4d15f6a43897f64480c2b8de2ad/prompt_toolkit-3.0.50-py3-none-any.whl", hash = "sha256:9b6427eb19e479d98acff65196a307c555eb567989e6d88ebbb1b509d9779198", size = 387816 },
 ]
 
+[[package]]
+name = "psutil"
+version = "7.0.0"
+source = { registry = "https://pypi.org/simple" }
+sdist = { url = "https://files.pythonhosted.org/packages/2a/80/336820c1ad9286a4ded7e845b2eccfcb27851ab8ac6abece774a6ff4d3de/psutil-7.0.0.tar.gz", hash = "sha256:7be9c3eba38beccb6495ea33afd982a44074b78f28c434a1f51cc07fd315c456", size = 497003 }
+wheels = [
+    { url = "https://files.pythonhosted.org/packages/ed/e6/2d26234410f8b8abdbf891c9da62bee396583f713fb9f3325a4760875d22/psutil-7.0.0-cp36-abi3-macosx_10_9_x86_64.whl", hash = "sha256:101d71dc322e3cffd7cea0650b09b3d08b8e7c4109dd6809fe452dfd00e58b25", size = 238051 },
+    { url = "https://files.pythonhosted.org/packages/04/8b/30f930733afe425e3cbfc0e1468a30a18942350c1a8816acfade80c005c4/psutil-7.0.0-cp36-abi3-macosx_11_0_arm64.whl", hash = "sha256:39db632f6bb862eeccf56660871433e111b6ea58f2caea825571951d4b6aa3da", size = 239535 },
+    { url = "https://files.pythonhosted.org/packages/2a/ed/d362e84620dd22876b55389248e522338ed1bf134a5edd3b8231d7207f6d/psutil-7.0.0-cp36-abi3-manylinux_2_12_i686.manylinux2010_i686.manylinux_2_17_i686.manylinux2014_i686.whl", hash = "sha256:1fcee592b4c6f146991ca55919ea3d1f8926497a713ed7faaf8225e174581e91", size = 275004 },
+    { url = "https://files.pythonhosted.org/packages/bf/b9/b0eb3f3cbcb734d930fdf839431606844a825b23eaf9a6ab371edac8162c/psutil-7.0.0-cp36-abi3-manylinux_2_12_x86_64.manylinux2010_x86_64.manylinux_2_17_x86_64.manylinux2014_x86_64.whl", hash = "sha256:4b1388a4f6875d7e2aff5c4ca1cc16c545ed41dd8bb596cefea80111db353a34", size = 277986 },
+    { url = "https://files.pythonhosted.org/packages/eb/a2/709e0fe2f093556c17fbafda93ac032257242cabcc7ff3369e2cb76a97aa/psutil-7.0.0-cp36-abi3-manylinux_2_17_aarch64.manylinux2014_aarch64.whl", hash = "sha256:a5f098451abc2828f7dc6b58d44b532b22f2088f4999a937557b603ce72b1993", size = 279544 },
+    { url = "https://files.pythonhosted.org/packages/50/e6/eecf58810b9d12e6427369784efe814a1eec0f492084ce8eb8f4d89d6d61/psutil-7.0.0-cp37-abi3-win32.whl", hash = "sha256:ba3fcef7523064a6c9da440fc4d6bd07da93ac726b5733c29027d7dc95b39d99", size = 241053 },
+    { url = "https://files.pythonhosted.org/packages/50/1b/6921afe68c74868b4c9fa424dad3be35b095e16687989ebbb50ce4fceb7c/psutil-7.0.0-cp37-abi3-win_amd64.whl", hash = "sha256:4cf3d4eb1aa9b348dec30105c55cd9b7d4629285735a102beb4441e38db90553", size = 244885 },
+]
+
 [[package]]
 name = "ptyprocess"
 version = "0.7.0"
@@ -1104,6 +1119,50 @@ wheels = [
     { url = "https://files.pythonhosted.org/packages/00/c0/8f5d070730d7836adc9c9b6408dec68c6ced86b304a9b26a14df072a6e8c/traitlets-5.14.3-py3-none-any.whl", hash = "sha256:b74e89e397b1ed28cc831db7aea759ba6640cb3de13090ca145426688ff1ac4f", size = 85359 },
 ]
 
+[[package]]
+name = "tree-sitter"
+version = "0.24.0"
+source = { registry = "https://pypi.org/simple" }
+sdist = { url = "https://files.pythonhosted.org/packages/a7/a2/698b9d31d08ad5558f8bfbfe3a0781bd4b1f284e89bde3ad18e05101a892/tree-sitter-0.24.0.tar.gz", hash = "sha256:abd95af65ca2f4f7eca356343391ed669e764f37748b5352946f00f7fc78e734", size = 168304 }
+wheels = [
+    { url = "https://files.pythonhosted.org/packages/66/08/82aaf7cbea7286ee2a0b43e9b75cb93ac6ac132991b7d3c26ebe5e5235a3/tree_sitter-0.24.0-cp311-cp311-macosx_10_9_x86_64.whl", hash = "sha256:de0fb7c18c6068cacff46250c0a0473e8fc74d673e3e86555f131c2c1346fb13", size = 140733 },
+    { url = "https://files.pythonhosted.org/packages/8c/bd/1a84574911c40734d80327495e6e218e8f17ef318dd62bb66b55c1e969f5/tree_sitter-0.24.0-cp311-cp311-macosx_11_0_arm64.whl", hash = "sha256:a7c9c89666dea2ce2b2bf98e75f429d2876c569fab966afefdcd71974c6d8538", size = 134243 },
+    { url = "https://files.pythonhosted.org/packages/46/c1/c2037af2c44996d7bde84eb1c9e42308cc84b547dd6da7f8a8bea33007e1/tree_sitter-0.24.0-cp311-cp311-manylinux_2_17_aarch64.manylinux2014_aarch64.whl", hash = "sha256:4ddb113e6b8b3e3b199695b1492a47d87d06c538e63050823d90ef13cac585fd", size = 562030 },
+    { url = "https://files.pythonhosted.org/packages/4c/aa/2fb4d81886df958e6ec7e370895f7106d46d0bbdcc531768326124dc8972/tree_sitter-0.24.0-cp311-cp311-manylinux_2_17_x86_64.manylinux2014_x86_64.whl", hash = "sha256:01ea01a7003b88b92f7f875da6ba9d5d741e0c84bb1bd92c503c0eecd0ee6409", size = 575585 },
+    { url = "https://files.pythonhosted.org/packages/e3/3c/5f997ce34c0d1b744e0f0c0757113bdfc173a2e3dadda92c751685cfcbd1/tree_sitter-0.24.0-cp311-cp311-musllinux_1_2_x86_64.whl", hash = "sha256:464fa5b2cac63608915a9de8a6efd67a4da1929e603ea86abaeae2cb1fe89921", size = 578203 },
+    { url = "https://files.pythonhosted.org/packages/d5/1f/f2bc7fa7c3081653ea4f2639e06ff0af4616c47105dbcc0746137da7620d/tree_sitter-0.24.0-cp311-cp311-win_amd64.whl", hash = "sha256:3b1f3cbd9700e1fba0be2e7d801527e37c49fc02dc140714669144ef6ab58dce", size = 120147 },
+    { url = "https://files.pythonhosted.org/packages/c0/4c/9add771772c4d72a328e656367ca948e389432548696a3819b69cdd6f41e/tree_sitter-0.24.0-cp311-cp311-win_arm64.whl", hash = "sha256:f3f08a2ca9f600b3758792ba2406971665ffbad810847398d180c48cee174ee2", size = 108302 },
+    { url = "https://files.pythonhosted.org/packages/e9/57/3a590f287b5aa60c07d5545953912be3d252481bf5e178f750db75572bff/tree_sitter-0.24.0-cp312-cp312-macosx_10_13_x86_64.whl", hash = "sha256:14beeff5f11e223c37be7d5d119819880601a80d0399abe8c738ae2288804afc", size = 140788 },
+    { url = "https://files.pythonhosted.org/packages/61/0b/fc289e0cba7dbe77c6655a4dd949cd23c663fd62a8b4d8f02f97e28d7fe5/tree_sitter-0.24.0-cp312-cp312-macosx_11_0_arm64.whl", hash = "sha256:26a5b130f70d5925d67b47db314da209063664585a2fd36fa69e0717738efaf4", size = 133945 },
+    { url = "https://files.pythonhosted.org/packages/86/d7/80767238308a137e0b5b5c947aa243e3c1e3e430e6d0d5ae94b9a9ffd1a2/tree_sitter-0.24.0-cp312-cp312-manylinux_2_17_aarch64.manylinux2014_aarch64.whl", hash = "sha256:5fc5c3c26d83c9d0ecb4fc4304fba35f034b7761d35286b936c1db1217558b4e", size = 564819 },
+    { url = "https://files.pythonhosted.org/packages/bf/b3/6c5574f4b937b836601f5fb556b24804b0a6341f2eb42f40c0e6464339f4/tree_sitter-0.24.0-cp312-cp312-manylinux_2_17_x86_64.manylinux2014_x86_64.whl", hash = "sha256:772e1bd8c0931c866b848d0369b32218ac97c24b04790ec4b0e409901945dd8e", size = 579303 },
+    { url = "https://files.pythonhosted.org/packages/0a/f4/bd0ddf9abe242ea67cca18a64810f8af230fc1ea74b28bb702e838ccd874/tree_sitter-0.24.0-cp312-cp312-musllinux_1_2_x86_64.whl", hash = "sha256:24a8dd03b0d6b8812425f3b84d2f4763322684e38baf74e5bb766128b5633dc7", size = 581054 },
+    { url = "https://files.pythonhosted.org/packages/8c/1c/ff23fa4931b6ef1bbeac461b904ca7e49eaec7e7e5398584e3eef836ec96/tree_sitter-0.24.0-cp312-cp312-win_amd64.whl", hash = "sha256:f9e8b1605ab60ed43803100f067eed71b0b0e6c1fb9860a262727dbfbbb74751", size = 120221 },
+    { url = "https://files.pythonhosted.org/packages/b2/2a/9979c626f303177b7612a802237d0533155bf1e425ff6f73cc40f25453e2/tree_sitter-0.24.0-cp312-cp312-win_arm64.whl", hash = "sha256:f733a83d8355fc95561582b66bbea92ffd365c5d7a665bc9ebd25e049c2b2abb", size = 108234 },
+    { url = "https://files.pythonhosted.org/packages/61/cd/2348339c85803330ce38cee1c6cbbfa78a656b34ff58606ebaf5c9e83bd0/tree_sitter-0.24.0-cp313-cp313-macosx_10_13_x86_64.whl", hash = "sha256:0d4a6416ed421c4210f0ca405a4834d5ccfbb8ad6692d4d74f7773ef68f92071", size = 140781 },
+    { url = "https://files.pythonhosted.org/packages/8b/a3/1ea9d8b64e8dcfcc0051028a9c84a630301290995cd6e947bf88267ef7b1/tree_sitter-0.24.0-cp313-cp313-macosx_11_0_arm64.whl", hash = "sha256:e0992d483677e71d5c5d37f30dfb2e3afec2f932a9c53eec4fca13869b788c6c", size = 133928 },
+    { url = "https://files.pythonhosted.org/packages/fe/ae/55c1055609c9428a4aedf4b164400ab9adb0b1bf1538b51f4b3748a6c983/tree_sitter-0.24.0-cp313-cp313-manylinux_2_17_aarch64.manylinux2014_aarch64.whl", hash = "sha256:57277a12fbcefb1c8b206186068d456c600dbfbc3fd6c76968ee22614c5cd5ad", size = 564497 },
+    { url = "https://files.pythonhosted.org/packages/ce/d0/f2ffcd04882c5aa28d205a787353130cbf84b2b8a977fd211bdc3b399ae3/tree_sitter-0.24.0-cp313-cp313-manylinux_2_17_x86_64.manylinux2014_x86_64.whl", hash = "sha256:d25fa22766d63f73716c6fec1a31ee5cf904aa429484256bd5fdf5259051ed74", size = 578917 },
+    { url = "https://files.pythonhosted.org/packages/af/82/aebe78ea23a2b3a79324993d4915f3093ad1af43d7c2208ee90be9273273/tree_sitter-0.24.0-cp313-cp313-musllinux_1_2_x86_64.whl", hash = "sha256:7d5d9537507e1c8c5fa9935b34f320bfec4114d675e028f3ad94f11cf9db37b9", size = 581148 },
+    { url = "https://files.pythonhosted.org/packages/a1/b4/6b0291a590c2b0417cfdb64ccb8ea242f270a46ed429c641fbc2bfab77e0/tree_sitter-0.24.0-cp313-cp313-win_amd64.whl", hash = "sha256:f58bb4956917715ec4d5a28681829a8dad5c342cafd4aea269f9132a83ca9b34", size = 120207 },
+    { url = "https://files.pythonhosted.org/packages/a8/18/542fd844b75272630229c9939b03f7db232c71a9d82aadc59c596319ea6a/tree_sitter-0.24.0-cp313-cp313-win_arm64.whl", hash = "sha256:23641bd25dcd4bb0b6fa91b8fb3f46cc9f1c9f475efe4d536d3f1f688d1b84c8", size = 108232 },
+]
+
+[[package]]
+name = "tree-sitter-bash"
+version = "0.23.3"
+source = { registry = "https://pypi.org/simple" }
+sdist = { url = "https://files.pythonhosted.org/packages/7b/e0/1e73a17c5427dc62fc42e29f1e58b3a3c95a8fa314983a37f25a0c15be1f/tree_sitter_bash-0.23.3.tar.gz", hash = "sha256:7b15ed89a1ea8e3e3c2399758746413e464d4c1c3a6d3b75d643ae2bc2fb356b", size = 420078 }
+wheels = [
+    { url = "https://files.pythonhosted.org/packages/99/4d/c3a1105be2386fd68f1fe8ab9dac5d0d24011d610af2904f48a836525a73/tree_sitter_bash-0.23.3-cp39-abi3-macosx_10_9_x86_64.whl", hash = "sha256:c1ee7a46fcbfca9937d01056be756631762f53c5afdb8c4ab64eb9fed060896b", size = 194714 },
+    { url = "https://files.pythonhosted.org/packages/3c/d0/e87e4569a82a29037c06694ed0116c60368edef2d7b822a38514e95360d8/tree_sitter_bash-0.23.3-cp39-abi3-macosx_11_0_arm64.whl", hash = "sha256:5a090118e887bf667d82ae445794906186216f5500e0d2cd58eb499f7502dc57", size = 204226 },
+    { url = "https://files.pythonhosted.org/packages/c5/a2/216ae6fffbcb0fb831865477f53115930dae5ad1b5a16ed12259c8aaecea/tree_sitter_bash-0.23.3-cp39-abi3-manylinux_2_17_aarch64.manylinux2014_aarch64.whl", hash = "sha256:aa4b5dde719291eea3a81b1f9ece6afeee2deadc2b2f769bee92f955da7595cf", size = 265254 },
+    { url = "https://files.pythonhosted.org/packages/2e/89/5031979dc0be9946e999f790b4ceef42810a819508f77fa4017aad0dd5e2/tree_sitter_bash-0.23.3-cp39-abi3-manylinux_2_17_x86_64.manylinux2014_x86_64.whl", hash = "sha256:2ff7bffc3d594e7f1054de051e19df1b24082963598a175dda64083c6b3eea1a", size = 242476 },
+    { url = "https://files.pythonhosted.org/packages/52/0a/af4df04efbf253a93cc869f710a351ab2462f218135277268865eb599f5d/tree_sitter_bash-0.23.3-cp39-abi3-musllinux_1_2_x86_64.whl", hash = "sha256:4427baccbd7549a2ebb1859b6d42cdab0739c05d53c2b3daad9cadc069a7b3f6", size = 230285 },
+    { url = "https://files.pythonhosted.org/packages/46/86/b1ecfe46a4dbae96a6900a3aa19ceb8d15815b29d924d36f02026780c719/tree_sitter_bash-0.23.3-cp39-abi3-win_amd64.whl", hash = "sha256:525c5cce28a7c5624fb016ac8f3ae33d32968567b718f7878c6351229d2e8394", size = 196940 },
+    { url = "https://files.pythonhosted.org/packages/e2/1e/cab42516fc72c136d388bfbefe34be9fc109ff5a1fbba3c30209970cd57e/tree_sitter_bash-0.23.3-cp39-abi3-win_arm64.whl", hash = "sha256:1f703d1bf6235355f6c900be64bf9f61fc4b1d0cfed6829b4eeb74a6b41ea910", size = 191919 },
+]
+
 [[package]]
 name = "typer"
 version = "0.15.1"
@@ -1128,6 +1187,15 @@ wheels = [
     { url = "https://files.pythonhosted.org/packages/37/c6/14c23e04bae6a83e051da86c1670684e59acadab333a8497384aa201defd/types_pexpect-4.9.0.20241208-py3-none-any.whl", hash = "sha256:1928f478528454f0fea3495c16cf1ee2e67fca5c9fe97d60b868ac48c1fd5633", size = 17085 },
 ]
 
+[[package]]
+name = "types-psutil"
+version = "7.0.0.20250218"
+source = { registry = "https://pypi.org/simple" }
+sdist = { url = "https://files.pythonhosted.org/packages/b5/7c/145600d30456e7ccbb499abcf718aab2bd830e604a0ae8eb32b67cd346a6/types_psutil-7.0.0.20250218.tar.gz", hash = "sha256:1e642cdafe837b240295b23b1cbd4691d80b08a07d29932143cbbae30eb0db9c", size = 19828 }
+wheels = [
+    { url = "https://files.pythonhosted.org/packages/50/c8/f4365293408da4a9bcb1849d3efd8c60427cffff68cbb98ab1b81851d8bb/types_psutil-7.0.0.20250218-py3-none-any.whl", hash = "sha256:1447a30c282aafefcf8941ece854e1100eee7b0296a9d9be9977292f0269b121", size = 22763 },
+]
+
 [[package]]
 name = "types-toml"
 version = "0.10.8.20240310"
@@ -1170,7 +1238,7 @@ wheels = [
 
 [[package]]
 name = "wcgw"
-version = "4.1.0"
+version = "4.1.1"
 source = { editable = "." }
 dependencies = [
     { name = "anthropic" },
@@ -1178,6 +1246,7 @@ dependencies = [
     { name = "openai" },
     { name = "petname" },
     { name = "pexpect" },
+    { name = "psutil" },
     { name = "pydantic" },
     { name = "pygit2" },
     { name = "pyte" },
@@ -1202,7 +1271,10 @@ dev = [
     { name = "pytest" },
     { name = "pytest-asyncio" },
     { name = "pytest-cov" },
+    { name = "tree-sitter" },
+    { name = "tree-sitter-bash" },
     { name = "types-pexpect" },
+    { name = "types-psutil" },
     { name = "types-toml" },
 ]
 
@@ -1213,6 +1285,7 @@ requires-dist = [
     { name = "openai", specifier = ">=1.46.0" },
     { name = "petname", specifier = ">=2.6" },
     { name = "pexpect", specifier = ">=4.9.0" },
+    { name = "psutil", specifier = ">=7.0.0" },
     { name = "pydantic", specifier = ">=2.9.2" },
     { name = "pygit2", specifier = ">=1.16.0" },
     { name = "pyte", specifier = ">=0.8.2" },
@@ -1237,7 +1310,10 @@ dev = [
     { name = "pytest", specifier = ">=8.0.0" },
     { name = "pytest-asyncio", specifier = ">=0.25.3" },
     { name = "pytest-cov", specifier = ">=4.1.0" },
+    { name = "tree-sitter", specifier = ">=0.24.0" },
+    { name = "tree-sitter-bash", specifier = ">=0.23.3" },
     { name = "types-pexpect", specifier = ">=4.9.0.20241208" },
+    { name = "types-psutil", specifier = ">=7.0.0.20250218" },
     { name = "types-toml", specifier = ">=0.10.8.20240310" },
 ]