rivetkit 2.0.18 → 2.0.20

package/dist/tsup/mod.cjs

@@ -10,27 +10,27 @@
 
 
 
-var _chunkBGBEE6N2cjs = require('./chunk-BGBEE6N2.cjs');
-require('./chunk-OPB45MKH.cjs');
+var _chunk2S7HJMMYcjs = require('./chunk-2S7HJMMY.cjs');
+require('./chunk-DUOTOMP7.cjs');
 
 
 
 
 
-var _chunkIXPZJCBEcjs = require('./chunk-IXPZJCBE.cjs');
-require('./chunk-ICPFPGOJ.cjs');
-require('./chunk-F5VXILGA.cjs');
-require('./chunk-4R5LOYEU.cjs');
+var _chunkJZD6FEOEcjs = require('./chunk-JZD6FEOE.cjs');
+require('./chunk-ZODINJWN.cjs');
+require('./chunk-KKRR7DSG.cjs');
+require('./chunk-FYP3TZXD.cjs');
 
 
 
 
-var _chunkSBBFWYNMcjs = require('./chunk-SBBFWYNM.cjs');
-require('./chunk-QMF6X2E2.cjs');
+var _chunkPV22ZBDEcjs = require('./chunk-PV22ZBDE.cjs');
+require('./chunk-RJVSNJO7.cjs');
 
 
 
-var _chunkFADDMRZPcjs = require('./chunk-FADDMRZP.cjs');
+var _chunkKUZWEM23cjs = require('./chunk-KUZWEM23.cjs');
 
 
 var _chunkKSRXX3Z4cjs = require('./chunk-KSRXX3Z4.cjs');
@@ -56,5 +56,5 @@ var _chunkKSRXX3Z4cjs = require('./chunk-KSRXX3Z4.cjs');
 
 
 
-exports.ALLOWED_PUBLIC_HEADERS = _chunkSBBFWYNMcjs.ALLOWED_PUBLIC_HEADERS; exports.InlineWebSocketAdapter2 = _chunkBGBEE6N2cjs.InlineWebSocketAdapter2; exports.PATH_CONNECT_WEBSOCKET = _chunkSBBFWYNMcjs.PATH_CONNECT_WEBSOCKET; exports.PATH_RAW_WEBSOCKET_PREFIX = _chunkSBBFWYNMcjs.PATH_RAW_WEBSOCKET_PREFIX; exports.Registry = _chunkBGBEE6N2cjs.Registry; exports.RegistryConfigSchema = _chunkBGBEE6N2cjs.RegistryConfigSchema; exports.UserError = _chunkKSRXX3Z4cjs.UserError; exports.actor = _chunkBGBEE6N2cjs.actor; exports.createActorRouter = _chunkBGBEE6N2cjs.createActorRouter; exports.createClientWithDriver = _chunkIXPZJCBEcjs.createClientWithDriver; exports.createEngineDriver = _chunkBGBEE6N2cjs.createEngineDriver; exports.createFileSystemDriver = _chunkBGBEE6N2cjs.createFileSystemDriver; exports.createMemoryDriver = _chunkBGBEE6N2cjs.createMemoryDriver; exports.generateConnId = _chunkIXPZJCBEcjs.generateConnId; exports.generateConnToken = _chunkIXPZJCBEcjs.generateConnToken; exports.handleRawWebSocketHandler = _chunkBGBEE6N2cjs.handleRawWebSocketHandler; exports.handleWebSocketConnect = _chunkBGBEE6N2cjs.handleWebSocketConnect; exports.lookupInRegistry = _chunkIXPZJCBEcjs.lookupInRegistry; exports.noopNext = _chunkFADDMRZPcjs.noopNext; exports.setup = _chunkBGBEE6N2cjs.setup; exports.toUint8Array = _chunkFADDMRZPcjs.toUint8Array;
+exports.ALLOWED_PUBLIC_HEADERS = _chunkPV22ZBDEcjs.ALLOWED_PUBLIC_HEADERS; exports.InlineWebSocketAdapter2 = _chunk2S7HJMMYcjs.InlineWebSocketAdapter2; exports.PATH_CONNECT_WEBSOCKET = _chunkPV22ZBDEcjs.PATH_CONNECT_WEBSOCKET; exports.PATH_RAW_WEBSOCKET_PREFIX = _chunkPV22ZBDEcjs.PATH_RAW_WEBSOCKET_PREFIX; exports.Registry = _chunk2S7HJMMYcjs.Registry; exports.RegistryConfigSchema = _chunk2S7HJMMYcjs.RegistryConfigSchema; exports.UserError = _chunkKSRXX3Z4cjs.UserError; exports.actor = _chunk2S7HJMMYcjs.actor; exports.createActorRouter = _chunk2S7HJMMYcjs.createActorRouter; exports.createClientWithDriver = _chunkJZD6FEOEcjs.createClientWithDriver; exports.createEngineDriver = _chunk2S7HJMMYcjs.createEngineDriver; exports.createFileSystemDriver = _chunk2S7HJMMYcjs.createFileSystemDriver; exports.createMemoryDriver = _chunk2S7HJMMYcjs.createMemoryDriver; exports.generateConnId = _chunkJZD6FEOEcjs.generateConnId; exports.generateConnToken = _chunkJZD6FEOEcjs.generateConnToken; exports.handleRawWebSocketHandler = _chunk2S7HJMMYcjs.handleRawWebSocketHandler; exports.handleWebSocketConnect = _chunk2S7HJMMYcjs.handleWebSocketConnect; exports.lookupInRegistry = _chunkJZD6FEOEcjs.lookupInRegistry; exports.noopNext = _chunkKUZWEM23cjs.noopNext; exports.setup = _chunk2S7HJMMYcjs.setup; exports.toUint8Array = _chunkKUZWEM23cjs.toUint8Array;
 //# sourceMappingURL=mod.cjs.map

