rivetkit 2.3.5-rc.1 → 2.3.6

package/dist/tsup/agent-os/index.d.cts

@@ -860,6 +860,7 @@ interface CoreRuntime {
     createRegistry(): RegistryHandle;
     registerActor(registry: RegistryHandle, name: string, factory: ActorFactoryHandle): void;
     serveRegistry(registry: RegistryHandle, config: RuntimeServeConfig): Promise<void>;
+    waitRegistryReady(registry: RegistryHandle): Promise<void>;
     shutdownRegistry(registry: RegistryHandle): Promise<void>;
     registryActorStopThresholdMs?(registry: RegistryHandle): Promise<number | undefined>;
     handleServerlessRequest(registry: RegistryHandle, req: RuntimeServerlessRequest, onStreamEvent: RuntimeServerlessStreamCallback, cancelToken: CancellationTokenHandle, config: RuntimeServeConfig): Promise<RuntimeServerlessResponseHead>;
@@ -1057,6 +1058,15 @@ declare class Registry<A extends RegistryActors> {
      */
     shutdown(): Promise<void>;
     startEnvoy(): void;
+    /**
+     * Starts the serverful registry if needed and waits until its envoy has
+     * registered with the Engine. Repeated and concurrent calls share one
+     * startup lifecycle and readiness promise.
+     *
+     * Unlike {@link start}, this reports startup failures and does not resolve
+     * merely because an HTTP health endpoint is listening.
+     */
+    startAndWait(): Promise<void>;
     /**
      * Starts the actor envoy for standalone server deployments.
      *
@@ -1101,7 +1111,7 @@ type ActorActionFunction<Args extends Array<unknown> = unknown[], Response = unk
 type IsAny<T> = 0 extends 1 & T ? true : false;
 type LooseEventSubscribe = (eventName: string, callback: (...args: any[]) => void) => () => void;
 type ActorActionMap<R> = {
-    [K in keyof NonNullable<R>]: NonNullable<R>[K] extends (...args: infer Args) => infer Return ? ActorActionFunction<Args, Awaited<Return>> : never;
+    [K in keyof NonNullable<R>]: NonNullable<R>[K] extends (...args: infer Args) => infer Return ? ActorActionFunction<Args, Awaited<Return>> : NonNullable<R>[K] extends Record<string, unknown> ? ActorActionMap<NonNullable<R>[K]> : never;
 };
 type ActionsOf<AD extends AnyActorDefinition> = AD extends BaseActorDefinition<any, any, any, any, any, any, any, any, infer R> ? R : never;
 interface ActorGatewayOptions {
@@ -1869,7 +1879,7 @@ declare const ActorConfigSchema: z$1.ZodObject<{
     onBeforeActionResponse: z$1.ZodOptional<z$1.ZodCustom<(...args: unknown[]) => unknown, (...args: unknown[]) => unknown>>;
     onRequest: z$1.ZodOptional<z$1.ZodCustom<(...args: unknown[]) => unknown, (...args: unknown[]) => unknown>>;
     onWebSocket: z$1.ZodOptional<z$1.ZodCustom<(...args: unknown[]) => unknown, (...args: unknown[]) => unknown>>;
-    actions: z$1.ZodDefault<z$1.ZodRecord<z$1.ZodString, z$1.ZodCustom<(...args: unknown[]) => unknown, (...args: unknown[]) => unknown>>>;
+    actions: z$1.ZodDefault<z$1.ZodCustom<Record<string, unknown>, Record<string, unknown>>>;
     actionInputSchemas: z$1.ZodOptional<z$1.ZodRecord<z$1.ZodString, z$1.ZodAny>>;
     connParamsSchema: z$1.ZodOptional<z$1.ZodAny>;
     events: z$1.ZodOptional<z$1.ZodRecord<z$1.ZodString, z$1.ZodAny>>;
@@ -1952,7 +1962,10 @@ type CreateVars<TState, _TConnParams, _TConnState, TVars, TInput, TDatabase exte
     createVars: (c: CreateVarsContext<TState, TInput, TDatabase, TEvents, TQueues>, driverCtx: any) => TVars | Promise<TVars>;
 } | Record<never, never>;
 interface Actions<TState, TConnParams, TConnState, TVars, TInput, TDatabase extends AnyDatabaseProvider, TEvents extends EventSchemaConfig = Record<never, never>, TQueues extends QueueSchemaConfig = Record<never, never>> {
-    [Action: string]: (c: ActionContext<TState, TConnParams, TConnState, TVars, TInput, TDatabase, TEvents, TQueues>, ...args: any[]) => any;
+    [Action: string]: ((c: ActionContext<TState, TConnParams, TConnState, TVars, TInput, TDatabase, TEvents, TQueues>, ...args: any[]) => any) | Actions<TState, TConnParams, TConnState, TVars, TInput, TDatabase, TEvents, TQueues>;
+}
+interface ActionInputSchemas {
+    [Action: string]: PrimitiveSchema | ActionInputSchemas;
 }
 interface BaseActorConfig<TState, TConnParams, TConnState, TVars, TInput, TDatabase extends AnyDatabaseProvider, TEvents extends EventSchemaConfig, TQueues extends QueueSchemaConfig, TActions extends Actions<TState, TConnParams, TConnState, TVars, TInput, TDatabase, TEvents, TQueues>> {
     /**
@@ -2104,9 +2117,10 @@ interface BaseActorConfig<TState, TConnParams, TConnState, TVars, TInput, TDatab
     onWebSocket?: (c: WebSocketContext<TState, TConnParams, TConnState, TVars, TInput, TDatabase, TEvents, TQueues>, websocket: UniversalWebSocket) => void | Promise<void>;
     actions?: TActions;
     /**
-     * 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.
      */
-    actionInputSchemas?: Record<string, PrimitiveSchema>;
+    actionInputSchemas?: ActionInputSchemas;
     /**
      * Optional schema for validating connection params in native runtimes.
      */

package/dist/tsup/agent-os/index.d.ts

@@ -860,6 +860,7 @@ interface CoreRuntime {
     createRegistry(): RegistryHandle;
     registerActor(registry: RegistryHandle, name: string, factory: ActorFactoryHandle): void;
     serveRegistry(registry: RegistryHandle, config: RuntimeServeConfig): Promise<void>;
+    waitRegistryReady(registry: RegistryHandle): Promise<void>;
     shutdownRegistry(registry: RegistryHandle): Promise<void>;
     registryActorStopThresholdMs?(registry: RegistryHandle): Promise<number | undefined>;
     handleServerlessRequest(registry: RegistryHandle, req: RuntimeServerlessRequest, onStreamEvent: RuntimeServerlessStreamCallback, cancelToken: CancellationTokenHandle, config: RuntimeServeConfig): Promise<RuntimeServerlessResponseHead>;
@@ -1057,6 +1058,15 @@ declare class Registry<A extends RegistryActors> {
      */
     shutdown(): Promise<void>;
     startEnvoy(): void;
+    /**
+     * Starts the serverful registry if needed and waits until its envoy has
+     * registered with the Engine. Repeated and concurrent calls share one
+     * startup lifecycle and readiness promise.
+     *
+     * Unlike {@link start}, this reports startup failures and does not resolve
+     * merely because an HTTP health endpoint is listening.
+     */
+    startAndWait(): Promise<void>;
     /**
      * Starts the actor envoy for standalone server deployments.
      *
@@ -1101,7 +1111,7 @@ type ActorActionFunction<Args extends Array<unknown> = unknown[], Response = unk
 type IsAny<T> = 0 extends 1 & T ? true : false;
 type LooseEventSubscribe = (eventName: string, callback: (...args: any[]) => void) => () => void;
 type ActorActionMap<R> = {
-    [K in keyof NonNullable<R>]: NonNullable<R>[K] extends (...args: infer Args) => infer Return ? ActorActionFunction<Args, Awaited<Return>> : never;
+    [K in keyof NonNullable<R>]: NonNullable<R>[K] extends (...args: infer Args) => infer Return ? ActorActionFunction<Args, Awaited<Return>> : NonNullable<R>[K] extends Record<string, unknown> ? ActorActionMap<NonNullable<R>[K]> : never;
 };
 type ActionsOf<AD extends AnyActorDefinition> = AD extends BaseActorDefinition<any, any, any, any, any, any, any, any, infer R> ? R : never;
 interface ActorGatewayOptions {
@@ -1869,7 +1879,7 @@ declare const ActorConfigSchema: z$1.ZodObject<{
     onBeforeActionResponse: z$1.ZodOptional<z$1.ZodCustom<(...args: unknown[]) => unknown, (...args: unknown[]) => unknown>>;
     onRequest: z$1.ZodOptional<z$1.ZodCustom<(...args: unknown[]) => unknown, (...args: unknown[]) => unknown>>;
     onWebSocket: z$1.ZodOptional<z$1.ZodCustom<(...args: unknown[]) => unknown, (...args: unknown[]) => unknown>>;
-    actions: z$1.ZodDefault<z$1.ZodRecord<z$1.ZodString, z$1.ZodCustom<(...args: unknown[]) => unknown, (...args: unknown[]) => unknown>>>;
+    actions: z$1.ZodDefault<z$1.ZodCustom<Record<string, unknown>, Record<string, unknown>>>;
     actionInputSchemas: z$1.ZodOptional<z$1.ZodRecord<z$1.ZodString, z$1.ZodAny>>;
     connParamsSchema: z$1.ZodOptional<z$1.ZodAny>;
     events: z$1.ZodOptional<z$1.ZodRecord<z$1.ZodString, z$1.ZodAny>>;
@@ -1952,7 +1962,10 @@ type CreateVars<TState, _TConnParams, _TConnState, TVars, TInput, TDatabase exte
     createVars: (c: CreateVarsContext<TState, TInput, TDatabase, TEvents, TQueues>, driverCtx: any) => TVars | Promise<TVars>;
 } | Record<never, never>;
 interface Actions<TState, TConnParams, TConnState, TVars, TInput, TDatabase extends AnyDatabaseProvider, TEvents extends EventSchemaConfig = Record<never, never>, TQueues extends QueueSchemaConfig = Record<never, never>> {
-    [Action: string]: (c: ActionContext<TState, TConnParams, TConnState, TVars, TInput, TDatabase, TEvents, TQueues>, ...args: any[]) => any;
+    [Action: string]: ((c: ActionContext<TState, TConnParams, TConnState, TVars, TInput, TDatabase, TEvents, TQueues>, ...args: any[]) => any) | Actions<TState, TConnParams, TConnState, TVars, TInput, TDatabase, TEvents, TQueues>;
+}
+interface ActionInputSchemas {
+    [Action: string]: PrimitiveSchema | ActionInputSchemas;
 }
 interface BaseActorConfig<TState, TConnParams, TConnState, TVars, TInput, TDatabase extends AnyDatabaseProvider, TEvents extends EventSchemaConfig, TQueues extends QueueSchemaConfig, TActions extends Actions<TState, TConnParams, TConnState, TVars, TInput, TDatabase, TEvents, TQueues>> {
     /**
@@ -2104,9 +2117,10 @@ interface BaseActorConfig<TState, TConnParams, TConnState, TVars, TInput, TDatab
     onWebSocket?: (c: WebSocketContext<TState, TConnParams, TConnState, TVars, TInput, TDatabase, TEvents, TQueues>, websocket: UniversalWebSocket) => void | Promise<void>;
     actions?: TActions;
     /**
-     * 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.
      */
-    actionInputSchemas?: Record<string, PrimitiveSchema>;
+    actionInputSchemas?: ActionInputSchemas;
     /**
      * Optional schema for validating connection params in native runtimes.
      */

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

@@ -1,10 +1,108 @@
 // src/agent-os/actor/index.ts
 import { AgentOs, createInMemoryFileSystem } from "@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 = nestedSchema ?? flatSchema;
+    if (schema !== void 0) {
+      flattened[name] = schema;
+    }
+  }
+  return flattened;
+}
+function collectActionEntries(actions) {
+  const entries = [];
+  const names = /* @__PURE__ */ new Set();
+  visitActionGroup(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
 import { z } from "zod/v4";
 var DEFAULT_SLEEP_GRACE_PERIOD = 15e3;
 var zFunction = () => z.custom((val) => typeof val === "function");
+var zActionTree = 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 = z.object({
   getHistory: zFunction(),
   onHistoryUpdated: zFunction().optional(),
@@ -141,7 +239,7 @@ var ActorConfigSchema = z.object({
   onBeforeActionResponse: zFunction().optional(),
   onRequest: zFunction().optional(),
   onWebSocket: zFunction().optional(),
-  actions: z.record(z.string(), zFunction()).default(() => ({})),
+  actions: zActionTree.default(() => ({})),
   actionInputSchemas: z.record(z.string(), z.any()).optional(),
   connParamsSchema: z.any().optional(),
   events: z.record(z.string(), z.any()).optional(),
@@ -285,10 +383,10 @@ var DocActorConfigSchema = z.object({
     "Called for raw WebSocket connections to /actors/{name}/websocket/* endpoints."
   ),
   actions: z.record(z.string(), 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: z.record(z.string(), 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: 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);
 }