rivet-design 0.20.0 → 0.21.0

package/dist/routes/agentVariants.js

@@ -4,6 +4,8 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
 };
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.createAgentVariantsRouter = exports.stampWatcherLiveness = exports.nestedFreshSessionWorkdirs = void 0;
+const variantRequestSchema_1 = require("./variantRequestSchema");
+const generationGuidance_1 = require("../services/generationGuidance");
 const RuntimeAdmission_1 = require("../services/RuntimeAdmission");
 const telemetry_1 = require("../config/telemetry");
 const runtimeAdmission_1 = require("./runtimeAdmission");
@@ -20,9 +22,10 @@ const express_1 = require("express");
 const v3_1 = require("zod/v3");
 const logger_1 = require("../utils/logger");
 const agent_variants_1 = require("../agent-variants");
-const contracts_2 = require("../agent-variants/contracts");
 const changeBatchClassification_1 = require("../mcp/changeBatchClassification");
 const createAgentVariantsOrchestrator_1 = require("../services/createAgentVariantsOrchestrator");
+const ConfigManager_1 = require("../services/ConfigManager");
+const guidance_1 = require("../types/guidance");
 const VariantHistoryService_1 = require("../services/VariantHistoryService");
 const PrototypeDeployService_1 = require("../services/PrototypeDeployService");
 const HistoryReplayService_1 = require("../services/HistoryReplayService");
@@ -36,8 +39,6 @@ const preview_bridge_1 = require("../proxy-middleware/preview-bridge");
 const displayLocale_1 = require("../utils/displayLocale");
 const htmlDocument_1 = require("../utils/htmlDocument");
 const log = (0, logger_1.createLogger)('AgentVariantsRouter');
