rivetkit 2.3.11-rc.9 → 2.3.11

package/dist/tsup/test/mod.cjs

@@ -1,13 +1,13 @@
 "use strict";Object.defineProperty(exports, "__esModule", {value: true}); function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }
 
-var _chunkV5CVATRUcjs = require('../chunk-V5CVATRU.cjs');
+var _chunkGLIPKKECcjs = require('../chunk-GLIPKKEC.cjs');
 
 
-var _chunkBZWKY6WOcjs = require('../chunk-BZWKY6WO.cjs');
+var _chunkRXXDNJXBcjs = require('../chunk-RXXDNJXB.cjs');
 require('../chunk-MYWZCDML.cjs');
-require('../chunk-T5XVTVQY.cjs');
-require('../chunk-UEULFXUI.cjs');
-require('../chunk-T56IZ42M.cjs');
+require('../chunk-KVFIONKQ.cjs');
+require('../chunk-GN2EUMEW.cjs');
+require('../chunk-GU7ML3KE.cjs');
 require('../chunk-4FC7TVS6.cjs');
 
 // src/test/ready.ts
@@ -36,8 +36,8 @@ async function setupTest(c, registry) {
   registry.config.noWelcome = true;
   registry.start();
   await waitForRegistryReady(registry);
-  const client = _chunkV5CVATRUcjs.createClient.call(void 0, {
-    ..._chunkBZWKY6WOcjs.convertRegistryConfigToClientConfig.call(void 0, registry.parseConfig()),
+  const client = _chunkGLIPKKECcjs.createClient.call(void 0, {
+    ..._chunkRXXDNJXBcjs.convertRegistryConfigToClientConfig.call(void 0, registry.parseConfig()),
     disableMetadataLookup: false
   });
   c.onTestFinished(async () => {

package/dist/tsup/test/mod.d.cts

@@ -1,11 +1,12 @@
 import { TestContext } from 'vitest';
-import { aC as Registry, a2 as Client } from '../config-rGWrumlO.cjs';
+import { aF as Registry, a3 as Client } from '../config-DTGoWdkG.cjs';
 import 'zod/v4';
-import '../config-BxOimoGX.cjs';
+import '../config-ey2I2zz5.cjs';
 import '@rivetkit/rivetkit-wasm';
 import 'zod';
 import 'pino';
 import '@rivetkit/virtual-websocket';
+import '../inspector/workflow.cjs';
 import '@standard-schema/spec';
 import '../utils-DQDCphrl.cjs';
 import 'hono';

package/dist/tsup/test/mod.d.ts

@@ -1,11 +1,12 @@
 import { TestContext } from 'vitest';
-import { aC as Registry, a2 as Client } from '../config-CZcjyT5B.js';
+import { aF as Registry, a3 as Client } from '../config-C4OSv0QP.js';
 import 'zod/v4';
-import '../config-BxOimoGX.js';
+import '../config-ey2I2zz5.js';
 import '@rivetkit/rivetkit-wasm';
 import 'zod';
 import 'pino';
 import '@rivetkit/virtual-websocket';
+import '../inspector/workflow.js';
 import '@standard-schema/spec';
 import '../utils-CV0akuDD.js';
 import 'hono';

package/dist/tsup/test/mod.js

@@ -1,13 +1,13 @@
 import {
   createClient
-} from "../chunk-IOLZ3ONU.js";
+} from "../chunk-JQ2AZ5FN.js";
 import {
   convertRegistryConfigToClientConfig
-} from "../chunk-HOZ3ZLZ2.js";
+} from "../chunk-3ZR542SK.js";
 import "../chunk-JI6GZ2C2.js";
-import "../chunk-XOOESJY7.js";
-import "../chunk-7AFZMFPQ.js";
-import "../chunk-LZ4TSBIP.js";
+import "../chunk-JTHHCZCZ.js";
+import "../chunk-FWROC5LA.js";
+import "../chunk-OUENKQHB.js";
 import "../chunk-ZZ3WBRPD.js";
 
 // src/test/ready.ts

package/dist/tsup/unstable/migrations.d.cts

@@ -1,4 +1,4 @@
-import { R as RawAccess } from '../config-BxOimoGX.cjs';
+import { R as RawAccess } from '../config-ey2I2zz5.cjs';
 
 declare const DEFAULT_MIGRATION_TABLE_NAME = "rivetkit_schema_version";
 type MigrationDatabase = Pick<RawAccess, "execute">;

package/dist/tsup/unstable/migrations.d.ts

@@ -1,4 +1,4 @@
-import { R as RawAccess } from '../config-BxOimoGX.js';
+import { R as RawAccess } from '../config-ey2I2zz5.js';
 
 declare const DEFAULT_MIGRATION_TABLE_NAME = "rivetkit_schema_version";
 type MigrationDatabase = Pick<RawAccess, "execute">;

package/dist/tsup/utils.cjs

@@ -19,7 +19,7 @@
 
 
 
-var _chunkT56IZ42Mcjs = require('./chunk-T56IZ42M.cjs');
+var _chunkGU7ML3KEcjs = require('./chunk-GU7ML3KE.cjs');
 require('./chunk-4FC7TVS6.cjs');
 
 
@@ -42,5 +42,5 @@ require('./chunk-4FC7TVS6.cjs');
 
 
 
-exports.EXTRA_ERROR_LOG = _chunkT56IZ42Mcjs.EXTRA_ERROR_LOG; exports.SinglePromiseQueue = _chunkT56IZ42Mcjs.SinglePromiseQueue; exports.VERSION = _chunkT56IZ42Mcjs.VERSION; exports.arrayBuffersEqual = _chunkT56IZ42Mcjs.arrayBuffersEqual; exports.assertUnreachable = _chunkT56IZ42Mcjs.assertUnreachable; exports.bufferToArrayBuffer = _chunkT56IZ42Mcjs.bufferToArrayBuffer; exports.combineUrlPath = _chunkT56IZ42Mcjs.combineUrlPath; exports.dbg = _chunkT56IZ42Mcjs.dbg; exports.detectRuntime = _chunkT56IZ42Mcjs.detectRuntime; exports.getEnvUniversal = _chunkT56IZ42Mcjs.getEnvUniversal; exports.httpUserAgent = _chunkT56IZ42Mcjs.httpUserAgent; exports.interval = _chunkT56IZ42Mcjs.interval; exports.joinSignals = _chunkT56IZ42Mcjs.joinSignals; exports.jsonParseCompat = _chunkT56IZ42Mcjs.jsonParseCompat; exports.jsonStringifyCompat = _chunkT56IZ42Mcjs.jsonStringifyCompat; exports.promiseWithResolvers = _chunkT56IZ42Mcjs.promiseWithResolvers; exports.setLongTimeout = _chunkT56IZ42Mcjs.setLongTimeout; exports.sleep = _chunkT56IZ42Mcjs.sleep; exports.stringifyError = _chunkT56IZ42Mcjs.stringifyError; exports.toUint8Array = _chunkT56IZ42Mcjs.toUint8Array;
+exports.EXTRA_ERROR_LOG = _chunkGU7ML3KEcjs.EXTRA_ERROR_LOG; exports.SinglePromiseQueue = _chunkGU7ML3KEcjs.SinglePromiseQueue; exports.VERSION = _chunkGU7ML3KEcjs.VERSION; exports.arrayBuffersEqual = _chunkGU7ML3KEcjs.arrayBuffersEqual; exports.assertUnreachable = _chunkGU7ML3KEcjs.assertUnreachable; exports.bufferToArrayBuffer = _chunkGU7ML3KEcjs.bufferToArrayBuffer; exports.combineUrlPath = _chunkGU7ML3KEcjs.combineUrlPath; exports.dbg = _chunkGU7ML3KEcjs.dbg; exports.detectRuntime = _chunkGU7ML3KEcjs.detectRuntime; exports.getEnvUniversal = _chunkGU7ML3KEcjs.getEnvUniversal; exports.httpUserAgent = _chunkGU7ML3KEcjs.httpUserAgent; exports.interval = _chunkGU7ML3KEcjs.interval; exports.joinSignals = _chunkGU7ML3KEcjs.joinSignals; exports.jsonParseCompat = _chunkGU7ML3KEcjs.jsonParseCompat; exports.jsonStringifyCompat = _chunkGU7ML3KEcjs.jsonStringifyCompat; exports.promiseWithResolvers = _chunkGU7ML3KEcjs.promiseWithResolvers; exports.setLongTimeout = _chunkGU7ML3KEcjs.setLongTimeout; exports.sleep = _chunkGU7ML3KEcjs.sleep; exports.stringifyError = _chunkGU7ML3KEcjs.stringifyError; exports.toUint8Array = _chunkGU7ML3KEcjs.toUint8Array;
 //# sourceMappingURL=utils.cjs.map

package/dist/tsup/utils.js

@@ -19,7 +19,7 @@ import {
   sleep,
   stringifyError,
   toUint8Array
-} from "./chunk-LZ4TSBIP.js";
+} from "./chunk-OUENKQHB.js";
 import "./chunk-ZZ3WBRPD.js";
 export {
   EXTRA_ERROR_LOG,

package/dist/tsup/workflow/mod.cjs

@@ -1,18 +1,19 @@
 "use strict";Object.defineProperty(exports, "__esModule", {value: true}); function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; } async function _asyncNullishCoalesce(lhs, rhsFn) { if (lhs != null) { return lhs; } else { return await rhsFn(); } } var _class; var _class2;
 
-var _chunk6SBPWTXJcjs = require('../chunk-6SBPWTXJ.cjs');
-require('../chunk-IMRDHZM2.cjs');
+var _chunkPUZY2XCDcjs = require('../chunk-PUZY2XCD.cjs');
+require('../chunk-B3W74FBB.cjs');
+require('../chunk-VBJQIBXI.cjs');
 
 
 
-var _chunkHNHLBM3Ycjs = require('../chunk-HNHLBM3Y.cjs');
+var _chunk4KGXFC6Mcjs = require('../chunk-4KGXFC6M.cjs');
 
 
 
 var _chunkMYWZCDMLcjs = require('../chunk-MYWZCDML.cjs');
 
 
-var _chunkT56IZ42Mcjs = require('../chunk-T56IZ42M.cjs');
+var _chunkGU7ML3KEcjs = require('../chunk-GU7ML3KE.cjs');
 
 
 
@@ -392,53 +393,50 @@ function normalizeSqlBlob(value) {
   }
   throw new Error("workflow sqlite value was not a blob");
 }
-function runtimeSqlFromContext(runCtx) {
-  const sql = runCtx == null ? void 0 : runCtx.sql;
-  if (sql && typeof sql === "object" && "query" in sql && "execute" in sql && "executeBatch" in sql && "run" in sql) {
-    return sql;
+function runtimeDbFromContext(runCtx) {
+  const db = runCtx == null ? void 0 : runCtx.db;
+  if (db && typeof db === "object" && "execute" in db && "transaction" in db) {
+    return db;
   }
   return void 0;
 }
 var WorkflowStorage = class {
-  #sql;
-  constructor(sql) {
-    if (!sql) {
+  #db;
+  constructor(db) {
+    if (!db) {
       throw new Error(
         "workflow storage requires embedded SQLite; actor KV workflow storage is no longer supported"
       );
     }
-    if (!sql.executeBatch) {
-      throw new Error(
-        "workflow storage requires a SQLite database with executeBatch support"
-      );
-    }
-    this.#sql = sql;
+    this.#db = db;
   }
   async get(key) {
     var _a;
     const prefixed = _chunkMYWZCDMLcjs.makeWorkflowKey.call(void 0, key);
-    const result = await this.#sql.query(
+    const rows = await this.#db.execute(
       "SELECT value FROM _rivet_wf_kv WHERE key = ?",
-      [prefixed]
+      prefixed
     );
-    return ((_a = result.rows[0]) == null ? void 0 : _a[0]) == null ? null : normalizeSqlBlob(result.rows[0][0]);
+    return ((_a = rows[0]) == null ? void 0 : _a.value) == null ? null : normalizeSqlBlob(rows[0].value);
   }
   async set(key, value) {
     await this.batch([{ key, value }]);
   }
   async delete(key) {
     const prefixed = _chunkMYWZCDMLcjs.makeWorkflowKey.call(void 0, key);
-    await this.#sql.run("DELETE FROM _rivet_wf_kv WHERE key = ?", [
+    await this.#db.execute(
+      "DELETE FROM _rivet_wf_kv WHERE key = ?",
       prefixed
-    ]);
+    );
   }
   async deletePrefix(prefix) {
     const start = _chunkMYWZCDMLcjs.makeWorkflowKey.call(void 0, prefix);
     const end = computeUpperBound(start);
     if (end) {
-      await this.#sql.run(
+      await this.#db.execute(
         "DELETE FROM _rivet_wf_kv WHERE key >= ? AND key < ?",
-        [start, end]
+        start,
+        end
       );
       return;
     }
@@ -450,9 +448,10 @@ var WorkflowStorage = class {
   async deleteRange(start, end) {
     const prefixedStart = _chunkMYWZCDMLcjs.makeWorkflowKey.call(void 0, start);
     const prefixedEnd = _chunkMYWZCDMLcjs.makeWorkflowKey.call(void 0, end);
-    await this.#sql.run(
+    await this.#db.execute(
       "DELETE FROM _rivet_wf_kv WHERE key >= ? AND key < ?",
-      [prefixedStart, prefixedEnd]
+      prefixedStart,
+      prefixedEnd
     );
   }
   async list(prefix) {
@@ -466,36 +465,45 @@ var WorkflowStorage = class {
   async batch(writes) {
     if (writes.length === 0) return;
     for (const chunk of chunkWorkflowWrites(writes)) {
-      await this.#sql.executeBatch(
-        chunk.map(({ key, value }) => ({
-          sql: "INSERT INTO _rivet_wf_kv (key, value) VALUES (?, ?) ON CONFLICT(key) DO UPDATE SET value = excluded.value",
-          params: [_chunkMYWZCDMLcjs.makeWorkflowKey.call(void 0, key), value]
-        }))
-      );
+      await this.#db.transaction(async (tx) => {
+        for (const { key, value } of chunk) {
+          await tx.execute(
+            "INSERT INTO _rivet_wf_kv (key, value) VALUES (?, ?) ON CONFLICT(key) DO UPDATE SET value = excluded.value",
+            _chunkMYWZCDMLcjs.makeWorkflowKey.call(void 0, key),
+            value
+          );
+        }
+      });
     }
   }
   async listRaw(prefix) {
     const end = computeUpperBound(prefix);
-    const result = end ? await this.#sql.query(
+    const rows = end ? await this.#db.execute(
       "SELECT key, value FROM _rivet_wf_kv WHERE key >= ? AND key < ? ORDER BY key ASC",
-      [prefix, end]
-    ) : await this.#sql.query(
+      prefix,
+      end
+    ) : await this.#db.execute(
       "SELECT key, value FROM _rivet_wf_kv WHERE key >= ? ORDER BY key ASC",
