rivet-design 0.13.5 → 0.14.1

package/dist/agent-variants/WorktreeOrchestrator.js

@@ -674,6 +674,20 @@ class AgentVariantsOrchestrator {
             ...(hostWork.length > 0 ? { hostWork } : {}),
         };
     }
+    /** User reference files stored in the work item's context bundle (uploaded
+     *  or copied local images). Surfaced on the snapshot so briefings can point
+     *  workers at the files directly — one Read instead of a bundle walk. */
+    static contextFilesOf(context) {
+        const files = (context?.userContext ?? [])
+            .filter((source) => source.transport === 'file')
+            .map((source) => ({
+            path: source.path,
+            ...(source.label ?? source.originalFilename
+                ? { label: source.label ?? source.originalFilename }
+                : {}),
+        }));
+        return files.length > 0 ? { contextFiles: files } : undefined;
+    }
     getHostWork(sessionId) {
         return this.store.getHostWork(sessionId).map((item) => {
             const kind = item.kind;
@@ -703,6 +717,7 @@ class AgentVariantsOrchestrator {
                 kind,
                 status: item.status,
                 leaseId: item.leaseId,
+                leaseOwner: item.leaseOwner,
                 attempt: item.attempt,
                 leaseExpiresAt: item.leaseExpiresAt,
                 workspace,
@@ -712,6 +727,7 @@ class AgentVariantsOrchestrator {
                     brief: input.briefBody ?? input.instruction ?? 'Create this variant',
                     ...(input.briefNotes ? { notes: input.briefNotes } : {}),
                 },
+                ...(input.fidelity ? { fidelity: input.fidelity } : {}),
                 ...(sourceLabel && sourceBody
                     ? { sourceDirection: { label: sourceLabel, brief: sourceBody } }
                     : {}),
@@ -724,6 +740,7 @@ class AgentVariantsOrchestrator {
                 ...(input.context?.contextBundlePath
                     ? { contextBundlePath: input.context.contextBundlePath }
                     : {}),
+                ...(AgentVariantsOrchestrator.contextFilesOf(input.context) ?? {}),
                 completion: kind === 'code_gen'
                     ? { kind: 'worktree_diff' }
                     : { kind: 'static_artifact' },
@@ -1908,7 +1925,12 @@ class AgentVariantsOrchestrator {
         // here, so this guard is an internal invariant. Existing-project sessions
         // keep the single-call path even with a bundle — they have no source_plan
         // stage, and the bundle's raw context is threaded straight into code_gen.
-        const isSourceGrounded = projectContext.kind === 'fresh' && Boolean(projectContext.variantContext);
+        // Brief-driven fresh starts are exempt for the same reason: the caller
+        // already did its source research when it authored the briefs, so the
+        // bundle is raw worker context (reference images), not research input.
+        const isSourceGrounded = projectContext.kind === 'fresh' &&
+            Boolean(projectContext.variantContext) &&
+            !args.briefs?.length;
         if (isSourceGrounded) {
             throw new errors_1.AgentVariantsError('INVALID_STAGE_ACTION', 'Source-grounded sessions must run through the source-research flow (propose), not the single-call startUnified path.');
         }
@@ -1991,6 +2013,10 @@ class AgentVariantsOrchestrator {
                 relevantFiles: args.relevantFiles,
                 projectContext,
                 runLabel: args.runLabel,
+                ...(args.fidelity ? { fidelity: args.fidelity } : {}),
+                // Briefs land via reportBriefs immediately below — never enter the
+                // source-plan stage, even for a fresh session carrying a bundle.
+                hasPreparedBriefs: true,
                 ...(args.sessionId ? { sessionId: args.sessionId } : {}),
             });
             if (proposeResult.stage !== 'awaiting_briefs' ||
@@ -2397,6 +2423,7 @@ class AgentVariantsOrchestrator {
                 tokensIn: args.tokensIn ?? null,
                 tokensOut: args.tokensOut ?? null,
                 model: args.model ?? null,
+                costUsd: args.costUsd ?? null,
                 outputBytes,
             });
             const refinementTelemetry = this.refinementTelemetryForWorkItem({