From 4612123ec41a0ec607e5ae3d18fc64dfec6ce802 Mon Sep 17 00:00:00 2001 From: iomgaa Date: Thu, 9 Jul 2026 00:25:36 -0400 Subject: [PATCH] style: ruff format telemetry + repair_trees, remove unused SRTEntry import --- adapters/telemetry.py | 4 +--- tools/repair_trees.py | 34 +++++++++++++++++++--------------- 2 files changed, 20 insertions(+), 18 deletions(-) diff --git a/adapters/telemetry.py b/adapters/telemetry.py index 5d24e50..ddc4f7e 100644 --- a/adapters/telemetry.py +++ b/adapters/telemetry.py @@ -124,9 +124,7 @@ class SQLiteTelemetryRecorder: finally: conn.close() except sqlite3.Error as exc: - logger.warning( - "遥测写入失败(已降级),call_id={}: {}", call_id, exc - ) + logger.warning("遥测写入失败(已降级),call_id={}: {}", call_id, exc) async def record_llm_call( self, diff --git a/tools/repair_trees.py b/tools/repair_trees.py index e87df42..c2690b7 100644 --- a/tools/repair_trees.py +++ b/tools/repair_trees.py @@ -38,7 +38,7 @@ from app.tree.index import TreeIndex from app.tree.repair.detector import detect_issues from app.tree.repair.regenerator import repair_tree from app.tree.repair.supplement import supplement_tree -from app.tree.subtitle import SRTEntry, parse_srt +from app.tree.subtitle import parse_srt from app.tree.verify import verify_tree # --------------------------------------------------------------------------- @@ -258,15 +258,15 @@ async def _repair_one_video( if srt_path.exists(): srt_entries = parse_srt(str(srt_path)) - repair_stats = await repair_tree( - index, empty_issues, vlm, llm, frames_dir, srt_entries - ) + repair_stats = await repair_tree(index, empty_issues, vlm, llm, frames_dir, srt_entries) stats["l3_repaired"] = repair_stats.l3_repaired stats["l2_regenerated"] = repair_stats.l2_regenerated stats["l1_regenerated"] = repair_stats.l1_regenerated logger.info( "[{}] 修复完成: L3={}, L2={}, L1={}", - vid, repair_stats.l3_repaired, repair_stats.l2_regenerated, + vid, + repair_stats.l3_repaired, + repair_stats.l2_regenerated, repair_stats.l1_regenerated, ) @@ -300,9 +300,7 @@ async def _repair_one_video( ) stats["facts_injected"] = supplement_stats.facts_injected if supplement_stats.facts_injected > 0: - logger.info( - "[{}] 补全注入 {} 个事实", vid, supplement_stats.facts_injected - ) + logger.info("[{}] 补全注入 {} 个事实", vid, supplement_stats.facts_injected) except Exception as exc: logger.error("[{}] Q&A 补全失败: {}", vid, exc) logger.info("[{}] Q&A 补全完成", vid) @@ -332,10 +330,7 @@ async def main_async(args: argparse.Namespace) -> None: reaggregate_all = args.reaggregate_all # 扫描所有视频 - vid_dirs = sorted( - d for d in videos_dir.iterdir() - if d.is_dir() and (d / "tree.json").exists() - ) + vid_dirs = sorted(d for d in videos_dir.iterdir() if d.is_dir() and (d / "tree.json").exists()) logger.info("发现 {} 个视频", len(vid_dirs)) # 加载 progress @@ -381,8 +376,14 @@ async def main_async(args: argparse.Namespace) -> None: logger.info("开始修复 {}", vid) stats = await _repair_one_video( - vid, tree_path, frames_dir, srt_dir, questions_dir, - llm, vlm, dry_run=args.dry_run, + vid, + tree_path, + frames_dir, + srt_dir, + questions_dir, + llm, + vlm, + dry_run=args.dry_run, ) async with stats_lock: @@ -399,7 +400,10 @@ async def main_async(args: argparse.Namespace) -> None: rate = completed / elapsed * 60 if elapsed > 0 else 0 logger.info( "进度: {}/{}, 已用 {:.0f}s, 速率 {:.1f} 视频/分钟", - completed, len(pending), elapsed, rate, + completed, + len(pending), + elapsed, + rate, ) tasks = [asyncio.create_task(_process(vd)) for vd in pending]