rivet-design 0.23.1 → 0.24.0

package/dist/cli/commands/mcpServe.js

@@ -36,58 +36,32 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
     return (mod && mod.__esModule) ? mod : { "default": mod };
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.handleMcpServe = exports.buildMcpServer = exports.isStaleAttachError = exports.isReferenceHost = void 0;
+exports.handleMcpServe = exports.buildMcpServer = exports.isStaleAttachError = void 0;
+const explorationTools_1 = require("../../mcp/explorationTools");
+const designContextTools_1 = require("../../mcp/designContextTools");
 const fs_1 = __importDefault(require("fs"));
 const path_1 = __importDefault(require("path"));
 const util_1 = __importDefault(require("util"));
 const v3_1 = require("zod/v3");
-const design_library_1 = require("../../types/design-library");
 const envelope_1 = require("../envelope");
 const flags_1 = require("../flags");
 const shared_1 = require("./shared");
 const client_1 = require("../client");
-const contracts_1 = require("../../agent-variants/contracts");
 const ConfigManager_1 = require("../../services/ConfigManager");
 const handoff_1 = require("../handoff");
-const callerModel_1 = require("../callerModel");
 const telemetry_1 = require("../telemetry");
-const displayLocale_1 = require("../../utils/displayLocale");
 const pluginContext_1 = require("../../hosts/pluginContext");
 const errorMessage_1 = require("../../utils/errorMessage");
-const GuidanceVerifierRunner_1 = require("../../services/GuidanceVerifierRunner");
 const toolPresentation = (title, behavior) => ({
     title,
     annotations: { title, ...behavior },
 });
-const STATUS_PRESENTATION = toolPresentation('Check Rivet status', {
-    readOnlyHint: true,
-    destructiveHint: false,
-    idempotentHint: true,
-    openWorldHint: true,
-});
 const LOGIN_PRESENTATION = toolPresentation('Sign in to Rivet', {
     readOnlyHint: false,
     destructiveHint: false,
-    idempotentHint: true,
-    openWorldHint: true,
-});
-const VARIANTS_PRESENTATION = toolPresentation('Generate design variants', {
-    readOnlyHint: false,
-    destructiveHint: true,
     idempotentHint: false,
     openWorldHint: true,
 });
