rivetkit 2.3.5 → 2.3.6

package/dist/browser/client.d.ts

@@ -397,7 +397,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 {
@@ -1253,7 +1253,7 @@ declare const ActorConfigSchema: z.ZodObject<{
     onBeforeActionResponse: z.ZodOptional<z.ZodCustom<(...args: unknown[]) => unknown, (...args: unknown[]) => unknown>>;
     onRequest: z.ZodOptional<z.ZodCustom<(...args: unknown[]) => unknown, (...args: unknown[]) => unknown>>;
     onWebSocket: z.ZodOptional<z.ZodCustom<(...args: unknown[]) => unknown, (...args: unknown[]) => unknown>>;
-    actions: z.ZodDefault<z.ZodRecord<z.ZodString, z.ZodCustom<(...args: unknown[]) => unknown, (...args: unknown[]) => unknown>>>;
+    actions: z.ZodDefault<z.ZodCustom<Record<string, unknown>, Record<string, unknown>>>;
     actionInputSchemas: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodAny>>;
     connParamsSchema: z.ZodOptional<z.ZodAny>;
     events: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodAny>>;
@@ -1336,7 +1336,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>> {
     /**
@@ -1488,9 +1491,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.
      */
@@ -2027,6 +2031,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>;
@@ -2224,6 +2229,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.
      *

package/dist/browser/client.js

@@ -194,7 +194,7 @@ function noopNext() {
 // package.json
 var package_default = {
   name: "rivetkit",
-  version: "2.3.5",
+  version: "2.3.6",
   description: "Lightweight libraries for building stateful actors on edge platforms",
   license: "Apache-2.0",
   keywords: [
@@ -266,6 +266,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",
@@ -365,7 +375,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",
@@ -6812,6 +6822,23 @@ function createClientWithDriver(driver, config = {}) {
 }
 function createActorProxy(handle) {
   const methodCache = /* @__PURE__ */ new Map();
+  const actionPath = (name) => {
+    let method = methodCache.get(name);
+    if (method) return method;
+    method = new Proxy(
+      (...args) => handle.action({ name, args }),
+      {
+        get(target, prop) {
+          if (typeof prop === "symbol")
+            return Reflect.get(target, prop);
+          if (prop === "then") return void 0;
+          return actionPath(`${name}.${prop}`);
+        }
+      }
+    );
+    methodCache.set(name, method);
+    return method;
+  };
   return new Proxy(handle, {
     get(target, prop, receiver) {
       if (typeof prop === "symbol") {
@@ -6826,12 +6853,7 @@ function createActorProxy(handle) {
       }
       if (typeof prop === "string") {
         if (prop === "then") return void 0;
-        let method = methodCache.get(prop);
-        if (!method) {
-          method = (...args) => target.action({ name: prop, args });
-          methodCache.set(prop, method);
-        }
-        return method;
+        return actionPath(prop);
       }
     },
     // Support for 'in' operator
@@ -6849,6 +6871,7 @@ function createActorProxy(handle) {
     },
     // Support proper property descriptors
     getOwnPropertyDescriptor(target, prop) {
+      if (prop === "then") return void 0;
       const targetDescriptor = Reflect.getOwnPropertyDescriptor(
         target,
         prop
@@ -6861,7 +6884,7 @@ function createActorProxy(handle) {
           configurable: true,
           enumerable: false,
           writable: false,
-          value: (...args) => target.action({ name: prop, args })
+          value: actionPath(prop)
         };
       }
       return void 0;