style: ruff format

Author: xerrors

server/routers/evaluation_router.py

@@ -14,11 +14,7 @@
 
 @evaluation.get("/databases/{db_id}/benchmarks/{benchmark_id}")
 async def get_evaluation_benchmark_by_db(
-    db_id: str,
-    benchmark_id: str,
-    page: int = 1,
-    page_size: int = 10,
-    current_user: User = Depends(get_admin_user)
+    db_id: str, benchmark_id: str, page: int = 1, page_size: int = 10, current_user: User = Depends(get_admin_user)
 ):
     """根据 db_id 获取评估基准详情（支持分页）"""
     from src.services.evaluation_service import EvaluationService
@@ -54,15 +50,14 @@ async def delete_evaluation_benchmark(benchmark_id: str, current_user: User = De
         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,
     page: int = 1,
     page_size: int = 20,
     error_only: bool = False,
-    current_user: User = Depends(get_admin_user)
+    current_user: User = Depends(get_admin_user),
 ):
     """获取评估结果（带 db_id，支持分页）"""
     from src.services.evaluation_service import EvaluationService

src/services/evaluation_service.py

@@ -7,7 +7,6 @@
 from typing import Any
 
 from server.services.tasker import TaskContext, tasker
-from src import config
 from src.knowledge import knowledge_base
 from src.models import select_model
 from src.utils import logger
@@ -136,7 +135,9 @@ async def get_benchmark_detail(self, benchmark_id: str) -> dict[str, Any]:
             logger.error(f"获取评估基准详情失败: {e}")
             raise
 
-    async def get_benchmark_detail_by_db(self, db_id: str, benchmark_id: str, page: int = 1, page_size: int = 10) -> dict[str, Any]:
+    async def get_benchmark_detail_by_db(
+        self, db_id: str, benchmark_id: str, page: int = 1, page_size: int = 10
+    ) -> dict[str, Any]:
         """根据 db_id 获取评估基准详情（支持分页）"""
         try:
             kb_instance = knowledge_base.get_kb(db_id)
@@ -174,17 +175,19 @@ async def get_benchmark_detail_by_db(self, db_id: str, benchmark_id: str, page:
             total_pages = (total_questions + page_size - 1) // page_size
 
             meta_with_q = meta.copy()
-            meta_with_q.update({
-                "questions": questions,
-                "pagination": {
-                    "current_page": page,
-                    "page_size": page_size,
-                    "total_questions": total_questions,
-                    "total_pages": total_pages,
-                    "has_next": page < total_pages,
-                    "has_prev": page > 1
+            meta_with_q.update(
+                {
+                    "questions": questions,
+                    "pagination": {
+                        "current_page": page,
+                        "page_size": page_size,
+                        "total_questions": total_questions,
+                        "total_pages": total_pages,
+                        "has_next": page < total_pages,
+                        "has_prev": page > 1,
+                    },
                 }
-            })
+            )
             return meta_with_q
         except Exception as e:
             logger.error(f"获取评估基准详情失败: {e}")
@@ -226,8 +229,8 @@ async def generate_benchmark(self, db_id: str, params: dict[str, Any], created_b
         return {"task_id": task_id, "message": "基准生成任务已提交"}
 
     async def _generate_benchmark_task(self, context: TaskContext):
-        import random
         import math
+        import random
 
         await context.set_progress(0, "初始化")
 
@@ -346,16 +349,16 @@ def cosine(a, b, na, nb):
                 prompt = (
                     "你将基于以下上下文生成一个可由上下文准确回答的问题与标准答案。"
                     "仅返回一个JSON对象，不要包含其他文字。"
-                    "键为 query、gold_answer、gold_chunk_ids。gold_chunk_ids 必须是上述上下文片段的ID子集。\n\n上下文：\n"
-                    + context_text
-                    + "\n"
+                    "键为 query、gold_answer、gold_chunk_ids。gold_chunk_ids 必须是上述上下文片段的ID子集。\n\n"
+                    "上下文：\n" + context_text + "\n"
                 )
 
                 try:
                     resp = await asyncio.to_thread(llm.call, prompt, False)
                     content = resp.content if resp else ""
 
                     import json_repair
+
                     obj = json_repair.loads(content)
                     q = obj.get("query")
                     a = obj.get("gold_answer")
@@ -587,7 +590,7 @@ def update_result_file(status="running", completed=0, metrics=None, interim=None
                         prompt = (
                             f"基于以下上下文信息，请回答用户的问题。\n\n"
                             f"上下文信息：{context_text}\n\n"
-                            f"用户问题：{question_data["query"]}\n\n"
+                            f"用户问题：{question_data['query']}\n\n"
                             "请根据上下文信息准确回答问题。\n\n"
                             "如果上下文中缺少相关信息，请回答“信息不足，无法回答”。\n\n"
                         )
@@ -762,12 +765,7 @@ async def get_evaluation_history(self, db_id: str) -> list[dict[str, Any]]:
         # 索引与回退逻辑已移除，统一通过 db_id 定位
 
     async def get_evaluation_results_by_db(
-        self,
-        db_id: str,
-        task_id: str,
-        page: int = 1,
-        page_size: int = 20,
-        error_only: bool = False
+        self, db_id: str, task_id: str, page: int = 1, page_size: int = 20, error_only: bool = False
     ) -> dict[str, Any]:
         result_file_path = os.path.join(self._get_result_dir(db_id), f"{task_id}.json")
         if not os.path.exists(result_file_path):
@@ -800,11 +798,7 @@ async def get_evaluation_results_by_db(
 
                     # 检查检索指标是否明显偏低
                     metrics = item.get("metrics", {})
-                    has_low_recall = any(
-                        metrics.get(k, 1.0) < 0.3
-                        for k in metrics
-                        if k.startswith("recall@")
-                    )
+                    has_low_recall = any(metrics.get(k, 1.0) < 0.3 for k in metrics if k.startswith("recall@"))
                     if has_low_recall:
                         filtered_results.append(item)
                 all_results = filtered_results
@@ -831,8 +825,8 @@ async def get_evaluation_results_by_db(
                     "page_size": page_size,
                     "total": total,
                     "total_pages": (total + page_size - 1) // page_size,
-                    "error_only": error_only
-                }
+                    "error_only": error_only,
+                },
             }
 
         # 非分页请求，返回完整数据（保持向后兼容）