experimental-agent 0.0.0 → 0.0.2

package/dist/{client-FCFZYOOB.mjs → client-4Y3UPWFR.mjs}

@@ -1,8 +1,8 @@
 import {
   getSandbox
-} from "./chunk-JQPR6M7D.mjs";
-import "./chunk-2YI7MQGZ.mjs";
-import "./chunk-DPPQO7DA.mjs";
+} from "./chunk-4RGMKC2M.mjs";
+import "./chunk-6ICYKNCC.mjs";
+import "./chunk-24DJSI7C.mjs";
 export {
   getSandbox
 };

package/dist/client-BBpD9kKL.d.ts

@@ -0,0 +1,193 @@
+import { f as SandboxRecord, d as SandboxError, k as Storage, e as SandboxNotFoundError, j as StorageError, i as StorageConfig, c as SandboxConfig, h as SessionNotFoundError, l as SessionError, b as MessageNotFoundError } from './types-DPXFq_r6.js';
+import * as ai from 'ai';
+import { GatewayModelId, UIMessage } from 'ai';
+import { z } from 'zod';
+
+/**
+ * Summary of a discovered skill, used in the system prompt
+ * to inform the LLM about available skills.
+ */
+type SkillSummary = {
+    name: string;
+    description: string;
+    /** Full path to the SKILL.md file inside the sandbox */
+    skillMdPath: string;
+};
+/**
+ * A file ready to be uploaded to the sandbox.
+ * Content can be text (string) or binary (Buffer).
+ */
+type UploadableFile = {
+    /** Relative path within the upload destination */
+    path: string;
+    content: string | Buffer;
+};
+/**
+ * Path or paths to skills directories inside the sandbox.
+ */
+type SkillsDir = string | string[];
+
+type SandboxLifecycleInput = {
+    id: string;
+    storageConfig: StorageConfig;
+    initialConfig: SandboxConfig;
+};
+type LogEntry = {
+    stream: "stdout" | "stderr";
+    data: string;
+};
+type ExecResult = {
+    commandId: string;
+    logs: () => AsyncIterable<LogEntry>;
+    result: Promise<{
+        stdout: string;
+        stderr: string;
+        exitCode: number;
+    }>;
+};
+type SandboxStatus = "pending" | "running" | "stopping" | "stopped" | "failed";
+interface SandboxLifecycle {
+    start: () => Promise<SandboxError | SandboxStatus>;
+    snapshot: () => Promise<SandboxError | {
+        snapshotId: string;
+    }>;
+    stop: () => Promise<SandboxError | undefined>;
+    getStatus: () => Promise<SandboxError | SandboxStatus>;
+    getCreatedAt: () => Promise<SandboxError | Date>;
+    getRemainingTimeout: () => Promise<SandboxError | number>;
+}
+interface Sandbox<TTags extends TagsSchema = TagsSchema> {
+    id: SandboxRecord["id"];
+    config: SandboxRecord["config"];
+    exec: (opts: {
+        command: string;
+        args?: string[];
+        signal?: AbortSignal;
+    }) => Promise<SandboxError | ExecResult>;
+    /**
+     * Get the public domain URL for an exposed port.
+     * Only available for Vercel sandboxes with exposed ports.
+     */
+    getDomain: (port: number) => Promise<SandboxError | string>;
+    kill: (opts: {
+        commandId: string;
+        storage: Storage;
+    }) => Promise<SandboxError | undefined>;
+    writeFiles: (opts: {
+        files: UploadableFile[];
+        destPath: string;
+    }) => Promise<void>;
+    lifecycle?: SandboxLifecycle;
+    tag: {
+        list: () => Promise<SandboxNotFoundError | StorageError | TTags>;
+        get: (key: string) => Promise<SandboxNotFoundError | StorageError | TTags[typeof key] | undefined>;
+        set: (key: string, value: unknown) => Promise<StorageError | undefined>;
+        setMany: (tags: Record<string, unknown>) => Promise<StorageError | undefined>;
+    };
+}
+
+type McpToolDef = {
+    description: string;
+    inputSchema: z.ZodType;
+    outputSchema?: z.ZodType;
+};
+type McpServerDef<TTools extends Record<string, McpToolDef> = Record<string, McpToolDef>, THeaders extends z.ZodType = z.ZodType> = {
+    url: string;
+    description?: string;
+    headersSchema?: THeaders;
+    tools: TTools;
+};
+type McpConfig = Record<string, McpServerDef>;
+type ExtractHeaders<T> = T extends McpServerDef<infer _TTools, infer THeaders> ? THeaders extends z.ZodType ? z.infer<THeaders> : Record<string, string> : Record<string, string>;
+type HasRequiredHeaders<T extends McpServerDef> = T extends {
+    headersSchema: z.ZodType;
+} ? true : false;
+type ServerSendOptions<T extends McpServerDef> = HasRequiredHeaders<T> extends true ? {
+    headers: ExtractHeaders<T>;
+    disabled?: false;
+} | {
+    disabled: true;
+    headers?: ExtractHeaders<T>;
+} : {
+    headers?: ExtractHeaders<T>;
+    disabled?: boolean;
+};
+type McpSendOptions<TMcp extends McpConfig> = {
+    [K in keyof TMcp]?: ServerSendOptions<TMcp[K]>;
+};
+declare function mcp<THeaders extends z.ZodType, TTools extends Record<string, McpToolDef>>(opts: {
+    url: string;
+    description?: string;
+    headersSchema?: THeaders;
+    tools: TTools;
+}): McpServerDef<TTools, THeaders>;
+
+type SendInput = string | {
+    role?: UIMessage["role"];
+    parts: UIMessage["parts"];
+    id?: string;
+};
+type SessionOptions<TMcp extends McpConfig = {}, TTags extends TagsSchema = TagsSchema> = {
+    model?: GatewayModelId;
+    instructions?: string;
+    sandbox?: SandboxConfig | string;
+    tags?: TTags;
+    skillsDir?: SkillsDir;
+    mcp?: TMcp;
+};
+type McpToolCallBody = {
+    server: string;
+    tool: string;
+    input: unknown;
+    sessionId: string;
+    messageId: string;
+    hookToken: string;
+};
+type AgentOptions<TMcp extends McpConfig = {}> = {
+    storage?: StorageConfig;
+    generateId?: (resource: "session" | "message" | "part" | "sandbox" | "command") => string;
+} & SessionOptions<TMcp>;
+type TagsSchema = Record<string, unknown>;
+declare const agent: <TMcp extends McpConfig = {}>(options: AgentOptions<TMcp>) => {
+    session: <TTags extends TagsSchema = TagsSchema, TMcpOverride extends McpConfig = {}>(sessionId: string, sessionOptions?: SessionOptions<TMcp & TMcpOverride, TTags>) => Promise<{
+        send: ({ input, mcpContext, interruptIfStreaming, }: {
+            input: SendInput | SendInput[];
+            mcpContext?: McpSendOptions<TMcp & TMcpOverride>;
+            interruptIfStreaming?: boolean;
+        }) => Promise<void | SessionNotFoundError | SessionError | StorageError | MessageNotFoundError>;
+        stream: (opts?: {
+            messageId?: string;
+        }) => Promise<SessionNotFoundError | SessionError | StorageError | ReadableStream<any>>;
+        ui: () => Promise<StorageError | {
+            messages: UIMessage<unknown, ai.UIDataTypes, ai.UITools>[];
+            streamingMessageId: string | null;
+        }>;
+        tag: {
+            list: () => Promise<SessionNotFoundError | StorageError | TTags>;
+            get: (key: string) => Promise<SessionNotFoundError | StorageError | TTags[string] | undefined>;
+            set: (key: string, value: unknown) => Promise<StorageError | undefined>;
+            setMany: (tags: Record<string, unknown>) => Promise<StorageError | undefined>;
+        };
+        sandbox: Sandbox<TagsSchema>;
+        interrupt: () => Promise<SessionNotFoundError | StorageError | MessageNotFoundError | undefined>;
+    }>;
+    sandbox: <TTags extends TagsSchema = TagsSchema>(sandboxId: string, opts?: {
+        config?: SandboxConfig;
+        tags?: TTags;
+    }) => Promise<Sandbox<TTags>>;
+    storage: Storage;
+    handleMcpToolCall: (body: McpToolCallBody, handlers: Record<string, (opts: {
+        input: unknown;
+        headers: Record<string, string>;
+        body: McpToolCallBody;
+    }) => Promise<unknown>>) => Promise<{
+        result: unknown;
+    } | {
+        error: {
+            code: string;
+            message: string;
+        };
+    }>;
+};
+
+export { type AgentOptions as A, type ExecResult as E, type Sandbox as S, type TagsSchema as T, type UploadableFile as U, type SendInput as a, type SessionOptions as b, type SkillSummary as c, agent as d, type SandboxLifecycleInput as e, mcp as m };