-      [prefix]
+      prefix
     );
-    return result.rows.map((row) => [
-      normalizeSqlBlob(row[0]),
-      normalizeSqlBlob(row[1])
+    return rows.map((row) => [
+      normalizeSqlBlob(row.key),
+      normalizeSqlBlob(row.value)
     ]);
   }
   async deleteRawKeys(keys) {
     for (let start = 0; start < keys.length; start += WORKFLOW_SQLITE_MAX_BATCH_ROWS) {
-      await this.#sql.executeBatch(
-        keys.slice(start, start + WORKFLOW_SQLITE_MAX_BATCH_ROWS).map((key) => ({
-          sql: "DELETE FROM _rivet_wf_kv WHERE key = ?",
-          params: [key]
-        }))
-      );
+      await this.#db.transaction(async (tx) => {
+        for (const key of keys.slice(
+          start,
+          start + WORKFLOW_SQLITE_MAX_BATCH_ROWS
+        )) {
+          await tx.execute(
+            "DELETE FROM _rivet_wf_kv WHERE key = ?",
+            key
+          );
+        }
+      });
     }
   }
 };
@@ -585,7 +593,7 @@ var ActorWorkflowDriver = (_class = class {
     this.#actor = actor;
     this.#runCtx = runCtx;
     this.messageDriver = new ActorWorkflowMessageDriver(actor, runCtx);
-    this.#storage = new WorkflowStorage(runtimeSqlFromContext(runCtx));
+    this.#storage = new WorkflowStorage(runtimeDbFromContext(runCtx));
   }
   async get(key) {
     return await this.#runCtx.internalKeepAwake(this.#storage.get(key));
@@ -648,7 +656,7 @@ var ActorWorkflowControlDriver = (_class2 = class {
   #storage;
   constructor(actor, runCtx) {;_class2.prototype.__init3.call(this);_class2.prototype.__init4.call(this);
     this.#actor = actor;
-    this.#storage = new WorkflowStorage(runtimeSqlFromContext(runCtx));
+    this.#storage = new WorkflowStorage(runtimeDbFromContext(runCtx));
   }
   async get(key) {
     return await this.#storage.get(key);
@@ -710,45 +718,53 @@ function workflowReplayInFlightError() {
 function isWorkflowReplayBlockedByRunningEntry(error) {
   return error instanceof Error && error.message === "Cannot replay a workflow while a step is currently running";
 }
+function isRunHandlerUnavailable(error) {
+  return error instanceof _chunk4FC7TVS6cjs.RivetError && error.group === "actor" && error.code === "run_handler_unavailable";
+}
 function workflow(fn, options = {}) {
   const onError = options.onError;
   const workflowInspectors = /* @__PURE__ */ new Map();
   function getWorkflowInspector(actorId) {
     let workflowInspector = workflowInspectors.get(actorId);
     if (!workflowInspector) {
-      workflowInspector = _chunk6SBPWTXJcjs.createWorkflowInspectorAdapter.call(void 0, );
+      workflowInspector = _chunkPUZY2XCDcjs.createWorkflowInspectorAdapter.call(void 0, );
       workflowInspectors.set(actorId, workflowInspector);
     }
     return workflowInspector;
   }
   async function run(runCtx) {
-    const actor = runCtx[_chunkHNHLBM3Ycjs.ACTOR_CONTEXT_INTERNAL_SYMBOL];
+    const actor = runCtx[_chunk4KGXFC6Mcjs.ACTOR_CONTEXT_INTERNAL_SYMBOL];
     _invariant2.default.call(void 0, actor, "workflow() requires an actor instance");
     const workflowInspector = getWorkflowInspector(actor.id);
     const driver = new ActorWorkflowDriver(actor, runCtx);
     const controlDriver = new ActorWorkflowControlDriver(actor, runCtx);
     workflowInspector.setReplayFromStep(async (entryId) => {
       const workflowState = await workflowInspector.adapter.getState();
-      if (actor.isRunHandlerActive() || workflowState === "pending" || workflowState === "running") {
+      if (workflowState === "pending" || workflowState === "running") {
         throw workflowReplayInFlightError();
       }
-      let snapshot;
+      const control = workflowInspector.control;
+      _invariant2.default.call(void 0, control, "workflow Inspector control is not initialized");
       try {
-        snapshot = await _workflowengine.replayWorkflowFromStep.call(void 0, 
-          actor.id,
-          controlDriver,
-          entryId,
-          { scheduleAlarm: false }
+        return await control.run.withInactive(
+          { restartOnSuccess: true },
+          async () => {
+            const snapshot = await _workflowengine.replayWorkflowFromStep.call(void 0, 
+              actor.id,
+              controlDriver,
+              entryId,
+              { scheduleAlarm: false }
+            );
+            workflowInspector.update(snapshot);
+            return workflowInspector.adapter.getHistory();
+          }
         );
       } catch (error) {
-        if (isWorkflowReplayBlockedByRunningEntry(error)) {
+        if (isWorkflowReplayBlockedByRunningEntry(error) || isRunHandlerUnavailable(error)) {
           throw workflowReplayInFlightError();
         }
         throw error;
       }
-      workflowInspector.update(snapshot);
-      await actor.restartRunHandler();
-      return workflowInspector.adapter.getHistory();
     });
     const handle = _workflowengine.runWorkflow.call(void 0, 
       actor.id,
@@ -784,56 +800,34 @@ function workflow(fn, options = {}) {
       if (shouldRethrowWorkflowError(error)) {
         runCtx.log.error({
           msg: "workflow run failed",
-          error: _chunkT56IZ42Mcjs.stringifyError.call(void 0, error)
+          error: _chunkGU7ML3KEcjs.stringifyError.call(void 0, error)
         });
         throw error;
       }
       runCtx.log.warn({
         msg: "workflow failed and will sleep until woken",
-        error: _chunkT56IZ42Mcjs.stringifyError.call(void 0, error)
+        error: _chunkGU7ML3KEcjs.stringifyError.call(void 0, error)
       });
     } finally {
       runCtx.abortSignal.removeEventListener("abort", onAbort);
     }
   }
-  const runWithConfig = run;
-  runWithConfig[_chunkHNHLBM3Ycjs.RUN_FUNCTION_CONFIG_SYMBOL] = {
+  return _chunk4KGXFC6Mcjs.defineRunHandler.call(void 0, run, {
     icon: "diagram-project",
-    // Drop the per-actor inspector when the actor is destroyed so this map
-    // does not retain one inspector (and its encoded history) per actor id
-    // for the process lifetime.
-    disposeInspector: (actorId) => {
-      workflowInspectors.delete(actorId);
-    },
-    inspectorFactory: (actor) => {
-      const actorId = resolveWorkflowInspectorActorId(actor);
+    inspectorKind: "workflow",
+    createInspector: ({ actorId, control }) => {
+      const workflowInspector = getWorkflowInspector(actorId);
+      workflowInspector.control = control;
       return {
-        workflow: actorId ? getWorkflowInspector(actorId).adapter : {
-          getHistory: () => null,
-          onHistoryUpdated: () => () => {
-          },
-          replayFromStep: async () => null
+        inspector: { workflow: workflowInspector.adapter },
+        dispose: () => {
+          if (workflowInspectors.get(actorId) === workflowInspector) {
+            workflowInspectors.delete(actorId);
+          }
         }
       };
     }
-  };
-  return runWithConfig;
-}
-function resolveWorkflowInspectorActorId(actor) {
-  if (typeof actor === "string" && actor.length > 0) {
-    return actor;
-  }
-  if (!actor || typeof actor !== "object") {
-    return void 0;
-  }
-  const candidate = actor;
-  if (typeof candidate.id === "string" && candidate.id.length > 0) {
-    return candidate.id;
-  }
-  if (typeof candidate.actorId === "string" && candidate.actorId.length > 0) {
-    return candidate.actorId;
-  }
-  return void 0;
+  });
 }