37 lines
1.2 KiB
Python
37 lines
1.2 KiB
Python
import json
|
|
import sys
|
|
from pathlib import Path
|
|
from fastapi import APIRouter, HTTPException
|
|
|
|
# Add parent directory to sys.path for relative imports
|
|
sys.path.insert(0, str(Path(__file__).parent.parent))
|
|
from models.db import get_db
|
|
|
|
router = APIRouter(prefix="/api/report", tags=["评测报告"])
|
|
|
|
|
|
@router.get("/{task_id}")
|
|
async def get_report(task_id: str):
|
|
async with get_db() as db:
|
|
rows = await db.execute_fetchall(
|
|
"SELECT * FROM eval_report WHERE task_id=?", (task_id,)
|
|
)
|
|
if not rows:
|
|
raise HTTPException(status_code=404, detail="Report not found. Task may still be running.")
|
|
return {"status": 0, "data": dict(rows[0])}
|
|
|
|
|
|
@router.get("/{task_id}/items")
|
|
async def get_report_items(task_id: str):
|
|
async with get_db() as db:
|
|
rows = await db.execute_fetchall(
|
|
"SELECT * FROM eval_result WHERE task_id=? ORDER BY rowid ASC", (task_id,)
|
|
)
|
|
items = []
|
|
for r in rows:
|
|
d = dict(r)
|
|
d["retrieved_chunks"] = json.loads(d["retrieved_chunks"] or "[]")
|
|
d["judge_detail"] = json.loads(d["judge_detail"] or "{}")
|
|
items.append(d)
|
|
return {"status": 0, "data": {"total": len(items), "records": items}}
|