rivet-design 0.14.21 → 0.16.0

package/dist/agent-variants/WorktreeOrchestrator.js

@@ -1158,7 +1158,26 @@ class AgentVariantsOrchestrator {
                 workItemId: item.id,
                 kind: item.kind,
             });
-            resources.contextAssets.set(item.id, (0, contextAssets_1.materializeContextAssets)({ assets, workspace }));
+            const previewKind = workspace.kind === 'artifact_directory'
+                ? 'static_artifact'
+                : (resources.contextAssetPreviewKind ?? 'dev_server');
+            resources.contextAssets.set(item.id, (0, contextAssets_1.materializeContextAssets)({ assets, workspace, previewKind }));
+        }
+    }
+    /**
+     * @effect Records how an existing project's worktrees will be previewed so
+     * context assets receive references that resolve on that serving surface.
+     */
+    async resolveContextAssetPreviewKind(sessionId) {
+        const resources = this.ensureResources(sessionId);
+        try {
+            const environment = await this.resolveEnv(sessionId);
+            resources.contextAssetPreviewKind =
+                environment.framework === 'static' ? 'static_artifact' : 'dev_server';
+        }
+        catch (error) {
+            resources.contextAssetPreviewKind = 'dev_server';
+            log.warn(`Failed to resolve context asset preview kind for ${sessionId}; using dev-server references`, error);
         }
     }
     /**
@@ -1528,9 +1547,7 @@ class AgentVariantsOrchestrator {
         }
         let hasAssets = false;
         try {
-            hasAssets = fs_1.default
-                .readdirSync(assetBase)
-                .some((name) => name !== entryPath);
+            hasAssets = fs_1.default.readdirSync(assetBase).some((name) => name !== entryPath);
         }
         catch {
             hasAssets = false;
@@ -1946,6 +1963,9 @@ class AgentVariantsOrchestrator {
         // agent has nothing useful to do between approve and request_work
         // anyway — running them serially is the contract the skill assumes.
         try {
+            if (projectContext.kind === 'existing') {
+                await this.resolveContextAssetPreviewKind(args.sessionId);
+            }
             await this.provisionWorktrees(args.sessionId, result);
             this.emitChange();
         }
@@ -2713,6 +2733,39 @@ class AgentVariantsOrchestrator {
             sessionStage: result.sessionStage,
         };
     }
+    /**
+     * Rename a run folder: re-home one request batch in a single store pass,
+     * persist each affected manifest best-effort, and push one updated snapshot.
+     */
+    async renameRun(args) {
+        const result = this.store.renameRun({
+            sessionId: args.sessionId,
+            ...(args.requestId ? { requestId: args.requestId } : {}),
+            runLabel: args.runLabel,
+            label: args.label,
+        });
+        this.telemetry.track('agent_variants.run_renamed', {
+            source: 'mcp',
+            sessionId: args.sessionId,
+            // Telemetry policy: never the raw label text, only its length.
+            runLabelLength: args.label.length,
+            renamedCount: result.workItemIds.length,
+            sessionStage: result.sessionStage,
+        });
+        for (const variantId of result.workItemIds) {
+            void this.markPersistedVariantRenamed(args.sessionId, variantId, undefined, args.label).catch((err) => {
+                log.warn(`markPersistedVariantRenamed failed for ${args.sessionId}/${variantId}`, err);
+            });
+        }
+        this.emitChange();
+        return {
+            sessionId: args.sessionId,
+            runLabel: args.runLabel,
+            label: args.label,
+            renamedCount: result.workItemIds.length,
+            sessionStage: result.sessionStage,
+        };
+    }
     /**
      * User has reviewed the rendered variants and picked one. Look up the
      * captured diff or generated project, apply it to the destination, and record
@@ -4751,7 +4804,7 @@ class AgentVariantsOrchestrator {
             status: 'removed',
         });
     }
-    async markPersistedVariantRenamed(sessionId, variantId, label) {
+    async markPersistedVariantRenamed(sessionId, variantId, label, runLabel) {
         if (!this.store.hasSession(sessionId))
             return;
         const projectPath = await this.resolveHistoryProjectPath(sessionId);
@@ -4765,6 +4818,7 @@ class AgentVariantsOrchestrator {
             sessionId,
             variantId,
             label,
+            runLabel,
         });
     }
     /**