|
|
--- |
|
|
|
|
|
|
|
|
|
|
|
app.py: |
|
|
```python |
|
|
from fastapi import FastAPI, WebSocket |
|
|
import asyncio, json |
|
|
|
|
|
app = FastAPI() |
|
|
|
|
|
@app.websocket("/ws/dummy") |
|
|
async def ws_dummy(ws: WebSocket): |
|
|
await ws.accept() |
|
|
await ws.send_json({"type":"ready","msg":"Dummy AI online"}) |
|
|
try: |
|
|
async for msg in ws.iter_text(): |
|
|
data = json.loads(msg) |
|
|
|
|
|
for i in range(3): |
|
|
await asyncio.sleep(1) |
|
|
await ws.send_json({"type":"step","text":f"Step {i+1} for task: {data}"}) |
|
|
await ws.send_json({"type":"done","result":f"Finished task: {data}"}) |
|
|
except Exception as e: |
|
|
await ws.send_json({"error":str(e)}) |