From a0149e528cd60a45885fe2f16187b7abca67ec5e Mon Sep 17 00:00:00 2001 From: chengke <404835780@qq.com> Date: Thu, 14 May 2026 18:04:04 +0800 Subject: [PATCH] fix: resolve pyright typecheck errors in probe_sample_pages and update git_sync skill --- .../services/document_agent/tools/probe_sample_pages.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/apps/worker/app/services/document_agent/tools/probe_sample_pages.py b/apps/worker/app/services/document_agent/tools/probe_sample_pages.py index e0f21ac88..1ac8018e8 100644 --- a/apps/worker/app/services/document_agent/tools/probe_sample_pages.py +++ b/apps/worker/app/services/document_agent/tools/probe_sample_pages.py @@ -166,9 +166,14 @@ def _sample_pages_worker( doc = pymupdf.open(pdf_path) try: page_count = int(doc.page_count) + safe_strategy: SampleStrategy = "stratified" + if strategy == "uniform": + safe_strategy = "uniform" + elif strategy == "key_pages": + safe_strategy = "key_pages" indices = _choose_sample_indices( page_count, - strategy=strategy if strategy in {"stratified", "uniform", "key_pages"} else "stratified", + strategy=safe_strategy, max_samples=max_samples, ) sampled_pages = [_extract_page_features(doc[idx], idx) for idx in indices]