45 lines
1.5 KiB
Python
45 lines
1.5 KiB
Python
|
"""Routes for triggering evaluations and polling background task status."""
|
||
|
|
|
||
|
|
from __future__ import annotations
|
||
|
|
|
||
|
|
from fastapi import APIRouter, HTTPException
|
||
|
|
|
||
|
|
from webapp.models import (
|
||
|
|
TaskStatus,
|
||
|
|
TriggerEvaluationRequest,
|
||
|
|
TriggerEvaluationResponse,
|
||
|
|
)
|
||
|
|
from webapp.services import scenario_scanner
|
||
|
|
from webapp.services.task_manager import task_manager
|
||
|
|
|
||
|
|
router = APIRouter(prefix="/api/evaluations", tags=["evaluations"])
|
||
|
|
|
||
|
|
|
||
|
|
@router.post("", response_model=TriggerEvaluationResponse)
|
||
|
|
def trigger_evaluation(request: TriggerEvaluationRequest) -> TriggerEvaluationResponse:
|
||
|
|
"""Validate the scenario path and queue a background evaluation task."""
|
||
|
|
resolved = scenario_scanner.resolve_scenario_path(request.scenario_path)
|
||
|
|
if resolved is None:
|
||
|
|
raise HTTPException(
|
||
|
|
status_code=400,
|
||
|
|
detail=f"无效或不允许的场景路径: {request.scenario_path}",
|
||
|
|
)
|
||
|
|
|
||
|
|
task_id = task_manager.submit(request.scenario_path)
|
||
|
|
return TriggerEvaluationResponse(task_id=task_id)
|
||
|
|
|
||
|
|
|
||
|
|
@router.get("/{task_id}", response_model=TaskStatus)
|
||
|
|
def get_task_status(task_id: str) -> TaskStatus:
|
||
|
|
"""Return the current status and logs for one evaluation task."""
|
||
|
|
status = task_manager.get(task_id)
|
||
|
|
if status is None:
|
||
|
|
raise HTTPException(status_code=404, detail=f"未找到任务: {task_id}")
|
||
|
|
return status
|
||
|
|
|
||
|
|
|
||
|
|
@router.get("", response_model=dict)
|
||
|
|
def list_tasks() -> dict[str, list]:
|
||
|
|
"""Return all known evaluation tasks for this server session."""
|
||
|
|
return {"tasks": [task.model_dump() for task in task_manager.list_tasks()]}
|