File size: 2,321 Bytes
f2c113d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 |
"""
REST API for case submission and result retrieval.
"""
from __future__ import annotations
import asyncio
import uuid
from typing import Dict
from fastapi import APIRouter, HTTPException
from app.agent.orchestrator import Orchestrator
from app.models.schemas import (
AgentState,
CaseResponse,
CaseResult,
CaseSubmission,
)
router = APIRouter()
# In-memory store for active/completed cases
# In production, use Redis or a database
_cases: Dict[str, Orchestrator] = {}
@router.post("/submit", response_model=CaseResponse)
async def submit_case(case: CaseSubmission):
"""
Submit a patient case for analysis.
The agent pipeline runs asynchronously. Use the WebSocket endpoint
or poll /api/cases/{case_id} for real-time updates.
"""
orchestrator = Orchestrator()
# Generate a case_id upfront so we can return it immediately
case_id = str(uuid.uuid4())[:8]
async def _run_pipeline():
async for _step in orchestrator.run(case):
pass # Steps are tracked in orchestrator state
# Once run() creates state, store the orchestrator under the real case_id
if orchestrator.state:
_cases[orchestrator.state.case_id] = orchestrator
asyncio.create_task(_run_pipeline())
# Wait briefly for the orchestrator to initialise its state
await asyncio.sleep(0.15)
# Use the orchestrator's actual case_id if available, otherwise the pre-generated one
actual_id = orchestrator.state.case_id if orchestrator.state else case_id
_cases[actual_id] = orchestrator
return CaseResponse(
case_id=actual_id,
status="running",
message="Agent pipeline started. Connect to WebSocket for real-time updates.",
)
@router.get("/{case_id}", response_model=CaseResult)
async def get_case(case_id: str):
"""Get the current state and results for a case."""
orchestrator = _cases.get(case_id)
if not orchestrator or not orchestrator.state:
raise HTTPException(status_code=404, detail=f"Case {case_id} not found")
return CaseResult(
case_id=case_id,
state=orchestrator.state,
report=orchestrator.get_result(),
)
@router.get("/", response_model=list[str])
async def list_cases():
"""List all case IDs."""
return list(_cases.keys())
|