package/dist/tsup/mod.d.cts

@@ -1,5 +1,5 @@
-import { U as UniversalWebSocket, R as RivetEvent, a as RivetCloseEvent, b as RivetMessageEvent, A as ActorDriver, c as RunnerConfig, E as Encoding, d as AnyDatabaseProvider, e as Actions, f as ActorConfigInput, g as ActorDefinition, D as DriverConfig } from './conn-CxCsTz2s.cjs';
-export { n as ActionContext, r as ActionContextOf, z as ActorConfig, x as ActorConfigSchema, q as ActorContext, s as ActorContextOf, m as ActorKey, l as ActorQuery, w as ActorTypes, t as AnyActorDefinition, v as AnyActorInstance, j as AnyClient, y as AuthIntent, C as Client, o as Conn, p as ConnectionStatus, I as InitContext, O as OnConnectOptions, K as Registry, G as RegistryActors, F as RegistryConfig, J as RegistryConfigSchema, H as RunConfigInput, S as ServerlessActorDriverBuilder, k as createClientWithDriver, h as generateConnId, i as generateConnToken, u as lookupInRegistry, L as setup, B as test } from './conn-CxCsTz2s.cjs';
+import { U as UniversalWebSocket, R as RivetEvent, a as RivetCloseEvent, b as RivetMessageEvent, A as ActorDriver, c as RunnerConfig, E as Encoding, d as AnyDatabaseProvider, e as Actions, f as ActorConfigInput, g as ActorDefinition, D as DriverConfig } from './conn-DU5EbfCu.cjs';
+export { n as ActionContext, r as ActionContextOf, z as ActorConfig, x as ActorConfigSchema, q as ActorContext, s as ActorContextOf, m as ActorKey, l as ActorQuery, w as ActorTypes, t as AnyActorDefinition, v as AnyActorInstance, j as AnyClient, y as AuthIntent, C as Client, o as Conn, p as ConnectionStatus, I as InitContext, O as OnConnectOptions, K as Registry, G as RegistryActors, F as RegistryConfig, J as RegistryConfigSchema, H as RunConfigInput, S as ServerlessActorDriverBuilder, k as createClientWithDriver, h as generateConnId, i as generateConnToken, u as lookupInRegistry, L as setup, B as test } from './conn-DU5EbfCu.cjs';
 import { WSContext } from 'hono/ws';
 export { n as noopNext } from './utils-fwx3o3K9.cjs';
 export { toUint8Array } from './utils.cjs';

package/dist/tsup/mod.d.ts

@@ -1,5 +1,5 @@
-import { U as UniversalWebSocket, R as RivetEvent, a as RivetCloseEvent, b as RivetMessageEvent, A as ActorDriver, c as RunnerConfig, E as Encoding, d as AnyDatabaseProvider, e as Actions, f as ActorConfigInput, g as ActorDefinition, D as DriverConfig } from './conn-DhdZYlSX.js';
-export { n as ActionContext, r as ActionContextOf, z as ActorConfig, x as ActorConfigSchema, q as ActorContext, s as ActorContextOf, m as ActorKey, l as ActorQuery, w as ActorTypes, t as AnyActorDefinition, v as AnyActorInstance, j as AnyClient, y as AuthIntent, C as Client, o as Conn, p as ConnectionStatus, I as InitContext, O as OnConnectOptions, K as Registry, G as RegistryActors, F as RegistryConfig, J as RegistryConfigSchema, H as RunConfigInput, S as ServerlessActorDriverBuilder, k as createClientWithDriver, h as generateConnId, i as generateConnToken, u as lookupInRegistry, L as setup, B as test } from './conn-DhdZYlSX.js';
+import { U as UniversalWebSocket, R as RivetEvent, a as RivetCloseEvent, b as RivetMessageEvent, A as ActorDriver, c as RunnerConfig, E as Encoding, d as AnyDatabaseProvider, e as Actions, f as ActorConfigInput, g as ActorDefinition, D as DriverConfig } from './conn-CmPcqOCF.js';
+export { n as ActionContext, r as ActionContextOf, z as ActorConfig, x as ActorConfigSchema, q as ActorContext, s as ActorContextOf, m as ActorKey, l as ActorQuery, w as ActorTypes, t as AnyActorDefinition, v as AnyActorInstance, j as AnyClient, y as AuthIntent, C as Client, o as Conn, p as ConnectionStatus, I as InitContext, O as OnConnectOptions, K as Registry, G as RegistryActors, F as RegistryConfig, J as RegistryConfigSchema, H as RunConfigInput, S as ServerlessActorDriverBuilder, k as createClientWithDriver, h as generateConnId, i as generateConnToken, u as lookupInRegistry, L as setup, B as test } from './conn-CmPcqOCF.js';
 import { WSContext } from 'hono/ws';
 export { n as noopNext } from './utils-fwx3o3K9.js';
 export { toUint8Array } from './utils.js';

package/dist/tsup/mod.js

@@ -10,27 +10,27 @@ import {
   handleRawWebSocketHandler,
   handleWebSocketConnect,
   setup
-} from "./chunk-S5K7CB2F.js";
-import "./chunk-KPFRD7KN.js";
+} from "./chunk-L3YPHXPE.js";
+import "./chunk-RI4YHZZW.js";
 import {
   createClientWithDriver,
   generateConnId,
   generateConnToken,
   lookupInRegistry
-} from "./chunk-DZEQUURK.js";
-import "./chunk-YVK2XQNR.js";
-import "./chunk-2P4ENWQ6.js";
-import "./chunk-2MUSMEMN.js";
+} from "./chunk-GQ5WTE64.js";
+import "./chunk-QRUGCDA5.js";
+import "./chunk-MGHPBNWB.js";
+import "./chunk-7ACKZS3T.js";
 import {
   ALLOWED_PUBLIC_HEADERS,
   PATH_CONNECT_WEBSOCKET,
   PATH_RAW_WEBSOCKET_PREFIX
-} from "./chunk-AR5ZY563.js";
-import "./chunk-6JGEHV56.js";
+} from "./chunk-YFFCPYHY.js";
+import "./chunk-ER5OT3SQ.js";
 import {
   noopNext,
   toUint8Array
-} from "./chunk-42UD33X4.js";
+} from "./chunk-QMVCFQ37.js";
 import {
   UserError
 } from "./chunk-G64QUEDJ.js";

package/dist/tsup/test/mod.cjs

@@ -1,19 +1,19 @@
 "use strict";Object.defineProperty(exports, "__esModule", {value: true});
 
 
-var _chunkO7NI7QQWcjs = require('../chunk-O7NI7QQW.cjs');
-require('../chunk-BGBEE6N2.cjs');
-require('../chunk-OPB45MKH.cjs');
-require('../chunk-IXPZJCBE.cjs');
-require('../chunk-ICPFPGOJ.cjs');
-require('../chunk-F5VXILGA.cjs');
-require('../chunk-4R5LOYEU.cjs');
-require('../chunk-SBBFWYNM.cjs');
-require('../chunk-QMF6X2E2.cjs');
-require('../chunk-FADDMRZP.cjs');
+var _chunk2I6L3VROcjs = require('../chunk-2I6L3VRO.cjs');
+require('../chunk-2S7HJMMY.cjs');
+require('../chunk-DUOTOMP7.cjs');
+require('../chunk-JZD6FEOE.cjs');
+require('../chunk-ZODINJWN.cjs');
+require('../chunk-KKRR7DSG.cjs');
+require('../chunk-FYP3TZXD.cjs');
+require('../chunk-PV22ZBDE.cjs');
+require('../chunk-RJVSNJO7.cjs');
+require('../chunk-KUZWEM23.cjs');
 require('../chunk-KSRXX3Z4.cjs');
 
 
 
-exports.getPort = _chunkO7NI7QQWcjs.getPort; exports.setupTest = _chunkO7NI7QQWcjs.setupTest;
+exports.getPort = _chunk2I6L3VROcjs.getPort; exports.setupTest = _chunk2I6L3VROcjs.setupTest;
 //# sourceMappingURL=mod.cjs.map

package/dist/tsup/test/mod.d.cts

@@ -1,5 +1,5 @@
 import { TestContext } from 'vitest';
-import { K as Registry, C as Client } from '../conn-CxCsTz2s.cjs';
+import { K as Registry, C as Client } from '../conn-DU5EbfCu.cjs';
 import 'hono/streaming';
 import 'hono/ws';
 import '../utils.cjs';

package/dist/tsup/test/mod.d.ts

@@ -1,5 +1,5 @@
 import { TestContext } from 'vitest';
-import { K as Registry, C as Client } from '../conn-DhdZYlSX.js';
+import { K as Registry, C as Client } from '../conn-CmPcqOCF.js';
 import 'hono/streaming';
 import 'hono/ws';
 import '../utils.js';

package/dist/tsup/test/mod.js