package/dist/client-BGJViybU.d.mts

@@ -0,0 +1,193 @@
+import { f as SandboxRecord, d as SandboxError, k as Storage, e as SandboxNotFoundError, j as StorageError, i as StorageConfig, c as SandboxConfig, h as SessionNotFoundError, l as SessionError, b as MessageNotFoundError } from './types-DPXFq_r6.mjs';
+import * as ai from 'ai';
+import { GatewayModelId, UIMessage } from 'ai';
+import { z } from 'zod';
+
+/**
+ * Summary of a discovered skill, used in the system prompt
+ * to inform the LLM about available skills.
+ */
+type SkillSummary = {
+    name: string;
+    description: string;
+    /** Full path to the SKILL.md file inside the sandbox */
+    skillMdPath: string;
+};
+/**
+ * A file ready to be uploaded to the sandbox.
+ * Content can be text (string) or binary (Buffer).
+ */
+type UploadableFile = {
+    /** Relative path within the upload destination */
+    path: string;
+    content: string | Buffer;
+};
+/**
+ * Path or paths to skills directories inside the sandbox.
+ */
+type SkillsDir = string | string[];
+
+type SandboxLifecycleInput = {
+    id: string;
+    storageConfig: StorageConfig;
+    initialConfig: SandboxConfig;
+};
+type LogEntry = {
+    stream: "stdout" | "stderr";
+    data: string;
+};
+type ExecResult = {
+    commandId: string;
+    logs: () => AsyncIterable<LogEntry>;
+    result: Promise<{
+        stdout: string;
+        stderr: string;
+        exitCode: number;
+    }>;
+};
+type SandboxStatus = "pending" | "running" | "stopping" | "stopped" | "failed";
+interface SandboxLifecycle {
+    start: () => Promise<SandboxError | SandboxStatus>;
+    snapshot: () => Promise<SandboxError | {
+        snapshotId: string;
+    }>;
+    stop: () => Promise<SandboxError | undefined>;
+    getStatus: () => Promise<SandboxError | SandboxStatus>;
+    getCreatedAt: () => Promise<SandboxError | Date>;
+    getRemainingTimeout: () => Promise<SandboxError | number>;
+}
+interface Sandbox<TTags extends TagsSchema = TagsSchema> {
+    id: SandboxRecord["id"];
+    config: SandboxRecord["config"];
+    exec: (opts: {
+        command: string;
+        args?: string[];
+        signal?: AbortSignal;
+    }) => Promise<SandboxError | ExecResult>;
+    /**
+     * Get the public domain URL for an exposed port.
+     * Only available for Vercel sandboxes with exposed ports.
+     */
+    getDomain: (port: number) => Promise<SandboxError | string>;
+    kill: (opts: {
+        commandId: string;
+        storage: Storage;
+    }) => Promise<SandboxError | undefined>;
+    writeFiles: (opts: {
+        files: UploadableFile[];
+        destPath: string;
+    }) => Promise<void>;
+    lifecycle?: SandboxLifecycle;
+    tag: {
+        list: () => Promise<SandboxNotFoundError | StorageError | TTags>;
+        get: (key: string) => Promise<SandboxNotFoundError | StorageError | TTags[typeof key] | undefined>;
+        set: (key: string, value: unknown) => Promise<StorageError | undefined>;
+        setMany: (tags: Record<string, unknown>) => Promise<StorageError | undefined>;
+    };
+}
+
+type McpToolDef = {
+    description: string;
+    inputSchema: z.ZodType;
+    outputSchema?: z.ZodType;
+};
+type McpServerDef<TTools extends Record<string, McpToolDef> = Record<string, McpToolDef>, THeaders extends z.ZodType = z.ZodType> = {
+    url: string;
+    description?: string;
+    headersSchema?: THeaders;
+    tools: TTools;
+};
+type McpConfig = Record<string, McpServerDef>;
+type ExtractHeaders<T> = T extends McpServerDef<infer _TTools, infer THeaders> ? THeaders extends z.ZodType ? z.infer<THeaders> : Record<string, string> : Record<string, string>;
+type HasRequiredHeaders<T extends McpServerDef> = T extends {
+    headersSchema: z.ZodType;
+} ? true : false;
+type ServerSendOptions<T extends McpServerDef> = HasRequiredHeaders<T> extends true ? {
+    headers: ExtractHeaders<T>;
+    disabled?: false;
+} | {
+    disabled: true;
+    headers?: ExtractHeaders<T>;
+} : {
+    headers?: ExtractHeaders<T>;
+    disabled?: boolean;
+};
+type McpSendOptions<TMcp extends McpConfig> = {
+    [K in keyof TMcp]?: ServerSendOptions<TMcp[K]>;
+};
+declare function mcp<THeaders extends z.ZodType, TTools extends Record<string, McpToolDef>>(opts: {
+    url: string;
+    description?: string;
+    headersSchema?: THeaders;
+    tools: TTools;
+}): McpServerDef<TTools, THeaders>;
+
+type SendInput = string | {
+    role?: UIMessage["role"];
+    parts: UIMessage["parts"];
+    id?: string;
+};
+type SessionOptions<TMcp extends McpConfig = {}, TTags extends TagsSchema = TagsSchema> = {
+    model?: GatewayModelId;
+    instructions?: string;
+    sandbox?: SandboxConfig | string;
+    tags?: TTags;
+    skillsDir?: SkillsDir;
+    mcp?: TMcp;
+};
+type McpToolCallBody = {
+    server: string;
+    tool: string;
+    input: unknown;
+    sessionId: string;
+    messageId: string;
+    hookToken: string;
+};
+type AgentOptions<TMcp extends McpConfig = {}> = {
+    storage?: StorageConfig;
+    generateId?: (resource: "session" | "message" | "part" | "sandbox" | "command") => string;
+} & SessionOptions<TMcp>;
+type TagsSchema = Record<string, unknown>;
+declare const agent: <TMcp extends McpConfig = {}>(options: AgentOptions<TMcp>) => {
+    session: <TTags extends TagsSchema = TagsSchema, TMcpOverride extends McpConfig = {}>(sessionId: string, sessionOptions?: SessionOptions<TMcp & TMcpOverride, TTags>) => Promise<{
+        send: ({ input, mcpContext, interruptIfStreaming, }: {
+            input: SendInput | SendInput[];
+            mcpContext?: McpSendOptions<TMcp & TMcpOverride>;
+            interruptIfStreaming?: boolean;
+        }) => Promise<void | SessionNotFoundError | SessionError | StorageError | MessageNotFoundError>;
+        stream: (opts?: {
+            messageId?: string;
+        }) => Promise<SessionNotFoundError | SessionError | StorageError | ReadableStream<any>>;
+        ui: () => Promise<StorageError | {
+            messages: UIMessage<unknown, ai.UIDataTypes, ai.UITools>[];
+            streamingMessageId: string | null;
+        }>;
+        tag: {
+            list: () => Promise<SessionNotFoundError | StorageError | TTags>;
+            get: (key: string) => Promise<SessionNotFoundError | StorageError | TTags[string] | undefined>;
+            set: (key: string, value: unknown) => Promise<StorageError | undefined>;
+            setMany: (tags: Record<string, unknown>) => Promise<StorageError | undefined>;
+        };
+        sandbox: Sandbox<TagsSchema>;
+        interrupt: () => Promise<SessionNotFoundError | StorageError | MessageNotFoundError | undefined>;
+    }>;
+    sandbox: <TTags extends TagsSchema = TagsSchema>(sandboxId: string, opts?: {
+        config?: SandboxConfig;
+        tags?: TTags;
+    }) => Promise<Sandbox<TTags>>;
+    storage: Storage;
+    handleMcpToolCall: (body: McpToolCallBody, handlers: Record<string, (opts: {
+        input: unknown;
+        headers: Record<string, string>;
+        body: McpToolCallBody;
+    }) => Promise<unknown>>) => Promise<{
+        result: unknown;
+    } | {
+        error: {
+            code: string;
+            message: string;
+        };
+    }>;
+};
+
+export { type AgentOptions as A, type ExecResult as E, type Sandbox as S, type TagsSchema as T, type UploadableFile as U, type SendInput as a, type SessionOptions as b, type SkillSummary as c, agent as d, type SandboxLifecycleInput as e, mcp as m };

package/dist/{client-RRX3GDQD.mjs → client-HUG4HT5L.mjs}

@@ -1,7 +1,7 @@
 import {
   getStorage,
   getStorageClient
-} from "./chunk-DPPQO7DA.mjs";
+} from "./chunk-24DJSI7C.mjs";
 export {
   getStorage,
   getStorageClient

package/dist/index.d.mts

@@ -1,10 +1,8 @@
-import { M as MethodName, S as StorageMethods, a as StorageConfig, b as SessionError, c as SessionNotFoundError, d as StorageError, e as SandboxConfig, f as Storage } from './types-vRxN1Qz1.mjs';
-export { L as ListResult, j as Message, g as MessageNotFoundError, P as Part, R as ResolvedStorage, h as SandboxError, i as SandboxNotFoundError, k as SandboxRecord, l as Session } from './types-vRxN1Qz1.mjs';
-import { S as Sandbox, a as SkillsDir } from './types-Lwut_0_u.mjs';
-export { E as ExecResult, b as SkillSummary, U as UploadableFile } from './types-Lwut_0_u.mjs';
-import * as ai from 'ai';
-import { UIMessage, GatewayModelId } from 'ai';
+export { A as AgentOptions, E as ExecResult, S as Sandbox, a as SendInput, b as SessionOptions, c as SkillSummary, T as TagsSchema, U as UploadableFile, d as agent, m as mcp } from './client-BGJViybU.mjs';
+import { M as MethodName, S as StorageMethods } from './types-DPXFq_r6.mjs';
+export { L as ListResult, a as Message, b as MessageNotFoundError, P as Part, R as ResolvedStorage, c as SandboxConfig, d as SandboxError, e as SandboxNotFoundError, f as SandboxRecord, g as Session, h as SessionNotFoundError, i as StorageConfig, j as StorageError } from './types-DPXFq_r6.mjs';
 import { z } from 'zod';
+import 'ai';
 import 'errore';
 
 type Handlers = {
@@ -26,103 +24,4 @@ type RpcErrorResponse = {
 type RpcResponse<T = unknown> = RpcSuccessResponse<T> | RpcErrorResponse;
 declare function handleStorageRpc(body: RpcRequest, handlers: Handlers): Promise<RpcResponse>;
 
-type McpToolDef = {
-    description: string;
-    inputSchema: z.ZodType;
-    outputSchema?: z.ZodType;
-};
-type McpServerDef<TTools extends Record<string, McpToolDef> = Record<string, McpToolDef>, THeaders extends z.ZodType = z.ZodType> = {
-    url: string;
-    description?: string;
-    headersSchema?: THeaders;
-    tools: TTools;
-};
-type McpConfig = Record<string, McpServerDef>;
-type ExtractHeaders<T> = T extends McpServerDef<infer _TTools, infer THeaders> ? THeaders extends z.ZodType ? z.infer<THeaders> : Record<string, string> : Record<string, string>;
-type HasRequiredHeaders<T extends McpServerDef> = T extends {
-    headersSchema: z.ZodType;
-} ? true : false;
-type ServerSendOptions<T extends McpServerDef> = HasRequiredHeaders<T> extends true ? {
-    headers: ExtractHeaders<T>;
-    disabled?: false;
-} | {
-    disabled: true;
-    headers?: ExtractHeaders<T>;
-} : {
-    headers?: ExtractHeaders<T>;
-    disabled?: boolean;
-};
-type McpSendOptions<TMcp extends McpConfig> = {
-    [K in keyof TMcp]?: ServerSendOptions<TMcp[K]>;
-};
-declare function mcp<THeaders extends z.ZodType, TTools extends Record<string, McpToolDef>>(opts: {
-    url: string;
-    description?: string;
-    headersSchema?: THeaders;
-    tools: TTools;
-}): McpServerDef<TTools, THeaders>;
-
-type SendInput = string | {
-    role?: UIMessage["role"];
-    parts: UIMessage["parts"];
-    id?: string;
-};
-type SessionOptions<TMcp extends McpConfig = {}, TTags extends TagsSchema = TagsSchema> = {
-    model?: GatewayModelId;
-    instructions?: string;
-    sandbox?: SandboxConfig | string;
-    tags?: TTags;
-    skillsDir?: SkillsDir;
-    mcp?: TMcp;
-};
-type McpToolCallBody = {
-    server: string;
-    tool: string;
-    input: unknown;
-    sessionId: string;
-    messageId: string;
-    hookToken: string;
-};
-type AgentOptions<TMcp extends McpConfig = {}> = {
-    storage?: StorageConfig;
-    generateId?: (resource: "session" | "message" | "part" | "sandbox" | "command") => string;
-} & SessionOptions<TMcp>;
-type TagsSchema = Record<string, unknown>;
-declare const agent: <TMcp extends McpConfig = {}>(options: AgentOptions<TMcp>) => {
-    session: <TTags extends TagsSchema = TagsSchema, TMcpOverride extends McpConfig = {}>(sessionId: string, sessionOptions?: SessionOptions<TMcp & TMcpOverride, TTags>) => Promise<{
-        send: ({ input, mcpContext, }: {
-            input: SendInput | SendInput[];
-            mcpContext?: McpSendOptions<TMcp & TMcpOverride>;
-        }) => Promise<void | SessionError>;
-        stream: (opts?: {
-            messageId?: string;
-        }) => Promise<SessionNotFoundError | SessionError | StorageError | ReadableStream<any>>;
-        ui: () => Promise<StorageError | {
-            messages: UIMessage<unknown, ai.UIDataTypes, ai.UITools>[];
-            streamingMessageId: string | null;
-        }>;
-        tag: {
-            list: () => Promise<SessionNotFoundError | StorageError | TTags>;
-            get: (key: string) => Promise<SessionNotFoundError | StorageError | TTags[string] | undefined>;
-            set: (key: string, value: unknown) => Promise<StorageError | undefined>;
-            setMany: (tags: Record<string, unknown>) => Promise<StorageError | undefined>;
-        };
-        sandbox: Sandbox;
-    }>;
-    sandbox: (sandboxId: string, config?: SandboxConfig) => Promise<Sandbox>;
-    storage: Storage;
-    handleMcpToolCall: (body: McpToolCallBody, handlers: Record<string, (opts: {
-        input: unknown;
-        headers: Record<string, string>;
-        body: McpToolCallBody;
-    }) => Promise<unknown>>) => Promise<{
-        result: unknown;
-    } | {
-        error: {
-            code: string;
-            message: string;
-        };
-    }>;
-};
-
-export { type AgentOptions, type Handlers, type RpcRequest, type RpcResponse, Sandbox, SandboxConfig, type SendInput, SessionNotFoundError, StorageConfig, StorageError, type TagsSchema, agent, handleStorageRpc, mcp };
+export { type Handlers, type RpcRequest, type RpcResponse, handleStorageRpc };

package/dist/index.d.ts

@@ -1,10 +1,8 @@
-import { M as MethodName, S as StorageMethods, a as StorageConfig, b as SessionError, c as SessionNotFoundError, d as StorageError, e as SandboxConfig, f as Storage } from './types-vRxN1Qz1.js';
-export { L as ListResult, j as Message, g as MessageNotFoundError, P as Part, R as ResolvedStorage, h as SandboxError, i as SandboxNotFoundError, k as SandboxRecord, l as Session } from './types-vRxN1Qz1.js';
-import { S as Sandbox, a as SkillsDir } from './types-ctZeJ3iQ.js';
-export { E as ExecResult, b as SkillSummary, U as UploadableFile } from './types-ctZeJ3iQ.js';
-import * as ai from 'ai';
-import { UIMessage, GatewayModelId } from 'ai';
+export { A as AgentOptions, E as ExecResult, S as Sandbox, a as SendInput, b as SessionOptions, c as SkillSummary, T as TagsSchema, U as UploadableFile, d as agent, m as mcp } from './client-BBpD9kKL.js';
+import { M as MethodName, S as StorageMethods } from './types-DPXFq_r6.js';
+export { L as ListResult, a as Message, b as MessageNotFoundError, P as Part, R as ResolvedStorage, c as SandboxConfig, d as SandboxError, e as SandboxNotFoundError, f as SandboxRecord, g as Session, h as SessionNotFoundError, i as StorageConfig, j as StorageError } from './types-DPXFq_r6.js';
 import { z } from 'zod';
+import 'ai';
 import 'errore';
 
 type Handlers = {
@@ -26,103 +24,4 @@ type RpcErrorResponse = {
 type RpcResponse<T = unknown> = RpcSuccessResponse<T> | RpcErrorResponse;
 declare function handleStorageRpc(body: RpcRequest, handlers: Handlers): Promise<RpcResponse>;
 
-type McpToolDef = {
-    description: string;
-    inputSchema: z.ZodType;
-    outputSchema?: z.ZodType;
-};
-type McpServerDef<TTools extends Record<string, McpToolDef> = Record<string, McpToolDef>, THeaders extends z.ZodType = z.ZodType> = {
-    url: string;
-    description?: string;
-    headersSchema?: THeaders;
-    tools: TTools;
-};
-type McpConfig = Record<string, McpServerDef>;
-type ExtractHeaders<T> = T extends McpServerDef<infer _TTools, infer THeaders> ? THeaders extends z.ZodType ? z.infer<THeaders> : Record<string, string> : Record<string, string>;
-type HasRequiredHeaders<T extends McpServerDef> = T extends {
-    headersSchema: z.ZodType;
-} ? true : false;
-type ServerSendOptions<T extends McpServerDef> = HasRequiredHeaders<T> extends true ? {
-    headers: ExtractHeaders<T>;
-    disabled?: false;
-} | {
-    disabled: true;
-    headers?: ExtractHeaders<T>;
-} : {
-    headers?: ExtractHeaders<T>;
-    disabled?: boolean;
-};
-type McpSendOptions<TMcp extends McpConfig> = {
-    [K in keyof TMcp]?: ServerSendOptions<TMcp[K]>;
-};
-declare function mcp<THeaders extends z.ZodType, TTools extends Record<string, McpToolDef>>(opts: {
-    url: string;
-    description?: string;
-    headersSchema?: THeaders;
-    tools: TTools;
-}): McpServerDef<TTools, THeaders>;
-
-type SendInput = string | {
-    role?: UIMessage["role"];
-    parts: UIMessage["parts"];
-    id?: string;
-};
-type SessionOptions<TMcp extends McpConfig = {}, TTags extends TagsSchema = TagsSchema> = {
-    model?: GatewayModelId;
-    instructions?: string;
-    sandbox?: SandboxConfig | string;
-    tags?: TTags;
-    skillsDir?: SkillsDir;
-    mcp?: TMcp;
-};
-type McpToolCallBody = {
-    server: string;
-    tool: string;
-    input: unknown;
-    sessionId: string;
-    messageId: string;
-    hookToken: string;
-};
-type AgentOptions<TMcp extends McpConfig = {}> = {
-    storage?: StorageConfig;
-    generateId?: (resource: "session" | "message" | "part" | "sandbox" | "command") => string;
-} & SessionOptions<TMcp>;
-type TagsSchema = Record<string, unknown>;
-declare const agent: <TMcp extends McpConfig = {}>(options: AgentOptions<TMcp>) => {
-    session: <TTags extends TagsSchema = TagsSchema, TMcpOverride extends McpConfig = {}>(sessionId: string, sessionOptions?: SessionOptions<TMcp & TMcpOverride, TTags>) => Promise<{
-        send: ({ input, mcpContext, }: {
-            input: SendInput | SendInput[];
-            mcpContext?: McpSendOptions<TMcp & TMcpOverride>;
-        }) => Promise<void | SessionError>;
-        stream: (opts?: {
-            messageId?: string;
-        }) => Promise<SessionNotFoundError | SessionError | StorageError | ReadableStream<any>>;
-        ui: () => Promise<StorageError | {
-            messages: UIMessage<unknown, ai.UIDataTypes, ai.UITools>[];
-            streamingMessageId: string | null;
-        }>;
-        tag: {
-            list: () => Promise<SessionNotFoundError | StorageError | TTags>;
-            get: (key: string) => Promise<SessionNotFoundError | StorageError | TTags[string] | undefined>;
-            set: (key: string, value: unknown) => Promise<StorageError | undefined>;
-            setMany: (tags: Record<string, unknown>) => Promise<StorageError | undefined>;
-        };
-        sandbox: Sandbox;
-    }>;
-    sandbox: (sandboxId: string, config?: SandboxConfig) => Promise<Sandbox>;
-    storage: Storage;
-    handleMcpToolCall: (body: McpToolCallBody, handlers: Record<string, (opts: {
-        input: unknown;
-        headers: Record<string, string>;
-        body: McpToolCallBody;
-    }) => Promise<unknown>>) => Promise<{
-        result: unknown;
-    } | {
-        error: {
-            code: string;
-            message: string;
-        };
-    }>;
-};
-
-export { type AgentOptions, type Handlers, type RpcRequest, type RpcResponse, Sandbox, SandboxConfig, type SendInput, SessionNotFoundError, StorageConfig, StorageError, type TagsSchema, agent, handleStorageRpc, mcp };
+export { type Handlers, type RpcRequest, type RpcResponse, handleStorageRpc };