Timeout related fixes: use custom fork of mcp

Author: Aman Rusia

pyproject.toml

@@ -1,7 +1,7 @@
 [project]
 authors = [{ name = "Aman Rusia", email = "[email protected]" }]
 name = "wcgw"
-version = "1.5.0"
+version = "1.5.1"
 description = "What could go wrong giving full shell access to chatgpt?"
 readme = "README.md"
 requires-python = ">=3.11, <3.13"
@@ -25,7 +25,7 @@ dependencies = [
     "semantic-version>=2.10.0",
     "nltk>=3.9.1",
     "anthropic>=0.39.0",
-    "mcp>=1.0.0",
+    "mcp",
 ]
 
 [project.urls]
@@ -49,3 +49,6 @@ dev-dependencies = [
     "ipython>=8.12.3",
     "gunicorn>=23.0.0",
 ]
+
+[tool.uv.sources]
+mcp = { git = "https://github.com/rusiaaman/python-sdk", rev = "53b69f397eae6ac81a51b84b34ff52b3119f11cb" }

src/wcgw/client/computer_use.py

@@ -26,7 +26,6 @@
 OUTPUT_DIR = "/tmp/outputs"
 TYPING_DELAY_MS = 12
 TYPING_GROUP_SIZE = 50
-TRUNCATED_MESSAGE: str = "<response clipped><NOTE>To save on context only part of this file has been shown to you.</NOTE>"
 
 Action = Literal[
     "key",

src/wcgw/client/mcp_server/server.py

@@ -30,6 +30,7 @@
 )
 from ..computer_use import Computer
 
+tools.TIMEOUT = 3
 
 server = Server("wcgw")
 

src/wcgw/client/sys_utils.py

@@ -1,6 +1,7 @@
 import subprocess
 
 MAX_RESPONSE_LEN: int = 16000
+TRUNCATED_MESSAGE: str = "<response clipped><NOTE>To save on context only part of this file has been shown to you.</NOTE>"
 
 
 def maybe_truncate(content: str, truncate_after: int | None = MAX_RESPONSE_LEN) -> str:
@@ -14,7 +15,7 @@ def maybe_truncate(content: str, truncate_after: int | None = MAX_RESPONSE_LEN)
 
 def command_run(
     cmd: str,
-    timeout: float | None = 30.0,  # seconds
+    timeout: float | None = 3.0,  # seconds
     truncate_after: int | None = MAX_RESPONSE_LEN,
     text: bool = True,
 ) -> tuple[int, str, str]:

src/wcgw/client/tools.py

@@ -989,7 +989,7 @@ def read_file(readfile: ReadFile, max_tokens: Optional[int]) -> str:
 
     else:
         return_code, content, stderr = command_run(
-            f"cat {readfile.file_path}",
+            f"cat {readfile.file_path}", timeout=TIMEOUT
         )
         if return_code != 0:
             raise Exception(