@@ -1,16 +1,16 @@
 import {
   getPort,
   setupTest
-} from "../chunk-JMSXK5ZO.js";
-import "../chunk-S5K7CB2F.js";
-import "../chunk-KPFRD7KN.js";
-import "../chunk-DZEQUURK.js";
-import "../chunk-YVK2XQNR.js";
-import "../chunk-2P4ENWQ6.js";
-import "../chunk-2MUSMEMN.js";
-import "../chunk-AR5ZY563.js";
-import "../chunk-6JGEHV56.js";
-import "../chunk-42UD33X4.js";
+} from "../chunk-5F6X4AFU.js";
+import "../chunk-L3YPHXPE.js";
+import "../chunk-RI4YHZZW.js";
+import "../chunk-GQ5WTE64.js";
+import "../chunk-QRUGCDA5.js";
+import "../chunk-MGHPBNWB.js";
+import "../chunk-7ACKZS3T.js";
+import "../chunk-YFFCPYHY.js";
+import "../chunk-ER5OT3SQ.js";
+import "../chunk-QMVCFQ37.js";
 import "../chunk-G64QUEDJ.js";
 export {
   getPort,

package/dist/tsup/utils.cjs

@@ -11,7 +11,7 @@
 
 
 
-var _chunkFADDMRZPcjs = require('./chunk-FADDMRZP.cjs');
+var _chunkKUZWEM23cjs = require('./chunk-KUZWEM23.cjs');
 require('./chunk-KSRXX3Z4.cjs');
 
 
@@ -26,5 +26,5 @@ require('./chunk-KSRXX3Z4.cjs');
 
 
 
-exports.SinglePromiseQueue = _chunkFADDMRZPcjs.SinglePromiseQueue; exports.VERSION = _chunkFADDMRZPcjs.VERSION; exports.assertUnreachable = _chunkFADDMRZPcjs.assertUnreachable; exports.bufferToArrayBuffer = _chunkFADDMRZPcjs.bufferToArrayBuffer; exports.combineUrlPath = _chunkFADDMRZPcjs.combineUrlPath; exports.dbg = _chunkFADDMRZPcjs.dbg; exports.getEnvUniversal = _chunkFADDMRZPcjs.getEnvUniversal; exports.httpUserAgent = _chunkFADDMRZPcjs.httpUserAgent; exports.promiseWithResolvers = _chunkFADDMRZPcjs.promiseWithResolvers; exports.setLongTimeout = _chunkFADDMRZPcjs.setLongTimeout; exports.stringifyError = _chunkFADDMRZPcjs.stringifyError; exports.toUint8Array = _chunkFADDMRZPcjs.toUint8Array;
+exports.SinglePromiseQueue = _chunkKUZWEM23cjs.SinglePromiseQueue; exports.VERSION = _chunkKUZWEM23cjs.VERSION; exports.assertUnreachable = _chunkKUZWEM23cjs.assertUnreachable; exports.bufferToArrayBuffer = _chunkKUZWEM23cjs.bufferToArrayBuffer; exports.combineUrlPath = _chunkKUZWEM23cjs.combineUrlPath; exports.dbg = _chunkKUZWEM23cjs.dbg; exports.getEnvUniversal = _chunkKUZWEM23cjs.getEnvUniversal; exports.httpUserAgent = _chunkKUZWEM23cjs.httpUserAgent; exports.promiseWithResolvers = _chunkKUZWEM23cjs.promiseWithResolvers; exports.setLongTimeout = _chunkKUZWEM23cjs.setLongTimeout; exports.stringifyError = _chunkKUZWEM23cjs.stringifyError; exports.toUint8Array = _chunkKUZWEM23cjs.toUint8Array;
 //# sourceMappingURL=utils.cjs.map

package/dist/tsup/utils.js

@@ -11,7 +11,7 @@ import {
   setLongTimeout,
   stringifyError,
   toUint8Array
-} from "./chunk-42UD33X4.js";
+} from "./chunk-QMVCFQ37.js";
 import "./chunk-G64QUEDJ.js";
 export {
   SinglePromiseQueue,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "rivetkit",
-  "version": "2.0.18",
+  "version": "2.0.20",
   "description": "Lightweight libraries for building stateful actors on edge platforms",
   "license": "Apache-2.0",
   "keywords": [

package/src/common/eventsource.ts

@@ -4,10 +4,12 @@ import { logger } from "@/client/log";
 // Global singleton promise that will be reused for subsequent calls
 let eventSourcePromise: Promise<typeof EventSource> | null = null;
 
-/**
- * Import `eventsource` from the custom `eventsource` library. We need a custom implemnetation since we need to attach our own custom headers to the request.
- **/
 export async function importEventSource(): Promise<typeof EventSource> {
+	// IMPORTANT: Import `eventsource` from the custom `eventsource` library. We need a custom implementation
+	// since we need to attach our own custom headers to the request.
+	//
+	// We can't use the browser-provided EventSource since it does not allow providing custom headers.
+
 	// Return existing promise if we already started loading
 	if (eventSourcePromise !== null) {
 		return eventSourcePromise;
@@ -19,7 +21,8 @@ export async function importEventSource(): Promise<typeof EventSource> {
 
 		// Node.js environment
 		try {
-			const es = await import("eventsource");
+			const moduleName = "eventsource";
+			const es = await import(/* webpackIgnore: true */ moduleName);
 			_EventSource = es.EventSource;
 			logger().debug("using eventsource from npm");
 		} catch (err) {
@@ -39,42 +42,3 @@ export async function importEventSource(): Promise<typeof EventSource> {
 
 	return eventSourcePromise;
 }
-
-//export async function importEventSource(): Promise<typeof EventSource> {
-//	// Return existing promise if we already started loading
-//	if (eventSourcePromise !== null) {
-//		return eventSourcePromise;
-//	}
-//
-//	// Create and store the promise
-//	eventSourcePromise = (async () => {
-//		let _EventSource: typeof EventSource;
-//
-//		if (typeof EventSource !== "undefined") {
-//			// Browser environment
-//			_EventSource = EventSource;
-//			logger().debug("using native eventsource");
-//		} else {
-//			// Node.js environment
-//			try {
-//				const es = await import("eventsource");
-//				_EventSource = es.EventSource;
-//				logger().debug("using eventsource from npm");
-//			} catch (err) {
-//				// EventSource not available
-//				_EventSource = class MockEventSource {
-//					constructor() {
-//						throw new Error(
-//							'EventSource support requires installing the "eventsource" peer dependency.',
-//						);
-//					}
-//				} as unknown as typeof EventSource;
-//				logger().debug("using mock eventsource");
-//			}
-//		}
-//
-//		return _EventSource;
-//	})();
-//
-//	return eventSourcePromise;
-//}

package/src/common/websocket.ts

@@ -19,7 +19,8 @@ export async function importWebSocket(): Promise<typeof WebSocket> {
 		} else {
 			// Node.js environment
 			try {
-				const ws = await import("ws");
+				const moduleName = "ws";
+				const ws = await import(/* webpackIgnore: true */ moduleName);
 				_WebSocket = ws.default as unknown as typeof WebSocket;
 				logger().debug("using websocket from npm");
 			} catch {

package/src/engine-process/mod.ts

@@ -1,4 +1,5 @@
 import { spawn } from "node:child_process";
+import { randomUUID } from "node:crypto";
 import { createWriteStream } from "node:fs";
 import * as fs from "node:fs/promises";
 import * as path from "node:path";
@@ -103,9 +104,11 @@ export async function ensureEngineProcess(
 
 	child.once("exit", (code, signal) => {
 		logger().warn({
-			msg: "engine process exited",
+			msg: "engine process exited, please report this error",
 			code,
 			signal,
+			issues: "https://github.com/rivet-dev/rivetkit/issues",
+			support: "https://rivet.dev/discord",
 		});
 		// Clean up log streams
 		stdoutStream.end();
@@ -168,22 +171,70 @@ async function downloadEngineBinaryIfNeeded(
 		);
 	}
 
-	const tempPath = `${binaryPath}.${process.pid}.tmp`;
-	await pipeline(response.body, createWriteStream(tempPath));
-	if (process.platform !== "win32") {
-		await fs.chmod(tempPath, 0o755);
-	}
-	await fs.rename(tempPath, binaryPath);
+	// Generate unique temp file name to prevent parallel download conflicts
+	const tempPath = `${binaryPath}.${randomUUID()}.tmp`;
+	const startTime = Date.now();
+
 	logger().debug({
-		msg: "engine binary download complete",
-		version,
-		path: binaryPath,
-	});
-	logger().info({
-		msg: "engine binary downloaded",
-		version,
-		path: binaryPath,
+		msg: "starting binary download",
+		tempPath,
+		contentLength: response.headers.get("content-length"),
 	});
+
+	// Warn user if download is taking a long time
+	const slowDownloadWarning = setTimeout(() => {
+		logger().warn({
+			msg: "engine binary download is taking longer than expected, please be patient",
+			version,
+		});
+	}, 5000);
+
+	try {
+		await pipeline(response.body, createWriteStream(tempPath));
+
+		// Clear the slow download warning
+		clearTimeout(slowDownloadWarning);
+
+		// Get file size to verify download
+		const stats = await fs.stat(tempPath);
+		const downloadDuration = Date.now() - startTime;
+
+		if (process.platform !== "win32") {
+			await fs.chmod(tempPath, 0o755);
+		}
+		await fs.rename(tempPath, binaryPath);
+
+		logger().debug({
+			msg: "engine binary download complete",
+			version,
+			path: binaryPath,
+			size: stats.size,
+			durationMs: downloadDuration,
+		});
+		logger().info({
+			msg: "engine binary downloaded",
+			version,
+			path: binaryPath,
+		});
+	} catch (error) {
+		// Clear the slow download warning
+		clearTimeout(slowDownloadWarning);
+
+		// Clean up partial temp file on error
+		logger().warn({
+			msg: "engine download failed, please report this error",
+			tempPath,
+			error,
+			issues: "https://github.com/rivet-dev/rivetkit/issues",
+			support: "https://rivet.dev/discord",
+		});
+		try {
+			await fs.unlink(tempPath);
+		} catch (unlinkError) {
+			// Ignore errors when cleaning up (file may not exist)
+		}
+		throw error;
+	}
 }
 
 function resolveTargetTriplet(): { targetTriplet: string; extension: string } {

package/src/registry/mod.ts

@@ -71,6 +71,25 @@ export class Registry<A extends RegistryActors> {
 		// Promise for any async operations we need to wait to complete
 		const readyPromises = [];
 
+		// Disable health check if using serverless
+		//
+		// This is because the endpoint will not be configured until we receive
+		// a serverless runner request, so we do not know what to health check
+		if (config.runnerKind === "serverless") {
+			logger().debug("disabling health check since using serverless");
+			config.disableHealthCheck = true;
+		}
+
+		// Auto-configure serverless runner if not in prod
+		if (
+			process.env.NODE_ENV !== "production" &&
+			config.runnerKind === "serverless"
+		) {
+			if (inputConfig?.runEngine === undefined) config.runEngine = true;
+			if (inputConfig?.autoConfigureServerless === undefined)
+				config.autoConfigureServerless = true;
+		}
+
 		// Start engine
 		if (config.runEngine) {
 			logger().debug({
@@ -185,6 +204,7 @@ export class Registry<A extends RegistryActors> {
 		// Even though we do not use the returned ActorDriver, this is required to start the code that will handle incoming actors
 		if (!config.disableActorDriver) {
 			Promise.all(readyPromises).then(async () => {
+				logger().debug("starting actor driver");
 				driver.actor(this.#config, config, managerDriver, client);
 			});
 		}
@@ -220,6 +240,8 @@ export class Registry<A extends RegistryActors> {
 }
 
 async function configureServerlessRunner(config: RunnerConfig): Promise<void> {
+	logger().debug("configuring serverless runner");
+
 	try {
 		// Ensure we have required config values
 		if (!config.runnerName) {
@@ -275,6 +297,7 @@ async function configureServerlessRunner(config: RunnerConfig): Promise<void> {
 				slots_per_runner:
 					customConfig.slotsPerRunner ?? config.totalSlots ?? 1000,
 			},
+			metadata: customConfig.metadata ?? {},
 		};
 		await updateRunnerConfig(clientConfig, config.runnerName, {
 			datacenters: Object.fromEntries(
@@ -289,10 +312,11 @@ async function configureServerlessRunner(config: RunnerConfig): Promise<void> {
 		});
 	} catch (error) {
 		logger().error({
-			msg: "failed to configure serverless runner",
+			msg: "failed to configure serverless runner, validate endpoint is configured correctly then restart this process",
 			error,
 		});
-		throw error;
+
+		// Don't throw, allow the runner to continue
 	}
 }
 

package/src/registry/run-config.ts

@@ -119,6 +119,7 @@ export const RunnerConfigSchema = z
 					requestLifespan: z.number().optional(),
 					runnersMargin: z.number().optional(),
 					slotsPerRunner: z.number().optional(),
+					metadata: z.record(z.unknown()).optional(),
 				}),
 			])
 			.optional(),

package/src/remote-manager-driver/mod.ts

@@ -16,7 +16,7 @@ import type {
 } from "@/driver-helpers/mod";
 import type { Encoding, UniversalWebSocket } from "@/mod";
 import { uint8ArrayToBase64 } from "@/serde";
-import { combineUrlPath } from "@/utils";
+import { combineUrlPath, getEnvUniversal } from "@/utils";
 import { sendHttpRequestToActor } from "./actor-http-client";
 import {
 	buildWebSocketProtocols,
@@ -56,13 +56,22 @@ export class RemoteManagerDriver implements ManagerDriver {
 	#metadataPromise: Promise<void> | undefined;
 
 	constructor(runConfig: ClientConfig) {
+		// Disable health check if in Next.js build phase since there is no `/metadata` endpoint
+		//
+		// See https://github.com/vercel/next.js/blob/5e6b008b561caf2710ab7be63320a3d549474a5b/packages/next/shared/lib/constants.ts#L19-L23
+		if (getEnvUniversal("NEXT_PHASE") === "phase-production-build") {
+			logger().info("detected next.js build phase, disabling health check");
+			runConfig.disableHealthCheck = true;
+		}
+
 		this.#config = runConfig;
 
 		// Perform metadata check if enabled
 		if (!runConfig.disableHealthCheck) {
 			this.#metadataPromise = this.#performMetadataCheck(runConfig);
 			this.#metadataPromise.catch((error) => {
-				logger().error({
+				// TODO: Promot to error after metadata endpoint merged on prod
+				logger().info({
 					msg: "metadata check failed",
 					error: error instanceof Error ? error.message : String(error),
 				});
@@ -101,7 +110,7 @@ export class RemoteManagerDriver implements ManagerDriver {
 				});
 			} catch (error) {
 				logger().error({
-					msg: "failed to connect to metadata endpoint",
+					msg: "health check failed, validate the Rivet endpoint is configured correctly",
 					endpoint,
 					error: stringifyError(error),
 				});