wcgw 3.0.5__py3-none-any.whl → 3.0.7__py3-none-any.whl

wcgw/client/bash_state/bash_state.py

@@ -78,7 +78,9 @@ def get_tmpdir() -> str:
 
 def check_if_screen_command_available() -> bool:
     try:
-        subprocess.run(["screen", "-v"], capture_output=True, check=True, timeout=0.2)
+        subprocess.run(
+            ["which", "screen"], capture_output=True, check=True, timeout=0.2
+        )
         return True
     except (subprocess.CalledProcessError, FileNotFoundError):
         return False
@@ -137,7 +139,7 @@ def cleanup_all_screens_with_name(name: str, console: Console) -> None:
 
 def start_shell(
     is_restricted_mode: bool, initial_dir: str, console: Console, over_screen: bool
-) -> tuple[pexpect.spawn, str]:  # type: ignore[type-arg]
+) -> tuple["pexpect.spawn[str]", str]:
     cmd = "/bin/bash"
     if is_restricted_mode:
         cmd += " -r"
@@ -232,6 +234,7 @@ class BashState:
         use_screen: bool,
         whitelist_for_overwrite: Optional[set[str]] = None,
     ) -> None:
+        self._last_command: str = ""
         self.console = console
         self._cwd = working_dir or os.getcwd()
         self._bash_command_mode: BashCommandMode = bash_command_mode or BashCommandMode(
@@ -253,13 +256,17 @@ class BashState:
         output = self._shell.expect(pattern, timeout)
         return output
 
-    def send(self, s: str | bytes) -> int:
+    def send(self, s: str | bytes, set_as_command: Optional[str]) -> int:
         self.close_bg_expect_thread()
+        if set_as_command is not None:
+            self._last_command = set_as_command
         output = self._shell.send(s)
         return output
 
-    def sendline(self, s: str | bytes) -> int:
+    def sendline(self, s: str | bytes, set_as_command: Optional[str]) -> int:
         self.close_bg_expect_thread()
+        if set_as_command is not None:
+            self._last_command = set_as_command
         output = self._shell.sendline(s)
         return output
 
@@ -273,7 +280,10 @@ class BashState:
 
     @property
     def before(self) -> Optional[str]:
-        return self._shell.before
+        before = self._shell.before
+        if before and before.startswith(self._last_command):
+            return before[len(self._last_command) :]
+        return before
 
     def run_bg_expect_thread(self) -> None:
         """
@@ -333,11 +343,11 @@ class BashState:
     def ensure_env_and_bg_jobs(self) -> Optional[int]:
         quick_timeout = 0.2 if not self.over_screen else 1
         # First reset the prompt in case venv was sourced or other reasons.
-        self.sendline(PROMPT_STATEMENT)
+        self.sendline(PROMPT_STATEMENT, set_as_command=PROMPT_STATEMENT)
         self.expect(PROMPT_CONST, timeout=quick_timeout)
         # Reset echo also if it was enabled
         command = "jobs | wc -l"
-        self.sendline(command)
+        self.sendline(command, set_as_command=command)
         before = ""
         counts = 0
         while not _is_int(before):  # Consume all previous output
@@ -365,6 +375,7 @@ class BashState:
 
     def _init_shell(self) -> None:
         self._state: Literal["repl"] | datetime.datetime = "repl"
+        self._last_command = ""
         # Ensure self._cwd exists
         os.makedirs(self._cwd, exist_ok=True)
         try:
@@ -404,6 +415,7 @@ class BashState:
     def set_repl(self) -> None:
         self._state = "repl"
         self._pending_output = ""
+        self._last_command = ""
 
     @property
     def state(self) -> BASH_CLF_OUTPUT:
@@ -420,7 +432,7 @@ class BashState:
         return PROMPT_CONST
 
     def update_cwd(self) -> str:
-        self.sendline("pwd")
+        self.sendline("pwd", set_as_command="pwd")
         self.expect(PROMPT_CONST, timeout=0.2)
         before_val = self.before
         if not isinstance(before_val, str):
@@ -505,8 +517,8 @@ class BashState:
 
 
 WAITING_INPUT_MESSAGE = """A command is already running. NOTE: You can't run multiple shell sessions, likely a previous program hasn't exited. 
-1. Get its output using `send_ascii: [10] or send_specials: ["Enter"]`
-2. Use `send_ascii` or `send_specials` to give inputs to the running program, don't use `BashCommand` OR
+1. Get its output using status check.
+2. Use `send_ascii` or `send_specials` to give inputs to the running program OR
 3. kill the previous program by sending ctrl+c first using `send_ascii` or `send_specials`
 4. Interrupt and run the process in background by re-running it using screen
 """
@@ -538,6 +550,11 @@ def _incremental_text(text: str, last_pending_output: str) -> str:
     # text = render_terminal_output(text[-100_000:])
     text = text[-100_000:]
 
+    if not last_pending_output:
+        # This is the first call. We need to offset the position where this program
+        # is being rendered for the new screen versions
+        # Caveat: no difference in output between a program with leading whitespace and one without.
+        return rstrip(render_terminal_output(text)).lstrip()
     last_rendered_lines = render_terminal_output(last_pending_output)
     last_pending_output_rendered = "\n".join(last_rendered_lines)
     if not last_rendered_lines:
@@ -580,7 +597,10 @@ def is_status_check(arg: BashCommand) -> bool:
             isinstance(arg.action_json, SendSpecials)
             and arg.action_json.send_specials == ["Enter"]
         )
-        or (isinstance(arg.action_json, SendAscii) and arg.action_json.send_ascii == [10])
+        or (
+            isinstance(arg.action_json, SendAscii)
+            and arg.action_json.send_ascii == [10]
+        )
     )
 
 
@@ -632,8 +652,8 @@ def _execute_bash(
                 )
 
             for i in range(0, len(command), 128):
-                bash_state.send(command[i : i + 128])
-            bash_state.send(bash_state.linesep)
+                bash_state.send(command[i : i + 128], set_as_command=None)
+            bash_state.send(bash_state.linesep, set_as_command=command)
 
         elif isinstance(command_data, StatusCheck):
             bash_state.console.print("Checking status")
@@ -646,8 +666,10 @@ def _execute_bash(
 
             bash_state.console.print(f"Interact text: {command_data.send_text}")
             for i in range(0, len(command_data.send_text), 128):
-                bash_state.send(command_data.send_text[i : i + 128])
-            bash_state.send(bash_state.linesep)
+                bash_state.send(
+                    command_data.send_text[i : i + 128], set_as_command=None
+                )
+            bash_state.send(bash_state.linesep, set_as_command=None)
 
         elif isinstance(command_data, SendSpecials):
             if not command_data.send_specials:
@@ -658,15 +680,15 @@ def _execute_bash(
             )
             for char in command_data.send_specials:
                 if char == "Key-up":
-                    bash_state.send("\033[A")
+                    bash_state.send("\033[A", set_as_command=None)
                 elif char == "Key-down":
-                    bash_state.send("\033[B")
+                    bash_state.send("\033[B", set_as_command=None)
                 elif char == "Key-left":
-                    bash_state.send("\033[D")
+                    bash_state.send("\033[D", set_as_command=None)
                 elif char == "Key-right":
-                    bash_state.send("\033[C")
+                    bash_state.send("\033[C", set_as_command=None)
                 elif char == "Enter":
-                    bash_state.send("\n")
+                    bash_state.send("\n", set_as_command=None)
                 elif char == "Ctrl-c":
                     bash_state.sendintr()
                     is_interrupt = True
@@ -674,7 +696,7 @@ def _execute_bash(
                     bash_state.sendintr()
                     is_interrupt = True
                 elif char == "Ctrl-z":
-                    bash_state.send("\x1a")
+                    bash_state.send("\x1a", set_as_command=None)
                 else:
                     raise Exception(f"Unknown special character: {char}")
 
@@ -686,7 +708,7 @@ def _execute_bash(
                 f"Sending ASCII sequence: {command_data.send_ascii}"
             )
             for ascii_char in command_data.send_ascii:
-                bash_state.send(chr(ascii_char))
+                bash_state.send(chr(ascii_char), set_as_command=None)
                 if ascii_char == 3:
                     is_interrupt = True
         else:
@@ -747,9 +769,9 @@ def _execute_bash(
                 incremental_text = (
                     incremental_text
                     + """---
-    ----
-    Failure interrupting.
-    You may want to try Ctrl-c again or program specific exit interactive commands.
+----
+Failure interrupting.
+You may want to try Ctrl-c again or program specific exit interactive commands.
     """
                 )
 

wcgw/client/tools.py

@@ -20,6 +20,7 @@ from typing import (
     TypeVar,
 )
 
+import rich
 from openai.types.chat import (
     ChatCompletionMessageParam,
 )
@@ -31,6 +32,7 @@ from wcgw.client.bash_state.bash_state import get_status
 from ..types_ import (
     BashCommand,
     CodeWriterMode,
+    Command,
     Console,
     ContextSave,
     FileEdit,
@@ -195,7 +197,10 @@ def initialize(
     if read_files_:
         if folder_to_start:
             read_files_ = [
-                os.path.join(folder_to_start, f) if not os.path.isabs(f) else f
+                # Expand the path before checking if it's absolute
+                os.path.join(folder_to_start, f)
+                if not os.path.isabs(expand_user(f))
+                else expand_user(f)
                 for f in read_files_
             ]
         initial_files = read_files(read_files_, max_tokens, context)
@@ -362,9 +367,10 @@ def truncate_if_over(content: str, max_tokens: Optional[int]) -> str:
 
 
 def read_image_from_shell(file_path: str, context: Context) -> ImageData:
-    # Expand the path
+    # Expand the path before checking if it's absolute
     file_path = expand_user(file_path)
 
+    # If not absolute after expansion, join with current working directory
     if not os.path.isabs(file_path):
         file_path = os.path.join(context.bash_state.cwd, file_path)
 
@@ -400,10 +406,10 @@ def write_file(
     max_tokens: Optional[int],
     context: Context,
 ) -> str:
-    if not os.path.isabs(writefile.file_path):
+    # Expand the path before checking if it's absolute
+    path_ = expand_user(writefile.file_path)
+    if not os.path.isabs(path_):
         return f"Failure: file_path should be absolute path, current working directory is {context.bash_state.cwd}"
-    else:
-        path_ = expand_user(writefile.file_path)
 
     error_on_exist_ = (
         error_on_exist and path_ not in context.bash_state.whitelist_for_overwrite
@@ -452,6 +458,9 @@ def write_file(
         syntax_errors = check.description
 
         if syntax_errors:
+            if extension in {"tsx", "ts"}:
+                syntax_errors += "\nNote: Ignore if 'tagged template literals' are used, they may raise false positive errors in tree-sitter."
+
             context_for_errors = get_context_for_errors(
                 check.errors, writefile.file_content, max_tokens
             )
@@ -502,12 +511,12 @@ def do_diff_edit(fedit: FileEdit, max_tokens: Optional[int], context: Context) -
 def _do_diff_edit(fedit: FileEdit, max_tokens: Optional[int], context: Context) -> str:
     context.console.log(f"Editing file: {fedit.file_path}")
 
-    if not os.path.isabs(fedit.file_path):
+    # Expand the path before checking if it's absolute
+    path_ = expand_user(fedit.file_path)
+    if not os.path.isabs(path_):
         raise Exception(
             f"Failure: file_path should be absolute path, current working directory is {context.bash_state.cwd}"
         )
-    else:
-        path_ = expand_user(fedit.file_path)
 
     # Validate using file_edit_mode
     allowed_globs = context.bash_state.file_edit_mode.allowed_globs
@@ -548,11 +557,13 @@ def _do_diff_edit(fedit: FileEdit, max_tokens: Optional[int], context: Context)
             context_for_errors = get_context_for_errors(
                 check.errors, apply_diff_to, max_tokens
             )
+            if extension in {"tsx", "ts"}:
+                syntax_errors += "\nNote: Ignore if 'tagged template literals' are used, they may raise false positive errors in tree-sitter."
 
             context.console.print(f"W: Syntax errors encountered: {syntax_errors}")
             return f"""{comments}
 ---
-Tree-sitter reported syntax errors, please re-read the file and fix if there are any errors.
+Warning: tree-sitter reported syntax errors, please re-read the file and fix if there are any errors.
 Syntax errors:
 {syntax_errors}
 
@@ -698,9 +709,12 @@ def get_tool_output(
         context.console.print("Calling task memory tool")
         relevant_files = []
         warnings = ""
+        # Expand user in project root path
         arg.project_root_path = os.path.expanduser(arg.project_root_path)
         for fglob in arg.relevant_file_globs:
+            # Expand user in glob pattern before checking if it's absolute
             fglob = expand_user(fglob)
+            # If not absolute after expansion, join with project root path
             if not os.path.isabs(fglob) and arg.project_root_path:
                 fglob = os.path.join(arg.project_root_path, fglob)
             globs = glob.glob(fglob, recursive=True)
@@ -794,6 +808,46 @@ def read_file(
             rest = save_out_of_context(
                 default_enc.decoder(tokens[max_tokens:]), Path(file_path).suffix
             )
-            content += f"\n(...truncated)\n---\nI've saved the continuation in a new file. Please read: `{rest}`"
+            content += f"\n(...truncated)\n---\nI've saved the continuation in a new file. You may want to read: `{rest}`"
             truncated = True
     return content, truncated, tokens_counts
+
+
+if __name__ == "__main__":
+    with BashState(
+        rich.console.Console(style="blue", highlight=False, markup=False),
+        "",
+        None,
+        None,
+        None,
+        None,
+        True,
+        None,
+    ) as BASH_STATE:
+        print(
+            get_tool_output(
+                Context(BASH_STATE, BASH_STATE.console),
+                Initialize(
+                    type="first_call",
+                    any_workspace_path="",
+                    initial_files_to_read=[],
+                    task_id_to_resume="",
+                    mode_name="wcgw",
+                    code_writer_config=None,
+                ),
+                default_enc,
+                0,
+                lambda x, y: ("", 0),
+                None,
+            )
+        )
+        print(
+            get_tool_output(
+                Context(BASH_STATE, BASH_STATE.console),
+                BashCommand(action_json=Command(command="pwd")),
+                default_enc,
+                0,
+                lambda x, y: ("", 0),
+                None,
+            )
+        )

wcgw/relay/serve.py

@@ -1,4 +1,5 @@
 import asyncio
+import os
 import threading
 import time
 from importlib import metadata
@@ -281,7 +282,7 @@ async def context_save(context_save_data: ContextSaveWithUUID) -> str:
     raise fastapi.HTTPException(status_code=500, detail="Timeout error")
 
 
-app.mount("/static", StaticFiles(directory="static"), name="static")
+app.mount("/static", StaticFiles(directory=os.path.join(os.path.dirname(__file__), "static")), name="static")
 
 
 def run() -> None:

{wcgw-3.0.5.dist-info → wcgw-3.0.7.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: wcgw
-Version: 3.0.5
+Version: 3.0.7
 Summary: Shell and coding agent on claude and chatgpt
 Project-URL: Homepage, https://github.com/rusiaaman/wcgw
 Author-email: Aman Rusia <gapypi@arcfu.com>
@@ -204,6 +204,32 @@ Read here: https://github.com/rusiaaman/wcgw/blob/main/openai.md
 
 ![example](https://github.com/rusiaaman/wcgw/blob/main/static/example.jpg?raw=true)
 
+
+## Using mcp server over docker
+
+First build the docker image `docker build -t wcgw https://github.com/rusiaaman/wcgw.git`
+
+Then you can update `/Users/username/Library/Application Support/Claude/claude_desktop_config.json` to have
+```
+{
+  "mcpServers": {
+    "filesystem": {
+      "command": "docker",
+      "args": [
+        "run",
+        "-i",
+        "--rm",
+        "--mount",
+        "type=bind,src=/Users/username/Desktop,dst=/workspace/Desktop",
+        "wcgw",
+      ]
+    }
+  }
+}
+```
+
+
+
 ## [Optional] Local shell access with openai API key or anthropic API key
 
 ### Openai
@@ -255,3 +281,4 @@ The server provides the following MCP tools:
   - Parameters: `id` (string), `project_root_path` (string), `description` (string), `relevant_file_globs` (string[])
 
 All tools support absolute paths and include built-in protections against common errors. See the [MCP specification](https://modelcontextprotocol.io/) for detailed protocol information.
+

{wcgw-3.0.5.dist-info → wcgw-3.0.7.dist-info}/RECORD

@@ -7,8 +7,8 @@ wcgw/client/diff-instructions.txt,sha256=tmJ9Fu9XdO_72lYXQQNY9RZyx91bjxrXJf9d_KB
 wcgw/client/memory.py,sha256=8LdYsOhvCOoC1kfvDr85kNy07WnhPMvE6B2FRM2w85Y,2902
 wcgw/client/modes.py,sha256=FjIQOjT5oI7dk9VG0oRemN1A6EHBvHnuSQGKuRbguJI,10352
 wcgw/client/tool_prompts.py,sha256=H36Sr3yH2YuHZTc08Y6rTUWlYWjKMFtIGP6vbAVNwZo,3988
-wcgw/client/tools.py,sha256=Sz6tCm9KXVKZHbUqGViYUZnhJ78oV4_k1USX205HLI0,25415
-wcgw/client/bash_state/bash_state.py,sha256=_qQC7dASIm4c1wAb6SKTdAdNYD5l3Km-WyA7U_lDt_k,26157
+wcgw/client/tools.py,sha256=rboKbRHtIl_4Mhypc23CrDaswXOHjyovyxWkwbXswXk,27387
+wcgw/client/bash_state/bash_state.py,sha256=aY_hW9y7TyPWHR5799v6rxzuWb1VzgjiGjA4K0cpGjE,27287
 wcgw/client/encoder/__init__.py,sha256=Y-8f43I6gMssUCWpX5rLYiAFv3D-JPRs4uNEejPlke8,1514
 wcgw/client/file_ops/diff_edit.py,sha256=sIwXSSkWYff_Dp3oHfefqtSuFqLrxbhKvzkCoLuLqDE,18679
 wcgw/client/file_ops/search_replace.py,sha256=Napa7IWaYPGMNdttunKyRDkb90elZE7r23B_o_htRxo,5585
@@ -21,13 +21,13 @@ wcgw/client/repo_ops/paths_model.vocab,sha256=M1pXycYDQehMXtpp-qAgU7rtzeBbCOiJo4
 wcgw/client/repo_ops/paths_tokens.model,sha256=jiwwE4ae8ADKuTZISutXuM5Wfyc_FBmN5rxTjoNnCos,1569052
 wcgw/client/repo_ops/repo_context.py,sha256=5NqRxBY0K-SBFXJ0Ybt7llzYOBD8pRkTpruMMJHWxv4,4336
 wcgw/relay/client.py,sha256=BUeEKUsWts8RpYxXwXcyFyjBJhOCS-CxThAlL_-VCOI,3618
-wcgw/relay/serve.py,sha256=xFAFiRENPae723_kwaaCgBmqL5sffYHKPu3esrMu57M,7896
+wcgw/relay/serve.py,sha256=UFaRlGTfeD9Oc2AcPF8rIdTRwROBKoT2HglOKrc-aoo,7947
 wcgw/relay/static/privacy.txt,sha256=s9qBdbx2SexCpC_z33sg16TptmAwDEehMCLz4L50JLc,529
 wcgw_cli/__init__.py,sha256=TNxXsTPgb52OhakIda9wTRh91cqoBqgQRx5TxjzQQFU,21
 wcgw_cli/__main__.py,sha256=wcCrL4PjG51r5wVKqJhcoJPTLfHW0wNbD31DrUN0MWI,28
-wcgw_cli/anthropic_client.py,sha256=gQc1opw-N5ecqcORbvvHGBW1Ac-Soe7c7APJD25HIfo,19887
+wcgw_cli/anthropic_client.py,sha256=-rfnw_tIx1VjhzOsPeTZkoF0VAXhRC9zBW2oLMFJcxg,19624
 wcgw_cli/cli.py,sha256=-7FBe_lahKyUOhf65iurTA1M1gXXXAiT0OVKQVcZKKo,948
-wcgw_cli/openai_client.py,sha256=o4vfhcfGKqoWYWnn5YxNw4muKyDnFEiPmWpZpufDGMA,16021
+wcgw_cli/openai_client.py,sha256=4NFv-d7u6B7vMD_w3Qqr-564ZcdvfBUGowUmQ6idglE,15758
 wcgw_cli/openai_utils.py,sha256=xGOb3W5ALrIozV7oszfGYztpj0FnXdD7jAxm5lEIVKY,2439
 mcp_wcgw/__init__.py,sha256=fKCgOdN7cn7gR3YGFaGyV5Goe8A2sEyllLcsRkN0i-g,2601
 mcp_wcgw/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -51,8 +51,8 @@ mcp_wcgw/shared/memory.py,sha256=dBsOghxHz8-tycdSVo9kSujbsC8xb_tYsGmuJobuZnw,281
 mcp_wcgw/shared/progress.py,sha256=ymxOsb8XO5Mhlop7fRfdbmvPodANj7oq6O4dD0iUcnw,1048
 mcp_wcgw/shared/session.py,sha256=e44a0LQOW8gwdLs9_DE9oDsxqW2U8mXG3d5KT95bn5o,10393
 mcp_wcgw/shared/version.py,sha256=d2LZii-mgsPIxpshjkXnOTUmk98i0DT4ff8VpA_kAvE,111
-wcgw-3.0.5.dist-info/METADATA,sha256=H1pRv12T5HGlwRjRd-RuMkRmpJEMkbCrVre1aeb93PE,14049
-wcgw-3.0.5.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-wcgw-3.0.5.dist-info/entry_points.txt,sha256=vd3tj1_Kzfp55LscJ8-6WFMM5hm9cWTfNGFCrWBnH3Q,124
-wcgw-3.0.5.dist-info/licenses/LICENSE,sha256=BvY8xqjOfc3X2qZpGpX3MZEmF-4Dp0LqgKBbT6L_8oI,11142
-wcgw-3.0.5.dist-info/RECORD,,
+wcgw-3.0.7.dist-info/METADATA,sha256=EUfU99SQ5awMMfYVqCS969KEJ_4sWvcd8Jn9DkbVq0s,14549
+wcgw-3.0.7.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+wcgw-3.0.7.dist-info/entry_points.txt,sha256=vd3tj1_Kzfp55LscJ8-6WFMM5hm9cWTfNGFCrWBnH3Q,124
+wcgw-3.0.7.dist-info/licenses/LICENSE,sha256=BvY8xqjOfc3X2qZpGpX3MZEmF-4Dp0LqgKBbT6L_8oI,11142
+wcgw-3.0.7.dist-info/RECORD,,

wcgw_cli/anthropic_client.py

@@ -227,17 +227,6 @@ def loop(
             mode="wcgw",
         )
 
-        with open(
-            os.path.join(
-                os.path.dirname(__file__),
-                "..",
-                "wcgw",
-                "client",
-                "diff-instructions.txt",
-            )
-        ) as f:
-            system += f.read()
-
         if history:
             if (
                 (last_msg := history[-1])["role"] == "user"

wcgw_cli/openai_client.py

@@ -191,17 +191,6 @@ def loop(
             mode="wcgw",
         )
 
-        with open(
-            os.path.join(
-                os.path.dirname(__file__),
-                "..",
-                "wcgw",
-                "client",
-                "diff-instructions.txt",
-            )
-        ) as f:
-            system += f.read()
-
         if not history:
             history = [{"role": "system", "content": system}]
         else: