rivetkit 2.3.5-rc.1 → 2.3.6

package/dist/tsup/mod.cjs

@@ -1,6 +1,6 @@
 "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(); } } async function _asyncNullishCoalesce(lhs, rhsFn) { if (lhs != null) { return lhs; } else { return await rhsFn(); } } var _class; var _class2; var _class3;
 
-var _chunkTR5NLR5Vcjs = require('./chunk-TR5NLR5V.cjs');
+var _chunkFNTTW25Qcjs = require('./chunk-FNTTW25Q.cjs');
 
 
 
@@ -10,9 +10,9 @@ var _chunkTR5NLR5Vcjs = require('./chunk-TR5NLR5V.cjs');
 
 
 
-var _chunkG3CCRCTKcjs = require('./chunk-G3CCRCTK.cjs');
 
 
+var _chunkHNHLBM3Ycjs = require('./chunk-HNHLBM3Y.cjs');
 
 
 
@@ -20,13 +20,15 @@ var _chunkG3CCRCTKcjs = require('./chunk-G3CCRCTK.cjs');
 
 
 
-var _chunk4XKED5P5cjs = require('./chunk-4XKED5P5.cjs');
+
+
+var _chunkAXEGEUYBcjs = require('./chunk-AXEGEUYB.cjs');
 
 
 
 var _chunkMYWZCDMLcjs = require('./chunk-MYWZCDML.cjs');
+require('./chunk-WS3PP253.cjs');
 require('./chunk-UEULFXUI.cjs');
-require('./chunk-M3FCXL3Q.cjs');
 
 
 
@@ -59,7 +61,7 @@ require('./chunk-M3FCXL3Q.cjs');
 
 
 
-var _chunk6HQCLWAWcjs = require('./chunk-6HQCLWAW.cjs');
+var _chunk234EFKN2cjs = require('./chunk-234EFKN2.cjs');
 
 
 
@@ -75,7 +77,7 @@ var _chunk4FC7TVS6cjs = require('./chunk-4FC7TVS6.cjs');
 
 // src/actor/log.ts
 function loggerWithoutContext() {
-  return _chunk6HQCLWAWcjs.getLogger.call(void 0, "actor-runtime");
+  return _chunk234EFKN2cjs.getLogger.call(void 0, "actor-runtime");
 }
 
 // src/actor/definition.ts
@@ -114,7 +116,9 @@ function actor(input) {
   warnDeprecatedOptionKeys(
     input == null ? void 0 : input.options
   );
-  const config = _chunkG3CCRCTKcjs.ActorConfigSchema.parse(input);
+  const config = _chunkHNHLBM3Ycjs.ActorConfigSchema.parse(input);
+  _chunkHNHLBM3Ycjs.flattenActionHandlers.call(void 0, config.actions);
+  _chunkHNHLBM3Ycjs.flattenActionInputSchemas.call(void 0, config.actions, config.actionInputSchemas);
   return new ActorDefinition(config);
 }
 function isStaticActorDefinition(definition) {
@@ -213,7 +217,7 @@ function validateSchemaSync(schemas, key, data) {
 var _virtualwebsocket = require('@rivetkit/virtual-websocket');
 var _ws = require('hono/ws');
 function logger() {
-  return _chunk6HQCLWAWcjs.getLogger.call(void 0, "inline-websocket-adapter");
+  return _chunk234EFKN2cjs.getLogger.call(void 0, "inline-websocket-adapter");
 }
 var InlineWebSocketAdapter = class {
   #handler;
@@ -384,7 +388,7 @@ function isLocalEngineEndpoint(endpoint) {
 
 // src/registry/log.ts
 function logger2() {
-  return _chunk6HQCLWAWcjs.getLogger.call(void 0, "registry");
+  return _chunk234EFKN2cjs.getLogger.call(void 0, "registry");
 }
 
 // src/serverless/configure.ts
@@ -430,8 +434,8 @@ async function configureServerlessPool(config) {
         );
       }
       const customConfig = config.configurePool;
-      const clientConfig = _chunk4XKED5P5cjs.convertRegistryConfigToClientConfig.call(void 0, config);
-      const dcsRes = await _chunk4XKED5P5cjs.getDatacenters.call(void 0, clientConfig);
+      const clientConfig = _chunkAXEGEUYBcjs.convertRegistryConfigToClientConfig.call(void 0, config);
+      const dcsRes = await _chunkAXEGEUYBcjs.getDatacenters.call(void 0, clientConfig);
       const poolName = _nullishCoalesce(customConfig.name, () => ( "default"));
       const serverlessToken = _nullishCoalesce(config.token, () => ( config.publicToken));
       const headers = {
@@ -453,7 +457,7 @@ async function configureServerlessPool(config) {
         metadata: _nullishCoalesce(customConfig.metadata, () => ( {})),
         drain_on_version_upgrade: _nullishCoalesce(customConfig.drainOnVersionUpgrade, () => ( true))
       };
-      await _chunk4XKED5P5cjs.updateRunnerConfig.call(void 0, clientConfig, poolName, {
+      await _chunkAXEGEUYBcjs.updateRunnerConfig.call(void 0, clientConfig, poolName, {
         datacenters: Object.fromEntries(
           dcsRes.datacenters.map((dc) => [dc.name, serverlessConfig])
         )
@@ -470,7 +474,7 @@ async function configureServerlessPool(config) {
       logger2().warn({
         msg: "serverless pool configuration attempt failed",
         attempts,
-        error: _chunk6HQCLWAWcjs.stringifyError.call(void 0, error)
+        error: _chunk234EFKN2cjs.stringifyError.call(void 0, error)
       });
       await sleep(CONFIGURE_RETRY_DELAY_MS);
     }
@@ -478,7 +482,7 @@ async function configureServerlessPool(config) {
   logger2().error({
     msg: "failed to configure serverless pool, validate endpoint is configured correctly then restart this process",
     attempts,
-    error: _chunk6HQCLWAWcjs.stringifyError.call(void 0, lastError)
+    error: _chunk234EFKN2cjs.stringifyError.call(void 0, lastError)
   });
   throw lastError;
 }
@@ -490,20 +494,20 @@ var _zod = require('zod');
 var _v4 = require('zod/v4');
 var warnedMissingVersion = false;
 var EnvoyConfigSchema = _v4.z.object({
-  poolName: _v4.z.string().default(() => _nullishCoalesce(_chunk6HQCLWAWcjs.getRivetPool.call(void 0, ), () => ( "default"))),
+  poolName: _v4.z.string().default(() => _nullishCoalesce(_chunk234EFKN2cjs.getRivetPool.call(void 0, ), () => ( "default"))),
   version: _v4.z.number().default(() => {
-    const version = _chunk6HQCLWAWcjs.getRivetEnvoyVersion.call(void 0, );
+    const version = _chunk234EFKN2cjs.getRivetEnvoyVersion.call(void 0, );
     if (version !== void 0) return version;
-    if (_chunk6HQCLWAWcjs.getNodeEnv.call(void 0, ) === "production" && !warnedMissingVersion) {
+    if (_chunk234EFKN2cjs.getNodeEnv.call(void 0, ) === "production" && !warnedMissingVersion) {
       warnedMissingVersion = true;
-      _chunk6HQCLWAWcjs.getLogger.call(void 0, "rivetkit").error(
+      _chunk234EFKN2cjs.getLogger.call(void 0, "rivetkit").error(
         "RIVET_ENVOY_VERSION is not set. Actors will not be versioned, which means they won't be drained on deploy. This is only needed when self-hosting or using a custom envoy (not needed for Rivet Compute). Set this as a build arg in your Dockerfile. See https://rivet.dev/docs/actors/versions"
       );
     }
     return 1;
   }),
   // Deprecated.
-  totalSlots: _v4.z.number().default(() => _nullishCoalesce(_chunk6HQCLWAWcjs.getRivetTotalSlots.call(void 0, ), () => ( 1e5))),
+  totalSlots: _v4.z.number().default(() => _nullishCoalesce(_chunk234EFKN2cjs.getRivetTotalSlots.call(void 0, ), () => ( 1e5))),
   envoyKey: _v4.z.string().optional()
 });
 
@@ -541,7 +545,7 @@ var ServerlessConfigSchema = _v4.z.object({
    *
    * Can also be set via RIVET_PUBLIC_ENDPOINT environment variable.
    */
-  publicEndpoint: _v4.z.string().optional().transform((val) => _nullishCoalesce(val, () => ( _chunk6HQCLWAWcjs.getRivetPublicEndpoint.call(void 0, )))),
+  publicEndpoint: _v4.z.string().optional().transform((val) => _nullishCoalesce(val, () => ( _chunk234EFKN2cjs.getRivetPublicEndpoint.call(void 0, )))),
   /**
    * Token that clients should use when connecting via the public endpoint.
    *
@@ -549,7 +553,7 @@ var ServerlessConfigSchema = _v4.z.object({
    *
    * Can also be specified in the publicEndpoint URL as `https://namespace:token@host`.
    */
-  publicToken: _v4.z.string().optional().transform((val) => _nullishCoalesce(val, () => ( _chunk6HQCLWAWcjs.getRivetPublicToken.call(void 0, ))))
+  publicToken: _v4.z.string().optional().transform((val) => _nullishCoalesce(val, () => ( _chunk234EFKN2cjs.getRivetPublicToken.call(void 0, ))))
   // There is no publicNamespace config option because the frontend and backend
   // cannot use different namespaces. The namespace is extracted from the
   // publicEndpoint URL auth syntax if provided.
@@ -603,7 +607,7 @@ var RegistryConfigSchema = _zod.z.object({
    * Runtime binding to use for RivetKit core.
    * */
   runtime: RuntimeKindSchema.optional().transform((val, ctx) => {
-    const rawRuntime = _nullishCoalesce(val, () => ( _chunk6HQCLWAWcjs.getRivetkitRuntime.call(void 0, )));
+    const rawRuntime = _nullishCoalesce(val, () => ( _chunk234EFKN2cjs.getRivetkitRuntime.call(void 0, )));
     if (rawRuntime === void 0) {
       return "auto";
     }
@@ -640,7 +644,7 @@ var RegistryConfigSchema = _zod.z.object({
    * */
   logging: _zod.z.object({
     baseLogger: _zod.z.custom().optional(),
-    level: _chunk6HQCLWAWcjs.LogLevelSchema.optional()
+    level: _chunk234EFKN2cjs.LogLevelSchema.optional()
   }).optional().default(() => ({})),
   // MARK: Routing
   // // This is a function to allow for lazy configuration of upgradeWebSocket on the
@@ -658,9 +662,9 @@ var RegistryConfigSchema = _zod.z.object({
    *
    * Can also be set via RIVET_ENDPOINT environment variables.
    */
-  endpoint: _zod.z.string().optional().transform((val) => _nullishCoalesce(_nullishCoalesce(val, () => ( _chunk6HQCLWAWcjs.getRivetEngine.call(void 0, ))), () => ( _chunk6HQCLWAWcjs.getRivetEndpoint.call(void 0, )))),
-  token: _zod.z.string().optional().transform((val) => _nullishCoalesce(val, () => ( _chunk6HQCLWAWcjs.getRivetToken.call(void 0, )))),
-  namespace: _zod.z.string().optional().transform((val) => _nullishCoalesce(val, () => ( _chunk6HQCLWAWcjs.getRivetNamespace.call(void 0, )))),
+  endpoint: _zod.z.string().optional().transform((val) => _nullishCoalesce(_nullishCoalesce(val, () => ( _chunk234EFKN2cjs.getRivetEngine.call(void 0, ))), () => ( _chunk234EFKN2cjs.getRivetEndpoint.call(void 0, )))),
+  token: _zod.z.string().optional().transform((val) => _nullishCoalesce(val, () => ( _chunk234EFKN2cjs.getRivetToken.call(void 0, )))),
+  namespace: _zod.z.string().optional().transform((val) => _nullishCoalesce(val, () => ( _chunk234EFKN2cjs.getRivetNamespace.call(void 0, )))),
   headers: _zod.z.record(_zod.z.string(), _zod.z.string()).optional().default({}),
   // MARK: Client
   // TODO:
@@ -700,21 +704,21 @@ var RegistryConfigSchema = _zod.z.object({
    *
    * Starts the full Rust engine process locally.
    */
-  startEngine: _zod.z.boolean().default(() => _chunk6HQCLWAWcjs.getRivetRunEngine.call(void 0, )),
+  startEngine: _zod.z.boolean().default(() => _chunk234EFKN2cjs.getRivetRunEngine.call(void 0, )),
   /**
    * @experimental
    *
    * Host to bind the spawned local engine process to.
    */
-  engineHost: _zod.z.string().optional().default(() => _nullishCoalesce(_chunk6HQCLWAWcjs.getRivetRunEngineHost.call(void 0, ), () => ( ENGINE_HOST))),
+  engineHost: _zod.z.string().optional().default(() => _nullishCoalesce(_chunk234EFKN2cjs.getRivetRunEngineHost.call(void 0, ), () => ( ENGINE_HOST))),
   /**
    * @experimental
    *
    * Port to bind the spawned local engine process to.
    */
-  enginePort: _zod.z.number().int().min(1).max(65535).optional().default(() => _nullishCoalesce(_chunk6HQCLWAWcjs.getRivetRunEnginePort.call(void 0, ), () => ( ENGINE_PORT))),
+  enginePort: _zod.z.number().int().min(1).max(65535).optional().default(() => _nullishCoalesce(_chunk234EFKN2cjs.getRivetRunEnginePort.call(void 0, ), () => ( ENGINE_PORT))),
   /** @experimental */
-  engineVersion: _zod.z.string().optional().default(() => _nullishCoalesce(_chunk6HQCLWAWcjs.getRivetRunEngineVersion.call(void 0, ), () => ( _chunk6HQCLWAWcjs.VERSION))),
+  engineVersion: _zod.z.string().optional().default(() => _nullishCoalesce(_chunk234EFKN2cjs.getRivetRunEngineVersion.call(void 0, ), () => ( _chunk234EFKN2cjs.VERSION))),
   /**
    * @experimental
    *
@@ -762,7 +766,7 @@ var RegistryConfigSchema = _zod.z.object({
   }))
 }).transform((config, ctx) => {
   var _a, _b, _c;
-  const isDevEnv = _chunk6HQCLWAWcjs.isDev.call(void 0, );
+  const isDevEnv = _chunk234EFKN2cjs.isDev.call(void 0, );
   const sqliteBackend = _nullishCoalesce(((_a = config.sqlite) == null ? void 0 : _a.backend), () => ( ((_b = config.test) == null ? void 0 : _b.sqliteBackend)));
   if (config.runtime === "wasm" && sqliteBackend === "local") {
     ctx.addIssue({
@@ -772,7 +776,7 @@ var RegistryConfigSchema = _zod.z.object({
     });
   }
   const sqlite = config.runtime === "wasm" && config.sqlite === void 0 ? { backend: "remote" } : config.sqlite;
-  const parsedEndpoint = config.endpoint ? _chunk4XKED5P5cjs.tryParseEndpoint.call(void 0, ctx, {
+  const parsedEndpoint = config.endpoint ? _chunkAXEGEUYBcjs.tryParseEndpoint.call(void 0, ctx, {
     endpoint: config.endpoint,
     path: ["endpoint"],
     namespace: config.namespace,
@@ -794,13 +798,13 @@ var RegistryConfigSchema = _zod.z.object({
     config.engineHost,
     config.enginePort
   );
-  const endpoint = config.startEngine ? localEngineEndpoint : _nullishCoalesce((parsedEndpoint == null ? void 0 : parsedEndpoint.endpoint), () => ( (isDevEnv ? buildEngineEndpoint(ENGINE_HOST, ENGINE_PORT) : void 0)));
+  const endpoint = config.startEngine ? localEngineEndpoint : _nullishCoalesce((parsedEndpoint == null ? void 0 : parsedEndpoint.endpoint), () => ( (isDevEnv ? localEngineEndpoint : void 0)));
   const validateServerlessEndpoint = Boolean(
     config.startEngine || parsedEndpoint
   );
   const namespace = _nullishCoalesce(_nullishCoalesce((parsedEndpoint == null ? void 0 : parsedEndpoint.namespace), () => ( config.namespace)), () => ( "default"));
   const token = _nullishCoalesce((parsedEndpoint == null ? void 0 : parsedEndpoint.token), () => ( config.token));
-  const parsedPublicEndpoint = config.serverless.publicEndpoint ? _chunk4XKED5P5cjs.tryParseEndpoint.call(void 0, ctx, {
+  const parsedPublicEndpoint = config.serverless.publicEndpoint ? _chunkAXEGEUYBcjs.tryParseEndpoint.call(void 0, ctx, {
     endpoint: config.serverless.publicEndpoint,
     path: ["serverless", "publicEndpoint"]
   }) : void 0;
@@ -811,7 +815,7 @@ var RegistryConfigSchema = _zod.z.object({
       path: ["serverless", "publicEndpoint"]
     });
   }
-  const publicEndpoint = _nullishCoalesce((parsedPublicEndpoint == null ? void 0 : parsedPublicEndpoint.endpoint), () => ( (isDevEnv && config.startEngine ? endpoint : void 0)));
+  const publicEndpoint = _nullishCoalesce((parsedPublicEndpoint == null ? void 0 : parsedPublicEndpoint.endpoint), () => ( (isDevEnv && !parsedEndpoint ? endpoint : void 0)));
   const publicNamespace = parsedPublicEndpoint == null ? void 0 : parsedPublicEndpoint.namespace;
   const publicToken = _nullishCoalesce((parsedPublicEndpoint == null ? void 0 : parsedPublicEndpoint.token), () => ( config.serverless.publicToken));
   return {
@@ -835,7 +839,7 @@ function buildActorNames(config) {
     Object.keys(config.use).map((actorName) => {
       const definition = config.use[actorName];
       const options = _nullishCoalesce(definition.config.options, () => ( {}));
-      const runMeta = _chunkG3CCRCTKcjs.getRunMetadata.call(void 0, definition.config.run);
+      const runMeta = _chunkHNHLBM3Ycjs.getRunMetadata.call(void 0, definition.config.run);
       const metadata = {};
       metadata.icon = _nullishCoalesce(options.icon, () => ( runMeta.icon));
       metadata.name = _nullishCoalesce(options.name, () => ( runMeta.name));
@@ -903,7 +907,7 @@ var DocRegistryConfigSchema = _zod.z.object({
   noWelcome: _zod.z.boolean().optional().describe("Disable the welcome message on startup. Default: false"),
   sqlite: DocSqliteConfigSchema,
   logging: _zod.z.object({
-    level: _chunk6HQCLWAWcjs.LogLevelSchema.optional().describe(
+    level: _chunk234EFKN2cjs.LogLevelSchema.optional().describe(
       "Log level for RivetKit. Default: 'warn'"
     )
   }).optional().describe("Logging configuration."),
@@ -1493,6 +1497,9 @@ var NapiCoreRuntime = (_class = class {
   async serveRegistry(registry, config) {
     await asNativeRegistry(registry).serve(config);
   }
+  async waitRegistryReady(registry) {
+    await asNativeRegistry(registry).waitReady();
+  }
   async shutdownRegistry(registry) {
     await asNativeRegistry(registry).shutdown();
   }
@@ -2161,6 +2168,11 @@ var WasmCoreRuntime = (_class2 = class {
   async serveRegistry(registry, config) {
     await callWasm(() => asWasmRegistry(registry).serve(config));
   }
+  async waitRegistryReady() {
+    throw new Error(
+      "registry.startAndWait() is not supported by the WASM runtime"
+    );
+  }
   async shutdownRegistry(registry) {
     await callWasm(() => asWasmRegistry(registry).shutdown());
   }
@@ -2972,7 +2984,7 @@ async function cleanupNativeSleepRuntimeState(runtime, ctx, afterTrackedWorkDrai
     }).catch((error) => {
       logger2().warn({
         msg: "deferred native sleep cleanup failed",
-        error: _chunk6HQCLWAWcjs.stringifyError.call(void 0, error)
+        error: _chunk234EFKN2cjs.stringifyError.call(void 0, error)
       });
     });
     return;
@@ -3099,10 +3111,10 @@ function decodeValue(value) {
   if (!value || value.length === 0) {
     return void 0;
   }
-  return _chunk6HQCLWAWcjs.decodeCborCompat.call(void 0, value);
+  return _chunk234EFKN2cjs.decodeCborCompat.call(void 0, value);
 }
 function encodeValue(value) {
-  return _chunk6HQCLWAWcjs.encodeCborCompat.call(void 0, value);
+  return _chunk234EFKN2cjs.encodeCborCompat.call(void 0, value);
 }
 function normalizeArgs(value) {
   return Array.isArray(value) ? value : value === void 0 || value === null ? [] : [value];
@@ -3138,7 +3150,7 @@ function isStructuredBridgeError(error) {
   return _chunk4FC7TVS6cjs.isRivetErrorLike.call(void 0, error) && "__type" in error && (error.__type === "RivetError" || error.__type === "ActorError");
 }
 function encodeNativeCallbackError(error) {
-  const structuredError = isStructuredBridgeError(error) ? error : _chunk6HQCLWAWcjs.deconstructError.call(void 0, error, true);
+  const structuredError = isStructuredBridgeError(error) ? error : _chunk234EFKN2cjs.deconstructError.call(void 0, error, true);
   const bridgeError = new Error(_chunk4FC7TVS6cjs.encodeBridgeRivetError.call(void 0, structuredError), {
     cause: error instanceof Error ? error : void 0
   });
@@ -3188,7 +3200,7 @@ function decodeWorkflowCbor(data) {
     return null;
   }
   try {
-    return _chunk6HQCLWAWcjs.decodeCborCompat.call(void 0, new Uint8Array(data));
+    return _chunk234EFKN2cjs.decodeCborCompat.call(void 0, new Uint8Array(data));
   } catch (e5) {
     return null;
   }
@@ -3301,7 +3313,7 @@ function serializeWorkflowHistoryForJson(data) {
   if (data === null) {
     return null;
   }
-  const history = _chunkTR5NLR5Vcjs.decodeWorkflowHistoryTransport.call(void 0, data);
+  const history = _chunkFNTTW25Qcjs.decodeWorkflowHistoryTransport.call(void 0, data);
   return jsonSafe({
     nameRegistry: [...history.nameRegistry],
     entries: history.entries.map((entry) => ({
@@ -3414,7 +3426,7 @@ var NativeConnAdapter = class {
     this.#schemas = schemas;
     this.#ctx = ctx;
     this.#queueHibernationRemoval = queueHibernationRemoval;
-    this[_chunkG3CCRCTKcjs.CONN_STATE_MANAGER_SYMBOL] = {
+    this[_chunkHNHLBM3Ycjs.CONN_STATE_MANAGER_SYMBOL] = {
       stateEnabled: true,
       get state() {
         return thisConn.state;
@@ -3431,7 +3443,7 @@ var NativeConnAdapter = class {
       decodeValue(this.#runtime.connParams(this.#conn))
     );
   }
-  [_chunkG3CCRCTKcjs.RAW_STATE_SYMBOL]() {
+  [_chunkHNHLBM3Ycjs.RAW_STATE_SYMBOL]() {
     return this.#readState();
   }
   get state() {
@@ -3442,12 +3454,12 @@ var NativeConnAdapter = class {
         this.#writeState(nextValue, { writeNative: true });
       },
       (newValue) => {
-        _chunk6HQCLWAWcjs.assertJsonCompatValue.call(void 0, newValue);
+        _chunk234EFKN2cjs.assertJsonCompatValue.call(void 0, newValue);
       }
     );
   }
   set state(value) {
-    _chunk6HQCLWAWcjs.assertJsonCompatValue.call(void 0, value);
+    _chunk234EFKN2cjs.assertJsonCompatValue.call(void 0, value);
     this.#writeState(value, { writeNative: true });
   }
   initializeState(value) {
@@ -4436,7 +4448,7 @@ var ActorContextHandleAdapter = class {
       this.#databaseProvider = databaseProvider;
     }
     this.#request = request;
-    this[_chunkG3CCRCTKcjs.ACTOR_CONTEXT_INTERNAL_SYMBOL] = new NativeWorkflowRuntimeAdapter(
+    this[_chunkHNHLBM3Ycjs.ACTOR_CONTEXT_INTERNAL_SYMBOL] = new NativeWorkflowRuntimeAdapter(
       this
     );
   }
@@ -4472,7 +4484,7 @@ var ActorContextHandleAdapter = class {
     }
     throw databaseClientNotReadyError();
   }
-  [_chunkG3CCRCTKcjs.RAW_STATE_SYMBOL]() {
+  [_chunkHNHLBM3Ycjs.RAW_STATE_SYMBOL]() {
     if (!this.#stateEnabled) {
       throw stateNotEnabledError();
     }
@@ -4493,7 +4505,7 @@ var ActorContextHandleAdapter = class {
         },
         (newValue) => {
           this.#assertCanMutateState();
-          _chunk6HQCLWAWcjs.assertJsonCompatValue.call(void 0, newValue);
+          _chunk234EFKN2cjs.assertJsonCompatValue.call(void 0, newValue);
         }
       );
     }
@@ -4504,7 +4516,7 @@ var ActorContextHandleAdapter = class {
       throw stateNotEnabledError();
     }
     this.#assertCanMutateState();
-    _chunk6HQCLWAWcjs.assertJsonCompatValue.call(void 0, value);
+    _chunk234EFKN2cjs.assertJsonCompatValue.call(void 0, value);
     this.#writeState(value, { scheduleSave: true });
   }
   initializeState(value) {
@@ -4780,7 +4792,7 @@ var ActorContextHandleAdapter = class {
     const trackedPromise = Promise.resolve(promise).catch((error) => {
       logger2().warn({
         msg: "keepAwake promise rejected",
-        error: _chunk6HQCLWAWcjs.stringifyError.call(void 0, error)
+        error: _chunk234EFKN2cjs.stringifyError.call(void 0, error)
       });
     }).then(() => null);
     try {
@@ -5155,7 +5167,7 @@ function buildActorConfig(definition, registryConfig, runtimeKind) {
     maxOutgoingMessageSize: registryConfig.maxOutgoingMessageSize,
     preloadMaxWorkflowBytes: options.preloadMaxWorkflowBytes,
     preloadMaxConnectionsBytes: options.preloadMaxConnectionsBytes,
-    actions: Object.keys(_nullishCoalesce(config.actions, () => ( {}))).sort().map((name) => ({ name })),
+    actions: Object.keys(_chunkHNHLBM3Ycjs.flattenActionHandlers.call(void 0, config.actions)).sort().map((name) => ({ name })),
     inspectorTabs: buildInspectorTabs(config.inspector, runtimeKind)
   };
 }
@@ -5232,18 +5244,19 @@ function buildNativeFactory(runtime, registryConfig, definition) {
   var _a;
   const config = definition.config;
   const databaseProvider = config.db;
+  const actionHandlers = _chunkHNHLBM3Ycjs.flattenActionHandlers.call(void 0, config.actions);
   const schemaConfig = {
-    actionInputSchemas: config.actionInputSchemas,
+    actionInputSchemas: _chunkHNHLBM3Ycjs.flattenActionInputSchemas.call(void 0, 
+      config.actions,
+      config.actionInputSchemas
+    ),
     connParamsSchema: config.connParamsSchema,
     events: config.events,
     queues: config.queues
   };
-  const actionHandlers = Object.fromEntries(
-    Object.entries(_nullishCoalesce(config.actions, () => ( {}))).map(([name, handler]) => [name, handler])
-  );
-  const createClient = () => _chunk4XKED5P5cjs.createClientWithDriver.call(void 0, 
-    new (0, _chunk4XKED5P5cjs.RemoteEngineControlClient)(
-      _chunk4XKED5P5cjs.convertRegistryConfigToClientConfig.call(void 0, registryConfig)
+  const createClient = () => _chunkAXEGEUYBcjs.createClientWithDriver.call(void 0, 
+    new (0, _chunkAXEGEUYBcjs.RemoteEngineControlClient)(
+      _chunkAXEGEUYBcjs.convertRegistryConfigToClientConfig.call(void 0, registryConfig)
     ),
     { encoding: "bare" }
   );
@@ -5253,7 +5266,7 @@ function buildNativeFactory(runtime, registryConfig, definition) {
   ), () => ( false));
   const getNativeWorkflowInspector = (ctx) => {
     var _a2;
-    return (_a2 = _chunkG3CCRCTKcjs.getRunInspectorConfig.call(void 0, 
+    return (_a2 = _chunkHNHLBM3Ycjs.getRunInspectorConfig.call(void 0, 
       config.run,
       callNativeSync(() => runtime.actorId(ctx))
     )) == null ? void 0 : _a2.workflow;
@@ -5426,7 +5439,7 @@ function buildNativeFactory(runtime, registryConfig, definition) {
             400
           );
         }
-        const cbor = _chunk6HQCLWAWcjs.encodeCborCompat.call(void 0, 
+        const cbor = _chunk234EFKN2cjs.encodeCborCompat.call(void 0, 
           _nullishCoalesce(body.body, () => ( null))
         );
         const message = await runtime.actorQueueSend(ctx, name, cbor);
@@ -5812,7 +5825,7 @@ function buildNativeFactory(runtime, registryConfig, definition) {
         } finally {
           const actorId = callNativeSync(() => runtime.actorId(ctx));
           nativeRunHandlerActiveByActorId.delete(actorId);
-          _chunkG3CCRCTKcjs.disposeRunInspector.call(void 0, config.run, actorId);
+          _chunkHNHLBM3Ycjs.disposeRunInspector.call(void 0, config.run, actorId);
           resolveNativeDestroy(runtime, ctx);
           await actorCtx.closeDatabase();
           clearNativeRuntimeState(runtime, ctx);
@@ -6008,7 +6021,7 @@ function buildNativeFactory(runtime, registryConfig, definition) {
               new Response(null, { status: 404 })
             );
           }
-          const rawConnParams = jsRequest.headers.get(_chunk4XKED5P5cjs.HEADER_CONN_PARAMS);
+          const rawConnParams = jsRequest.headers.get(_chunkAXEGEUYBcjs.HEADER_CONN_PARAMS);
           let requestCtx;
           let conn;
           try {
@@ -6073,7 +6086,7 @@ function buildNativeFactory(runtime, registryConfig, definition) {
       }
     ) : void 0,
     run: (() => {
-      const run = _chunkG3CCRCTKcjs.getRunFunction.call(void 0, config.run);
+      const run = _chunkHNHLBM3Ycjs.getRunFunction.call(void 0, config.run);
       if (!run) {
         return void 0;
       }
@@ -6092,7 +6105,7 @@ function buildNativeFactory(runtime, registryConfig, definition) {
         }
       );
     })(),
-    getWorkflowHistory: _chunkG3CCRCTKcjs.getRunInspectorConfig.call(void 0, config.run) !== void 0 ? wrapNativeCallback(
+    getWorkflowHistory: _chunkHNHLBM3Ycjs.getRunInspectorConfig.call(void 0, config.run) !== void 0 ? wrapNativeCallback(
       async (error, payload) => {
         var _a2;
         const { ctx } = unwrapTsfnPayload(error, payload);
@@ -6100,7 +6113,7 @@ function buildNativeFactory(runtime, registryConfig, definition) {
         return history == null ? void 0 : encodeValue(history);
       }
     ) : void 0,
-    replayWorkflow: _chunkG3CCRCTKcjs.getRunInspectorConfig.call(void 0, config.run) !== void 0 ? wrapNativeCallback(
+    replayWorkflow: _chunkHNHLBM3Ycjs.getRunInspectorConfig.call(void 0, config.run) !== void 0 ? wrapNativeCallback(
       async (error, payload) => {
         const { ctx, entryId } = unwrapTsfnPayload(
           error,
@@ -6236,7 +6249,7 @@ async function buildServeConfig(config) {
     poolName: config.envoy.poolName,
     handleInspectorHttpInRuntime: true,
     serverlessBasePath: config.serverless.basePath,
-    serverlessPackageVersion: _chunk6HQCLWAWcjs.VERSION,
+    serverlessPackageVersion: _chunk234EFKN2cjs.VERSION,
     serverlessClientEndpoint: config.publicEndpoint,
     serverlessClientNamespace: config.publicNamespace,
     serverlessClientToken: config.publicToken,
@@ -6249,19 +6262,19 @@ async function buildServeConfig(config) {
   } catch (error) {
     logger2().warn({
       msg: "could not resolve a local engine binary; if a local engine must be spawned it will fail with engine.binary_unavailable \u2014 set RIVET_ENGINE_BINARY_PATH or install the @rivetkit/engine-cli platform package",
-      error: _chunk6HQCLWAWcjs.stringifyError.call(void 0, error)
+      error: _chunk234EFKN2cjs.stringifyError.call(void 0, error)
     });
   }
   serveConfig.engineHost = config.engineHost;
   serveConfig.enginePort = config.enginePort;
   if ((_a = config.test) == null ? void 0 : _a.enabled) {
-    serveConfig.inspectorTestToken = _nullishCoalesce(_chunk6HQCLWAWcjs.getEnvUniversal.call(void 0, "_RIVET_TEST_INSPECTOR_TOKEN"), () => ( "token"));
+    serveConfig.inspectorTestToken = _nullishCoalesce(_chunk234EFKN2cjs.getEnvUniversal.call(void 0, "_RIVET_TEST_INSPECTOR_TOKEN"), () => ( "token"));
   }
   return serveConfig;
 }
 async function buildRegistryWithRuntime(config, runtime) {
   var _a;
-  if (((_a = config.test) == null ? void 0 : _a.enabled) && _chunk6HQCLWAWcjs.getEnvUniversal.call(void 0, "_RIVET_TEST_INSPECTOR_TOKEN") === void 0) {
+  if (((_a = config.test) == null ? void 0 : _a.enabled) && _chunk234EFKN2cjs.getEnvUniversal.call(void 0, "_RIVET_TEST_INSPECTOR_TOKEN") === void 0) {
     trySetProcessEnv("_RIVET_TEST_INSPECTOR_TOKEN", "token");
   }
   if (runtime.kind === "napi") {
@@ -6290,6 +6303,7 @@ async function buildConfiguredRegistry(config) {
 }
 
 // src/registry/index.ts
+var REGISTRY_READY_TIMEOUT_MS = 3e4;
 function signalExitCode(signal) {
   switch (signal) {
     case "SIGINT":
@@ -6315,6 +6329,8 @@ var Registry = class {
     return RegistryConfigSchema.parse(this.#config);
   }
   #runtimeServePromise;
+  #runtimeServeLifecyclePromise;
+  #runtimeReadyPromise;
   #runtimeServeConfiguredPromise;
   #runtimeServerlessPromise;
   #configureServerlessPoolPromise;
@@ -6326,7 +6342,7 @@ var Registry = class {
     var _a;
     this.#config = config;
     if ((_a = config.logging) == null ? void 0 : _a.baseLogger) {
-      _chunk6HQCLWAWcjs.configureBaseLogger.call(void 0, config.logging.baseLogger);
+      _chunk234EFKN2cjs.configureBaseLogger.call(void 0, config.logging.baseLogger);
     }
     this.#buildConfiguredRegistry = _nullishCoalesce((deps == null ? void 0 : deps.buildConfiguredRegistry), () => ( buildConfiguredRegistry));
     this.routes = {
@@ -6550,8 +6566,8 @@ var Registry = class {
    * ```
    */
   async listen(opts = {}) {
-    const port = _nullishCoalesce(_nullishCoalesce(opts.port, () => ( _chunk6HQCLWAWcjs.parsePortEnv.call(void 0, process.env.RIVET_PORT))), () => ( 3e3));
-    const publicDir = _nullishCoalesce(opts.publicDir, () => ( _chunk6HQCLWAWcjs.getRivetkitPublicDir.call(void 0, )));
+    const port = _nullishCoalesce(_nullishCoalesce(opts.port, () => ( _chunk234EFKN2cjs.parsePortEnv.call(void 0, process.env.RIVET_PORT))), () => ( 3e3));
+    const publicDir = _nullishCoalesce(opts.publicDir, () => ( _chunk234EFKN2cjs.getRivetkitPublicDir.call(void 0, )));
     const config = this.parseConfig();
     const configuredRegistryPromise = buildConfiguredRegistry(config);
     this.#runtimeServeConfiguredPromise = configuredRegistryPromise;
@@ -6579,7 +6595,7 @@ var Registry = class {
       return jsonRouteResponse(503, {
         status: "not_started",
         runtime: "rivetkit",
-        version: _chunk6HQCLWAWcjs.VERSION
+        version: _chunk234EFKN2cjs.VERSION
       });
     }
     const { runtime, registry } = configured;
@@ -6587,7 +6603,7 @@ var Registry = class {
       return jsonRouteResponse(501, {
         status: "unsupported",
         runtime: "rivetkit",
-        version: _chunk6HQCLWAWcjs.VERSION
+        version: _chunk234EFKN2cjs.VERSION
       });
     }
     const response = await runtime.registryHealth(registry);
@@ -6661,11 +6677,16 @@ var Registry = class {
     if (!this.#runtimeServePromise) {
       const configuredRegistryPromise = this.#buildConfiguredRegistry(config);
       this.#runtimeServeConfiguredPromise = configuredRegistryPromise;
-      this.#runtimeServePromise = configuredRegistryPromise.then(async ({ runtime, registry, serveConfig }) => {
-        await runtime.serveRegistry(registry, serveConfig);
-      }).catch((error) => {
-        logger2().warn({ error }, "runtime registry serve errored");
-      });
+      this.#runtimeServeLifecyclePromise = configuredRegistryPromise.then(
+        async ({ runtime, registry, serveConfig }) => {
+          await runtime.serveRegistry(registry, serveConfig);
+        }
+      );
+      this.#runtimeServePromise = this.#runtimeServeLifecyclePromise.catch(
+        (error) => {
+          logger2().warn({ error }, "runtime registry serve errored");
+        }
+      );
       this.#installSignalHandlers(config);
     }
     if (printWelcome) {
@@ -6811,6 +6832,75 @@ var Registry = class {
   startEnvoy() {
     this.#startEnvoy(this.parseConfig(), true);
   }
+  /**
+   * 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() {
+    if (this.#shutdownInFlight !== null) {
+      return Promise.reject(
+        new Error("registry.startAndWait() cannot run after shutdown has begun")
+      );
+    }
+    if (this.#runtimeReadyPromise) return this.#runtimeReadyPromise;
+    if (_chunk234EFKN2cjs.getRivetkitRuntimeMode.call(void 0, ) === "serverless") {
+      return Promise.reject(
+        new Error(
+          "registry.startAndWait() requires envoy runtime mode; serverless registries become ready per request"
+        )
+      );
+    }
+    const config = this.parseConfig();
+    if (config.runtime === "wasm") {
+      return Promise.reject(
+        new Error(
+          "registry.startAndWait() requires the native runtime; WebAssembly registries do not host an envoy"
+        )
+      );
+    }
+    this.#startEnvoy(config, true);
+    const configuredRegistryPromise = this.#runtimeServeConfiguredPromise;
+    const serveLifecyclePromise = this.#runtimeServeLifecyclePromise;
+    if (!configuredRegistryPromise || !serveLifecyclePromise) {
+      throw new Error("registry envoy startup did not initialize");
+    }
+    let timeout;
+    const timeoutPromise = new Promise((_resolve, reject) => {
+      var _a;
+      timeout = setTimeout(
+        () => reject(
+          new Error(
+            `RivetKit registry did not register with the Engine within ${REGISTRY_READY_TIMEOUT_MS}ms`
+          )
+        ),
+        REGISTRY_READY_TIMEOUT_MS
+      );
+      (_a = timeout.unref) == null ? void 0 : _a.call(timeout);
+    });
+    const readinessPromise = (async () => {
+      const { runtime, registry } = await configuredRegistryPromise;
+      const stoppedBeforeReady = serveLifecyclePromise.then(() => {
+        throw new Error("RivetKit registry stopped before becoming ready");
+      });
+      await Promise.race([
+        runtime.waitRegistryReady(registry),
+        stoppedBeforeReady
+      ]);
+    })();
+    this.#runtimeReadyPromise = Promise.race([
+      readinessPromise,
+      timeoutPromise
+    ]).finally(() => {
+      if (timeout !== void 0) clearTimeout(timeout);
+    });
+    this.#runtimeReadyPromise.catch(() => {
+    });
+    return this.#runtimeReadyPromise;
+  }
   /**
    * Starts the actor envoy for standalone server deployments.
    *
@@ -6825,8 +6915,8 @@ var Registry = class {
    * ```
    */
   start() {
-    if (_chunk6HQCLWAWcjs.getRivetkitRuntimeMode.call(void 0, ) === "serverless") {
-      const publicDir = _nullishCoalesce(_chunk6HQCLWAWcjs.getRivetkitPublicDir.call(void 0, ), () => ( "/public"));
+    if (_chunk234EFKN2cjs.getRivetkitRuntimeMode.call(void 0, ) === "serverless") {
+      const publicDir = _nullishCoalesce(_chunk234EFKN2cjs.getRivetkitPublicDir.call(void 0, ), () => ( "/public"));
       this.listen({ publicDir }).catch((error) => {
         logger2().error({ error }, "auto-listen failed; exiting");
         if (typeof process !== "undefined" && typeof process.exit === "function") {
@@ -6847,7 +6937,7 @@ var Registry = class {
     };
     console.log();
     console.log(
-      `  RivetKit ${_chunk6HQCLWAWcjs.VERSION} (Engine - ${kind === "serverless" ? "Serverless" : "Serverful"})`
+      `  RivetKit ${_chunk234EFKN2cjs.VERSION} (Engine - ${kind === "serverless" ? "Serverless" : "Serverful"})`
     );
     if (config.namespace !== "default") {
       logLine("Namespace", config.namespace);
@@ -6922,5 +7012,5 @@ function setup(input) {
 
 
 
-exports.ALLOWED_PUBLIC_HEADERS = _chunk4XKED5P5cjs.ALLOWED_PUBLIC_HEADERS; exports.ActorDefinition = ActorDefinition; exports.ActorError = _chunk4FC7TVS6cjs.RivetError; exports.ActorsSchema = ActorsSchema; exports.DocConfigurePoolSchema = DocConfigurePoolSchema; exports.DocEnvoyConfigSchema = DocEnvoyConfigSchema; exports.DocRegistryConfigSchema = DocRegistryConfigSchema; exports.DocServerlessConfigSchema = DocServerlessConfigSchema; exports.DocSqliteConfigSchema = DocSqliteConfigSchema; exports.InlineWebSocketAdapter = InlineWebSocketAdapter; exports.Registry = Registry; exports.RegistryConfigSchema = RegistryConfigSchema; exports.RivetError = _chunk4FC7TVS6cjs.RivetError; exports.RuntimeKindSchema = RuntimeKindSchema; exports.SqliteBackendSchema = SqliteBackendSchema; exports.SqliteConfigSchema = SqliteConfigSchema; exports.TestConfigSchema = TestConfigSchema; exports.UserError = _chunk4FC7TVS6cjs.UserError; exports.WasmRuntimeConfigSchema = WasmRuntimeConfigSchema; exports.actor = actor; exports.buildActorNames = buildActorNames; exports.createClientWithDriver = _chunk4XKED5P5cjs.createClientWithDriver; exports.event = event; exports.isStaticActorDefinition = isStaticActorDefinition; exports.isStaticActorInstance = isStaticActorInstance; exports.lookupInRegistry = lookupInRegistry; exports.noopNext = _chunk6HQCLWAWcjs.noopNext; exports.queue = queue; exports.setup = setup; exports.toUint8Array = _chunk6HQCLWAWcjs.toUint8Array;
+exports.ALLOWED_PUBLIC_HEADERS = _chunkAXEGEUYBcjs.ALLOWED_PUBLIC_HEADERS; exports.ActorDefinition = ActorDefinition; exports.ActorError = _chunk4FC7TVS6cjs.RivetError; exports.ActorsSchema = ActorsSchema; exports.DocConfigurePoolSchema = DocConfigurePoolSchema; exports.DocEnvoyConfigSchema = DocEnvoyConfigSchema; exports.DocRegistryConfigSchema = DocRegistryConfigSchema; exports.DocServerlessConfigSchema = DocServerlessConfigSchema; exports.DocSqliteConfigSchema = DocSqliteConfigSchema; exports.InlineWebSocketAdapter = InlineWebSocketAdapter; exports.Registry = Registry; exports.RegistryConfigSchema = RegistryConfigSchema; exports.RivetError = _chunk4FC7TVS6cjs.RivetError; exports.RuntimeKindSchema = RuntimeKindSchema; exports.SqliteBackendSchema = SqliteBackendSchema; exports.SqliteConfigSchema = SqliteConfigSchema; exports.TestConfigSchema = TestConfigSchema; exports.UserError = _chunk4FC7TVS6cjs.UserError; exports.WasmRuntimeConfigSchema = WasmRuntimeConfigSchema; exports.actor = actor; exports.buildActorNames = buildActorNames; exports.createClientWithDriver = _chunkAXEGEUYBcjs.createClientWithDriver; exports.event = event; exports.isStaticActorDefinition = isStaticActorDefinition; exports.isStaticActorInstance = isStaticActorInstance; exports.lookupInRegistry = lookupInRegistry; exports.noopNext = _chunk234EFKN2cjs.noopNext; exports.queue = queue; exports.setup = setup; exports.toUint8Array = _chunk234EFKN2cjs.toUint8Array;
 //# sourceMappingURL=mod.cjs.map