cryptorugmunch
9db3f26f95
Some checks failed
- AST-extract all remaining routes from api.py into routers/*.py - Group routes by OpenAPI tag (AI, Advanced, Agency, ..., x402) - Keep existing auth/health/scraping/templates routers; add *_api.py for remaining routes of those tags - Move shared helpers/models/variables with the routes that need them - Update tests/test_api.py to import vision helpers from routers.vision - Regenerate openapi.json (186 paths) - All 497 tests pass; ruff clean
30 lines
823 B
Python
30 lines
823 B
Python
"""Pry — Untagged router (remaining api.py routes).
|
|
|
|
Auto-extracted from api.py during the router-split refactor.
|
|
"""
|
|
|
|
# SPDX-License-Identifier: MIT
|
|
# Copyright (c) 2026 Rug Munch Media LLC
|
|
|
|
from __future__ import annotations
|
|
|
|
import logging
|
|
|
|
from fastapi import APIRouter, WebSocket, WebSocketDisconnect
|
|
|
|
from pryextras import streams
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
router = APIRouter(tags=["Untagged"])
|
|
|
|
@router.websocket("/v1/stream/{job_id}")
|
|
async def stream_endpoint(websocket: WebSocket, job_id: str = "live") -> None:
|
|
await websocket.accept()
|
|
streams.register(job_id, websocket)
|
|
try:
|
|
while True:
|
|
data = await websocket.receive_text()
|
|
await streams.broadcast(job_id, {"echo": data})
|
|
except WebSocketDisconnect:
|
|
streams.unregister(job_id, websocket)
|