rivetkit 2.3.11-rc.8 → 2.3.11

package/dist/tsup/agent-os/index.js

@@ -1,6 +1,9 @@
 // src/agent-os/actor/index.ts
 import { AgentOs, createInMemoryFileSystem } from "@rivet-dev/agent-os-core";
 
+// src/actor/config.ts
+import { z } from "zod/v4";
+
 // src/actor/actions.ts
 function flattenActionHandlers(actions) {
   const flattened = /* @__PURE__ */ Object.create(null);
@@ -84,7 +87,6 @@ function formatActionPath(path) {
 }
 
 // src/actor/config.ts
-import { z } from "zod/v4";
 var DEFAULT_SLEEP_GRACE_PERIOD = 15e3;
 var zFunction = () => z.custom((val) => typeof val === "function");
 var zActionTree = z.custom((value) => {
@@ -105,6 +107,7 @@ var zActionTree = z.custom((value) => {
 });
 var WorkflowInspectorConfigSchema = z.object({
   getHistory: zFunction(),
+  getState: zFunction().optional(),
   onHistoryUpdated: zFunction().optional(),
   replayFromStep: zFunction().optional()
 });
@@ -417,7 +420,7 @@ function isRivetErrorCode(error, group, code) {
 // package.json
 var package_default = {
   name: "rivetkit",
-  version: "2.3.11-rc.8",
+  version: "2.3.11",
   description: "Lightweight libraries for building stateful actors on edge platforms",
   license: "Apache-2.0",
   keywords: [
@@ -553,6 +556,16 @@ var package_default = {
         default: "./dist/tsup/inspector/mod.cjs"
       }
     },
+    "./experimental/inspector/workflow": {
+      import: {
+        types: "./dist/tsup/inspector/workflow.d.ts",
+        default: "./dist/tsup/inspector/workflow.js"
+      },
+      require: {
+        types: "./dist/tsup/inspector/workflow.d.cts",
+        default: "./dist/tsup/inspector/workflow.cjs"
+      }
+    },
     "./inspector-tab": {
       import: {
         types: "./dist/tsup/inspector-tab/mod.d.ts",
@@ -598,7 +611,7 @@ var package_default = {
     "./dist/tsup/chunk-*.cjs"
   ],
   scripts: {
-    build: "tsup src/mod.ts src/client/mod.ts src/common/log.ts src/common/websocket.ts src/actor/errors.ts src/utils.ts src/workflow/mod.ts src/test/mod.ts src/inspector/mod.ts src/inspector-tab/mod.ts src/db/mod.ts src/db/drizzle.ts src/dynamic/mod.ts src/unstable/migrations.ts && tsup src/agent-os/index.ts --no-clean --out-dir dist/tsup/agent-os",
+    build: "tsup src/mod.ts src/client/mod.ts src/common/log.ts src/common/websocket.ts src/actor/errors.ts src/utils.ts src/workflow/mod.ts src/test/mod.ts src/inspector/mod.ts src/inspector/workflow.ts src/inspector-tab/mod.ts src/db/mod.ts src/db/drizzle.ts src/dynamic/mod.ts src/unstable/migrations.ts && tsup src/agent-os/index.ts --no-clean --out-dir dist/tsup/agent-os && node scripts/check-built-commonjs.mjs",
     "build:browser": "tsup --config tsup.browser.config.ts",
     "check-types": "tsc --noEmit",
     lint: "biome check . && pnpm run check:test-skips && pnpm run check:wait-for-comments",
@@ -899,6 +912,14 @@ function validateTransactionTimeout(timeout) {
     );
   }
 }
+function validateTransactionName(name) {
+  if (name === void 0) {
+    return;
+  }
+  if (name.length === 0) {
+    throw new Error("db.transaction() name must not be empty");
+  }
+}
 function isSqliteBindingValue(value) {
   if (value === null || typeof value === "number" || typeof value === "string" || typeof value === "bigint" || value instanceof Uint8Array) {
     return true;
@@ -938,15 +959,19 @@ function toSqliteBindings(input) {
 }
 
 // src/common/database/mod.ts
+var nativeStateTransactionOpeners = /* @__PURE__ */ new WeakMap();
+var nativeStateTransactionClientBinders = /* @__PURE__ */ new WeakMap();
 function hasMultipleStatements(query) {
   const trimmed = query.trim().replace(/;+$/, "").trimEnd();
   return trimmed.includes(";");
 }
 function db({
   onMigrate,
-  warnOnManualTransactions = true
+  warnOnManualTransactions = true,
+  profiling
 } = {}) {
-  return {
+  const provider = {
+    sqliteProfiling: profiling,
     createClient: async (ctx) => {
       const nativeDatabaseProvider = ctx.nativeDatabaseProvider;
       if (!nativeDatabaseProvider) {
@@ -964,86 +989,127 @@ function db({
           );
         }
       };
-      const createClient = (target, transactionScoped = false) => ({
-        execute: async (query, ...args) => {
-          var _a, _b, _c, _d;
-          ensureOpen();
-          if (!transactionScoped && warnOnManualTransactions && !manualTransactionWarned && !hasMultipleStatements(query) && isManualTransactionControl(query)) {
-            manualTransactionWarned = true;
-            getLogger("database").warn(
-              { actorId: ctx.actorId },
-              "Manual cross-call SQLite transactions can interleave with other actor work. Use db.transaction() for coordinated transactions. Set warnOnManualTransactions: false in your db(...) configuration to disable this warning."
-            );
-          }
-          const kvReadsBefore = ((_a = ctx.metrics) == null ? void 0 : _a.totalKvReads) ?? 0;
-          const kvWritesBefore = ((_b = ctx.metrics) == null ? void 0 : _b.totalKvWrites) ?? 0;
-          const start = performance.now();
-          try {
-            if (args.length > 0) {
-              const bindings = args.length === 1 && isSqliteBindingObject(args[0]) ? toSqliteBindings(args[0]) : toSqliteBindings(args);
-              const { rows, columns } = await target.execute(
-                query,
-                bindings
+      const createClient = (target, transactionScoped = false, stateTransactionContext) => {
+        const client2 = {
+          execute: async (query, ...args) => {
+            var _a, _b, _c, _d;
+            ensureOpen();
+            if (!transactionScoped && warnOnManualTransactions && !manualTransactionWarned && !hasMultipleStatements(query) && isManualTransactionControl(query)) {
+              manualTransactionWarned = true;
+              getLogger("database").warn(
+                { actorId: ctx.actorId },
+                "Manual cross-call SQLite transactions can interleave with other actor work. Use db.transaction() for coordinated transactions. Set warnOnManualTransactions: false in your db(...) configuration to disable this warning."
               );
-              return rows.map(
-                (row) => rowToObject(row, columns)
+            }
+            const kvReadsBefore = ((_a = ctx.metrics) == null ? void 0 : _a.totalKvReads) ?? 0;
+            const kvWritesBefore = ((_b = ctx.metrics) == null ? void 0 : _b.totalKvWrites) ?? 0;
+            const start = performance.now();
+            try {
+              if (args.length > 0) {
+                const bindings = args.length === 1 && isSqliteBindingObject(args[0]) ? toSqliteBindings(args[0]) : toSqliteBindings(args);
+                const { rows, columns } = await target.execute(
+                  query,
+                  bindings
+                );
+                return rows.map(
+                  (row) => rowToObject(row, columns)
+                );
+              }
+              if (!hasMultipleStatements(query)) {
+                const { rows, columns } = await target.execute(
+                  query,
+                  void 0
+                );
+                return rows.map(
+                  (row) => rowToObject(row, columns)
+                );
+              }
+              return await execMultiStatement(
+                target,
+                query
               );
+            } finally {
+              const durationMs = performance.now() - start;
+              (_c = ctx.metrics) == null ? void 0 : _c.trackSql(query, durationMs);
+              if (ctx.metrics) {
+                const kvReads = ctx.metrics.totalKvReads - kvReadsBefore;
+                const kvWrites = ctx.metrics.totalKvWrites - kvWritesBefore;
+                (_d = ctx.log) == null ? void 0 : _d.debug({
+                  msg: "sql query",
+                  query: query.slice(0, 120),
+                  durationMs,
+                  kvReads,
+                  kvWrites
+                });
+              }
             }
-            if (!hasMultipleStatements(query)) {
-              const { rows, columns } = await target.execute(
-                query,
-                void 0
+          },
+          transaction: async (callback, options) => {
+            var _a, _b;
+            validateTransactionTimeout(options == null ? void 0 : options.timeout);
+            validateTransactionName(options == null ? void 0 : options.name);
+            if (transactionScoped && ((_a = options == null ? void 0 : options.experimental) == null ? void 0 : _a.includeState)) {
+              throw new Error(
+                "experimental.includeState is not supported for nested transactions"
               );
-              return rows.map(
-                (row) => rowToObject(row, columns)
+            }
+            const includeState = ((_b = options == null ? void 0 : options.experimental) == null ? void 0 : _b.includeState) === true;
+            const stateScope = includeState && stateTransactionContext ? await stateTransactionContext.enter() : void 0;
+            try {
+              const transaction = includeState ? await (() => {
+                const beginStateTransaction = ctx.nativeDatabaseProvider && nativeStateTransactionOpeners.get(
+                  ctx.nativeDatabaseProvider
+                );
+                if (!beginStateTransaction) {
+                  throw new Error(
+                    "experimental.includeState is only supported by RivetKit's embedded database provider"
+                  );
+                }
+                return beginStateTransaction(
+                  options == null ? void 0 : options.timeout,
+                  stateScope
+                );
+              })() : await db2.beginTransaction(
+                options == null ? void 0 : options.timeout,
+                options == null ? void 0 : options.name
               );
+              const tx = createClient(transaction, true);
+              try {
+                const result = await callback(tx);
+                await transaction.commit();
+                return result;
+              } catch (error) {
+                try {
+                  await transaction.rollback();
+                } catch {
+                }
+                throw error;
+              }
+            } finally {
+              if (stateScope !== void 0) {
+                stateTransactionContext == null ? void 0 : stateTransactionContext.exit(stateScope);
+              }
             }
-            return await execMultiStatement(target, query);
-          } finally {
-            const durationMs = performance.now() - start;
-            (_c = ctx.metrics) == null ? void 0 : _c.trackSql(query, durationMs);
-            if (ctx.metrics) {
-              const kvReads = ctx.metrics.totalKvReads - kvReadsBefore;
-              const kvWrites = ctx.metrics.totalKvWrites - kvWritesBefore;
-              (_d = ctx.log) == null ? void 0 : _d.debug({
-                msg: "sql query",
-                query: query.slice(0, 120),
-                durationMs,
-                kvReads,
-                kvWrites
-              });
+          },
+          close: async () => {
+            if (!closed) {
+              closed = true;
+              await db2.close();
             }
+          },
+          nativeMetrics: () => {
+            var _a;
+            return ((_a = db2.nativeMetrics) == null ? void 0 : _a.call(db2)) ?? null;
           }
-        },
-        transaction: async (callback, options) => {
-          validateTransactionTimeout(options == null ? void 0 : options.timeout);
-          const transaction = await db2.beginTransaction(
-            options == null ? void 0 : options.timeout
+        };
+        if (!transactionScoped) {
+          nativeStateTransactionClientBinders.set(
+            client2,
+            (context) => createClient(target, false, context)
           );
-          const tx = createClient(transaction, true);
-          try {
-            const result = await callback(tx);
-            await transaction.commit();
-            return result;
-          } catch (error) {
-            try {
-              await transaction.rollback();
-            } catch {
-            }
-            throw error;
-          }
-        },
-        close: async () => {
-          if (!closed) {
-            closed = true;
-            await db2.close();
-          }
-        },
-        nativeMetrics: () => {
-          var _a;
-          return ((_a = db2.nativeMetrics) == null ? void 0 : _a.call(db2)) ?? null;
         }
-      });
+        return client2;
+      };
       const client = createClient(db2);
       return client;
     },
@@ -1056,6 +1122,7 @@ function db({
       }
     }
   };
+  return provider;
 }
 function rowToObject(row, columns) {
   const rowObj = {};
@@ -1098,7 +1165,10 @@ async function withMigrationSavepoint(client, callback) {
         throw error;
       }
     },
-    { timeout: MIGRATION_TRANSACTION_TIMEOUT_MS }
+    {
+      name: "rivetkit-migration",
+      timeout: MIGRATION_TRANSACTION_TIMEOUT_MS
+    }
   );
 }