17 lines
541 B
Python
17 lines
541 B
Python
|
#!/usr/bin/env python3
|
||
|
|
"""Stub worker for testing. Same JSON protocol as tinygrad_worker.py."""
|
||
|
|
import json, os, sys
|
||
|
|
|
||
|
|
print(json.dumps({"status": "ready", "pid": os.getpid()}), flush=True)
|
||
|
|
for line in sys.stdin:
|
||
|
|
line = line.strip()
|
||
|
|
if not line:
|
||
|
|
continue
|
||
|
|
try:
|
||
|
|
req = json.loads(line)
|
||
|
|
if req.get("prompt") == "__crash__":
|
||
|
|
os._exit(1)
|
||
|
|
print(json.dumps({"response": f"echo: {req['prompt']}"}), flush=True)
|
||
|
|
except Exception as e:
|
||
|
|
print(json.dumps({"error": str(e)}), flush=True)
|