mirror/computer
1
0
Fork 0
mirror of https://github.com/trycua/computer.git synced 2025-12-31 02:19:58 -06:00
Code Issues Packages Projects Releases Wiki Activity

Added sync command support

Browse Source
This commit is contained in:
Dillon DuPont
2025-07-10 15:22:01 -04:00
parent 9eb6af8569
commit d4a154059d
1 changed files with 10 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
libs/python/computer-server/computer_server/main.py
View File

@@ -233,7 +233,11 @@ async def websocket_endpoint(websocket: WebSocket):
sig = inspect.signature(handler_func)
filtered_params = {k: v for k, v in params.items() if k in sig.parameters}
result = await handler_func(**filtered_params)
# Handle both sync and async functions
if asyncio.iscoroutinefunction(handler_func):
result = await handler_func(**filtered_params)
else:
result = handler_func(**filtered_params)
await websocket.send_json({"success": True, **result})
except Exception as cmd_error:
logger.error(f"Error executing command {command}: {str(cmd_error)}")
@@ -356,8 +360,11 @@ async def cmd_endpoint(
sig = inspect.signature(handler_func)
filtered_params = {k: v for k, v in params.items() if k in sig.parameters}
# Execute the command
result = await handler_func(**filtered_params)
# Handle both sync and async functions
if asyncio.iscoroutinefunction(handler_func):
result = await handler_func(**filtered_params)
else:
result = handler_func(**filtered_params)
# Stream the successful result
response_data = {"success": True, **result}