rivetkit 2.3.27 → 2.3.28

package/src/agent-os/actor/shell.ts

@@ -1,60 +0,0 @@
-import type { OpenShellOptions } from "@rivet-dev/agent-os-core";
-import type { AgentOsActorConfig } from "../config";
-import type { AgentOsActionContext } from "../types";
-import { ensureVm, syncPreventSleep } from "./index";
-
-// Build shell actions for the actor factory.
-export function buildShellActions<TConnParams>(
-	config: AgentOsActorConfig<TConnParams>,
-) {
-	return {
-		openShell: async (
-			c: AgentOsActionContext<TConnParams>,
-			options?: OpenShellOptions,
-		): Promise<{ shellId: string }> => {
-			const agentOs = await ensureVm(c, config);
-			const { shellId } = agentOs.openShell(options);
-
-			// Wire shell data to actor events.
-			agentOs.onShellData(shellId, (data: Uint8Array) => {
-				c.broadcast("shellData", { shellId, data });
-			});
-
-			c.vars.activeShells.add(shellId);
-			syncPreventSleep(c);
-			c.log.info({ msg: "agent-os shell opened", shellId });
-
-			return { shellId };
-		},
-
-		writeShell: async (
-			c: AgentOsActionContext<TConnParams>,
-			shellId: string,
-			data: string | Uint8Array,
-		): Promise<void> => {
-			const agentOs = await ensureVm(c, config);
-			agentOs.writeShell(shellId, data);
-		},
-
-		resizeShell: async (
-			c: AgentOsActionContext<TConnParams>,
-			shellId: string,
-			cols: number,
-			rows: number,
-		): Promise<void> => {
-			const agentOs = await ensureVm(c, config);
-			agentOs.resizeShell(shellId, cols, rows);
-		},
-
-		closeShell: async (
-			c: AgentOsActionContext<TConnParams>,
-			shellId: string,
-		): Promise<void> => {
-			const agentOs = await ensureVm(c, config);
-			agentOs.closeShell(shellId);
-			c.vars.activeShells.delete(shellId);
-			syncPreventSleep(c);
-			c.log.info({ msg: "agent-os shell closed", shellId });
-		},
-	};
-}

package/src/agent-os/config.ts

@@ -1,79 +0,0 @@
-import type {
-	AgentOsOptions,
-	JsonRpcNotification,
-	PermissionRequest,
-} from "@rivet-dev/agent-os-core";
-import { z } from "zod/v4";
-import type { ActorContext, BeforeConnectContext } from "@/actor/config";
-import type { AgentOsActorState, AgentOsActorVars } from "./types";
-
-const zFunction = <
-	T extends (...args: any[]) => any = (...args: unknown[]) => unknown,
->() => z.custom<T>((val) => typeof val === "function");
-
-const AgentOsOptionsSchema = z.custom<AgentOsOptions>(
-	(val) => typeof val === "object" && val !== null,
-);
-
-export const agentOsActorConfigSchema = z
-	.object({
-		options: AgentOsOptionsSchema.optional(),
-		preview: z
-			.object({
-				defaultExpiresInSeconds: z.number().positive().default(3600),
-				maxExpiresInSeconds: z.number().positive().default(86400),
-			})
-			.strict()
-			.prefault(() => ({})),
-		onBeforeConnect: zFunction().optional(),
-		onSessionEvent: zFunction().optional(),
-		onPermissionRequest: zFunction().optional(),
-	})
-	.strict();
-
-// --- Typed config types (generic callbacks overlaid on the Zod schema) ---
-
-type AgentOsActorContext<TConnParams> = ActorContext<
-	AgentOsActorState,
-	TConnParams,
-	undefined,
-	AgentOsActorVars,
-	undefined,
-	any
->;
-
-interface AgentOsActorConfigCallbacks<TConnParams> {
-	onBeforeConnect?: (
-		c: BeforeConnectContext<
-			AgentOsActorState,
-			AgentOsActorVars,
-			undefined,
-			any
-		>,
-		params: TConnParams,
-	) => void | Promise<void>;
-	onSessionEvent?: (
-		c: AgentOsActorContext<TConnParams>,
-		sessionId: string,
-		event: JsonRpcNotification,
-	) => void | Promise<void>;
-	onPermissionRequest?: (
-		c: AgentOsActorContext<TConnParams>,
-		sessionId: string,
-		request: PermissionRequest,
-	) => void | Promise<void>;
-}
-
-// Parsed config (after Zod defaults/transforms applied).
-export type AgentOsActorConfig<TConnParams = undefined> = Omit<
-	z.infer<typeof agentOsActorConfigSchema>,
-	"onBeforeConnect" | "onSessionEvent" | "onPermissionRequest"
-> &
-	AgentOsActorConfigCallbacks<TConnParams>;
-
-// Input config (what users pass in before Zod transforms).
-export type AgentOsActorConfigInput<TConnParams = undefined> = Omit<
-	z.input<typeof agentOsActorConfigSchema>,
-	"onBeforeConnect" | "onSessionEvent" | "onPermissionRequest"
-> &
-	AgentOsActorConfigCallbacks<TConnParams>;