chore: update sdk (#1182)

### TL;DR

Updated the engine-runner dependency and added TypeScript type annotations to fix type errors.

### What changed?

- Updated `@rivetkit/engine-runner` dependency from version `472` to `7f23f3a`
- Added explicit type annotations in `actor-driver.ts`:
    - Added type `any` to the `input` variable
    - Added type `Promise<UpgradeWebSocketArgs>` to the `wsHandlerPromise` variable
- Added import for `UpgradeWebSocketArgs` type

### How to test?

1. Run the TypeScript compiler to verify no type errors remain
2. Test WebSocket connections to ensure they still function properly
3. Verify that actor input handling works as expected

### Why make this change?

This change addresses TypeScript type errors in the engine actor driver by adding proper type annotations. The update to the engine-runner dependency likely includes fixes or improvements that complement these type changes, ensuring better type safety and code reliability.

Author: jog1t

packages/core/package.json

@@ -172,7 +172,7 @@
     "@hono/node-server": "^1.18.2",
     "@hono/node-ws": "^1.1.1",
     "@rivet-gg/actor-core": "^25.1.0",
-    "@rivetkit/engine-runner": "https://pkg.pr.new/rivet-gg/engine/@rivetkit/engine-runner@472",
+    "@rivetkit/engine-runner": "https://pkg.pr.new/rivet-gg/engine/@rivetkit/engine-runner@7f23f3a",
     "@types/invariant": "^2",
     "@types/node": "^22.13.1",
     "@types/ws": "^8",

packages/core/src/drivers/engine/actor-driver.ts

@@ -4,17 +4,9 @@ import type {
 } from "@rivetkit/engine-runner";
 import { Runner } from "@rivetkit/engine-runner";
 import * as cbor from "cbor-x";
-import { WSContext, WSContextInit } from "hono/ws";
+import { WSContext } from "hono/ws";
 import invariant from "invariant";
-import { ActionContext } from "@/actor/action";
-import { generateConnId, generateConnToken } from "@/actor/connection";
-import {
-	CONN_DRIVER_GENERIC_HTTP,
-	type GenericHttpDriverState,
-} from "@/actor/generic-conn-driver";
-import * as protoHttpAction from "@/actor/protocol/http/action";
-import { deserialize, EncodingSchema, serialize } from "@/actor/protocol/serde";
-import { ActorHandle } from "@/client/actor-handle";
+import { EncodingSchema } from "@/actor/protocol/serde";
 import type { Client } from "@/client/client";
 import {
 	type ActorDriver,
@@ -27,10 +19,10 @@ import {
 } from "@/driver-helpers/mod";
 import type {
 	ActorRouter,
-	AnyActorInstance as CoreAnyActorInstance,
 	RegistryConfig,
 	RunConfig,
 	UniversalWebSocket,
+	UpgradeWebSocketArgs,
 } from "@/mod";
 import {
 	createActorRouter,
@@ -51,7 +43,6 @@ import { logger } from "./log";
 interface ActorHandler {
 	actor?: AnyActorInstance;
 	actorStartPromise?: PromiseWithResolvers<void>;
-	metadata: RunnerActorConfig["metadata"];
 	genericConnGlobalState: GenericConnGlobalState;
 	persistedData?: Uint8Array;
 }
@@ -93,6 +84,7 @@ export class EngineActorDriver implements ActorDriver {
 			addresses: config.addresses,
 			totalSlots: config.totalSlots,
 			runnerName: config.runnerName,
+			runnerKey: config.runnerKey,
 			prepopulateActorNames: Object.keys(this.#registryConfig.use),
 			onConnected: () => {
 				if (hasDisconnected) {
@@ -201,13 +193,13 @@ export class EngineActorDriver implements ActorDriver {
 	): Promise<void> {
 		logger().debug("runner actor starting", {
 			actorId,
-			name: config.metadata.actor.name,
-			keys: config.metadata.actor.keys,
+			name: config.name,
+			key: config.key,
 			generation,
 		});
 
 		// Deserialize input
-		let input;
+		let input: any;
 		if (config.input) {
 			input = cbor.decode(config.input);
 		}
@@ -218,19 +210,19 @@ export class EngineActorDriver implements ActorDriver {
 			handler = {
 				genericConnGlobalState: new GenericConnGlobalState(),
 				actorStartPromise: Promise.withResolvers(),
-				metadata: config.metadata,
 				persistedData: serializeEmptyPersistData(input),
 			};
 			this.#actors.set(actorId, handler);
 		}
 
-		const name = config.metadata.actor.name as string;
-		const key = deserializeActorKey(config.metadata.actor.keys[0]);
+		const name = config.name as string;
+		invariant(config.key, "actor should have a key");
+		const key = deserializeActorKey(config.key);
 
 		// Create actor instance
 		const definition = lookupInRegistry(
 			this.#registryConfig,
-			config.metadata.actor.name as string, // TODO: Remove cast
+			config.name as string, // TODO: Remove cast
 		);
 		handler.actor = definition.instantiate();
 
@@ -299,7 +291,7 @@ export class EngineActorDriver implements ActorDriver {
 		// Fetch WS handler
 		//
 		// We store the promise since we need to add WebSocket event listeners immediately that will wait for the promise to resolve
-		let wsHandlerPromise;
+		let wsHandlerPromise: Promise<UpgradeWebSocketArgs>;
 		if (url.pathname === PATH_CONNECT_WEBSOCKET) {
 			wsHandlerPromise = handleWebSocketConnect(
 				request,

packages/core/src/drivers/engine/config.ts

@@ -15,6 +15,10 @@ export const ConfigSchema = z
 		runnerName: z
 			.string()
 			.default(getEnvUniversal("RIVET_RUNNER") ?? "rivetkit"),
+		// TODO: Automatically attempt ot determine key by common env vars (e.g. k8s pod name)
+		runnerKey: z
+			.string()
+			.default(getEnvUniversal("RIVET_RUNNER_KEY") ?? crypto.randomUUID()),
 		totalSlots: z.number().default(100_000),
 		addresses: z
 			.record(

packages/core/src/drivers/engine/kv.ts

@@ -1,3 +1,3 @@
 export const KEYS = {
-	PERSIST_DATA: [Uint8Array.from([1, 1])],
+	PERSIST_DATA: Uint8Array.from([1, 1]),
 };

packages/core/tsup.config.ts

@@ -3,5 +3,4 @@ import defaultConfig from "../../tsup.base.ts";
 
 export default defineConfig({
 	...defaultConfig,
-	noExternal: ["@rivetkit/engine-runner", "@rivetkit/engine-runner-protocol"],
 });