-const DIRECTION_TITLE_MAX_WORDS = 5;
-const DIRECTION_SUBTITLE_MAX_WORDS = 10;
 const PREVIEW_CDN_HOSTS = [
     'https://unpkg.com',
     'https://cdn.jsdelivr.net',
@@ -121,77 +122,6 @@ const inlineStaticPreviewTargetSchema = v3_1.z.object({
     sessionId: v3_1.z.string().min(1),
     variantId: v3_1.z.string().min(1),
 });
-const inlineVariantBriefSchema = contracts_2.startVariantsBriefContent.superRefine((brief, context) => {
-    const titleWordCount = brief.label.trim().split(/\s+/).length;
-    const subtitleWordCount = brief.body.trim().split(/\s+/).length;
-    if (titleWordCount > DIRECTION_TITLE_MAX_WORDS) {
-        context.addIssue({
-            code: v3_1.z.ZodIssueCode.custom,
-            path: ['label'],
-            message: 'direction titles must contain at most 5 words',
-        });
-    }
-    if (subtitleWordCount > DIRECTION_SUBTITLE_MAX_WORDS) {
-        context.addIssue({
-            code: v3_1.z.ZodIssueCode.custom,
-            path: ['body'],
-            message: 'direction subtitles must contain at most 10 words',
-        });
-    }
-});
-const inlineVariantRequestSchema = v3_1.z
-    .object({
-    count: v3_1.z.number().int().min(contracts_2.VARIANT_COUNT_MIN).max(contracts_2.VARIANT_COUNT_MAX),
-    mode: v3_1.z.enum(['existing', 'zero_to_one']).optional(),
-    profile: v3_1.z.enum(['auto', 'off']).optional(),
-    runLabel: contracts_2.runLabelSchema,
-    fidelity: v3_1.z.enum(['low', 'medium', 'high']).optional(),
-    briefs: v3_1.z
-        .array(inlineVariantBriefSchema)
-        .min(contracts_2.VARIANT_COUNT_MIN)
-        .max(contracts_2.VARIANT_COUNT_MAX)
-        .optional(),
-    contextAssets: v3_1.z
-        .array(contracts_2.contextAssetInputSchema)
-        .min(1)
-        .max(contracts_2.CONTEXT_ASSET_MAX_ITEMS)
-        .optional(),
-    referencePages: v3_1.z
-        .array(v3_1.z.object({
-        url: v3_1.z.string().url().max(2048),
-        note: v3_1.z.string().min(1).max(500).optional(),
-    }))
-        .min(1)
-        .max(3)
-        .optional(),
-    referenceVideos: v3_1.z
-        .array(v3_1.z.object({
-        path: v3_1.z.string().min(1).max(2048),
-        note: v3_1.z.string().min(1).max(500).optional(),
-    }))
-        .min(1)
-        .max(2)
-        .optional(),
-})
-    .superRefine((request, context) => {
-    if (request.briefs && request.briefs.length !== request.count) {
-        context.addIssue({
-            code: v3_1.z.ZodIssueCode.custom,
-            path: ['briefs'],
-            message: 'briefs length must match count',
-        });
-    }
-    if (request.briefs) {
-        const labels = request.briefs.map((brief) => brief.label.trim().toLowerCase().replace(/\s+/g, ' '));
-        if (new Set(labels).size !== labels.length) {
-            context.addIssue({
-                code: v3_1.z.ZodIssueCode.custom,
-                path: ['briefs'],
-                message: 'brief labels must be distinct',
-            });
-        }
-    }
-});
 const inlineInitiatorSchema = v3_1.z
     .discriminatedUnion('kind', [
     v3_1.z.object({ kind: v3_1.z.literal('human') }),
@@ -212,7 +142,7 @@ const inlineVariantBody = v3_1.z.object({
         instruction: v3_1.z.string().min(1),
         images: v3_1.z.array(v3_1.z.string()).optional(),
         references: v3_1.z.array(commentReferenceItemSchema).max(8).optional(),
-        variantRequest: inlineVariantRequestSchema,
+        variantRequest: variantRequestSchema_1.inlineVariantRequestSchema,
     })
         .passthrough())
         .min(1),
@@ -805,6 +735,8 @@ const createAgentVariantsRouter = (orchestrator, bridge, telemetry, variantHisto
             acceptedLanguages: req.acceptsLanguages(),
         });
         let request = {
+            guidanceAccountId: req.header('X-Rivet-Account-Id'),
+            guidanceSelection: parsed.data.changes[0]?.variantRequest.guidanceSelection,
             changes: parsed.data.changes,
             sourceFiles: parsed.data.sourceFiles ?? [],
             displayLocale,
@@ -816,39 +748,41 @@ const createAgentVariantsRouter = (orchestrator, bridge, telemetry, variantHisto
             initiator: parsed.data.initiator,
             workerSelection: parsed.data.workerSelection,
         };
-        bridge?.updateTelemetryContext(request.telemetryContext);
-        if ((0, changeBatchClassification_1.isMixedVariantBatch)(request) || !(0, changeBatchClassification_1.isSingleVariantRequestBatch)(request)) {
-            res.status(400).json({
-                error: 'Invalid inline variants batch',
-                details: 'Inline variants require exactly one comment with a variantRequest.',
-                reason: 'mixed_batch',
-            });
-            return;
-        }
-        const activeVariantsSession = orchestrator.buildActiveSnapshot();
-        let effectiveTarget = batchTarget ?? staticPreviewTarget;
-        const isPersistedTargetRequest = Boolean(effectiveTarget &&
-            (activeVariantsSession.active !== true ||
-                effectiveTarget.sessionId !== activeVariantsSession.sessionId));
-        if ((isPersistedTargetRequest || !effectiveTarget) &&
-            rejectBlockedVariantSession(res, activeVariantsSession)) {
-            return;
-        }
-        if ((isPersistedTargetRequest || !effectiveTarget) &&
-            inlineStartTracker.isStarting()) {
-            res.status(400).json({
-                error: 'Variants start already in progress',
-                details: 'A variants start for this project is still being processed (provisioning can take a while) — status reports it as `variants.starting`. Poll status until the session registers or the start settles, then retry. Do not cancel anything.',
-                reason: 'start_in_flight',
-            });
-            return;
-        }
         let trackedStart = false;
-        if (isPersistedTargetRequest || !effectiveTarget) {
-            inlineStartTracker.begin();
-            trackedStart = true;
-        }
         try {
+            (0, generationGuidance_1.assertGenerationGuidance)(request);
+            bridge?.updateTelemetryContext(request.telemetryContext);
+            if ((0, changeBatchClassification_1.isMixedVariantBatch)(request) ||
+                !(0, changeBatchClassification_1.isSingleVariantRequestBatch)(request)) {
+                res.status(400).json({
+                    error: 'Invalid inline variants batch',
+                    details: 'Inline variants require exactly one comment with a variantRequest.',
+                    reason: 'mixed_batch',
+                });
+                return;
+            }
+            const activeVariantsSession = orchestrator.buildActiveSnapshot();
+            let effectiveTarget = batchTarget ?? staticPreviewTarget;
+            const isPersistedTargetRequest = Boolean(effectiveTarget &&
+                (activeVariantsSession.active !== true ||
+                    effectiveTarget.sessionId !== activeVariantsSession.sessionId));
+            if ((isPersistedTargetRequest || !effectiveTarget) &&
+                rejectBlockedVariantSession(res, activeVariantsSession)) {
+                return;
+            }
+            if ((isPersistedTargetRequest || !effectiveTarget) &&
+                inlineStartTracker.isStarting()) {
+                res.status(400).json({
+                    error: 'Variants start already in progress',
+                    details: 'A variants start for this project is still being processed (provisioning can take a while) — status reports it as `variants.starting`. Poll status until the session registers or the start settles, then retry. Do not cancel anything.',
+                    reason: 'start_in_flight',
+                });
+                return;
+            }
+            if (isPersistedTargetRequest || !effectiveTarget) {
+                inlineStartTracker.begin();
+                trackedStart = true;
+            }
             if (isPersistedTargetRequest && effectiveTarget) {
                 const staleTarget = effectiveTarget;
                 let adoption;
@@ -938,14 +872,20 @@ const createAgentVariantsRouter = (orchestrator, bridge, telemetry, variantHisto
                 });
                 return;
             }
+            const { profileContext, guidance = [], guidanceWarning } = result;
             res.status(200).json({
                 success: true,
-                ...(result.profileContext
-                    ? { profileContext: result.profileContext }
-                    : {}),
+                guidance,
+                ...(guidanceWarning ? { guidanceWarning } : {}),
+                ...(profileContext ? { profileContext } : {}),
             });
         }
         catch (err) {
+            if (err instanceof guidance_1.GuidanceError &&
+                err.code === 'guidance_account_changed') {
+                res.status(409).json({ error: err.message, code: err.code });
+                return;
+            }
             log.error('Inline variants run failed:', err);
             res.status(500).json({
                 error: err instanceof Error ? err.message : 'Inline variants run failed',
@@ -1002,6 +942,56 @@ const createAgentVariantsRouter = (orchestrator, bridge, telemetry, variantHisto
             });
         }
     });
