wcgw 1.1.1__py3-none-any.whl → 1.1.2__py3-none-any.whl

wcgw/client/tools.py

@@ -9,6 +9,7 @@ import re
 import sys
 import threading
 import importlib.metadata
+import time
 import traceback
 from typing import (
     Callable,
@@ -755,16 +756,16 @@ def register_client(server_url: str, client_uuid: str = "") -> None:
         client_uuid = str(uuid.uuid4())
 
     # Create the WebSocket connection
-    with syncconnect(f"{server_url}/{client_uuid}") as websocket:
-        server_version = str(websocket.recv())
-        print(f"Server version: {server_version}")
-        client_version = importlib.metadata.version("wcgw")
-        websocket.send(client_version)
-
-        print(
-            f"Connected. Share this user id with the chatbot: {client_uuid} \nLink: https://chatgpt.com/g/g-Us0AAXkRh-wcgw-giving-shell-access"
-        )
-        try:
+    try:
+        with syncconnect(f"{server_url}/{client_uuid}") as websocket:
+            server_version = str(websocket.recv())
+            print(f"Server version: {server_version}")
+            client_version = importlib.metadata.version("wcgw")
+            websocket.send(client_version)
+
+            print(
+                f"Connected. Share this user id with the chatbot: {client_uuid} \nLink: https://chatgpt.com/g/g-Us0AAXkRh-wcgw-giving-shell-access"
+            )
             while True:
                 # Wait to receive data from the server
                 message = websocket.recv()
@@ -783,9 +784,10 @@ def register_client(server_url: str, client_uuid: str = "") -> None:
                 assert isinstance(output, str)
                 websocket.send(output)
 
-        except (websockets.ConnectionClosed, ConnectionError, OSError):
-            print(f"Connection closed for UUID: {client_uuid}, retrying")
-            register_client(server_url, client_uuid)
+    except (websockets.ConnectionClosed, ConnectionError, OSError):
+        print(f"Connection closed for UUID: {client_uuid}, retrying")
+        time.sleep(0.5)
+        register_client(server_url, client_uuid)
 
 
 run = Typer(pretty_exceptions_show_locals=False, no_args_is_help=True)

{wcgw-1.1.1.dist-info → wcgw-1.1.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: wcgw
-Version: 1.1.1
+Version: 1.1.2
 Summary: What could go wrong giving full shell access to chatgpt?
 Project-URL: Homepage, https://github.com/rusiaaman/wcgw
 Author-email: Aman Rusia <gapypi@arcfu.com>

{wcgw-1.1.1.dist-info → wcgw-1.1.2.dist-info}/RECORD

@@ -8,10 +8,10 @@ wcgw/client/common.py,sha256=grH-yV_4tnTQZ29xExn4YicGLxEq98z-HkEZwH0ReSg,1410
 wcgw/client/diff-instructions.txt,sha256=IcoPUZS5Y_fDlA5BuryIEGYykpnDtGUAP_oUNrZyC8U,1564
 wcgw/client/openai_client.py,sha256=Qu5xjfL3zP7YkdDwowtyq2kAGYY1-YvNOt8FwdScoU8,16972
 wcgw/client/openai_utils.py,sha256=YNwCsA-Wqq7jWrxP0rfQmBTb1dI0s7dWXzQqyTzOZT4,2629
-wcgw/client/tools.py,sha256=ElFLxeEezLmIzmEmvts6BdtcWWjjK-R4VUS2BXA6qWw,24781
+wcgw/client/tools.py,sha256=-Of_Kknaf27fi76rhDdMeDTuHjTw_AKSesPKCDDqAVM,24833
 wcgw/relay/serve.py,sha256=hUMNnf4KxF-clUYvky09bWQtuRvqJny71D93PppSpec,7236
 wcgw/relay/static/privacy.txt,sha256=s9qBdbx2SexCpC_z33sg16TptmAwDEehMCLz4L50JLc,529
-wcgw-1.1.1.dist-info/METADATA,sha256=g91AStAolRmCDBXzhjKkTwljoHoJTv9uuS0iqGXpbHM,5255
-wcgw-1.1.1.dist-info/WHEEL,sha256=C2FUgwZgiLbznR-k0b_5k3Ai_1aASOXDss3lzCUsUug,87
-wcgw-1.1.1.dist-info/entry_points.txt,sha256=WlIB825-Vm9ZtNzgENQsbHj4DRMkbpVR7uSkQyBlaPA,93
-wcgw-1.1.1.dist-info/RECORD,,
+wcgw-1.1.2.dist-info/METADATA,sha256=f5t2PzR0Oz0BRTAMs2BwQC8Y5p7hXVUobut-PS1HIVQ,5255
+wcgw-1.1.2.dist-info/WHEEL,sha256=C2FUgwZgiLbznR-k0b_5k3Ai_1aASOXDss3lzCUsUug,87
+wcgw-1.1.2.dist-info/entry_points.txt,sha256=WlIB825-Vm9ZtNzgENQsbHj4DRMkbpVR7uSkQyBlaPA,93
+wcgw-1.1.2.dist-info/RECORD,,