imprint-mcp 0.6.1 → 0.6.3

package/src/cli.ts

@@ -25,7 +25,6 @@ function loadDotenv(): void {
     if (!(key in process.env)) process.env[key] = value;
   }
 }
-
 loadDotenv();
 
 /** Parse a duration string: "5m" → 300000, "1h" → 3600000, "30s" → 30000, "5000" → 5000.
@@ -113,7 +112,7 @@ export const VERB_HELP: Record<string, VerbHelp> = {
     summary:
       'Record a workflow, compile both artifacts, emit the tool, and connect to your AI platform — all in one interactive flow. Supports resuming incomplete runs and multiple workflows per site.',
     usage: [
-      'imprint teach <site> [--url <url>] [--from-session <path>] [--persist-profile] [--no-interactive] [--all-tools] [--provider <name>] [--model <name>] [--timeout <duration>] [--keep-test] [--skip-replay] [--from-step <step>] [--to-step <step>] [--only <step>]',
+      'imprint teach <site> [--url <url>] [--from-session <path>] [--persist-profile] [--no-interactive] [--all-tools] [--tool <toolName>] [--provider <name>] [--model <name>] [--timeout <duration>] [--keep-test] [--skip-replay] [--from-step <step>] [--to-step <step>] [--only <step>]',
     ],
     flags: [
       { name: '--url <url>', description: 'Starting URL (else about:blank).' },
@@ -132,6 +131,11 @@ export const VERB_HELP: Record<string, VerbHelp> = {
         description:
           'With --no-interactive, compile every detected candidate tool instead of only the primary.',
       },
+      {
+        name: '--tool <toolName>',
+        description:
+          'With --from-step/--only, resume exactly one persisted tool from the retained recording.',
+      },
       {
         name: '--provider <name>',
         description:
@@ -406,10 +410,14 @@ export const VERB_HELP: Record<string, VerbHelp> = {
   },
   'mcp-server': {
     summary: "Serve one site's generated tools as MCP (stdio default).",
-    usage: ['imprint mcp-server <site> [--http] [--port <num>]'],
+    usage: ['imprint mcp-server <site> [--http] [--port <num>] [--include-tools <names>]'],
     flags: [
       { name: '--http', description: 'Use Streamable HTTP transport instead of stdio.' },
       { name: '--port <num>', description: 'Port for HTTP transport (default 8765).' },
+      {
+        name: '--include-tools <names>',
+        description: 'Expose only these comma-separated generated tool names.',
+      },
     ],
     example: 'imprint mcp-server southwest',
   },
@@ -1118,6 +1126,7 @@ async function main(argv: string[]): Promise<number> {
         options: {
           http: { type: 'boolean' },
           port: { type: 'string' },
+          'include-tools': { type: 'string' },
         },
         allowPositionals: false,
       });
@@ -1126,6 +1135,10 @@ async function main(argv: string[]): Promise<number> {
         site,
         http: values.http,
         port: values.port ? Number(values.port) : undefined,
+        includeToolNames: values['include-tools']
+          ?.split(',')
+          .map((name) => name.trim())
+          .filter(Boolean),
       });
       return 0;
     }
@@ -1410,6 +1423,7 @@ async function main(argv: string[]): Promise<number> {
           'persist-profile': { type: 'boolean' },
           'no-interactive': { type: 'boolean' },
           'all-tools': { type: 'boolean' },
+          tool: { type: 'string' },
           provider: { type: 'string' },
           model: { type: 'string' },
           timeout: { type: 'string' },
@@ -1428,7 +1442,9 @@ async function main(argv: string[]): Promise<number> {
       // mutual exclusion with --from-session, and --to-step ≥ redact when
       // --from-session enters the chain at redact), returning the resolved window
       // or the exact error message to print. Extracted for unit-testing.
-      const { resolveTeachPhaseWindow } = await import('./imprint/teach-state.ts');
+      const { resolveTeachPhaseWindow, validateToolScopedResumeStep } = await import(
+        './imprint/teach-state.ts'
+      );
       const phaseWindow = resolveTeachPhaseWindow(values);
       if ('error' in phaseWindow) {
         console.error(phaseWindow.error);
@@ -1437,6 +1453,22 @@ async function main(argv: string[]): Promise<number> {
       const fromStepArg = phaseWindow.fromStep;
       const toStepArg = phaseWindow.toStep;
 
+      if (values.tool && !fromStepArg) {
+        console.error('error: --tool requires --from-step or --only');
+        return 2;
+      }
+      if (values.tool) {
+        const toolWindowError = validateToolScopedResumeStep(fromStepArg);
+        if (toolWindowError) {
+          console.error(toolWindowError);
+          return 2;
+        }
+      }
+      if (values.tool && values['all-tools']) {
+        console.error('error: --tool cannot be combined with --all-tools');
+        return 2;
+      }
+
       if (!site && values['no-interactive']) {
         console.error(
           'error: `imprint teach` requires a <site> argument in --no-interactive mode.\n  <site> is a label you choose — it names the output folder under ~/.imprint/.\n\n  example: imprint teach google-flights --url https://flights.google.com\n→ run `imprint teach --help` for usage.',
@@ -1482,6 +1514,7 @@ async function main(argv: string[]): Promise<number> {
             'imprint.model': values.model ?? 'auto',
             'imprint.timeout_ms': teachTimeoutMs ?? 'default',
             'imprint.all_tools': values['all-tools'] ?? false,
+            'imprint.tool': values.tool,
             'imprint.no_interactive': values['no-interactive'] ?? false,
             'imprint.skip_replay': values['skip-replay'] ?? false,
           },
@@ -1498,6 +1531,7 @@ async function main(argv: string[]): Promise<number> {
               maxDurationMs: teachTimeoutMs,
               keepTest: values['keep-test'] || process.env.IMPRINT_KEEP_TEST === '1',
               allTools: values['all-tools'],
+              tool: values.tool,
               skipReplay: values['skip-replay'],
               fromStep: fromStepArg,
               toStep: toStepArg,
@@ -1533,6 +1567,8 @@ async function main(argv: string[]): Promise<number> {
           'build-plan-path': { type: 'string' },
           'shared-modules-json': { type: 'string' },
           'auth-plan-json': { type: 'string' },
+          'revision-mode': { type: 'boolean' },
+          provider: { type: 'string' },
           site: { type: 'string' },
         },
         allowPositionals: false,
@@ -1571,11 +1607,92 @@ async function main(argv: string[]): Promise<number> {
         buildPlanPath: values['build-plan-path'],
         sharedModules,
         authToolPlan: authToolPlan ?? undefined,
+        revisionMode: values['revision-mode'],
+        provider: values.provider as
+          | 'anthropic-api'
+          | 'claude-cli'
+          | 'codex-cli'
+          | 'cursor-cli'
+          | undefined,
         site: values.site,
       });
       return 0;
     }
 
+    // Hidden semantic-verifier MCP server. It exposes backend preparation,
+    // verifier-owned suite/targeted calls, and structured report submission.
+    case '__mcp-live-verifier-server': {
+      const { values } = parseArgs({
+        args: argv.slice(1),
+        options: {
+          'workflow-path': { type: 'string' },
+          'report-path': { type: 'string' },
+          'log-path': { type: 'string' },
+          attempt: { type: 'string' },
+          'session-label': { type: 'string' },
+        },
+        allowPositionals: false,
+      });
+      if (!values['workflow-path'] || !values['report-path'] || !values['session-label']) {
+        console.error(
+          'error: __mcp-live-verifier-server requires --workflow-path <path>, --report-path <path>, and --session-label <label>',
+        );
+        return 2;
+      }
+      const { runLiveVerifierMcpServer } = await import('./imprint/mcp-live-verifier-server.ts');
+      await runLiveVerifierMcpServer({
+        workflowPath: values['workflow-path'],
+        reportPath: values['report-path'],
+        sessionLabel: values['session-label'],
+        logPath: values['log-path'],
+        attempt: values.attempt ? Number.parseInt(values.attempt, 10) : undefined,
+      });
+      return 0;
+    }
+
+    // Hidden one-shot worker for compile-time live verification. The MCP
+    // verifier owns this worker as a detached process group, so a wedged
+    // browser-backed probe can be terminated without killing the verifier that
+    // must still write an inconclusive/changes-required report.
+    case '__probe-live-verification-backends': {
+      const { values } = parseArgs({
+        args: argv.slice(1),
+        options: {
+          'workflow-path': { type: 'string' },
+          'out-path': { type: 'string' },
+        },
+        allowPositionals: false,
+      });
+      if (!values['workflow-path'] || !values['out-path']) {
+        console.error(
+          'error: __probe-live-verification-backends requires --workflow-path <path> and --out-path <path>',
+        );
+        return 2;
+      }
+      const rawParams = await Bun.stdin.text();
+      const params = rawParams.trim()
+        ? (JSON.parse(rawParams) as Record<string, string | number | boolean>)
+        : {};
+      const { resolveWorkflowTool } = await import('./imprint/backend-ladder.ts');
+      const { probeResolvedTool } = await import('./imprint/probe-backends.ts');
+      const workflowPath = values['workflow-path'];
+      const tool = resolveWorkflowTool(workflowPath);
+      const assetRoot = resolve(dirname(workflowPath), '..', '..');
+      const result = await probeResolvedTool(
+        { site: tool.site, paramOverrides: params },
+        assetRoot,
+        tool,
+        values['out-path'],
+      );
+      console.log(
+        JSON.stringify({
+          outPath: result.outPath,
+          preferredBackend: result.cache.preferredOrder[0],
+        }),
+      );
+      return 0;
+    }
+
     case 'update': {
       const { checkForUpdate, performUpdate } = await import('./imprint/update.ts');
       const checkOnly = argv.slice(1).includes('--check');

package/src/imprint/agent.ts

@@ -49,6 +49,10 @@ export type OnDeadlineReached = () => Promise<number | null>;
 export interface AgentResult {
   outcome: 'done' | 'give_up' | 'timeout' | 'soft_cap' | 'error';
   doneSummary?: string;
+  /** Raw arguments supplied to the reserved done tool. Callers with a richer
+   * done contract can make a post-verification decision without teaching the
+   * generic agent loop about that contract. */
+  doneInput?: Record<string, unknown>;
   giveUpReason?: string;
   giveUpDetail?: string;
   errorMessage?: string;
@@ -294,12 +298,13 @@ export async function runAgentLoop(opts: AgentLoopOptions): Promise<AgentResult>
           // Check for done() or give_up() first
           for (const block of toolUseBlocks) {
             if (block.name === 'done') {
-              const input = block.input as { summary?: string };
+              const input = block.input as Record<string, unknown>;
               return {
                 action: 'return',
                 result: {
                   outcome: 'done',
-                  doneSummary: input.summary ?? 'Task completed',
+                  doneSummary: typeof input.summary === 'string' ? input.summary : 'Task completed',
+                  doneInput: input,
                   turns: turn,
                   durationMs: Date.now() - startTime,
                   inputTokens,