u1s1-cli 1.13.2 → 1.14.1

package/dist/workflow/tool.js

@@ -1,179 +0,0 @@
-import { existsSync, mkdirSync, readFileSync, writeFileSync } from "node:fs";
-import { dirname, resolve } from "node:path";
-import { defineTool } from "@earendil-works/pi-coding-agent";
-import { Text } from "@earendil-works/pi-tui";
-import { Type } from "typebox";
-import { tr } from "../i18n.js";
-import { contextProjectTrust } from "../subagent.js";
-import { compactResultRender, truncate } from "../tools.js";
-import { runWorkflow, saveWorkflowScript, validateScript, workflowsDir, WORKFLOW_DEFAULT_BUDGET_TOKENS, WORKFLOW_TIMEOUT_MS, } from "./runner.js";
-import { buildFromTemplate, TEMPLATES } from "./templates.js";
-function assertValidWorkflowCode(code) {
-    const errors = validateScript(code);
-    if (errors.length > 0) {
-        throw new Error(tr(`脚本未通过静态校验,请修复后重试:\n- ${errors.join("\n- ")}`, `The script failed static validation; fix it and try again:\n- ${errors.join("\n- ")}`));
-    }
-}
-function buildTemplateSource(template, input) {
-    try {
-        return buildFromTemplate(template, input);
-    }
-    catch (error) {
-        throw new Error(tr(`${error.message}\n模板 input 说明:${TEMPLATES[template]?.inputHint ?? "—"}`, `${error.message}\nTemplate input: ${TEMPLATES[template]?.inputHint ?? "—"}`), { cause: error });
-    }
-}
-function readWorkflowSource(scriptPath) {
-    try {
-        return readFileSync(scriptPath, "utf8");
-    }
-    catch {
-        throw new Error(tr(`读不到脚本文件:${scriptPath}`, `Can't read the script file: ${scriptPath}`));
-    }
-}
-function resolveWorkflowSource(params) {
-    const saveAs = params.save_as?.trim();
-    if (saveAs && !/^[a-zA-Z0-9_-]+$/.test(saveAs)) {
-        throw new Error(tr("save_as 只允许字母、数字、横线和下划线", "save_as may only contain letters, digits, hyphens, and underscores"));
-    }
-    if (saveAs && params.script_path?.trim()) {
-        throw new Error(tr("script_path 指向的脚本已经落盘,不能再配 save_as;要另存新名字请用 script 传入源码", "The script at script_path is already saved, so save_as can't be used with it; to save under a new name, pass the source via script"));
-    }
-    const inlineCode = params.script?.trim();
-    if (inlineCode) {
-        assertValidWorkflowCode(inlineCode);
-        return { code: inlineCode, saveAs };
-    }
-    const template = params.template?.trim();
-    if (template) {
-        return { code: buildTemplateSource(template, params.input ?? {}), saveAs };
-    }
-    const requestedPath = params.script_path?.trim();
-    if (requestedPath) {
-        const scriptPath = resolve(requestedPath);
-        return { code: readWorkflowSource(scriptPath), scriptPath };
-    }
-    throw new Error(tr("script、template、script_path 至少填一个", "Provide at least one of script, template, or script_path"));
-}
-function persistWorkflowSource(source) {
-    if (source.scriptPath)
-        return source.scriptPath;
-    if (!source.saveAs)
-        return saveWorkflowScript(source.code);
-    const scriptPath = resolve(workflowsDir(), `${source.saveAs}.mjs`);
-    if (existsSync(scriptPath)) {
-        throw new Error(tr(`工作流 "${source.saveAs}" 已存在(${scriptPath})。要重跑它请用 script_path 指向该文件(可加 resume:true 续跑);要新建请换个名字`, `Workflow "${source.saveAs}" already exists (${scriptPath}). To rerun it, point script_path at that file (add resume:true to continue); to create a new one, pick another name`));
-    }
-    mkdirSync(dirname(scriptPath), { recursive: true });
-    writeFileSync(scriptPath, source.code);
-    return scriptPath;
-}
-/**
- * run_workflow 工具:主 agent 把模型生成的编排脚本交给 WorkflowRunner 执行。
- * 脚本在 vm 沙箱里跑,只能用注入的 subagent/parallel/pipeline 原语。
- */
-export function createRunWorkflowTool(getParentModel) {
-    return defineTool({
-        name: "run_workflow",
-        label: tr("工作流", "Workflow"),
-        description: "Execute a JavaScript orchestration script that runs many sub-agents in structured patterns (fan-out/pipeline/loop). " +
-            "The script runs in a sandbox with ONLY these APIs: `subagent(task | {task, model})` returns the sub-agent's final output; " +
-            "`parallel([thunks])` fans out concurrently, a failed item becomes null (barrier); " +
-            "`pipeline(items, [stageFns])` streams each item through stages independently; plus `log(...)` and standard JS builtins. " +
-            "No require/import/process/fetch/fs access. Use for large parallelizable work (e.g. translate N locale files); " +
-            "don't use it for small tasks — write the script, pass it as `script`, then verify the report.",
-        promptSnippet: "Run a generated orchestration script that drives many sub-agents in parallel/pipeline patterns",
-        promptGuidelines: [
-            "Use run_workflow only when a task decomposes into many independent self-contained pieces; small jobs are cheaper done directly.",
-            "Every task string inside the script must be fully self-contained — sub-agents start with zero conversation context.",
-            "If the report shows failures, fix the script or call again with script_path + resume:true to continue from saved progress.",
-        ],
-        parameters: Type.Object({
-            script: Type.Optional(Type.String({ description: "Full JavaScript source of the orchestration script (top-level await allowed)." })),
-            template: Type.Optional(Type.String({
-                description: `Use a built-in template instead of writing a script: ${Object.keys(TEMPLATES).join(" / ")}. ` +
-                    "Provide `input` to parametrize it. Prefer this over hand-writing a script when a template fits.",
-            })),
-            input: Type.Optional(Type.Record(Type.String(), Type.Unknown(), {
-                description: 'Parameters for `template`, e.g. i18n: { "langs": ["en", "ja"] }; review/refactor: { "files": [...] } or omit for git-diff files; research: { "questions": [...] }.',
-            })),
-            script_path: Type.Optional(Type.String({
-                description: "Path to an existing workflow script under .u1s1/workflows/ — rerun it instead of passing `script`.",
-            })),
-            save_as: Type.Optional(Type.String({
-                description: 'Save this run\'s script as .u1s1/workflows/<name>.mjs (letters/digits/-/_) for easy reuse; rerun later via script_path. Fails if the name already exists. Only valid with `script` or `template`, not `script_path`.',
-                pattern: "^[a-zA-Z0-9_-]+$",
-            })),
-            resume: Type.Optional(Type.Boolean({
-                description: "Skip sub-tasks already recorded as succeeded in the progress file (for crash/interrupt recovery).",
-            })),
-            timeout_minutes: Type.Optional(Type.Number({
-                description: `Wall-clock limit for the whole run in minutes (default ${WORKFLOW_TIMEOUT_MS / 60_000}, max 180).`,
-                minimum: 1,
-                maximum: 180,
-            })),
-            budget_tokens: Type.Optional(Type.Number({
-                description: `Total token ceiling for the whole run across all sub-agents (default ${WORKFLOW_DEFAULT_BUDGET_TOKENS}). Once exceeded, no new sub-agents are spawned; running ones finish. Use 0 for unlimited.`,
-                minimum: 0,
-                maximum: 1_000_000_000,
-            })),
-        }),
-        // 精简展示与 spawn_subagent 同款:收起一行摘要,ctrl+o 展开报告
-        renderShell: "self",
-        renderCall() {
-            return new Text("", 0, 0);
-        },
-        renderResult(...args) {
-            const [result, options, theme, context] = args;
-            const d = result.details;
-            const summary = d && typeof d.ok === "number"
-                ? tr("🧭 工作流", "🧭 Workflow") +
-                    ` · ✓${d.ok}${d.failed ? ` ✗${d.failed}` : ""}` +
-                    (typeof d.seconds === "number" ? ` · ${d.seconds}s` : "") +
-                    (d.aborted ? tr(" · 已中止", " · aborted") : "")
-                : tr("🧭 工作流", "🧭 Workflow");
-            return compactResultRender({ result, options, theme, context, summaryLine: summary });
-        },
-        async execute(...args) {
-            const [, params, signal, onUpdate, ctx] = args;
-            const source = resolveWorkflowSource(params);
-            const scriptPath = persistWorkflowSource(source);
-            const progressPath = scriptPath.replace(/\.mjs$/, ".progress.jsonl");
-            // 流式进度:onProgress 高频触发,节流到 ≥2s 一次才推给 TUI
-            let lastPush = 0;
-            const onProgress = onUpdate
-                ? (p) => {
-                    const now = Date.now();
-                    if (now - lastPush < 2_000)
-                        return;
-                    lastPush = now;
-                    onUpdate({
-                        content: [
-                            {
-                                type: "text",
-                                text: tr(`🧭 子任务 ✓${p.ok} ✗${p.failed}${p.cached ? ` 缓存${p.cached}` : ""} · 在跑 ${p.running}`, `🧭 Subtasks ✓${p.ok} ✗${p.failed}${p.cached ? ` cached ${p.cached}` : ""} · running ${p.running}`) +
-                                    ` · ${(p.spentTokens / 1000).toFixed(0)}k${p.budgetTokens > 0 ? ` / ${(p.budgetTokens / 1000).toFixed(0)}k tokens` : ""}`,
-                            },
-                        ],
-                        details: p,
-                    });
-                }
-                : undefined;
-            const result = await runWorkflow({
-                code: source.code,
-                progressPath,
-                resume: params.resume === true,
-                parentModel: getParentModel(),
-                // 子 agent(含临时 worktree)沿用主会话的项目信任裁决
-                projectTrusted: contextProjectTrust(ctx),
-                signal,
-                timeoutMs: Math.min(180, params.timeout_minutes ?? WORKFLOW_TIMEOUT_MS / 60_000) * 60_000,
-                budgetTokens: params.budget_tokens,
-                onProgress,
-            });
-            return {
-                content: [{ type: "text", text: truncate(result.report) }],
-                details: { ...result.details, scriptPath, workflowsDir: workflowsDir() },
-            };
-        },
-    });
-}