rivet-design 0.9.1 → 0.9.3

package/dist/mcp/agent-variants/SessionStore.d.ts

@@ -1,4 +1,4 @@
-import { type Brief, type LeasedWorkItemDescriptor, type ProjectContext, type Stage, type Target, type TerminalSummary, type VariantPickEnvelope, type WorkItem, type WorkItemStatus } from './contracts';
+import { type Brief, type ActiveVariantSnapshot, type LeasedWorkItemDescriptor, type ProjectContext, type SourceContextArtifact, type Stage, type Target, type TerminalSummary, type VariantPickEnvelope, type WorkItem, type WorkItemStatus } from './contracts';
 /** TTL for a lease on a code_gen work item. Bumped to 10min after observing
  *  agents legitimately take longer than 3min when running 4+ Task sub-agents
  *  in parallel — the recovery path (re-lease + re-report) worked, but the UX
@@ -19,7 +19,9 @@ export interface ProposeArgs {
 }
 export interface ProposeResult {
     sessionId: string;
-    briefWorkItem: WorkItem;
+    stage: 'awaiting_source_research' | 'awaiting_briefs';
+    briefWorkItem?: WorkItem;
+    sourceResearchWorkItem?: WorkItem;
 }
 export interface ApprovedBriefSelection {
     briefId: string;
@@ -58,6 +60,15 @@ export declare class SessionStore {
     }): {
         briefs: Brief[];
     };
+    reportSourceContext(args: {
+        sessionId: string;
+        workItemId: string;
+        attempt: number;
+        leaseId: string;
+        sourceContext: SourceContextArtifact;
+    }): {
+        briefWorkItem: WorkItem;
+    };
     approve(args: {
         sessionId: string;
         selections: ApprovedBriefSelection[];
@@ -79,6 +90,33 @@ export declare class SessionStore {
             message: string;
         };
     }): ReportCompleteResult;
+    /**
+     * Mark a server-owned work item as succeeded without going through the
+     * lease + agent-report path. Used for `scaffold_base` items where the
+     * orchestrator runs the scaffold + npm install in the background and needs
+     * to unblock dependent code_gen items when it's done.
+     *
+     * Bypasses lease validation because no lease was ever issued for internal
+     * items. Idempotent: completing an already-succeeded item is a no-op.
+     */
+    completeInternal(args: {
+        sessionId: string;
+        workItemId: string;
+        output?: unknown;
+    }): ReportCompleteResult;
+    /**
+     * Mark a server-owned work item as failed, cascading DEPENDENCY_FAILED to
+     * its dependents. Used when the orchestrator's background scaffold work
+     * (npm install, file writes) fails and we need the session to degrade.
+     */
+    failInternal(args: {
+        sessionId: string;
+        workItemId: string;
+        error: {
+            code: string;
+            message: string;
+        };
+    }): ReportCompleteResult;
     cancel(args: {
         sessionId: string;
         reason?: string;
@@ -86,6 +124,24 @@ export declare class SessionStore {
         stage: 'cancelled';
         cleanupStatus: 'pending' | 'complete' | 'partial';
     };
+    /**
+     * Cancel a single work item without tearing down its sibling variants or
+     * the session itself. Idempotent on terminal states — already-cancelled,
+     * succeeded, or failed items return their current status with
+     * `alreadyTerminal: true` and no state change. Triggers
+     * `evaluateTerminal` so the session can reach a terminal stage if this
+     * was the last in-flight item.
+     */
+    cancelWorkItem(args: {
+        sessionId: string;
+        workItemId: string;
+        reason?: string;
+    }): {
+        workItemId: string;
+        finalStatus: WorkItemStatus;
+        sessionStage: Stage;
+        alreadyTerminal: boolean;
+    };
     recordVariantPick(args: {
         sessionId: string;
         envelope: VariantPickEnvelope;
@@ -99,11 +155,13 @@ export declare class SessionStore {
         total: number;
     };
     getSummary(sessionId: string): TerminalSummary;
+    getVariants(sessionId: string): ActiveVariantSnapshot[];
     getVariantPick(sessionId: string): VariantPickEnvelope | undefined;
     hasSession(sessionId: string): boolean;
     /** Read a work item's input blob — used by the orchestrator to recover
      *  briefLabel / target / projectContext when auto-enqueueing the diff. */
     getWorkItemInput(sessionId: string, workItemId: string): unknown;
+    getWorkItemOutput(sessionId: string, workItemId: string): unknown;
     /** Read the projectContext for a session (existing | fresh). */
     getProjectContext(sessionId: string): ProjectContext;
     private requireSession;
@@ -112,10 +170,13 @@ export declare class SessionStore {
     private validateLease;
     private expireStaleLeases;
     private leasableItems;
+    private hasLeasableCodeGenForStaticPreview;
     private dependenciesMet;
     private cascadeDependencyFailure;
+    private terminalItems;
     private evaluateTerminal;
     private progressOf;
     private summaryOf;
+    private variantsOf;
 }
 //# sourceMappingURL=SessionStore.d.ts.map

package/dist/mcp/agent-variants/SessionStore.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"SessionStore.d.ts","sourceRoot":"","sources":["../../../src/mcp/agent-variants/SessionStore.ts"],"names":[],"mappings":"AAEA,OAAO,EAEL,KAAK,KAAK,EACV,KAAK,wBAAwB,EAC7B,KAAK,cAAc,EACnB,KAAK,KAAK,EACV,KAAK,MAAM,EACX,KAAK,eAAe,EACpB,KAAK,mBAAmB,EACxB,KAAK,QAAQ,EAEb,KAAK,cAAc,EACpB,MAAM,aAAa,CAAC;AAIrB;;;2DAG2D;AAC3D,eAAO,MAAM,oBAAoB,QAAiB,CAAC;AACnD;oEACoE;AACpE,eAAO,MAAM,qBAAqB,QAAiB,CAAC;AAEpD,MAAM,WAAW,iBAAiB;IAChC,GAAG,IAAI,MAAM,CAAC;CACf;AAID,MAAM,WAAW,WAAW;IAC1B,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,MAAM,EAAE,MAAM,CAAC;IACf,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,cAAc,CAAC,EAAE,cAAc,CAAC;CACjC;AAED,MAAM,WAAW,aAAa;IAC5B,SAAS,EAAE,MAAM,CAAC;IAClB,aAAa,EAAE,QAAQ,CAAC;CACzB;AAED,MAAM,WAAW,sBAAsB;IACrC,OAAO,EAAE,MAAM,CAAC;IAChB,YAAY,CAAC,EAAE,MAAM,CAAC;CACvB;AAED,MAAM,WAAW,aAAa;IAC5B,aAAa,EAAE,MAAM,CAAC;IACtB,UAAU,EAAE,MAAM,CAAC;IACnB,kBAAkB,EAAE,MAAM,EAAE,CAAC;IAC7B,sBAAsB,CAAC,EAAE,MAAM,CAAC;CACjC;AAED,MAAM,WAAW,iBAAiB;IAChC,OAAO,EAAE,MAAM,CAAC;IAChB,UAAU,EAAE,MAAM,CAAC;IACnB,eAAe,EAAE,wBAAwB,EAAE,CAAC;CAC7C;AAED,MAAM,WAAW,oBAAoB;IACnC,KAAK,EAAE,KAAK,CAAC;IACb,QAAQ,EAAE;QAAE,KAAK,EAAE,MAAM,CAAC;QAAC,KAAK,EAAE,MAAM,CAAA;KAAE,CAAC;IAC3C,OAAO,CAAC,EAAE,eAAe,CAAC;CAC3B;AAmDD,qBAAa,YAAY;IACvB,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAoC;IAC7D,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAoB;gBAE9B,KAAK,GAAE,iBAAgC;IAMnD,OAAO,CAAC,IAAI,EAAE,WAAW,GAAG,aAAa;IA+CzC,YAAY,CAAC,IAAI,EAAE;QACjB,SAAS,EAAE,MAAM,CAAC;QAClB,UAAU,EAAE,MAAM,CAAC;QACnB,OAAO,EAAE,MAAM,CAAC;QAChB,OAAO,CAAC,EAAE,MAAM,CAAC;QACjB,MAAM,EAAE,KAAK,EAAE,CAAC;KACjB,GAAG;QAAE,MAAM,EAAE,KAAK,EAAE,CAAA;KAAE;IA4CvB,OAAO,CAAC,IAAI,EAAE;QACZ,SAAS,EAAE,MAAM,CAAC;QAClB,UAAU,EAAE,sBAAsB,EAAE,CAAC;KACtC,GAAG,aAAa;IA6EjB,WAAW,CAAC,IAAI,EAAE;QAChB,SAAS,EAAE,MAAM,CAAC;QAClB,UAAU,EAAE,MAAM,CAAC;QACnB,mBAAmB,CAAC,EAAE,MAAM,CAAC;KAC9B,GAAG,iBAAiB;IAyCrB,cAAc,CAAC,IAAI,EAAE;QACnB,SAAS,EAAE,MAAM,CAAC;QAClB,UAAU,EAAE,MAAM,CAAC;QACnB,OAAO,EAAE,MAAM,CAAC;QAChB,OAAO,EAAE,MAAM,CAAC;QAChB,MAAM,EAAE,cAAc,CAAC;QACvB,MAAM,CAAC,EAAE,OAAO,CAAC;QACjB,KAAK,CAAC,EAAE;YAAE,IAAI,EAAE,MAAM,CAAC;YAAC,OAAO,EAAE,MAAM,CAAA;SAAE,CAAC;KAC3C,GAAG,oBAAoB;IAoExB,MAAM,CAAC,IAAI,EAAE;QAAE,SAAS,EAAE,MAAM,CAAC;QAAC,MAAM,CAAC,EAAE,MAAM,CAAA;KAAE,GAAG;QACpD,KAAK,EAAE,WAAW,CAAC;QACnB,aAAa,EAAE,SAAS,GAAG,UAAU,GAAG,SAAS,CAAC;KACnD;IAwBD,iBAAiB,CAAC,IAAI,EAAE;QACtB,SAAS,EAAE,MAAM,CAAC;QAClB,QAAQ,EAAE,mBAAmB,CAAC;KAC/B,GAAG;QAAE,SAAS,EAAE,OAAO,CAAA;KAAE;IA0B1B,QAAQ,CAAC,SAAS,EAAE,MAAM,GAAG,KAAK;IAIlC,SAAS,CAAC,SAAS,EAAE,MAAM,GAAG,KAAK,EAAE;IAIrC,WAAW,CAAC,SAAS,EAAE,MAAM,GAAG;QAAE,KAAK,EAAE,MAAM,CAAC;QAAC,KAAK,EAAE,MAAM,CAAA;KAAE;IAIhE,UAAU,CAAC,SAAS,EAAE,MAAM,GAAG,eAAe;IAK9C,cAAc,CAAC,SAAS,EAAE,MAAM,GAAG,mBAAmB,GAAG,SAAS;IAIlE,UAAU,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO;IAItC;8EAC0E;IAC1E,gBAAgB,CAAC,SAAS,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,GAAG,OAAO;IAMhE,gEAAgE;IAChE,iBAAiB,CAAC,SAAS,EAAE,MAAM,GAAG,cAAc;IAMpD,OAAO,CAAC,cAAc;IAWtB,OAAO,CAAC,eAAe;IAWvB,OAAO,CAAC,WAAW;IASnB,OAAO,CAAC,aAAa;IA+BrB,OAAO,CAAC,iBAAiB;IAczB,OAAO,CAAC,aAAa;IAYrB,OAAO,CAAC,eAAe;IAQvB,OAAO,CAAC,wBAAwB;IAehC,OAAO,CAAC,gBAAgB;IAmCxB,OAAO,CAAC,UAAU;IAWlB,OAAO,CAAC,SAAS;CAkBlB"}
+{"version":3,"file":"SessionStore.d.ts","sourceRoot":"","sources":["../../../src/mcp/agent-variants/SessionStore.ts"],"names":[],"mappings":"AAEA,OAAO,EAEL,KAAK,KAAK,EACV,KAAK,qBAAqB,EAE1B,KAAK,wBAAwB,EAC7B,KAAK,cAAc,EACnB,KAAK,qBAAqB,EAC1B,KAAK,KAAK,EACV,KAAK,MAAM,EACX,KAAK,eAAe,EACpB,KAAK,mBAAmB,EACxB,KAAK,QAAQ,EAEb,KAAK,cAAc,EACpB,MAAM,aAAa,CAAC;AAKrB;;;2DAG2D;AAC3D,eAAO,MAAM,oBAAoB,QAAiB,CAAC;AACnD;oEACoE;AACpE,eAAO,MAAM,qBAAqB,QAAiB,CAAC;AAEpD,MAAM,WAAW,iBAAiB;IAChC,GAAG,IAAI,MAAM,CAAC;CACf;AAID,MAAM,WAAW,WAAW;IAC1B,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,MAAM,EAAE,MAAM,CAAC;IACf,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,cAAc,CAAC,EAAE,cAAc,CAAC;CACjC;AAED,MAAM,WAAW,aAAa;IAC5B,SAAS,EAAE,MAAM,CAAC;IAClB,KAAK,EAAE,0BAA0B,GAAG,iBAAiB,CAAC;IACtD,aAAa,CAAC,EAAE,QAAQ,CAAC;IACzB,sBAAsB,CAAC,EAAE,QAAQ,CAAC;CACnC;AAED,MAAM,WAAW,sBAAsB;IACrC,OAAO,EAAE,MAAM,CAAC;IAChB,YAAY,CAAC,EAAE,MAAM,CAAC;CACvB;AAED,MAAM,WAAW,aAAa;IAC5B,aAAa,EAAE,MAAM,CAAC;IACtB,UAAU,EAAE,MAAM,CAAC;IACnB,kBAAkB,EAAE,MAAM,EAAE,CAAC;IAC7B,sBAAsB,CAAC,EAAE,MAAM,CAAC;CACjC;AAED,MAAM,WAAW,iBAAiB;IAChC,OAAO,EAAE,MAAM,CAAC;IAChB,UAAU,EAAE,MAAM,CAAC;IACnB,eAAe,EAAE,wBAAwB,EAAE,CAAC;CAC7C;AAED,MAAM,WAAW,oBAAoB;IACnC,KAAK,EAAE,KAAK,CAAC;IACb,QAAQ,EAAE;QAAE,KAAK,EAAE,MAAM,CAAC;QAAC,KAAK,EAAE,MAAM,CAAA;KAAE,CAAC;IAC3C,OAAO,CAAC,EAAE,eAAe,CAAC;CAC3B;AA+GD,qBAAa,YAAY;IACvB,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAoC;IAC7D,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAoB;gBAE9B,KAAK,GAAE,iBAAgC;IAMnD,OAAO,CAAC,IAAI,EAAE,WAAW,GAAG,aAAa;IAsEzC,YAAY,CAAC,IAAI,EAAE;QACjB,SAAS,EAAE,MAAM,CAAC;QAClB,UAAU,EAAE,MAAM,CAAC;QACnB,OAAO,EAAE,MAAM,CAAC;QAChB,OAAO,CAAC,EAAE,MAAM,CAAC;QACjB,MAAM,EAAE,KAAK,EAAE,CAAC;KACjB,GAAG;QAAE,MAAM,EAAE,KAAK,EAAE,CAAA;KAAE;IA4CvB,mBAAmB,CAAC,IAAI,EAAE;QACxB,SAAS,EAAE,MAAM,CAAC;QAClB,UAAU,EAAE,MAAM,CAAC;QACnB,OAAO,EAAE,MAAM,CAAC;QAChB,OAAO,EAAE,MAAM,CAAC;QAChB,aAAa,EAAE,qBAAqB,CAAC;KACtC,GAAG;QAAE,aAAa,EAAE,QAAQ,CAAA;KAAE;IAsD/B,OAAO,CAAC,IAAI,EAAE;QACZ,SAAS,EAAE,MAAM,CAAC;QAClB,UAAU,EAAE,sBAAsB,EAAE,CAAC;KACtC,GAAG,aAAa;IAoFjB,WAAW,CAAC,IAAI,EAAE;QAChB,SAAS,EAAE,MAAM,CAAC;QAClB,UAAU,EAAE,MAAM,CAAC;QACnB,mBAAmB,CAAC,EAAE,MAAM,CAAC;KAC9B,GAAG,iBAAiB;IAkDrB,cAAc,CAAC,IAAI,EAAE;QACnB,SAAS,EAAE,MAAM,CAAC;QAClB,UAAU,EAAE,MAAM,CAAC;QACnB,OAAO,EAAE,MAAM,CAAC;QAChB,OAAO,EAAE,MAAM,CAAC;QAChB,MAAM,EAAE,cAAc,CAAC;QACvB,MAAM,CAAC,EAAE,OAAO,CAAC;QACjB,KAAK,CAAC,EAAE;YAAE,IAAI,EAAE,MAAM,CAAC;YAAC,OAAO,EAAE,MAAM,CAAA;SAAE,CAAC;KAC3C,GAAG,oBAAoB;IAkExB;;;;;;;;OAQG;IACH,gBAAgB,CAAC,IAAI,EAAE;QACrB,SAAS,EAAE,MAAM,CAAC;QAClB,UAAU,EAAE,MAAM,CAAC;QACnB,MAAM,CAAC,EAAE,OAAO,CAAC;KAClB,GAAG,oBAAoB;IAkBxB;;;;OAIG;IACH,YAAY,CAAC,IAAI,EAAE;QACjB,SAAS,EAAE,MAAM,CAAC;QAClB,UAAU,EAAE,MAAM,CAAC;QACnB,KAAK,EAAE;YAAE,IAAI,EAAE,MAAM,CAAC;YAAC,OAAO,EAAE,MAAM,CAAA;SAAE,CAAC;KAC1C,GAAG,oBAAoB;IAqBxB,MAAM,CAAC,IAAI,EAAE;QAAE,SAAS,EAAE,MAAM,CAAC;QAAC,MAAM,CAAC,EAAE,MAAM,CAAA;KAAE,GAAG;QACpD,KAAK,EAAE,WAAW,CAAC;QACnB,aAAa,EAAE,SAAS,GAAG,UAAU,GAAG,SAAS,CAAC;KACnD;IAsBD;;;;;;;OAOG;IACH,cAAc,CAAC,IAAI,EAAE;QACnB,SAAS,EAAE,MAAM,CAAC;QAClB,UAAU,EAAE,MAAM,CAAC;QACnB,MAAM,CAAC,EAAE,MAAM,CAAC;KACjB,GAAG;QACF,UAAU,EAAE,MAAM,CAAC;QACnB,WAAW,EAAE,cAAc,CAAC;QAC5B,YAAY,EAAE,KAAK,CAAC;QACpB,eAAe,EAAE,OAAO,CAAC;KAC1B;IA+BD,iBAAiB,CAAC,IAAI,EAAE;QACtB,SAAS,EAAE,MAAM,CAAC;QAClB,QAAQ,EAAE,mBAAmB,CAAC;KAC/B,GAAG;QAAE,SAAS,EAAE,OAAO,CAAA;KAAE;IA0B1B,QAAQ,CAAC,SAAS,EAAE,MAAM,GAAG,KAAK;IAIlC,SAAS,CAAC,SAAS,EAAE,MAAM,GAAG,KAAK,EAAE;IAIrC,WAAW,CAAC,SAAS,EAAE,MAAM,GAAG;QAAE,KAAK,EAAE,MAAM,CAAC;QAAC,KAAK,EAAE,MAAM,CAAA;KAAE;IAIhE,UAAU,CAAC,SAAS,EAAE,MAAM,GAAG,eAAe;IAK9C,WAAW,CAAC,SAAS,EAAE,MAAM,GAAG,qBAAqB,EAAE;IAKvD,cAAc,CAAC,SAAS,EAAE,MAAM,GAAG,mBAAmB,GAAG,SAAS;IAIlE,UAAU,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO;IAItC;8EAC0E;IAC1E,gBAAgB,CAAC,SAAS,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,GAAG,OAAO;IAMhE,iBAAiB,CAAC,SAAS,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,GAAG,OAAO;IAMjE,gEAAgE;IAChE,iBAAiB,CAAC,SAAS,EAAE,MAAM,GAAG,cAAc;IAMpD,OAAO,CAAC,cAAc;IAWtB,OAAO,CAAC,eAAe;IAWvB,OAAO,CAAC,WAAW;IASnB,OAAO,CAAC,aAAa;IA+BrB,OAAO,CAAC,iBAAiB;IAczB,OAAO,CAAC,aAAa;IAoBrB,OAAO,CAAC,kCAAkC;IAgB1C,OAAO,CAAC,eAAe;IAQvB,OAAO,CAAC,wBAAwB;IAehC,OAAO,CAAC,aAAa;IAOrB,OAAO,CAAC,gBAAgB;IA+BxB,OAAO,CAAC,UAAU;IAgBlB,OAAO,CAAC,SAAS;IAiBjB,OAAO,CAAC,UAAU;CAoBnB"}