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

wcgw/client/tools.py

@@ -744,6 +744,7 @@ class Mdata(BaseModel):
         | ResetShell
         | FileEditFindReplace
         | FullFileEdit
+        | str
     )
 
 
@@ -768,6 +769,8 @@ def register_client(server_url: str, client_uuid: str = "") -> None:
                 # Wait to receive data from the server
                 message = websocket.recv()
                 mdata = Mdata.model_validate_json(message)
+                if isinstance(mdata.data, str):
+                    raise Exception(mdata)
                 try:
                     output, cost = get_tool_output(
                         mdata.data, default_enc, 0.0, lambda x, y: ("", 0), None
@@ -780,7 +783,7 @@ def register_client(server_url: str, client_uuid: str = "") -> None:
                 assert isinstance(output, str)
                 websocket.send(output)
 
-        except (websockets.ConnectionClosed, ConnectionError):
+        except (websockets.ConnectionClosed, ConnectionError, OSError):
             print(f"Connection closed for UUID: {client_uuid}, retrying")
             register_client(server_url, client_uuid)
 

wcgw/relay/serve.py

@@ -35,6 +35,7 @@ class Mdata(BaseModel):
         | ResetShell
         | FileEditFindReplace
         | FullFileEdit
+        | str
     )
     user_id: UUID
 
@@ -48,7 +49,7 @@ gpts: dict[UUID, Callable[[str], None]] = {}
 images: DefaultDict[UUID, dict[str, dict[str, Any]]] = DefaultDict(dict)
 
 
-CLIENT_VERSION_MINIMUM = "1.0.0"
+CLIENT_VERSION_MINIMUM = "1.1.0"
 
 
 @app.websocket("/v1/register/{uuid}")
@@ -65,7 +66,10 @@ async def register_websocket(websocket: WebSocket, uuid: UUID) -> None:
     sem_version_server = semantic_version.Version.coerce(CLIENT_VERSION_MINIMUM)
     if sem_version_client < sem_version_server:
         await websocket.send_text(
-            f"Client version {client_version} is outdated. Please upgrade to {CLIENT_VERSION_MINIMUM} or higher."
+            Mdata(
+                user_id=uuid,
+                data=f"Client version {client_version} is outdated. Please upgrade to {CLIENT_VERSION_MINIMUM} or higher.",
+            ).model_dump_json()
         )
         await websocket.close(
             reason="Client version outdated. Please upgrade to the latest version.",

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: wcgw
-Version: 1.1.0
+Version: 1.1.1
 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.0.dist-info → wcgw-1.1.1.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=KxMd8aPJmnkH-R-IisEAJ6KHAX95PRt2kLh9Z1t28hE,24667
-wcgw/relay/serve.py,sha256=Ch-vdUK-W60fkVVwFZIfy2L1VT7Te4Yt4Z8Bb0ykAh8,7131
+wcgw/client/tools.py,sha256=ElFLxeEezLmIzmEmvts6BdtcWWjjK-R4VUS2BXA6qWw,24781
+wcgw/relay/serve.py,sha256=hUMNnf4KxF-clUYvky09bWQtuRvqJny71D93PppSpec,7236
 wcgw/relay/static/privacy.txt,sha256=s9qBdbx2SexCpC_z33sg16TptmAwDEehMCLz4L50JLc,529
-wcgw-1.1.0.dist-info/METADATA,sha256=uvRAnlHl03aoR4uhhF0qVTK3Et3GF6DHIofx1jHUNQQ,5255
-wcgw-1.1.0.dist-info/WHEEL,sha256=C2FUgwZgiLbznR-k0b_5k3Ai_1aASOXDss3lzCUsUug,87
-wcgw-1.1.0.dist-info/entry_points.txt,sha256=WlIB825-Vm9ZtNzgENQsbHj4DRMkbpVR7uSkQyBlaPA,93
-wcgw-1.1.0.dist-info/RECORD,,
+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,,