ade5445cae
重构评估服务接口和实现，统一通过知识库ID管理评估基准和结果： 1. 新增带db_id的评估基准和结果查询/删除接口 2. 将基准元数据集中存储到知识库metadata中 3. 实现自动生成评估基准功能 4. 优化评估结果表格展示生成答案列 5. 简化基准生成弹窗配置项 移除旧接口并标记为废弃，所有前端调用迁移至新接口
178 lines
7.0 KiB
Python
178 lines
7.0 KiB
Python
import traceback
|
||
|
||
from fastapi import APIRouter, HTTPException, Depends, File, Form, Body, UploadFile
|
||
from src.storage.db.models import User
|
||
from server.utils.auth_middleware import get_admin_user
|
||
from src.utils import logger
|
||
|
||
# 创建路由器
|
||
evaluation = APIRouter(prefix="/evaluation", tags=["evaluation"])
|
||
|
||
|
||
# 移除旧详情接口，统一使用带 db_id 的接口
|
||
|
||
|
||
@evaluation.get("/databases/{db_id}/benchmarks/{benchmark_id}")
|
||
async def get_evaluation_benchmark_by_db(
|
||
db_id: str, benchmark_id: str, current_user: User = Depends(get_admin_user)
|
||
):
|
||
"""根据 db_id 获取评估基准详情"""
|
||
from src.services.evaluation_service import EvaluationService
|
||
|
||
try:
|
||
service = EvaluationService()
|
||
benchmark = await service.get_benchmark_detail_by_db(db_id, benchmark_id)
|
||
return {"message": "success", "data": benchmark}
|
||
except Exception as e:
|
||
logger.error(f"获取评估基准详情失败: {e}, {traceback.format_exc()}")
|
||
raise HTTPException(status_code=500, detail=f"获取评估基准详情失败: {str(e)}")
|
||
|
||
|
||
@evaluation.delete("/benchmarks/{benchmark_id}")
|
||
async def delete_evaluation_benchmark(benchmark_id: str, current_user: User = Depends(get_admin_user)):
|
||
"""删除评估基准"""
|
||
from src.services.evaluation_service import EvaluationService
|
||
|
||
try:
|
||
service = EvaluationService()
|
||
await service.delete_benchmark(benchmark_id)
|
||
return {"message": "success", "data": None}
|
||
except Exception as e:
|
||
logger.error(f"删除评估基准失败: {e}, {traceback.format_exc()}")
|
||
raise HTTPException(status_code=500, detail=f"删除评估基准失败: {str(e)}")
|
||
|
||
|
||
|
||
@evaluation.get("/databases/{db_id}/results/{task_id}")
|
||
async def get_evaluation_results_by_db(db_id: str, task_id: str, current_user: User = Depends(get_admin_user)):
|
||
"""获取评估结果（带 db_id）"""
|
||
from src.services.evaluation_service import EvaluationService
|
||
|
||
try:
|
||
service = EvaluationService()
|
||
results = await service.get_evaluation_results_by_db(db_id, task_id)
|
||
return {"message": "success", "data": results}
|
||
except Exception as e:
|
||
logger.error(f"获取评估结果失败: {e}, {traceback.format_exc()}")
|
||
raise HTTPException(status_code=500, detail=f"获取评估结果失败: {str(e)}")
|
||
|
||
|
||
@evaluation.delete("/databases/{db_id}/results/{task_id}")
|
||
async def delete_evaluation_result_by_db(db_id: str, task_id: str, current_user: User = Depends(get_admin_user)):
|
||
"""删除评估结果（带 db_id）"""
|
||
from src.services.evaluation_service import EvaluationService
|
||
|
||
try:
|
||
service = EvaluationService()
|
||
await service.delete_evaluation_result_by_db(db_id, task_id)
|
||
return {"message": "success", "data": None}
|
||
except Exception as e:
|
||
logger.error(f"删除评估结果失败: {e}, {traceback.format_exc()}")
|
||
raise HTTPException(status_code=500, detail=f"删除评估结果失败: {str(e)}")
|
||
|
||
|
||
# ============================================================================
|
||
# === Knowledge-specific evaluation endpoints ===
|
||
# ============================================================================
|
||
|
||
|
||
@evaluation.post("/databases/{db_id}/benchmarks/upload")
|
||
async def upload_evaluation_benchmark(
|
||
db_id: str,
|
||
file: UploadFile = File(...),
|
||
name: str = Form(...),
|
||
description: str = Form(""),
|
||
current_user: User = Depends(get_admin_user),
|
||
):
|
||
"""上传评估基准文件"""
|
||
from src.services.evaluation_service import EvaluationService
|
||
|
||
try:
|
||
# 验证文件格式
|
||
if not file.filename.endswith(".jsonl"):
|
||
raise HTTPException(status_code=400, detail="仅支持JSONL格式文件")
|
||
|
||
# 读取文件内容
|
||
content = await file.read()
|
||
|
||
# 调用评估服务处理上传
|
||
service = EvaluationService()
|
||
result = await service.upload_benchmark(
|
||
db_id=db_id,
|
||
file_content=content,
|
||
filename=file.filename,
|
||
name=name,
|
||
description=description,
|
||
created_by=current_user.user_id,
|
||
)
|
||
|
||
return {"message": "success", "data": result}
|
||
except HTTPException:
|
||
raise
|
||
except Exception as e:
|
||
logger.error(f"上传评估基准失败: {e}, {traceback.format_exc()}")
|
||
raise HTTPException(status_code=500, detail=f"上传评估基准失败: {str(e)}")
|
||
|
||
|
||
@evaluation.get("/databases/{db_id}/benchmarks")
|
||
async def get_evaluation_benchmarks(db_id: str, current_user: User = Depends(get_admin_user)):
|
||
"""获取知识库的评估基准列表"""
|
||
from src.services.evaluation_service import EvaluationService
|
||
|
||
try:
|
||
service = EvaluationService()
|
||
benchmarks = await service.get_benchmarks(db_id)
|
||
return {"message": "success", "data": benchmarks}
|
||
except Exception as e:
|
||
logger.error(f"获取评估基准列表失败: {e}, {traceback.format_exc()}")
|
||
raise HTTPException(status_code=500, detail=f"获取评估基准列表失败: {str(e)}")
|
||
|
||
|
||
@evaluation.post("/databases/{db_id}/benchmarks/generate")
|
||
async def generate_evaluation_benchmark(
|
||
db_id: str, params: dict = Body(...), current_user: User = Depends(get_admin_user)
|
||
):
|
||
"""自动生成评估基准"""
|
||
from src.services.evaluation_service import EvaluationService
|
||
|
||
try:
|
||
service = EvaluationService()
|
||
result = await service.generate_benchmark(db_id=db_id, params=params, created_by=current_user.user_id)
|
||
return {"message": "success", "data": result}
|
||
except Exception as e:
|
||
logger.error(f"生成评估基准失败: {e}, {traceback.format_exc()}")
|
||
raise HTTPException(status_code=500, detail=f"生成评估基准失败: {str(e)}")
|
||
|
||
|
||
@evaluation.post("/databases/{db_id}/run")
|
||
async def run_evaluation(db_id: str, params: dict = Body(...), current_user: User = Depends(get_admin_user)):
|
||
"""运行RAG评估"""
|
||
from src.services.evaluation_service import EvaluationService
|
||
|
||
try:
|
||
service = EvaluationService()
|
||
task_id = await service.run_evaluation(
|
||
db_id=db_id,
|
||
benchmark_id=params.get("benchmark_id"),
|
||
model_config=params.get("model_config", {}),
|
||
created_by=current_user.user_id,
|
||
)
|
||
return {"message": "success", "data": {"task_id": task_id}}
|
||
except Exception as e:
|
||
logger.error(f"启动评估失败: {e}, {traceback.format_exc()}")
|
||
raise HTTPException(status_code=500, detail=f"启动评估失败: {str(e)}")
|
||
|
||
|
||
@evaluation.get("/databases/{db_id}/history")
|
||
async def get_evaluation_history(db_id: str, current_user: User = Depends(get_admin_user)):
|
||
"""获取知识库的评估历史记录"""
|
||
from src.services.evaluation_service import EvaluationService
|
||
|
||
try:
|
||
service = EvaluationService()
|
||
history = await service.get_evaluation_history(db_id)
|
||
return {"message": "success", "data": history}
|
||
except Exception as e:
|
||
logger.error(f"获取评估历史失败: {e}, {traceback.format_exc()}")
|
||
raise HTTPException(status_code=500, detail=f"获取评估历史失败: {str(e)}")
|