wcgw 2.3.0__py3-none-any.whl → 2.3.2__py3-none-any.whl

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/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)
 

{wcgw-2.3.0.dist-info → wcgw-2.3.2.dist-info}/METADATA

@@ -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.dist-info → wcgw-2.3.2.dist-info}/RECORD

@@ -10,13 +10,13 @@ wcgw/client/diff-instructions.txt,sha256=s5AJKG23JsjwRYhFZFQVvwDpF67vElawrmdXwvu
 wcgw/client/openai_client.py,sha256=uJ2l9NXsZuipUcJYR_bFcNNmNlfnCvPm6-M-LiVSVts,17942
 wcgw/client/openai_utils.py,sha256=YNwCsA-Wqq7jWrxP0rfQmBTb1dI0s7dWXzQqyTzOZT4,2629
 wcgw/client/sys_utils.py,sha256=GajPntKhaTUMn6EOmopENWZNR2G_BJyuVbuot0x6veI,1376
-wcgw/client/tools.py,sha256=YVzEQeJdrUpmGmEgNNRsIyjbP2TFWmpO8EjXaB28Ih8,40692
+wcgw/client/tools.py,sha256=99v3uZyTU5MDiPHm1GZ3Atpl9UZ8Ju4rAn2TCbFOjZk,40975
 wcgw/client/mcp_server/Readme.md,sha256=I8N4dHkTUVGNQ63BQkBMBhCCBTgqGOSF_pUR6iOEiUk,2495
 wcgw/client/mcp_server/__init__.py,sha256=hyPPwO9cabAJsOMWhKyat9yl7OlSmIobaoAZKHu3DMc,381
-wcgw/client/mcp_server/server.py,sha256=GfGOl1baqV7gmSLQtiJ_px-KyERNuG7-XCaLgkHdcEA,11858
+wcgw/client/mcp_server/server.py,sha256=y9oYFLrURxGySvbxcfiMiQMNvLpSMWuURi_UPxdM61k,11907
 wcgw/relay/serve.py,sha256=KLYjTvM9CfqdxgFOfHM8LUkFGZ9kKyyJunpNdEIFQUk,8766
 wcgw/relay/static/privacy.txt,sha256=s9qBdbx2SexCpC_z33sg16TptmAwDEehMCLz4L50JLc,529
-wcgw-2.3.0.dist-info/METADATA,sha256=w8yIw6ra2ianLHW54ToRzWbZ_oNx4fxaXjbDD2e0wQA,7950
-wcgw-2.3.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-wcgw-2.3.0.dist-info/entry_points.txt,sha256=eKo1omwbAggWlQ0l7GKoR7uV1-j16nk9tK0BhC2Oz_E,120
-wcgw-2.3.0.dist-info/RECORD,,
+wcgw-2.3.2.dist-info/METADATA,sha256=JmrhbMAmsmgCF4aF_fsSwRol-CQ_gDSx6-c5wFhbzRc,7950
+wcgw-2.3.2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+wcgw-2.3.2.dist-info/entry_points.txt,sha256=eKo1omwbAggWlQ0l7GKoR7uV1-j16nk9tK0BhC2Oz_E,120
+wcgw-2.3.2.dist-info/RECORD,,