wcgw 2.3.0__tar.gz → 2.3.2__tar.gz

{wcgw-2.3.0 → wcgw-2.3.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: wcgw
-Version: 2.3.0
+Version: 2.3.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>

{wcgw-2.3.0 → wcgw-2.3.2}/pyproject.toml

@@ -1,7 +1,7 @@
 [project]
 authors = [{ name = "Aman Rusia", email = "gapypi@arcfu.com" }]
 name = "wcgw"
-version = "2.3.0"
+version = "2.3.2"
 description = "Shell and coding agent on claude and chatgpt"
 readme = "README.md"
 requires-python = ">=3.11, <3.13"

{wcgw-2.3.0 → wcgw-2.3.2}/src/wcgw/client/mcp_server/server.py

@@ -30,8 +30,6 @@ from ...types_ import (
 )
 from ..computer_use import SLEEP_TIME_MAX_S
 
-tools.TIMEOUT = SLEEP_TIME_MAX_S
-
 COMPUTER_USE_ON_DOCKER_ENABLED = False
 
 server = Server("wcgw")
@@ -275,6 +273,11 @@ async def handle_call_tool(
 
 async def main(computer_use: bool) -> None:
     global COMPUTER_USE_ON_DOCKER_ENABLED
+
+    tools.TIMEOUT = SLEEP_TIME_MAX_S
+
+    tools.console = tools.DisableConsole()
+
     if computer_use:
         COMPUTER_USE_ON_DOCKER_ENABLED = True
 

{wcgw-2.3.0 → wcgw-2.3.2}/src/wcgw/client/tools.py

@@ -72,7 +72,18 @@ from .common import CostData, Models, discard_input
 from .sys_utils import command_run
 from .openai_utils import get_input_cost, get_output_cost
 
-console = rich.console.Console(style="magenta", highlight=False, markup=False)
+
+class DisableConsole:
+    def print(self, *args, **kwargs):  # type: ignore
+        pass
+
+    def log(self, *args, **kwargs):  # type: ignore
+        pass
+
+
+console: rich.console.Console | DisableConsole = rich.console.Console(
+    style="magenta", highlight=False, markup=False
+)
 
 TIMEOUT = 5
 
@@ -119,7 +130,7 @@ def start_shell() -> pexpect.spawn:  # type: ignore
         )
         shell.sendline(f"export PS1={PROMPT}")
     except Exception as e:
-        traceback.print_exc()
+        console.print(traceback.format_exc())
         console.log(f"Error starting shell: {e}. Retrying without rc ...")
 
         shell = pexpect.spawn(
@@ -159,7 +170,7 @@ def _get_exit_code(shell: pexpect.spawn) -> int:  # type: ignore
         try:
             shell.expect(PROMPT, timeout=0.2)
         except pexpect.TIMEOUT:
-            print(f"Couldn't get exit code, before: {before}")
+            console.print(f"Couldn't get exit code, before: {before}")
             raise
         assert isinstance(shell.before, str)
         # Render because there could be some anscii escape sequences still set like in google colab env
@@ -283,12 +294,12 @@ def update_repl_prompt(command: str) -> bool:
         PROMPT = before.split("\n")[-1].strip()
         # Escape all regex
         PROMPT = re.escape(PROMPT)
-        print(f"Trying to update prompt to: {PROMPT.encode()!r}")
+        console.print(f"Trying to update prompt to: {PROMPT.encode()!r}")
         index = 0
         while index == 0:
             # Consume all REPL prompts till now
             index = BASH_STATE.shell.expect([PROMPT, pexpect.TIMEOUT], timeout=0.2)
-        print(f"Prompt updated to: {PROMPT}")
+        console.print(f"Prompt updated to: {PROMPT}")
         return True
     return False
 
@@ -484,7 +495,7 @@ Otherwise, you may want to try Ctrl-c again or program specific exit interactive
         output += exit_status
     except ValueError as e:
         console.print(output)
-        traceback.print_exc()
+        console.print(traceback.format_exc())
         console.print("Malformed output, restarting shell", style="red")
         # Malformed output, restart shell
         BASH_STATE.reset()
@@ -508,7 +519,7 @@ def serve_image_in_bg(file_path: str, client_uuid: str, name: str) -> None:
         try:
             websocket.send(json.dumps(uu))
         except websockets.ConnectionClosed:
-            print(f"Connection closed for UUID: {client_uuid}, retrying")
+            console.print(f"Connection closed for UUID: {client_uuid}, retrying")
             serve_image_in_bg(file_path, client_uuid, name)
 
 
@@ -1139,7 +1150,7 @@ def register_client(server_url: str, client_uuid: str = "") -> None:
                     print(f"{curr_cost=}")
                 except Exception as e:
                     output = f"GOT EXCEPTION while calling tool. Error: {e}"
-                    traceback.print_exc()
+                    console.print(traceback.format_exc())
                 assert isinstance(output, str)
                 websocket.send(output)