rivetkit 2.3.5-rc.1 → 2.3.6

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

@@ -1,10 +1,108 @@
 "use strict";Object.defineProperty(exports, "__esModule", {value: true}); function _interopRequireWildcard(obj) { if (obj && obj.__esModule) { return obj; } else { var newObj = {}; if (obj != null) { for (var key in obj) { if (Object.prototype.hasOwnProperty.call(obj, key)) { newObj[key] = obj[key]; } } } newObj.default = obj; return newObj; } } function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; } function _nullishCoalesce(lhs, rhsFn) { if (lhs != null) { return lhs; } else { return rhsFn(); } }// src/agent-os/actor/index.ts
 var _agentoscore = require('@rivet-dev/agent-os-core');
 
+// src/actor/actions.ts
+function flattenActionHandlers(actions) {
+  const flattened = /* @__PURE__ */ Object.create(null);
+  for (const { name, handler } of collectActionEntries(actions)) {
+    flattened[name] = handler;
+  }
+  return flattened;
+}
+function flattenActionInputSchemas(actions, schemas) {
+  if (schemas === void 0) return void 0;
+  if (!isRecord(schemas)) {
+    throw new TypeError("actionInputSchemas must be an object");
+  }
+  const flattened = /* @__PURE__ */ Object.create(null);
+  for (const { name, path } of collectActionEntries(actions)) {
+    const nestedSchema = lookupNestedSchema(schemas, path);
+    const flatSchema = schemas[name];
+    if (nestedSchema !== void 0 && flatSchema !== void 0 && nestedSchema !== flatSchema) {
+      throw new TypeError(
+        `Action input schema \`${name}\` is defined by both a nested path and a dotted key`
+      );
+    }
+    const schema = _nullishCoalesce(nestedSchema, () => ( flatSchema));
+    if (schema !== void 0) {
+      flattened[name] = schema;
+    }
+  }
+  return flattened;
+}
+function collectActionEntries(actions) {
+  const entries = [];
+  const names = /* @__PURE__ */ new Set();
+  visitActionGroup(_nullishCoalesce(actions, () => ( {})), [], entries, names);
+  return entries;
+}
+function visitActionGroup(value, path, entries, names) {
+  if (!isRecord(value)) {
+    throw new TypeError(
+      `${formatActionPath(path)} must be an action handler or group`
+    );
+  }
+  for (const [segment, child] of Object.entries(value)) {
+    const childPath = [...path, segment];
+    if (typeof child === "function") {
+      const name = childPath.join(".");
+      if (names.has(name)) {
+        throw new TypeError(
+          `Multiple action definitions flatten to \`${name}\``
+        );
+      }
+      names.add(name);
+      entries.push({
+        name,
+        path: childPath,
+        handler: child
+      });
+    } else {
+      visitActionGroup(child, childPath, entries, names);
+    }
+  }
+}
+function lookupNestedSchema(schemas, path) {
+  let value = schemas;
+  for (const segment of path) {
+    if (!isRecord(value) || !Object.hasOwn(value, segment)) {
+      return void 0;
+    }
+    value = value[segment];
+  }
+  return value;
+}
+function isRecord(value) {
+  if (typeof value !== "object" || value === null || Array.isArray(value)) {
+    return false;
+  }
+  const prototype = Object.getPrototypeOf(value);
+  return prototype === Object.prototype || prototype === null;
+}
+function formatActionPath(path) {
+  return path.length === 0 ? "actions" : `Action \`${path.join(".")}\``;
+}
+
 // src/actor/config.ts
 var _v4 = require('zod/v4');
 var DEFAULT_SLEEP_GRACE_PERIOD = 15e3;
 var zFunction = () => _v4.z.custom((val) => typeof val === "function");
+var zActionTree = _v4.z.custom((value) => {
+  if (typeof value !== "object" || value === null || Array.isArray(value)) {
+    return false;
+  }
+  const prototype = Object.getPrototypeOf(value);
+  return prototype === Object.prototype || prototype === null;
+}).superRefine((actions, ctx) => {
+  try {
+    flattenActionHandlers(actions);
+  } catch (error) {
+    ctx.addIssue({
+      code: "custom",
+      message: error instanceof Error ? error.message : "Invalid action definition"
+    });
+  }
+});
 var WorkflowInspectorConfigSchema = _v4.z.object({
   getHistory: zFunction(),
   onHistoryUpdated: zFunction().optional(),
@@ -141,7 +239,7 @@ var ActorConfigSchema = _v4.z.object({
   onBeforeActionResponse: zFunction().optional(),
   onRequest: zFunction().optional(),
   onWebSocket: zFunction().optional(),
-  actions: _v4.z.record(_v4.z.string(), zFunction()).default(() => ({})),
+  actions: zActionTree.default(() => ({})),
   actionInputSchemas: _v4.z.record(_v4.z.string(), _v4.z.any()).optional(),
   connParamsSchema: _v4.z.any().optional(),
   events: _v4.z.record(_v4.z.string(), _v4.z.any()).optional(),
@@ -285,10 +383,10 @@ var DocActorConfigSchema = _v4.z.object({
     "Called for raw WebSocket connections to /actors/{name}/websocket/* endpoints."
   ),
   actions: _v4.z.record(_v4.z.string(), _v4.z.unknown()).optional().describe(
-    "Map of action name to handler function. Defaults to an empty object."
+    "Tree of action names or nested groups to handler functions. Nested paths use dot-separated low-level action names. Defaults to an empty object."
   ),
   actionInputSchemas: _v4.z.record(_v4.z.string(), _v4.z.unknown()).optional().describe(
-    "Optional schema map for validating action argument tuples in native runtimes."
+    "Optional schemas for validating action argument tuples in native runtimes. May mirror nested action groups or use dot-separated low-level action names."
   ),
   connParamsSchema: _v4.z.unknown().optional().describe(
     "Optional schema for validating connection params in native runtimes."
@@ -319,7 +417,7 @@ function isRivetErrorCode(error, group, code) {
 // package.json
 var package_default = {
   name: "rivetkit",
-  version: "2.3.5-rc.1",
+  version: "2.3.6",
   description: "Lightweight libraries for building stateful actors on edge platforms",
   license: "Apache-2.0",
   keywords: [
@@ -391,6 +489,16 @@ var package_default = {
         default: "./dist/tsup/db/drizzle.cjs"
       }
     },
+    "./unstable/migrations": {
+      import: {
+        types: "./dist/tsup/unstable/migrations.d.ts",
+        default: "./dist/tsup/unstable/migrations.js"
+      },
+      require: {
+        types: "./dist/tsup/unstable/migrations.d.cts",
+        default: "./dist/tsup/unstable/migrations.cjs"
+      }
+    },
     "./dynamic": {
       import: {
         types: "./dist/tsup/dynamic/mod.d.ts",
@@ -490,7 +598,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 && 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-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: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",
@@ -769,6 +877,8 @@ function actor(input) {
     input == null ? void 0 : input.options
   );
   const config = ActorConfigSchema.parse(input);
+  flattenActionHandlers(config.actions);
+  flattenActionInputSchemas(config.actions, config.actionInputSchemas);
   return new ActorDefinition(config);
 }