hotfix: tower cannot parse host message

sbip-sg · Jul 9, 2024 · 90a8d8a · 90a8d8a
1 parent 2963c9a
commit 90a8d8a
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 5 deletions.
diff --git a/pymeca-actors/src/functions/host_functions.py b/pymeca-actors/src/functions/host_functions.py
@@ -201,8 +201,7 @@ async def connect_to_tower(
                     )
                 except ValueError as e:
                     print(e)
-                    await websocket.send(str(e))
-                    return
+                    await websocket.send(sign_message(host_encryption_private_key, repr(e).encode()))
 
                 try:
                     # run the task
@@ -253,5 +252,4 @@ async def connect_to_tower(
                         print("Task output sent")
                 except Exception as e:
                     print(e)
-                    await websocket.send(str(e))
-                    return
+                    await websocket.send(sign_message(host_encryption_private_key, repr(e).encode()))
diff --git a/tower/src/main.py b/tower/src/main.py
@@ -147,7 +147,7 @@ async def websocket_endpoint(websocket: WebSocket):
 
     try:
         while True:
-            task_output = await websocket.recv()
+            task_output = await websocket.receive_bytes()
             task_id = "0x" + task_output[0:32].hex()
             signature = task_output[-65:]
             verify = pymeca.utils.verify_signature_pub_key(