-const DESIGN_CONTEXT_PRESENTATION = toolPresentation('Gather design context', {
-    readOnlyHint: false,
-    destructiveHint: true,
-    idempotentHint: false,
-    openWorldHint: true,
-});
-const DIRECTION_TITLE_MAX_WORDS = 5;
-const DIRECTION_SUBTITLE_MAX_WORDS = 10;
-const RELEVANT_FILE_MAX_COUNT = 12;
-const SESSION_ID_SEARCH_MAX_DEPTH = 2;
-const DISPLAY_LOCALE = (0, displayLocale_1.resolveDisplayLocale)();
 const readPackageVersion = () => {
     try {
         const pkg = JSON.parse(fs_1.default.readFileSync(path_1.default.resolve(__dirname, '../../../package.json'), 'utf8'));
@@ -97,15 +71,6 @@ const readPackageVersion = () => {
         return '0.0.0';
     }
 };
-const isReferenceHost = (hostname) => {
-    const host = hostname.toLowerCase();
-    return (host === 'are.na' ||
-        host.endsWith('.are.na') ||
-        host === 'pin.it' ||
-        host === 'pinterest.com' ||
-        host.endsWith('.pinterest.com'));
-};
-exports.isReferenceHost = isReferenceHost;
 const isStaleAttachError = (envelope) => {
     if (envelope.ok) {
         return false;
@@ -140,31 +105,12 @@ const sessionIdInValue = (value, depth = 0) => {
     return undefined;
 };
 const sessionIdInEnvelope = (envelope) => envelope.ok ? sessionIdInValue(envelope.data) : undefined;
-const STATUS_DESCRIPTION = 'Current Rivet session state for this project: server health, active work with workIds, per-variant progress. ' +
-    'Read-only, fast (~10ms); poll after starting work — there is no blocking wait. ' +
-    'When the response has a nextAction, follow its instructions exactly. ' +
-    '`variants.starting: true` = a start is still provisioning (can take minutes) — keep polling, never retry or cancel. ' +
-    "include: ['history'] adds past variant sets; ['auth'] adds signed-in state.";
-const VARIANTS_DESCRIPTION = "Drive the variant lifecycle ('directions' in the UI). Actions: 'start' (generate N directions), " +
-    "'complete' (report one work item done), 'commit' (write a chosen variant to the working tree), 'cancel'. " +
-    "Before every 'start', including fresh apps, resolve saved instructions with rivet_design_context contextAction=resolve, then author a 1\u20133 word runLabel summarizing the batch and one brief per direction: label = 1\u20135 word title naming the concrete change; body = \u226410-word subtitle; " +
-    `Write runLabel, label, and body naturally in ${DISPLAY_LOCALE}. ` +
-    'optional notes = richer implementation intent (tokens, motion specs, constraints) delivered verbatim to that direction\u2019s worker. briefs.length sets the count. ' +
-    "Rivet normally builds directions with its own server-side workers: variants in flight with NO nextAction means watch rivet_status until terminal — implement nothing. When a response DOES carry nextAction.action='complete_host_variant_work' (executor unavailable), follow its parallelism instructions EXACTLY: they name the worker agent type when one applies, and each item's workerPrompt points at a server-authored briefing containing the full packet and completion command \u2014 compose nothing, never re-plan briefs, and let each worker self-report the moment it finishes. " +
-    'If a completion returns requeuedForRegeneration, poll rivet_status for the renewed lease, address the QA summary, and complete that item again. ' +
-    "NEVER cancel to unblock a rejected start: 'start_in_flight' = still provisioning (poll until variants.starting clears); 'active_session' = the user has live work \u2014 report the conflict or refine via target. Cancel only when the user explicitly asks.";
 const LOGIN_DESCRIPTION = 'Sign the user in to Rivet with Google. Opens a browser sign-in, then waits (up to waitSeconds, default 45) for the user to finish and returns status signed_in with their email when they do — so a single call usually completes the whole sign-in. ' +
     'Call when any tool fails with AUTH_REQUIRED. If it returns status browser_opened instead, the user had not finished within the wait window: tell them to complete the browser sign-in, then call this again (it keeps waiting; the finished sign-in is never lost). ' +
     'Already signed in returns signed_in immediately.';
+const SESSION_ID_SEARCH_MAX_DEPTH = 2;
 const LOGIN_WAIT_DEFAULT_SECONDS = 45;
 const LOGIN_WAIT_MAX_SECONDS = 240;
-const DESIGN_CONTEXT_DESCRIPTION = 'Resolve copied Design library references in one batch with profileAction reference, profileId, and referenceKeys (shortId@revision). Default results contain only key, title, imageCount, and hasAnalysis; no image reads or model work. Request isDetailed=true only for references whose full text, analysis, sources, and contextImages you need. View those images and attach them explicitly with rivet_variants contextImages when generating. ' +
-    'For reference onboarding, discover a public Cosmos profile URL or referenceProvider pinterest, arena, or x. If authorization_required, show authUrl and repeat discovery after sign-in. Plan selected collectionIds using discoveryId, show the scope and usage disclosure, then import planId with its planRevision and isApproved=true only after explicit user consent. Use onboardingAction later or dismiss with referenceProvider to set one provider preference, or omit referenceProvider for all. ' +
-    'Save Pinterest, Are.na, public Cosmos collections (including nested collections), or X bookmarks (https://x.com/i/bookmarks) with profileAction connect. If X returns authorization_required, show authUrl and repeat connect after sign-in. ' +
-    'Gather design context from a URL. Pinterest and Are.na URLs return the connected account’s ' +
-    'reference data (boards, pins, blocks); any other URL is rendered live and captured as visual ' +
-    'evidence (screenshot + rendered-page metadata). When Pinterest needs authorization, present the returned authUrl to the user and retry after they connect. Routing is automatic — just pass the URL. ' +
-    'Slow (seconds — network/render bound).';
 const buildMcpServer = async (options = {}) => {
     const { McpServer } = await Promise.resolve().then(() => __importStar(require('@modelcontextprotocol/sdk/server/mcp.js')));
     const { dispatch } = await Promise.resolve().then(() => __importStar(require('../router')));
@@ -174,7 +120,6 @@ const buildMcpServer = async (options = {}) => {
         ? path_1.default.resolve(options.projectPath)
         : undefined;
     (0, client_1.enableAttachCache)();
-    void (0, client_1.attachToServer)(serveProject ?? process.cwd()).catch(() => null);
     void Promise.allSettled([
         Promise.resolve().then(() => __importStar(require('./status'))),
         Promise.resolve().then(() => __importStar(require('./variants'))),
@@ -196,7 +141,6 @@ const buildMcpServer = async (options = {}) => {
         const project = toolProject ?? serveProject;
         return project ? ['--project', project] : [];
     };
-    const verificationAbort = new AbortController();
     const runCommand = async (command, rest, commandOptions = {}) => {
         const progress = [];
         const io = {
@@ -205,13 +149,13 @@ const buildMcpServer = async (options = {}) => {
         };
         const argv = [
             ...rest,
+            ...(options.noTelemetry ? ['--no-telemetry'] : []),
             ...handoffArgs,
             ...(commandOptions.handoffArgs ?? []),
         ];
         const startedAt = Date.now();
         let result;
         try {
-            await maybeResumePendingLogin();
             result = await dispatch(command, argv, io, {
                 signal: commandOptions.signal,
             });
@@ -245,29 +189,13 @@ const buildMcpServer = async (options = {}) => {
         catch { }
         return { envelope: result.envelope, progress };
     };
-    const maybeResumePendingLogin = async () => {
-        try {
-            const { readPendingLogin } = await Promise.resolve().then(() => __importStar(require('../../services/pendingLogin')));
-            if (!readPendingLogin()) {
-                return;
-            }
-            const { AuthService } = await Promise.resolve().then(() => __importStar(require('../../services/AuthService')));
-            const resumed = await new AuthService().resumePendingLogin();
-            if (resumed?.success) {
-                telemetry?.track('mcp_login_completed', {
-                    resumed: true,
-                });
-            }
-        }
-        catch { }
-    };
     const toToolResult = (envelope, progress = []) => {
         const enriched = !envelope.ok && envelope.error.code === 'AUTH_REQUIRED'
             ? {
                 ...envelope,
                 error: {
                     ...envelope.error,
-                    signIn: 'Call the rivet_login tool — it opens a browser sign-in for the user and their session is picked up automatically afterwards.',
+                    signIn: 'Call the rivet_login tool — it opens a browser sign-in for the user and call rivet_login again if the browser sign-in is still pending.',
                 },
             }
             : envelope;
@@ -347,73 +275,11 @@ const buildMcpServer = async (options = {}) => {
             return result;
         });
     };
-    mcp.registerTool('rivet_status', {
-        ...STATUS_PRESENTATION,
-        description: STATUS_DESCRIPTION,
-        inputSchema: {
-            project: v3_1.z
-                .string()
-                .optional()
-                .describe('Project root path; defaults to the server for the current directory'),
-            include: v3_1.z
-                .array(v3_1.z.enum(['history', 'auth']))
-                .optional()
-                .describe('history = committed/past variant sets; auth = signed-in state'),
-        },
-    }, async ({ project, include }) => {
-        const startedAt = Date.now();
-        const wantHistory = include?.includes('history') ?? false;
-        const wantAuth = include?.includes('auth') ?? false;
-        const [primary, history, auth] = await Promise.all([
-            runCommand('status', projectArgs(project)),
-            wantHistory
-                ? runCommand('variants', [
-                    'list',
-                    '--history',
-                    ...projectArgs(project),
-                ])
-                : Promise.resolve(null),
-            wantAuth ? runCommand('auth', ['status']) : Promise.resolve(null),
-        ]);
-        if (!primary.envelope.ok) {
-            return completeToolCall({
-                toolName: 'rivet_status',
-                startedAt,
-                envelope: primary.envelope,
-                progress: primary.progress,
-            });
-        }
-        const data = { status: primary.envelope.data };
-        if (history) {
-            data.history = history.envelope.ok
-                ? history.envelope.data
-                : { error: history.envelope.error };
-        }
-        if (auth) {
-            data.auth = auth.envelope.ok
-                ? auth.envelope.data
-                : { error: auth.envelope.error };
-            const authData = data.auth;
-            if (authData && authData.authenticated === false) {
-                data.auth.signIn =
-                    'Call the rivet_login tool — it opens a browser sign-in for the user and returns immediately; the finished sign-in is picked up automatically on the next call.';
-            }
-        }
-        return completeToolCall({
-            toolName: 'rivet_status',
-            startedAt,
-            envelope: {
-                schemaVersion: envelope_1.CLI_SCHEMA_VERSION,
-                ok: true,
-                command: 'rivet_status',
-                data,
-            },
-        });
-    });
     mcp.registerTool('rivet_login', {
         ...LOGIN_PRESENTATION,
         description: LOGIN_DESCRIPTION,
-        inputSchema: {
+        inputSchema: v3_1.z
+            .object({
             waitSeconds: v3_1.z
                 .number()
                 .int()
@@ -421,7 +287,8 @@ const buildMcpServer = async (options = {}) => {
                 .max(LOGIN_WAIT_MAX_SECONDS)
                 .optional()
                 .describe('How long to wait for the user to finish the browser sign-in before returning (default 45). Pass a higher value when the user says they need a moment; 0 returns immediately after opening the browser.'),
-        },
+        })
+            .strict(),
     }, async ({ waitSeconds }) => {
         const startedAt = Date.now();
         const loginResult = (data, ok = true) => {
@@ -516,607 +383,19 @@ const buildMcpServer = async (options = {}) => {
             }, false);
         }
     });
-    mcp.registerTool('rivet_variants', {
-        ...VARIANTS_PRESENTATION,
-        description: VARIANTS_DESCRIPTION,
-        inputSchema: {
-            action: v3_1.z.enum(['start', 'complete', 'commit', 'cancel']),
-            instruction: v3_1.z
-                .string()
-                .optional()
-                .describe('start: what to vary and how'),
-            runLabel: contracts_1.runLabelSchema.describe('start: AI-authored 1–3 word summary shared by every direction in this batch'),
-            count: v3_1.z
-                .number()
-                .int()
-                .positive()
-                .optional()
-                .describe('start: number of variants'),
-            briefs: v3_1.z
-                .array(v3_1.z.object({
-                label: v3_1.z
-                    .string()
-                    .trim()
-                    .min(1)
-                    .max(48)
-                    .refine((label) => label.split(/\s+/).length <= DIRECTION_TITLE_MAX_WORDS, {
-                    message: 'Direction titles must contain at most 5 words',
-                })
-                    .describe('AI-authored 1–5 word direction title'),
-                body: v3_1.z
-                    .string()
-                    .trim()
-                    .min(1)
-                    .max(600)
-                    .refine((body) => body.split(/\s+/).length <= DIRECTION_SUBTITLE_MAX_WORDS, {
-                    message: 'Direction subtitles must contain at most 10 words',
-                })
-                    .describe('Concise direction subtitle of at most 10 words'),
-                notes: v3_1.z
-                    .string()
-                    .trim()
-                    .min(1)
-                    .max(4000)
-                    .optional()
-                    .describe('Optional implementation notes for this direction, passed verbatim to its worker (design tokens, motion specs, constraints). Not shown in UI chips.'),
-                visualReferenceUrl: v3_1.z.string().url().optional(),
-            }))
-                .min(contracts_1.VARIANT_COUNT_MIN)
-                .max(contracts_1.VARIANT_COUNT_MAX)
-                .superRefine((briefs, context) => {
-                const labels = briefs.map((brief) => brief.label.toLowerCase());
-                if (new Set(labels).size !== labels.length) {
-                    context.addIssue({
-                        code: v3_1.z.ZodIssueCode.custom,
-                        message: 'Direction titles must be distinct',
-                    });
-                }
-            })
-                .optional()
-                .describe('start: one AI-authored title/subtitle brief per direction; use one for a single visualization or distinct briefs for divergent ideas; briefs.length sets the variant count'),
-            element: v3_1.z
-                .string()
-                .optional()
-                .describe('start: CSS selector scoping the variation'),
-            files: v3_1.z
-                .array(v3_1.z.string().trim().min(1))
-                .max(RELEVANT_FILE_MAX_COUNT)
-                .optional()
-                .describe('start: known project files for workers to inspect before broader discovery'),
-            fresh: v3_1.z
-                .boolean()
-                .optional()
-                .describe('start: ignore prior variant context'),
-            profile: v3_1.z
-                .enum(['auto', 'off'])
-                .optional()
-                .describe('start: saved references are off by default; auto selects relevant references when the user asks for them'),
-            guidance: v3_1.z
-                .enum(['auto', 'off'])
-                .optional()
-                .describe('start: apply saved instructions, or turn them off for this request only'),
-            excludedGuidanceIds: v3_1.z
-                .array(v3_1.z.string().min(1).max(200))
-                .optional()
-                .describe('start: saved instruction document IDs to exclude for this request only'),
-            fidelity: v3_1.z
-                .enum(['low', 'medium', 'high'])
-                .optional()
-                .describe("start: the user's design-fidelity intent. low = fast sketch-grade workers (quick ideas/explorations); medium = balanced; high = the default with maximum deliberation, up to a few minutes (polished/pixel-perfect/final asks). Omit when unstated."),
-            workerProvider: v3_1.z
-                .enum(['claude', 'codex', 'cursor'])
-                .optional()
-                .describe('start: run the workers on this provider CLI instead of the one matching your caller agent. Pass ONLY when the user explicitly asks ("use opus for the workers") — it spends on that provider\'s subscription, so never infer it.'),
-            workerModel: v3_1.z
-                .string()
-                .trim()
-                .min(1)
-                .max(200)
-                .optional()
-                .describe('start: pin the worker model id for the selected (or caller-matching) provider. Pass ONLY on an explicit user request; a pinned model never silently falls back.'),
-            contextImages: v3_1.z
-                .array(v3_1.z.string().trim().min(1))
-                .min(1)
-                .optional()
-                .describe('start: local reference image file paths (png/jpg/webp/gif/avif). Rivet copies them into the run context and every worker views them natively — pass the actual images instead of describing them in notes.'),
-            contextAssets: v3_1.z
-                .array(contracts_1.contextAssetInputSchema)
-                .min(1)
-                .max(contracts_1.CONTEXT_ASSET_MAX_ITEMS)
-                .optional()
-                .describe('start: local regular files copied into every variant workspace and exposed to workers with stable file and reference paths.'),
-            referencePages: v3_1.z
-                .array(v3_1.z.string().trim().min(1))
-                .min(1)
-                .max(3)
-                .optional()
-                .describe('start (fresh only): reference page URLs. Rivet renders each page server-side (design evidence + screenshot) — pass the raw URLs, never capture evidence or screenshot pages yourself.'),
-            referenceVideos: v3_1.z
-                .array(v3_1.z.string().trim().min(1))
-                .min(1)
-                .max(2)
-                .optional()
-                .describe('start (fresh only): local reference video file paths (mp4/mov/webm/m4v). Rivet samples keyframes server-side — pass the raw paths, never run ffmpeg yourself.'),
-            target: v3_1.z
-                .string()
-                .optional()
-                .describe('start: sessionId:variantId — refine an existing variant in place'),
-            workId: v3_1.z.string().optional().describe('complete: host work item id'),
-            status: v3_1.z
-                .enum(['succeeded', 'failed', 'cancelled'])
-                .optional()
-                .describe('complete: terminal work status'),
-            output: v3_1.z
-                .unknown()
-                .optional()
-                .describe('complete: optional orchestrator output payload'),
-            error: v3_1.z.string().optional().describe('complete: failure message'),
-            title: v3_1.z
-                .string()
-                .optional()
-                .describe('complete: optional variant title'),
-            description: v3_1.z
-                .string()
-                .optional()
-                .describe('complete: optional variant description'),
-            variantId: v3_1.z
-                .string()
-                .optional()
-                .describe('commit: the variant to write into the working tree'),
-            sessionId: v3_1.z
-                .string()
-                .optional()
-                .describe('complete/commit: source session (optional) · cancel: session to cancel (required)'),
-            project: v3_1.z.string().optional().describe('Project root path'),
-        },
-    }, async ({ action, instruction, runLabel, count, briefs, element, files, fresh, fidelity, profile, guidance, excludedGuidanceIds, workerProvider, workerModel, contextImages, contextAssets, referencePages, referenceVideos, target, workId, status, output, error, title, description, variantId, sessionId, project, }, extra) => {
-        const startedAt = Date.now();
-        let rest;
-        if (action === 'start') {
-            if (!briefs) {
-                return completeToolCall({
-                    toolName: 'rivet_variants',
-                    operation: action,
-                    startedAt,
-                    envelope: (0, envelope_1.errorResult)('rivet_variants', 'USAGE_ERROR', 'rivet_variants action=start requires briefs with one AI-authored title/subtitle per direction, including fresh starts.').envelope,
-                });
-            }
-            if (!runLabel) {
-                return completeToolCall({
-                    toolName: 'rivet_variants',
-                    operation: action,
-                    startedAt,
-                    envelope: (0, envelope_1.errorResult)('rivet_variants', 'USAGE_ERROR', 'rivet_variants action=start requires runLabel with an AI-authored 1–3 word batch summary, including fresh starts.').envelope,
-                });
-            }
-            rest = [
-                'start',
-                ...(instruction !== undefined ? ['--instruction', instruction] : []),
-                ...(runLabel !== undefined ? ['--run-label', runLabel] : []),
-                ...(count !== undefined ? ['--count', String(count)] : []),
-                ...(briefs !== undefined ? ['--briefs', JSON.stringify(briefs)] : []),
-                ...(element !== undefined ? ['--element', element] : []),
-                ...(files !== undefined ? ['--files', JSON.stringify(files)] : []),
-                ...(fresh ? ['--fresh'] : []),
-                ...(profile ? ['--profile', profile] : []),
-                ...(guidance ? ['--guidance', guidance] : []),
-                ...(excludedGuidanceIds !== undefined
-                    ? ['--exclude-guidance', JSON.stringify(excludedGuidanceIds)]
-                    : []),
-                ...(fidelity !== undefined ? ['--fidelity', fidelity] : []),
-                ...(workerProvider !== undefined
-                    ? ['--worker-provider', workerProvider]
-                    : []),
-                ...(workerModel !== undefined ? ['--worker-model', workerModel] : []),
-                ...(contextImages !== undefined
-                    ? ['--context-images', JSON.stringify(contextImages)]
-                    : []),
-                ...(contextAssets !== undefined
-                    ? ['--context-assets', JSON.stringify(contextAssets)]
-                    : []),
-                ...(referencePages !== undefined
-                    ? ['--reference-pages', JSON.stringify(referencePages)]
-                    : []),
-                ...(referenceVideos !== undefined
-                    ? ['--reference-videos', JSON.stringify(referenceVideos)]
-                    : []),
-                ...(target !== undefined ? ['--target', target] : []),
-            ];
-        }
-        else if (action === 'commit') {
-            rest = [
-                'commit',
-                ...(variantId !== undefined ? [variantId] : []),
-                ...(sessionId !== undefined ? ['--session', sessionId] : []),
-            ];
-        }
-        else if (action === 'complete') {
-            rest = [
-                'complete',
-                ...(workId !== undefined ? [workId] : []),
-                ...(sessionId !== undefined ? ['--session', sessionId] : []),
-                ...(status !== undefined ? ['--status', status] : []),
-                ...(output !== undefined ? ['--output', JSON.stringify(output)] : []),
-                ...(error !== undefined ? ['--error', error] : []),
-                ...(title !== undefined ? ['--title', title] : []),
-                ...(description !== undefined ? ['--description', description] : []),
-            ];
-        }
-        else {
-            rest = ['cancel', ...(sessionId !== undefined ? [sessionId] : [])];
-        }
-        const { envelope, progress } = await runCommand('variants', [...rest, ...projectArgs(project)], { handoffArgs: (0, callerModel_1.callerModelArgsFromToolMeta)(extra._meta) });
-        return completeToolCall({
-            toolName: 'rivet_variants',
-            operation: action,
-            startedAt,
-            envelope,
-            progress,
-            sessionId: sessionId ??
-                (target !== undefined ? target.split(':')[0] : undefined),
-        });
-    });
-    mcp.registerTool('rivet_design_context', {
-        ...DESIGN_CONTEXT_PRESENTATION,
-        description: DESIGN_CONTEXT_DESCRIPTION,
-        inputSchema: {
-            contextAction: v3_1.z
-                .enum([
-                'add',
-                'list',
-                'show',
-                'update',
-                'enable',
-                'disable',
-                'remove',
-                'resolve',
-                'inspect',
-                'verifier',
-                'verify',
-            ])
-                .optional()
-                .describe('Manage saved instructions applied across projects. verifier retrieves one source and its checking recipe without execution; verify explicitly runs a saved executable package locally against project and contextVerifierInput. No automatic verification. Separate from optional reference profiles; lookup uses no model.'),
-            contextId: v3_1.z
-                .string()
-                .min(1)
-                .max(200)
-                .optional()
-                .describe('Saved instruction document ID for show, update, enable, disable, remove, verifier, or verify'),
-            contextFile: v3_1.z
-                .string()
-                .min(1)
-                .optional()
-                .describe('Local UTF-8 Markdown file copied when adding or updating saved instructions'),
-            contextVerifier: v3_1.z
-                .string()
-                .min(1)
-                .optional()
-                .describe('update: local UTF-8 Markdown verifier recipe, saved only after the user agrees to reuse it; requires contextSourceDigest'),
-            contextVerifierPackage: v3_1.z
-                .string()
-                .min(1)
-                .optional()
-                .describe('update: local JSON manifest {runtime:"node",entrypoint:"verify.mjs",files:[{name,path}]} for trusted local executable code retained beside contextVerifier; no dependency installation'),
-            contextVerifierInput: v3_1.z
-                .string()
-                .min(1)
-                .optional()
-                .describe('verify: required local JSON object with package-specific target URL, selectors, modes and prerequisites; explicit local execution, not a sandbox'),
-            contextSourceDigest: v3_1.z
-                .string()
-                .regex(/^[a-f0-9]{64}$/)
-                .optional()
-                .describe('update: exact sourceDigest from contextAction verifier, binding the recipe to the source inspected'),
-            removeContextVerifier: v3_1.z
-                .boolean()
-                .optional()
-                .describe('update: remove only the saved verifier recipe, preserving its source'),
-            contextReferences: v3_1.z
-                .string()
-                .min(1)
-                .optional()
-                .describe('Local JSON reference manifest for context add/update. Entries are {kind:"url"|"repository",url,intent,revision?,files?:[{name,path}]} or {kind:"directory",path,intent}. Repository URLs default to URL-only; existing local repositories or installed libraries use directory pointers without cloning or copying contents. Directory paths resolve relative to the manifest and remain live read-only sources, not snapshots. Omit files and additional setup Markdown unless the user explicitly requests a snapshot or approves a file selection. URLs are not fetched; this does not authorize source ingestion.'),
-            guidance: v3_1.z
-                .enum(['auto', 'off'])
-                .optional()
-                .describe('resolve: selection mode for this read only'),
-            excludedGuidanceIds: v3_1.z
-                .array(v3_1.z.string().min(1).max(200))
-                .optional()
-                .describe('resolve: saved instruction document IDs to exclude for this read only'),
-            url: v3_1.z
-                .string()
-                .optional()
-                .describe('The URL to gather design context from'),
-            profileAction: v3_1.z
-                .enum([
-                'status',
-                'diagnostics',
-                'rebuild-index',
-                'enable',
-                'disable',
-                'connect',
-                'refresh',
-                'pause',
-                'resume',
-                'retry',
-                'add-text',
-                'add-images',
-                'pins',
-                'add-variant',
-                'references',
-                'reference',
-                'include',
-                'exclude',
-                'remove',
-                'suggest',
-                'inspect',
-                'discover',
-                'plan',
-                'import',
-                'onboarding',
-            ])
-                .optional()
-                .describe('Manage the local cross-project design profile instead of fetching a URL. sourceId also identifies a job for pause/resume/retry.'),
-            title: v3_1.z.string().optional(),
-            text: v3_1.z.string().optional(),
-            files: v3_1.z.array(v3_1.z.string()).optional(),
-            query: v3_1.z.string().optional(),
-            sourceId: v3_1.z.string().optional(),
-            profileId: v3_1.z
-                .string()
-                .uuid()
-                .optional()
-                .describe('Library identity from the copied command'),
-            referenceKeys: v3_1.z
-                .array(v3_1.z.string())
-                .min(1)
-                .max(design_library_1.DESIGN_REFERENCE_BATCH_LIMIT)
-                .optional()
-                .describe('One or more copied shortId@revision keys to resolve in a single query'),
-            isDetailed: v3_1.z
-                .boolean()
-                .optional()
-                .describe('Include full saved text, analysis, sources, and validated image paths; default metadata only'),
-            discoveryId: v3_1.z
-                .string()
-                .uuid()
-                .optional()
-                .describe('Saved discovery to plan selected collections from'),
-            planId: v3_1.z
-                .string()
-                .uuid()
-                .optional()
-                .describe('Saved plan to inspect or explicitly approve'),
-            planRevision: v3_1.z
-                .string()
-                .optional()
-                .describe('Exact revision shown in the reviewed plan'),
-            collectionIds: v3_1.z.array(v3_1.z.string()).optional(),
-            isApproved: v3_1.z
-                .boolean()
-                .optional()
-                .describe('Set true only after the user approves the displayed scope and agent usage'),
-            onboardingAction: v3_1.z.enum(['later', 'dismiss', 'reset']).optional(),
-            referenceProvider: v3_1.z
-                .enum(['cosmos', 'pinterest', 'arena', 'x'])
-                .optional()
-                .describe('Provider to discover or set onboarding preferences for. Cosmos discovery requires its public profile URL.'),
-            requestId: v3_1.z
-                .string()
-                .min(1)
-                .max(200)
-                .optional()
-                .describe('Top-level workId from variants start to inspect, or suggestion request identity'),
-            projectId: v3_1.z
-                .string()
-                .optional()
-                .describe('Stable library project ID for pins or add-variant'),
-            variantRef: v3_1.z
-                .string()
-                .optional()
-                .describe('Exact sessionId:variantId to save as profile context'),
-            note: v3_1.z.string().optional(),
-            imageFile: v3_1.z
-                .string()
-                .optional()
-                .describe('Supplied screenshot for a direction without a saved static preview'),
-            runtime: v3_1.z.enum(['codex', 'claude', 'cursor']).optional(),
-            screenshot: v3_1.z
-                .boolean()
-                .optional()
-                .describe('Capture a screenshot (render path only)'),
-            width: v3_1.z
-                .number()
-                .int()
-                .positive()
-                .optional()
-                .describe('Viewport width (render path only)'),
-            height: v3_1.z
-                .number()
-                .int()
-                .positive()
-                .optional()
-                .describe('Viewport height (render path only)'),
-            project: v3_1.z.string().optional().describe('Project root path'),
-        },
-    }, async ({ url, contextAction, contextId, contextFile, contextReferences, contextVerifier, contextVerifierPackage, contextVerifierInput, contextSourceDigest, removeContextVerifier, guidance, excludedGuidanceIds, screenshot, width, height, project, profileAction, title, text, files, query, sourceId, profileId, referenceKeys, isDetailed, discoveryId, planId, planRevision, collectionIds, isApproved, onboardingAction, referenceProvider, requestId, projectId, variantRef, note, imageFile, runtime, }, extra) => {
-        const startedAt = Date.now();
-        const hasVerifierUpdate = contextVerifier !== undefined ||
-            contextVerifierPackage !== undefined ||
-            contextSourceDigest !== undefined ||
-            removeContextVerifier === true;
-        if (contextVerifierInput !== undefined && contextAction !== 'verify')
-            return completeToolCall({
-                toolName: 'rivet_design_context',
-                operation: 'context.verify',
-                startedAt,
-                envelope: (0, envelope_1.errorResult)('context.verify', 'USAGE_ERROR', 'contextVerifierInput requires contextAction verify').envelope,
-            });
-        if (hasVerifierUpdate && contextAction !== 'update')
-            return completeToolCall({
-                toolName: 'rivet_design_context',
-                operation: 'context.update',
-                startedAt,
-                envelope: (0, envelope_1.errorResult)('context.update', 'USAGE_ERROR', 'Verifier changes require contextAction update').envelope,
-            });
-        if (contextAction) {
-            if (profileAction || url) {
-                return completeToolCall({
-                    toolName: 'rivet_design_context',
-                    operation: `context.${contextAction}`,
-                    startedAt,
-                    envelope: (0, envelope_1.errorResult)('context', 'USAGE_ERROR', 'Choose contextAction without profileAction or url').envelope,
-                });
-            }
-            const isDocumentAction = [
-                'show',
-                'update',
-                'enable',
-                'disable',
-                'remove',
-                'verifier',
-                'verify',
-            ].includes(contextAction);
-            const isWriteAction = contextAction === 'add' || contextAction === 'update';
-            const isResolveAction = contextAction === 'resolve';
-            const result = await runCommand('context', [
-                contextAction,
-                ...(isDocumentAction && contextId !== undefined ? [contextId] : []),
-                ...(contextAction === 'inspect' && requestId !== undefined
-                    ? [requestId]
-                    : []),
-                ...(isWriteAction && title !== undefined ? ['--title', title] : []),
-                ...(isWriteAction && contextFile !== undefined
-                    ? ['--file', contextFile]
-                    : []),
-                ...(isWriteAction && contextReferences !== undefined
-                    ? ['--references', contextReferences]
-                    : []),
-                ...(contextVerifier !== undefined
-                    ? ['--verifier', contextVerifier]
-                    : []),
-                ...(contextVerifierPackage !== undefined
-                    ? ['--verifier-package', contextVerifierPackage]
-                    : []),
-                ...(contextVerifierInput !== undefined
-                    ? ['--input', contextVerifierInput]
-                    : []),
-                ...(contextSourceDigest !== undefined
-                    ? ['--source-digest', contextSourceDigest]
-                    : []),
-                ...(removeContextVerifier ? ['--remove-verifier'] : []),
-                ...(isResolveAction && guidance !== undefined
-                    ? ['--guidance', guidance]
-                    : []),
-                ...(isResolveAction && excludedGuidanceIds !== undefined
-                    ? ['--exclude-guidance', JSON.stringify(excludedGuidanceIds)]
-                    : []),
-                ...projectArgs(project),
-            ], { signal: AbortSignal.any([extra.signal, verificationAbort.signal]) });
-            return completeToolCall({
-                toolName: 'rivet_design_context',
-                operation: `context.${contextAction}`,
-                startedAt,
-                envelope: result.envelope,
-                progress: result.progress,
-            });
-        }
-        if (profileAction) {
-            const result = await runCommand('profile', [
-                profileAction,
-                ...(profileAction === 'reference'
-                    ? [
-                        ...(profileId ? [profileId] : []),
-                        ...(referenceKeys ?? []),
-                        ...(isDetailed ? ['--detail'] : []),
-                    ]
-                    : []),
-                ...(profileAction === 'connect' && url ? [url] : []),
-                ...(profileAction === 'discover' && (url ?? referenceProvider)
-                    ? [String(url ?? referenceProvider)]
-                    : []),
-                ...(profileAction === 'onboarding' && referenceProvider
-                    ? ['--provider', referenceProvider]
-                    : []),
-                ...(profileAction === 'plan' && (discoveryId ?? planId)
-                    ? [String(discoveryId ?? planId)]
-                    : []),
-                ...(profileAction === 'import' && planId ? [planId] : []),
-                ...(profileAction === 'onboarding' && onboardingAction
-                    ? [onboardingAction]
-                    : []),
-                ...(collectionIds
-                    ? ['--collections', JSON.stringify(collectionIds)]
-                    : []),
-                ...(planRevision ? ['--revision', planRevision] : []),
-                ...(isApproved === true ? ['--approve'] : []),
-                ...(sourceId ? [sourceId] : []),
-                ...(profileAction === 'inspect' && requestId ? [requestId] : []),
-                ...(profileAction === 'suggest' && requestId
-                    ? ['--request-id', requestId]
-                    : []),
-                ...(profileAction === 'add-variant' && variantRef
-                    ? [variantRef]
-                    : []),
-                ...(projectId ? ['--project-id', projectId] : []),
-                ...(note !== undefined ? ['--note', note] : []),
-                ...(imageFile ? ['--image-file', imageFile] : []),
-                ...(runtime ? ['--runtime', runtime] : []),
-                ...(title ? ['--title', title] : []),
-                ...(text ? ['--text', text] : []),
-                ...(files ? ['--files', JSON.stringify(files)] : []),
-                ...(query ? ['--query', query] : []),
-                ...projectArgs(project),
-            ]);
-            return completeToolCall({
-                toolName: 'rivet_design_context',
-                operation: `profile.${profileAction}`,
-                startedAt,
-                envelope: result.envelope,
-                progress: result.progress,
-            });
-        }
-        let hostname;
-        try {
-            if (!url)
-                throw new Error('A URL or profileAction is required');
-            hostname = new URL(url).hostname;
-        }
-        catch {
-            return completeToolCall({
-                toolName: 'rivet_design_context',
-                startedAt,
-                envelope: (0, envelope_1.errorResult)('design_context', 'USAGE_ERROR', `Not a valid URL: ${url}`).envelope,
-            });
-        }
-        const { envelope, progress } = (0, exports.isReferenceHost)(hostname)
-            ? await runCommand('reference', ['fetch', url, ...projectArgs(project)])
-            : await runCommand('evidence', [
-                'capture',
-                '--url',
-                url,
-                ...(screenshot ? ['--screenshot'] : []),
-                ...(width !== undefined ? ['--width', String(width)] : []),
-                ...(height !== undefined ? ['--height', String(height)] : []),
-                ...projectArgs(project),
-            ]);
-        return completeToolCall({
-            toolName: 'rivet_design_context',
-            operation: (0, telemetry_1.designContextKindForEnvelope)(envelope) ??
-                ((0, exports.isReferenceHost)(hostname) ? 'reference' : 'website'),
-            startedAt,
-            envelope,
-            progress,
-        });
-    });
+    const dependencies = {
+        mcp,
+        runCommand,
+        projectArgs,
+        completeToolCall,
+        serveProject,
+        callerAgent: callerFields.callerAgent,
+    };
+    (0, explorationTools_1.registerExplorationTools)(dependencies);
+    (0, designContextTools_1.registerDesignContextTools)(dependencies);
     return {
         mcp,
         telemetry,
-        cancelVerifications: async () => {
-            verificationAbort.abort();
-            await (0, GuidanceVerifierRunner_1.cancelGuidanceVerifications)();
-        },
     };
 };
 exports.buildMcpServer = buildMcpServer;
@@ -1181,9 +460,7 @@ const handleMcpServe = async (rest, io) => {
         if (isClosing)
             return;
         isClosing = true;
-        void built
-            .cancelVerifications()
-            .then(() => built.telemetry?.shutdown())
+        void Promise.resolve(built.telemetry?.shutdown())
             .catch(() => { })
             .finally(() => process.exit(0));
     };