Function _finish_reading_task

deployments/desktop/shared/agents.py:2673–2750

(task_id: str)

def _reading_language_notice(response_language: str) -> str:
    if _normalize_response_language(response_language) == "en":
        return "The GUI language is set to English; this reading report will be written in English."
    return "GUI 页面当前选择语言为中文；本次精读报告将按中文生成。"


def _reading_task_payload(task: Optional[Dict[str, Any]]) -> Optional[Dict[str, Any]]:
    if not task:
        return None
    return {key: value for key, value in task.items() if key not in {"thread", "internal_docs"}}


def _finish_reading_task(task_id: str) -> None:
    with _READING_TASK_LOCK:
        task = _READING_TASKS.get(task_id)
        if not task:
            return
        task["status"] = "running"
        task["progress_phase"] = "running"
        task["updated_at"] = _task_timestamp()
    user_id = str(task.get("user_id") or "")
    push_id = str(task.get("push_id") or "")
    response_language = str(task.get("response_language") or "zh")
    write_feishu = task.get("write_feishu")
    selected_numbers = list(task.get("selected_numbers") or [])
    docs = list(task.get("internal_docs") or [])
    completed_docs: List[Dict[str, Any]] = []
    errors: List[str] = []

    for index, number in enumerate(selected_numbers):
        doc = docs[index] if index < len(docs) else {}
        paper = doc.get("paper_raw") or doc.get("paper") or {}
        report_path = Path(str(doc.get("report_path") or ""))
        try:
            with _READING_TASK_LOCK:
                task["current_index"] = index + 1
                task["progress_phase"] = "reading"
                task["updated_at"] = _task_timestamp()
            _write_reading_placeholder(
                user_id=user_id,
                paper=paper,
                report_path=report_path,
                status="running",
                steps=[
                    "已创建精读报告占位文件。",
                    _reading_language_notice(response_language),
                    "正在解析 PDF；如已配置 MinerU，会优先使用 MinerU。",
                    "正在调用 LLM 生成 Q1-Q7 精读内容。",
                ],
                response_language=response_language,
            )
            result = create_reading_reports(
                user_id=user_id,
                push_id=push_id,
                paper_numbers=[number],
                write_feishu=write_feishu,
                response_language=response_language,
            )
            result_docs = list(result.get("created_docs") or [])
            completed_docs.extend(result_docs)