codebee 0.1.2 → 0.1.4

package/app/core/jobs.py

@@ -1,424 +1,441 @@
-# -*- coding: utf-8 -*-
-"""任务队列：可并发 worker 池（默认 3，1-6 可配）执行编排任务与管理操作。
-
-多个任务同时跑、互不打扰：每个 job 一条独立线程，run/step 数据按 run_id
-隔离，store 层有全局锁。目标并发数可在设置页调整；调小后多余线程在取到
-新任务前自行退出，调大即时补齐。安装/升级失败时自动触发 AI 诊断修复：
-由真实智能体读取失败日志与本机环境给出修正命令；仅当命令命中白名单前缀
-（npm/winget/pip 安装类）才自动执行，否则把建议命令记录在运行记录里等人工确认。
-"""
-from __future__ import annotations
-
-import queue
-import threading
-import traceback
-
-_QUEUE = queue.Queue()
-CANCELS = {}
-_started = False
-_alive = 0            # 活跃 worker 线程数
-_target = 3           # 目标并发数（settings.max_concurrent_jobs）
-_pool_lock = threading.Lock()
-_seq = 0
-
-AI_REPAIR_PROMPT = """你是环境工程师。在 Windows 上执行下面的安装命令失败了，请诊断原因并给出修正命令。
-只输出一个 ```json 代码块，不要输出其他内容。JSON 结构：
-{"diagnosis": "失败原因（一句话）", "command": "修正后的完整安装命令", "safe": true/false}
-硬性约束：command 只能是本机包管理器的安装命令，前缀必须是 npm install / winget install /
-py -3.13 -m pip install 之一。给不出符合约束的安全命令时，safe 设为 false 且 command 留空。
-
-## 失败的命令
-__CMD__
-
-## 失败输出（尾部）
-__LOG__
-
-## 本机环境
-__ENV__"""
-
-# AI 修复命令白名单：只放行包管理器的安装类命令
-AI_REPAIR_ALLOW = ("npm install ", "winget install", "py -3.13 -m pip install")
-
-
-def _repair_command_allowed(cmd):
-    cmd = (cmd or "").strip()
-    return cmd.startswith(AI_REPAIR_ALLOW) and "|" not in cmd and "&" not in cmd and ">" not in cmd
-
-
-def configure(max_workers):
-    """设置目标并发数（1-6）：扩容立即补线程，缩容由空闲线程自行退出。"""
-    global _target
-    _target = max(1, min(6, int(max_workers)))
-    if _started:
-        _resize()
-    return _target
-
-
-def _resize():
-    global _seq
-    with _pool_lock:
-        while _alive < _target:
-            _seq += 1
-            try:
-                threading.Thread(target=_worker, name="job-worker-%d" % _seq,
-                                 daemon=True).start()
-            except RuntimeError:
-                break  # 资源受限起不了新线程：保持现有 worker，不影响任务执行
-
-
-def start_worker():
-    global _started
-    if _started:
-        return
-    _started = True
-    try:
-        from . import settings
-        configure(settings.load()["max_concurrent_jobs"])
-        return
-    except Exception:
-        pass
-    configure(3)
-
-
-def enqueue(job):
-    if not _started:
-        start_worker()
-    _QUEUE.put(job)
-
-
-def cancel(run_id):
-    ev = CANCELS.get(run_id)
-    if ev:
-        ev.set()
-        return True
-    # 事件不存在=任务还在队列里没被 worker 拿起：直接落终态（取消事件在
-    # worker 起跑时才创建，排队任务点取消会在这里漏掉——起跑后再杀一遍）。
-    try:
-        from . import store
-        run = store.get_run(run_id)
-        if not run:
-            return False
-        if run.get("status") == "queued":
-            store.update_run(run_id, expected_status="queued", status="cancelled",
-                             ended_at=_now())
-            return True
-        if run.get("status") == "running" and run.get("cancelled_by_user"):
-            return True   # 上一轮取消已标记，等起跑时的兜底检查收口
-    except Exception:
-        pass
-    return False
-
-
-def cancel_event_for(run_id):
-    ev = CANCELS.get(run_id)   # get-or-create：排队期置位的取消不因重建事件而丢失
-    if ev is None:
-        ev = threading.Event()
-        CANCELS[run_id] = ev
-    return ev
-
-
-AUTO_RESUME_MAX = 2      # 连载任务自动续跑上限（超时/中断后自动接着写，无需人工）
-
-
-def _maybe_auto_resume(run_id):
-    """连载任务失败自动续跑：继承已完成章继续，最多 AUTO_RESUME_MAX 次。
-
-    真实长篇单次运行常因供应商拥堵超时中断；这里在 worker 收尾时自动重排一次
-    续跑（store.retry_task 会带上 inherit），让整个流程真正无人值守。
-    """
-    try:
-        from . import store
-        run = store.get_run(run_id)
-        if not run or run.get("status") not in ("failed", "cancelled"):
-            return False
-        task = store.get_task(run.get("task_id")) if run.get("task_id") else None
-        if not task or not task.get("serial"):
-            return False
-        if run.get("cancelled_by_user") or run.get("status") == "cancelled":
-            return False   # 用户主动取消的运行绝不自动续跑
-        if int(run.get("auto_resumes") or 0) >= AUTO_RESUME_MAX:
-            return False
-        ok, err, new_run = store.retry_task(task["id"])
-        if not ok or not new_run:
-            return False
-        store.update_run(new_run["id"], auto_resumes=int(run.get("auto_resumes") or 0) + 1,
-                         auto_resumed_from=run_id)
-        _QUEUE.put({"kind": "orchestration", "run_id": new_run["id"], "task_id": task["id"]})
-        return True
-    except Exception:
-        return False
-
-
-RESUME_WINDOW_HOURS = 24   # 启动恢复只看最近 24h 内中断的运行（更早的视为已放弃）
-
-
-def _recent(run):
-    """运行创建时间是否在恢复窗口内（时间格式 %Y-%m-%d %H:%M:%S）。"""
-    import time as _t
-    try:
-        ts = _t.mktime(_t.strptime(run.get("created_at") or "", "%Y-%m-%d %H:%M:%S"))
-    except Exception:
-        return False
-    return (_t.time() - ts) <= RESUME_WINDOW_HOURS * 3600
-
-
-def resume_interrupted(limit=3):
-    """启动恢复：把**近期**中断的连载任务重新入队（继承已完成章）。返回恢复条数。
-
-    服务被外部杀掉/崩溃时 worker 的 finally 不会执行；这里在启动时补一次。
-    只在 RESUME_WINDOW_HOURS 窗口内、且该任务没有更新的终态运行时恢复——
-    避免复活用户早已放弃或已完成任务的旧运行。用户手动取消的一律跳过。
-    """
-    try:
-        from . import store
-    except Exception:
-        return 0
-    runs = store.list_runs(200)
-    latest_by_task = {}
-    for r in runs:   # list_runs 已按 id 倒序：首次出现即该 task 最新运行
-        tid = r.get("task_id")
-        if tid and tid not in latest_by_task:
-            latest_by_task[tid] = r["id"]
-    n = 0
-    try:
-        for run in sorted(runs, key=lambda r: r["id"]):
-            if n >= limit:
-                break
-            if run.get("status") != "failed" or not run.get("task_id"):
-                continue
-            if run.get("cancelled_by_user"):
-                continue
-            if not _recent(run):
-                continue
-            if latest_by_task.get(run["task_id"]) != run["id"]:
-                continue   # 该任务已有更新的运行（如用户重试/已完结），不复活旧中断
-            task = store.get_task(run["task_id"])
-            if not task or not task.get("serial"):
-                continue
-            if int(run.get("auto_resumes") or 0) >= AUTO_RESUME_MAX:
-                continue
-            ok, err, new_run = store.retry_task(task["id"])
-            if not ok or not new_run:
-                continue
-            store.update_run(new_run["id"],
-                             auto_resumes=int(run.get("auto_resumes") or 0) + 1,
-                             auto_resumed_from=run["id"])
-            _QUEUE.put({"kind": "orchestration", "run_id": new_run["id"], "task_id": task["id"]})
-            n += 1
-    except Exception:
-        return n
-    return n
-
-
-def _worker():
-    global _alive
-    with _pool_lock:
-        _alive += 1
-    try:
-        while True:
-            with _pool_lock:
-                if _alive > _target:   # 缩容：多余的线程在空闲检查点自行退出
-                    return
-            try:
-                job = _QUEUE.get(timeout=5)  # 定期醒来检查并发数是否被调小
-            except queue.Empty:
-                continue
-            run_id = job.get("run_id")
-            ev = cancel_event_for(run_id) if run_id else threading.Event()
-            try:
-                # 出队后再兜一次底：排队期取消（cancel 已直接落终态）的任务
-                # 不再进流水线，避免 execute_run 又把 cancelled 改回 running。
-                # 查不到的 run（如测试 mock）不拦，保持原行为。
-                if run_id:
-                    from . import store
-                    r0 = store.get_run(run_id)
-                    if r0 and r0.get("status") == "cancelled":
-                        continue   # task_done 由 finally 统一收口，不能在此重复
-                if job.get("kind") == "orchestration":
-                    from . import pipeline
-                    pipeline.execute_run(run_id)
-                elif job.get("kind") == "mgmt":
-                    _do_mgmt(job, ev)
-                elif job.get("kind") == "selfupgrade":
-                    _do_selfupgrade(job)
-            except Exception:
-                try:
-                    from . import store
-                    err = traceback.format_exc()
-                    store.update_run(run_id, status="failed", error=err[-1500:], ended_at=_now())
-                except Exception:
-                    pass
-            finally:
-                if run_id:
-                    CANCELS.pop(run_id, None)
-                    try:
-                        _maybe_auto_resume(run_id)   # 连载失败自动续跑（继承已完成章）
-                    except Exception:
-                        pass
-                _QUEUE.task_done()
-    finally:
-        with _pool_lock:
-            _alive -= 1
-
-
-def workers_info():
-    with _pool_lock:
-        return {"target": _target, "alive": _alive}
-
-
-def _now():
-    import time
-    return time.strftime("%Y-%m-%d %H:%M:%S")
-
-
-def _do_mgmt(job, ev):
-    from . import catalog, manager, store
-    run_id = job["run_id"]
-    entry = catalog.by_id(job.get("entry_id"))
-    op = job.get("op") or ""
-    store.update_run(run_id, status="running", started_at=_now())
-    if entry is None:
-        store.update_run(run_id, status="failed", error="catalog 中找不到 %s" % job.get("entry_id"),
-                         ended_at=_now())
-        return
-    step, log_abs = store.add_step(run_id, op or "mgmt", entry["id"], entry.get("name", entry["id"]))
-    ok = False
-    if op in ("install", "upgrade", "uninstall"):
-        res = manager.run_mgmt_command(entry, op, cancel_event=ev, log_path=str(log_abs))
-        ok = res["ok"]
-        store.finish_step(run_id, step["n"],
-                          "done" if ok else "failed",
-                          summary=("完成" if ok else "失败") + (": " + res["error"][:300] if res.get("error") else ""),
-                          exit_code=res.get("exit_code"))
-        # AI 修复只针对安装类失败；卸载失败多为权限/程序占用，留给用户看日志处理
-        if not ok and op in ("install", "upgrade"):
-            ok = _ai_repair(run_id, entry, ev, entry.get(op), log_abs)
-    elif op == "smoke":
-        from . import runner as _r
-        from . import registry
-        from . import usage as _usage
-        agents = registry.effective_agents(catalog.load(), manager.detect_all())
-        agent = next((a for a in agents if a["id"] == entry["id"]), None)
-        if agent is None:
-            store.finish_step(run_id, step["n"], "failed", summary="该智能体未安装或未启用编排")
-            store.update_run(run_id, status="failed", error="未启用", ended_at=_now())
-            return
-        res = _r.run_agent(agent, "连通性测试：请只回复两个字：OK", readonly=True,
-                           timeout=180, cancel_event=ev, log_path=str(log_abs))
-        ok = res["ok"] and "OK" in (res.get("text") or "").upper()
-        try:
-            _usage.record(source="smoke", run_id=run_id, step=step["n"], role="smoke",
-                          agent=agent.get("id", ""), agent_label=agent.get("label", ""),
-                          tool=agent.get("kind", ""), model=res.get("model") or "",
-                          ok=bool(res.get("ok")),
-                          duration_s=float((res.get("raw") or {}).get("duration") or 0.0),
-                          cost_usd=float(res.get("cost_usd") or 0.0),
-                          usage=res.get("usage"))
-        except Exception:
-            pass
-        store.finish_step(run_id, step["n"], "done" if ok else "failed",
-                          summary=("连通正常：%s" % (res.get("text") or "")[:120]) if ok
-                          else ("异常：%s" % (res.get("error") or (res.get("text") or "")[:120])),
-                          exit_code=res["raw"].get("exit_code"),
-                          cost_usd=res.get("cost_usd", 0.0), tokens=res.get("tokens", 0))
-    else:
-        store.finish_step(run_id, step["n"], "failed", summary="未知操作 %s" % op)
-    # 以步骤状态汇总 run 状态
-    run = store.get_run(run_id)
-    statuses = [s["status"] for s in (run.get("steps") if run else [])] or ["failed"]
-    final = "done" if all(s == "done" for s in statuses) else "failed"
-    suffix = "（AI 修复成功）" if ok and len((run.get("steps") if run else [])) > 1 else ""
-    store.update_run(run_id, status=final, ended_at=_now(),
-                     summary=("%s %s %s%s" % (entry.get("name"), op,
-                                              "完成" if final == "done" else "失败", suffix)))
-
-
-def _do_selfupgrade(job):
-    """CodeBee 自升级：在 mgmt run 里跑 npm install -g @latest，日志实时落盘。"""
-    from . import selfupdate, store
-    run_id = job["run_id"]
-    store.update_run(run_id, status="running", started_at=_now())
-    step, log_abs = store.add_step(run_id, "selfupgrade", "__self__", "CodeBee")
-    try:
-        res = selfupdate.run_upgrade(run_id, str(log_abs))
-    except Exception as e:
-        res = {"ok": False, "exit_code": None, "error": repr(e)}
-    store.finish_step(run_id, step["n"], "done" if res["ok"] else "failed",
-                      summary="升级完成，点「重启」生效" if res["ok"]
-                      else ("升级失败: " + res["error"][:300]),
-                      exit_code=res.get("exit_code"))
-    store.update_run(run_id, status="done" if res["ok"] else "failed", ended_at=_now(),
-                     summary="CodeBee selfupgrade %s" % ("完成" if res["ok"] else "失败"))
-
-
-def _ai_repair(run_id, entry, ev, failed_cmd, orig_log):
-    """安装失败后的 AI 诊断修复：诊断 → 白名单校验 → 执行 → 复检。"""
-    from . import catalog, manager, registry, router, runner, store
-    from . import usage as _usage
-    agents = registry.effective_agents(catalog.load(), manager.detect_all())
-    agent, _reason = router.pick(agents, "repair", "mgmt")
-    if agent is None or agent.get("mode") != "real":
-        return False  # 无真实智能体可用，维持原失败
-    try:
-        log_tail = runner.tail_decoded(orig_log.read_bytes(), 2000) if orig_log.exists() else "（无输出）"
-    except Exception:
-        log_tail = "（日志不可读）"
-    import shutil
-    env_lines = [
-        "OS: Windows",
-        "node: %s" % (shutil.which("node") or "缺失"),
-        "npm: %s" % (shutil.which("npm") or "缺失"),
-        "pnpm: %s" % (shutil.which("pnpm") or "缺失"),
-        "python: %s" % (shutil.which("python") or "缺失"),
-    ]
-    prompt = (AI_REPAIR_PROMPT.replace("__CMD__", failed_cmd or "（未知）")
-              .replace("__LOG__", log_tail)
-              .replace("__ENV__", "\n".join(env_lines)))
-    step, log_abs = store.add_step(run_id, "ai-repair", agent["id"], agent.get("label"),
-                                   note="自动诊断修复")
-    res = runner.run_agent(agent, prompt, readonly=True, timeout=300,
-                           cancel_event=ev, log_path=str(log_abs))
-    try:
-        _usage.record(source="repair", run_id=run_id, step=step["n"], role="ai-repair",
-                      agent=agent.get("id", ""), agent_label=agent.get("label", ""),
-                      tool=agent.get("kind", ""), model=res.get("model") or "",
-                      ok=bool(res.get("ok")),
-                      duration_s=float((res.get("raw") or {}).get("duration") or 0.0),
-                      cost_usd=float(res.get("cost_usd") or 0.0),
-                      usage=res.get("usage"))
-    except Exception:
-        pass
-    if not res["ok"]:
-        store.finish_step(run_id, step["n"], "failed",
-                          summary="诊断调用失败：%s" % (res.get("error") or "")[:200])
-        return False
-    import re
-    data = runner.extract_json(res.get("text") or "")
-    diagnosis = str((data or {}).get("diagnosis") or "（无诊断）")[:200]
-    cmd = str((data or {}).get("command") or "").strip()
-    safe = bool((data or {}).get("safe")) and _repair_command_allowed(cmd)
-    try:
-        log_abs.write_text(
-            ("\n[AI 诊断] %s\n[AI 建议] %s\n[白名单] %s\n" %
-             (diagnosis, cmd or "（无）", "通过" if safe else "不通过，拒绝自动执行")).encode("utf-8"))
-    except Exception:
-        pass
-    if not safe:
-        store.finish_step(run_id, step["n"], "failed",
-                          summary="AI 建议命令未过白名单，需人工执行：%s（诊断：%s）" % (cmd, diagnosis))
-        return False
-    from . import paths
-    fix = runner.run_process(shell_cmd=cmd, cwd=str(paths.ROOT), timeout=1800,
-                             cancel_event=ev, log_path=str(log_abs))
-    manager.detect_all(force=True)
-    with manager._LOCK:
-        manager._STATE["versions"].pop(entry["id"], None)
-    installed = manager.detect_entry(entry)["installed"]
-    store.finish_step(run_id, step["n"],
-                      "done" if (fix["ok"] and installed) else "failed",
-                      summary="诊断：%s → 执行 %r：%s，检测安装状态：%s" % (
-                          diagnosis, cmd,
-                          "命令成功" if fix["ok"] else "命令失败",
-                          "已安装" if installed else "未检出"),
-                      exit_code=fix.get("exit_code"))
-    return bool(fix["ok"] and installed)
+# -*- coding: utf-8 -*-
+"""任务队列：可并发 worker 池（默认 3，1-6 可配）执行编排任务与管理操作。
+
+多个任务同时跑、互不打扰：每个 job 一条独立线程，run/step 数据按 run_id
+隔离，store 层有全局锁。目标并发数可在设置页调整；调小后多余线程在取到
+新任务前自行退出，调大即时补齐。安装/升级失败时自动触发 AI 诊断修复：
+由真实智能体读取失败日志与本机环境给出修正命令；仅当命令命中白名单前缀
+（npm/winget/pip 安装类）才自动执行，否则把建议命令记录在运行记录里等人工确认。
+"""
+from __future__ import annotations
+
+import queue
+import threading
+import traceback
+
+_QUEUE = queue.Queue()
+CANCELS = {}
+_started = False
+_alive = 0            # 活跃 worker 线程数
+_target = 3           # 目标并发数（settings.max_concurrent_jobs）
+_pool_lock = threading.Lock()
+_seq = 0
+
+AI_REPAIR_PROMPT = """你是环境工程师。在 Windows 上执行下面的安装命令失败了，请诊断原因并给出修正命令。
+只输出一个 ```json 代码块，不要输出其他内容。JSON 结构：
+{"diagnosis": "失败原因（一句话）", "command": "修正后的完整安装命令", "safe": true/false}
+硬性约束：command 只能是本机包管理器的安装命令，前缀必须是 npm install / winget install /
+py -3.13 -m pip install 之一。给不出符合约束的安全命令时，safe 设为 false 且 command 留空。
+
+## 失败的命令
+__CMD__
+
+## 失败输出（尾部）
+__LOG__
+
+## 本机环境
+__ENV__"""
+
+# AI 修复命令白名单：只放行包管理器的安装类命令
+AI_REPAIR_ALLOW = ("npm install ", "winget install", "py -3.13 -m pip install")
+
+
+def _repair_command_allowed(cmd):
+    cmd = (cmd or "").strip()
+    return cmd.startswith(AI_REPAIR_ALLOW) and "|" not in cmd and "&" not in cmd and ">" not in cmd
+
+
+def configure(max_workers):
+    """设置目标并发数（1-6）：扩容立即补线程，缩容由空闲线程自行退出。"""
+    global _target
+    _target = max(1, min(6, int(max_workers)))
+    if _started:
+        _resize()
+    return _target
+
+
+def _resize():
+    global _seq
+    with _pool_lock:
+        # 上限 50 次尝试：Thread.start() 返回到 _worker 真正执行之间有调度间隙，
+        # 极端环境下（杀软挂起新线程）_alive 迟迟不涨，无界循环会转着圈造线程。
+        attempts = 0
+        while _alive < _target and attempts < 50:
+            attempts += 1
+            _seq += 1
+            try:
+                threading.Thread(target=_worker, name="job-worker-%d" % _seq,
+                                 daemon=True).start()
+            except RuntimeError:
+                break  # 资源受限起不了新线程：保持现有 worker，不影响任务执行
+
+
+def start_worker():
+    """标记队列可用并加载并发配置。worker 线程**不在启动期创建**（真实装机
+    案例：某些杀软环境下启动期 Thread.start() 挂死，进程停在任务队列一步），
+    推迟到首次 enqueue 时由 _ensure_workers 创建——服务就绪不再依赖线程。"""
+    global _started
+    if _started:
+        return
+    _started = True
+    try:
+        from . import settings
+        configure(settings.load()["max_concurrent_jobs"])
+        return
+    except Exception:
+        pass
+    configure(3)
+
+
+def enqueue(job):
+    if not _started:
+        start_worker()
+    _ensure_workers()
+    _QUEUE.put(job)
+
+
+def _ensure_workers():
+    """队列里积压超过空闲 worker 数时补线程（惰性扩容，替代启动期预建）。"""
+    with _pool_lock:
+        pending = _QUEUE.qsize()
+        need = max(_target, 1) - _alive + pending
+    if need > 0:
+        _resize()
+
+
+def cancel(run_id):
+    ev = CANCELS.get(run_id)
+    if ev:
+        ev.set()
+        return True
+    # 事件不存在=任务还在队列里没被 worker 拿起：直接落终态（取消事件在
+    # worker 起跑时才创建，排队任务点取消会在这里漏掉——起跑后再杀一遍）。
+    try:
+        from . import store
+        run = store.get_run(run_id)
+        if not run:
+            return False
+        if run.get("status") == "queued":
+            store.update_run(run_id, expected_status="queued", status="cancelled",
+                             ended_at=_now())
+            return True
+        if run.get("status") == "running" and run.get("cancelled_by_user"):
+            return True   # 上一轮取消已标记，等起跑时的兜底检查收口
+    except Exception:
+        pass
+    return False
+
+
+def cancel_event_for(run_id):
+    ev = CANCELS.get(run_id)   # get-or-create：排队期置位的取消不因重建事件而丢失
+    if ev is None:
+        ev = threading.Event()
+        CANCELS[run_id] = ev
+    return ev
+
+
+AUTO_RESUME_MAX = 2      # 连载任务自动续跑上限（超时/中断后自动接着写，无需人工）
+
+
+def _maybe_auto_resume(run_id):
+    """连载任务失败自动续跑：继承已完成章继续，最多 AUTO_RESUME_MAX 次。
+
+    真实长篇单次运行常因供应商拥堵超时中断；这里在 worker 收尾时自动重排一次
+    续跑（store.retry_task 会带上 inherit），让整个流程真正无人值守。
+    """
+    try:
+        from . import store
+        run = store.get_run(run_id)
+        if not run or run.get("status") not in ("failed", "cancelled"):
+            return False
+        task = store.get_task(run.get("task_id")) if run.get("task_id") else None
+        if not task or not task.get("serial"):
+            return False
+        if run.get("cancelled_by_user") or run.get("status") == "cancelled":
+            return False   # 用户主动取消的运行绝不自动续跑
+        if int(run.get("auto_resumes") or 0) >= AUTO_RESUME_MAX:
+            return False
+        ok, err, new_run = store.retry_task(task["id"])
+        if not ok or not new_run:
+            return False
+        store.update_run(new_run["id"], auto_resumes=int(run.get("auto_resumes") or 0) + 1,
+                         auto_resumed_from=run_id)
+        _QUEUE.put({"kind": "orchestration", "run_id": new_run["id"], "task_id": task["id"]})
+        return True
+    except Exception:
+        return False
+
+
+RESUME_WINDOW_HOURS = 24   # 启动恢复只看最近 24h 内中断的运行（更早的视为已放弃）
+
+
+def _recent(run):
+    """运行创建时间是否在恢复窗口内（时间格式 %Y-%m-%d %H:%M:%S）。"""
+    import time as _t
+    try:
+        ts = _t.mktime(_t.strptime(run.get("created_at") or "", "%Y-%m-%d %H:%M:%S"))
+    except Exception:
+        return False
+    return (_t.time() - ts) <= RESUME_WINDOW_HOURS * 3600
+
+
+def resume_interrupted(limit=3):
+    """启动恢复：把**近期**中断的连载任务重新入队（继承已完成章）。返回恢复条数。
+
+    服务被外部杀掉/崩溃时 worker 的 finally 不会执行；这里在启动时补一次。
+    只在 RESUME_WINDOW_HOURS 窗口内、且该任务没有更新的终态运行时恢复——
+    避免复活用户早已放弃或已完成任务的旧运行。用户手动取消的一律跳过。
+    """
+    try:
+        from . import store
+    except Exception:
+        return 0
+    runs = store.list_runs(200)
+    latest_by_task = {}
+    for r in runs:   # list_runs 已按 id 倒序：首次出现即该 task 最新运行
+        tid = r.get("task_id")
+        if tid and tid not in latest_by_task:
+            latest_by_task[tid] = r["id"]
+    n = 0
+    try:
+        for run in sorted(runs, key=lambda r: r["id"]):
+            if n >= limit:
+                break
+            if run.get("status") != "failed" or not run.get("task_id"):
+                continue
+            if run.get("cancelled_by_user"):
+                continue
+            if not _recent(run):
+                continue
+            if latest_by_task.get(run["task_id"]) != run["id"]:
+                continue   # 该任务已有更新的运行（如用户重试/已完结），不复活旧中断
+            task = store.get_task(run["task_id"])
+            if not task or not task.get("serial"):
+                continue
+            if int(run.get("auto_resumes") or 0) >= AUTO_RESUME_MAX:
+                continue
+            ok, err, new_run = store.retry_task(task["id"])
+            if not ok or not new_run:
+                continue
+            store.update_run(new_run["id"],
+                             auto_resumes=int(run.get("auto_resumes") or 0) + 1,
+                             auto_resumed_from=run["id"])
+            _QUEUE.put({"kind": "orchestration", "run_id": new_run["id"], "task_id": task["id"]})
+            n += 1
+    except Exception:
+        return n
+    return n
+
+
+def _worker():
+    global _alive
+    with _pool_lock:
+        _alive += 1
+    try:
+        while True:
+            with _pool_lock:
+                if _alive > _target:   # 缩容：多余的线程在空闲检查点自行退出
+                    return
+            try:
+                job = _QUEUE.get(timeout=5)  # 定期醒来检查并发数是否被调小
+            except queue.Empty:
+                continue
+            run_id = job.get("run_id")
+            ev = cancel_event_for(run_id) if run_id else threading.Event()
+            try:
+                # 出队后再兜一次底：排队期取消（cancel 已直接落终态）的任务
+                # 不再进流水线，避免 execute_run 又把 cancelled 改回 running。
+                # 查不到的 run（如测试 mock）不拦，保持原行为。
+                if run_id:
+                    from . import store
+                    r0 = store.get_run(run_id)
+                    if r0 and r0.get("status") == "cancelled":
+                        continue   # task_done 由 finally 统一收口，不能在此重复
+                if job.get("kind") == "orchestration":
+                    from . import pipeline
+                    pipeline.execute_run(run_id)
+                elif job.get("kind") == "mgmt":
+                    _do_mgmt(job, ev)
+                elif job.get("kind") == "selfupgrade":
+                    _do_selfupgrade(job)
+            except Exception:
+                try:
+                    from . import store
+                    err = traceback.format_exc()
+                    store.update_run(run_id, status="failed", error=err[-1500:], ended_at=_now())
+                except Exception:
+                    pass
+            finally:
+                if run_id:
+                    CANCELS.pop(run_id, None)
+                    try:
+                        _maybe_auto_resume(run_id)   # 连载失败自动续跑（继承已完成章）
+                    except Exception:
+                        pass
+                _QUEUE.task_done()
+    finally:
+        with _pool_lock:
+            _alive -= 1
+
+
+def workers_info():
+    with _pool_lock:
+        return {"target": _target, "alive": _alive}
+
+
+def _now():
+    import time
+    return time.strftime("%Y-%m-%d %H:%M:%S")
+
+
+def _do_mgmt(job, ev):
+    from . import catalog, manager, store
+    run_id = job["run_id"]
+    entry = catalog.by_id(job.get("entry_id"))
+    op = job.get("op") or ""
+    store.update_run(run_id, status="running", started_at=_now())
+    if entry is None:
+        store.update_run(run_id, status="failed", error="catalog 中找不到 %s" % job.get("entry_id"),
+                         ended_at=_now())
+        return
+    step, log_abs = store.add_step(run_id, op or "mgmt", entry["id"], entry.get("name", entry["id"]))
+    ok = False
+    if op in ("install", "upgrade", "uninstall"):
+        res = manager.run_mgmt_command(entry, op, cancel_event=ev, log_path=str(log_abs))
+        ok = res["ok"]
+        store.finish_step(run_id, step["n"],
+                          "done" if ok else "failed",
+                          summary=("完成" if ok else "失败") + (": " + res["error"][:300] if res.get("error") else ""),
+                          exit_code=res.get("exit_code"))
+        # AI 修复只针对安装类失败；卸载失败多为权限/程序占用，留给用户看日志处理
+        if not ok and op in ("install", "upgrade"):
+            ok = _ai_repair(run_id, entry, ev, entry.get(op), log_abs)
+    elif op == "smoke":
+        from . import runner as _r
+        from . import registry
+        from . import usage as _usage
+        agents = registry.effective_agents(catalog.load(), manager.detect_all())
+        agent = next((a for a in agents if a["id"] == entry["id"]), None)
+        if agent is None:
+            store.finish_step(run_id, step["n"], "failed", summary="该智能体未安装或未启用编排")
+            store.update_run(run_id, status="failed", error="未启用", ended_at=_now())
+            return
+        res = _r.run_agent(agent, "连通性测试：请只回复两个字：OK", readonly=True,
+                           timeout=180, cancel_event=ev, log_path=str(log_abs))
+        ok = res["ok"] and "OK" in (res.get("text") or "").upper()
+        try:
+            _usage.record(source="smoke", run_id=run_id, step=step["n"], role="smoke",
+                          agent=agent.get("id", ""), agent_label=agent.get("label", ""),
+                          tool=agent.get("kind", ""), model=res.get("model") or "",
+                          ok=bool(res.get("ok")),
+                          duration_s=float((res.get("raw") or {}).get("duration") or 0.0),
+                          cost_usd=float(res.get("cost_usd") or 0.0),
+                          usage=res.get("usage"))
+        except Exception:
+            pass
+        store.finish_step(run_id, step["n"], "done" if ok else "failed",
+                          summary=("连通正常：%s" % (res.get("text") or "")[:120]) if ok
+                          else ("异常：%s" % (res.get("error") or (res.get("text") or "")[:120])),
+                          exit_code=res["raw"].get("exit_code"),
+                          cost_usd=res.get("cost_usd", 0.0), tokens=res.get("tokens", 0))
+    else:
+        store.finish_step(run_id, step["n"], "failed", summary="未知操作 %s" % op)
+    # 以步骤状态汇总 run 状态
+    run = store.get_run(run_id)
+    statuses = [s["status"] for s in (run.get("steps") if run else [])] or ["failed"]
+    final = "done" if all(s == "done" for s in statuses) else "failed"
+    suffix = "（AI 修复成功）" if ok and len((run.get("steps") if run else [])) > 1 else ""
+    store.update_run(run_id, status=final, ended_at=_now(),
+                     summary=("%s %s %s%s" % (entry.get("name"), op,
+                                              "完成" if final == "done" else "失败", suffix)))
+
+
+def _do_selfupgrade(job):
+    """CodeBee 自升级：在 mgmt run 里跑 npm install -g @latest，日志实时落盘。"""
+    from . import selfupdate, store
+    run_id = job["run_id"]
+    store.update_run(run_id, status="running", started_at=_now())
+    step, log_abs = store.add_step(run_id, "selfupgrade", "__self__", "CodeBee")
+    try:
+        res = selfupdate.run_upgrade(run_id, str(log_abs))
+    except Exception as e:
+        res = {"ok": False, "exit_code": None, "error": repr(e)}
+    store.finish_step(run_id, step["n"], "done" if res["ok"] else "failed",
+                      summary="升级完成，点「重启」生效" if res["ok"]
+                      else ("升级失败: " + res["error"][:300]),
+                      exit_code=res.get("exit_code"))
+    store.update_run(run_id, status="done" if res["ok"] else "failed", ended_at=_now(),
+                     summary="CodeBee selfupgrade %s" % ("完成" if res["ok"] else "失败"))
+
+
+def _ai_repair(run_id, entry, ev, failed_cmd, orig_log):
+    """安装失败后的 AI 诊断修复：诊断 → 白名单校验 → 执行 → 复检。"""
+    from . import catalog, manager, registry, router, runner, store
+    from . import usage as _usage
+    agents = registry.effective_agents(catalog.load(), manager.detect_all())
+    agent, _reason = router.pick(agents, "repair", "mgmt")
+    if agent is None or agent.get("mode") != "real":
+        return False  # 无真实智能体可用，维持原失败
+    try:
+        log_tail = runner.tail_decoded(orig_log.read_bytes(), 2000) if orig_log.exists() else "（无输出）"
+    except Exception:
+        log_tail = "（日志不可读）"
+    import shutil
+    env_lines = [
+        "OS: Windows",
+        "node: %s" % (shutil.which("node") or "缺失"),
+        "npm: %s" % (shutil.which("npm") or "缺失"),
+        "pnpm: %s" % (shutil.which("pnpm") or "缺失"),
+        "python: %s" % (shutil.which("python") or "缺失"),
+    ]
+    prompt = (AI_REPAIR_PROMPT.replace("__CMD__", failed_cmd or "（未知）")
+              .replace("__LOG__", log_tail)
+              .replace("__ENV__", "\n".join(env_lines)))
+    step, log_abs = store.add_step(run_id, "ai-repair", agent["id"], agent.get("label"),
+                                   note="自动诊断修复")
+    res = runner.run_agent(agent, prompt, readonly=True, timeout=300,
+                           cancel_event=ev, log_path=str(log_abs))
+    try:
+        _usage.record(source="repair", run_id=run_id, step=step["n"], role="ai-repair",
+                      agent=agent.get("id", ""), agent_label=agent.get("label", ""),
+                      tool=agent.get("kind", ""), model=res.get("model") or "",
+                      ok=bool(res.get("ok")),
+                      duration_s=float((res.get("raw") or {}).get("duration") or 0.0),
+                      cost_usd=float(res.get("cost_usd") or 0.0),
+                      usage=res.get("usage"))
+    except Exception:
+        pass
+    if not res["ok"]:
+        store.finish_step(run_id, step["n"], "failed",
+                          summary="诊断调用失败：%s" % (res.get("error") or "")[:200])
+        return False
+    import re
+    data = runner.extract_json(res.get("text") or "")
+    diagnosis = str((data or {}).get("diagnosis") or "（无诊断）")[:200]
+    cmd = str((data or {}).get("command") or "").strip()
+    safe = bool((data or {}).get("safe")) and _repair_command_allowed(cmd)
+    try:
+        log_abs.write_text(
+            ("\n[AI 诊断] %s\n[AI 建议] %s\n[白名单] %s\n" %
+             (diagnosis, cmd or "（无）", "通过" if safe else "不通过，拒绝自动执行")).encode("utf-8"))
+    except Exception:
+        pass
+    if not safe:
+        store.finish_step(run_id, step["n"], "failed",
+                          summary="AI 建议命令未过白名单，需人工执行：%s（诊断：%s）" % (cmd, diagnosis))
+        return False
+    from . import paths
+    fix = runner.run_process(shell_cmd=cmd, cwd=str(paths.ROOT), timeout=1800,
+                             cancel_event=ev, log_path=str(log_abs))
+    manager.detect_all(force=True)
+    with manager._LOCK:
+        manager._STATE["versions"].pop(entry["id"], None)
+    installed = manager.detect_entry(entry)["installed"]
+    store.finish_step(run_id, step["n"],
+                      "done" if (fix["ok"] and installed) else "failed",
+                      summary="诊断：%s → 执行 %r：%s，检测安装状态：%s" % (
+                          diagnosis, cmd,
+                          "命令成功" if fix["ok"] else "命令失败",
+                          "已安装" if installed else "未检出"),
+                      exit_code=fix.get("exit_code"))
+    return bool(fix["ok"] and installed)