+    router.get('/variants/guidance/:workId', async (req, res) => {
+        if (rejectNonLocalRequest(req, res))
+            return;
+        const projectPath = resolveProjectPath(req.query.projectPath);
+        const workId = req.params.workId;
+        if (!projectPath ||
+            typeof workId !== 'string' ||
+            !/^[A-Za-z0-9_-]+$/.test(workId)) {
+            res.status(400).json({
+                error: 'A project and valid work ID are required to inspect saved context.',
+            });
+            return;
+        }
+        const config = (0, ConfigManager_1.getConfigManager)();
+        const accountId = config.isAuthenticated() ? config.getUserId() : undefined;
+        const assertAccount = () => {
+            const currentAccountId = config.isAuthenticated()
+                ? config.getUserId()
+                : undefined;
+            if (currentAccountId !== accountId ||
+                (req.query.accountId !== undefined &&
+                    req.query.accountId !== currentAccountId))
+                throw new guidance_1.GuidanceError('guidance_account_changed', 'The saved context account changed; retry inspection.');
+        };
+        try {
+            assertAccount();
+            for (const candidate of resolveHistoryProjectPaths({ projectPath })) {
+                const result = await variantHistory.inspectGuidance({
+                    projectPath: candidate,
+                    workId,
+                    accountId: accountId ?? undefined,
+                });
+                if (result.isRecorded) {
+                    assertAccount();
+                    res.json(result);
+                    return;
+                }
+            }
+            assertAccount();
+            res.json({ isRecorded: false, guidance: [] });
+        }
+        catch (error) {
+            res.status(error instanceof guidance_1.GuidanceError ? 409 : 500).json({
+                ...(error instanceof guidance_1.GuidanceError ? { code: error.code } : {}),
+                error: error instanceof Error
+                    ? error.message
+                    : 'Could not inspect saved context.',
+            });
+        }
+    });
     router.get('/variants/active', (req, res) => {
         if (rejectNonLocalRequest(req, res))
             return;