agents 0.0.0-956c772 → 0.0.0-96a8138

package/dist/index.js

@@ -1,22 +1,21 @@
 import {
   Agent,
   StreamingResponse,
-  WorkflowEntrypoint,
   getAgentByName,
+  getCurrentAgent,
   routeAgentEmail,
   routeAgentRequest,
-  unstable_callable,
-  unstable_context
-} from "./chunk-YMUU7QHV.js";
+  unstable_callable
+} from "./chunk-HD4VEHBA.js";
+import "./chunk-Q5ZBHY4Z.js";
 import "./chunk-HMLY7DHA.js";
 export {
   Agent,
   StreamingResponse,
-  WorkflowEntrypoint,
   getAgentByName,
+  getCurrentAgent,
   routeAgentEmail,
   routeAgentRequest,
-  unstable_callable,
-  unstable_context
+  unstable_callable
 };
 //# sourceMappingURL=index.js.map

package/dist/mcp/client.d.ts

@@ -1,14 +1,22 @@
 import * as zod from 'zod';
-import { Tool, Prompt, Resource, ResourceTemplate, ServerCapabilities, ClientCapabilities, CallToolRequest, CallToolResultSchema, CompatibilityCallToolResultSchema, ReadResourceRequest, GetPromptRequest } from '@modelcontextprotocol/sdk/types.js';
+import { ClientCapabilities, Tool, Prompt, Resource, ResourceTemplate, ServerCapabilities, CallToolRequest, CallToolResultSchema, CompatibilityCallToolResultSchema, ReadResourceRequest, GetPromptRequest } from '@modelcontextprotocol/sdk/types.js';
 import { Client } from '@modelcontextprotocol/sdk/client/index.js';
 import { SSEClientTransportOptions } from '@modelcontextprotocol/sdk/client/sse.js';
+import { AgentsOAuthProvider } from './do-oauth-client-provider.js';
 import { RequestOptions } from '@modelcontextprotocol/sdk/shared/protocol.js';
-import { OAuthClientProvider } from '@modelcontextprotocol/sdk/client/auth.js';
+import { ToolSet } from 'ai';
+import '@modelcontextprotocol/sdk/client/auth.js';
+import '@modelcontextprotocol/sdk/shared/auth.js';
 
 declare class MCPClientConnection {
     url: URL;
-    private info;
-    private options;
+    options: {
+        transport: SSEClientTransportOptions & {
+            authProvider?: AgentsOAuthProvider;
+        };
+        client: ConstructorParameters<typeof Client>[1];
+        capabilities: ClientCapabilities;
+    };
     client: Client;
     connectionState: "authenticating" | "connecting" | "ready" | "discovering" | "failed";
     instructions?: string;
@@ -18,7 +26,9 @@ declare class MCPClientConnection {
     resourceTemplates: ResourceTemplate[];
     serverCapabilities: ServerCapabilities | undefined;
     constructor(url: URL, info: ConstructorParameters<typeof Client>[0], options?: {
-        transport: SSEClientTransportOptions;
+        transport: SSEClientTransportOptions & {
+            authProvider?: AgentsOAuthProvider;
+        };
         client: ConstructorParameters<typeof Client>[1];
         capabilities: ClientCapabilities;
     });
@@ -50,10 +60,10 @@ declare class MCPClientConnection {
     }[]>;
     fetchResources(): Promise<{
         [x: string]: unknown;
-        name: string;
         uri: string;
-        description?: string | undefined;
+        name: string;
         mimeType?: string | undefined;
+        description?: string | undefined;
     }[]>;
     fetchPrompts(): Promise<{
         [x: string]: unknown;
@@ -70,33 +80,25 @@ declare class MCPClientConnection {
         [x: string]: unknown;
         name: string;
         uriTemplate: string;
-        description?: string | undefined;
         mimeType?: string | undefined;
+        description?: string | undefined;
     }[]>;
 }
 
-interface AgentsOAuthProvider extends OAuthClientProvider {
-    authUrl: string | undefined;
-    clientId: string | undefined;
-}
-
 /**
  * Utility class that aggregates multiple MCP clients into one
  */
 declare class MCPClientManager {
     private name;
     private version;
-    private auth?;
     mcpConnections: Record<string, MCPClientConnection>;
+    private callbackUrls;
     /**
      * @param name Name of the MCP client
      * @param version Version of the MCP Client
      * @param auth Auth paramters if being used to create a DurableObjectOAuthClientProvider
      */
-    constructor(name: string, version: string, auth?: {
-        baseCallbackUri: string;
-        storage: DurableObjectStorage;
-    } | undefined);
+    constructor(name: string, version: string);
     /**
      * Connect to and register an MCP server
      *
@@ -104,14 +106,14 @@ declare class MCPClientManager {
      * @param clientConfig Client config
      * @param capabilities Client capabilities (i.e. if the client supports roots/sampling)
      */
-    connect(url: string, opts?: {
+    connect(url: string, options?: {
         reconnect?: {
             id: string;
             oauthClientId?: string;
             oauthCode?: string;
         };
         transport?: SSEClientTransportOptions & {
-            authProvider: AgentsOAuthProvider;
+            authProvider?: AgentsOAuthProvider;
         };
         client?: ConstructorParameters<typeof Client>[1];
         capabilities?: ClientCapabilities;
@@ -127,6 +129,19 @@ declare class MCPClientManager {
      * @returns namespaced list of tools
      */
     listTools(): NamespacedData["tools"];
+    /**
+     * @returns a set of tools that you can use with the AI SDK
+     */
+    unstable_getAITools(): ToolSet;
+    /**
+     * Closes all connections to MCP servers
+     */
+    closeAllConnections(): Promise<void[]>;
+    /**
+     * Closes a connection to an MCP server
+     * @param id The id of the connection to close
+     */
+    closeConnection(id: string): Promise<void>;
     /**
      * @returns namespaced list of prompts
      */
@@ -144,9 +159,9 @@ declare class MCPClientManager {
      */
     callTool(params: CallToolRequest["params"] & {
         serverId: string;
-    }, resultSchema: typeof CallToolResultSchema | typeof CompatibilityCallToolResultSchema, options: RequestOptions): Promise<zod.objectOutputType<zod.objectUtil.extendShape<{
+    }, resultSchema?: typeof CallToolResultSchema | typeof CompatibilityCallToolResultSchema, options?: RequestOptions): Promise<zod.objectOutputType<{
         _meta: zod.ZodOptional<zod.ZodObject<{}, "passthrough", zod.ZodTypeAny, zod.objectOutputType<{}, zod.ZodTypeAny, "passthrough">, zod.objectInputType<{}, zod.ZodTypeAny, "passthrough">>>;
-    }, {
+    } & {
         content: zod.ZodArray<zod.ZodUnion<[zod.ZodObject<{
             type: zod.ZodLiteral<"text">;
             text: zod.ZodString;
@@ -168,6 +183,18 @@ declare class MCPClientManager {
             type: zod.ZodLiteral<"image">;
             data: zod.ZodString;
             mimeType: zod.ZodString;
+        }, zod.ZodTypeAny, "passthrough">>, zod.ZodObject<{
+            type: zod.ZodLiteral<"audio">;
+            data: zod.ZodString;
+            mimeType: zod.ZodString;
+        }, "passthrough", zod.ZodTypeAny, zod.objectOutputType<{
+            type: zod.ZodLiteral<"audio">;
+            data: zod.ZodString;
+            mimeType: zod.ZodString;
+        }, zod.ZodTypeAny, "passthrough">, zod.objectInputType<{
+            type: zod.ZodLiteral<"audio">;
+            data: zod.ZodString;
+            mimeType: zod.ZodString;
         }, zod.ZodTypeAny, "passthrough">>, zod.ZodObject<{
             type: zod.ZodLiteral<"resource">;
             resource: zod.ZodUnion<[zod.ZodObject<zod.objectUtil.extendShape<{
@@ -269,19 +296,19 @@ declare class MCPClientManager {
             }>, zod.ZodTypeAny, "passthrough">>]>;
         }, zod.ZodTypeAny, "passthrough">>]>, "many">;
         isError: zod.ZodOptional<zod.ZodDefault<zod.ZodBoolean>>;
-    }>, zod.ZodTypeAny, "passthrough"> | zod.objectOutputType<zod.objectUtil.extendShape<{
+    }, zod.ZodTypeAny, "passthrough"> | zod.objectOutputType<{
         _meta: zod.ZodOptional<zod.ZodObject<{}, "passthrough", zod.ZodTypeAny, zod.objectOutputType<{}, zod.ZodTypeAny, "passthrough">, zod.objectInputType<{}, zod.ZodTypeAny, "passthrough">>>;
-    }, {
+    } & {
         toolResult: zod.ZodUnknown;
-    }>, zod.ZodTypeAny, "passthrough">>;
+    }, zod.ZodTypeAny, "passthrough">>;
     /**
      * Namespaced version of readResource
      */
     readResource(params: ReadResourceRequest["params"] & {
         serverId: string;
-    }, options: RequestOptions): Promise<zod.objectOutputType<zod.objectUtil.extendShape<{
+    }, options: RequestOptions): Promise<zod.objectOutputType<{
         _meta: zod.ZodOptional<zod.ZodObject<{}, "passthrough", zod.ZodTypeAny, zod.objectOutputType<{}, zod.ZodTypeAny, "passthrough">, zod.objectInputType<{}, zod.ZodTypeAny, "passthrough">>>;
-    }, {
+    } & {
         contents: zod.ZodArray<zod.ZodUnion<[zod.ZodObject<zod.objectUtil.extendShape<{
             uri: zod.ZodString;
             mimeType: zod.ZodOptional<zod.ZodString>;
@@ -313,15 +340,15 @@ declare class MCPClientManager {
         }, {
             blob: zod.ZodString;
         }>, zod.ZodTypeAny, "passthrough">>]>, "many">;
-    }>, zod.ZodTypeAny, "passthrough">>;
+    }, zod.ZodTypeAny, "passthrough">>;
     /**
      * Namespaced version of getPrompt
      */
     getPrompt(params: GetPromptRequest["params"] & {
         serverId: string;
-    }, options: RequestOptions): Promise<zod.objectOutputType<zod.objectUtil.extendShape<{
+    }, options: RequestOptions): Promise<zod.objectOutputType<{
         _meta: zod.ZodOptional<zod.ZodObject<{}, "passthrough", zod.ZodTypeAny, zod.objectOutputType<{}, zod.ZodTypeAny, "passthrough">, zod.objectInputType<{}, zod.ZodTypeAny, "passthrough">>>;
-    }, {
+    } & {
         description: zod.ZodOptional<zod.ZodString>;
         messages: zod.ZodArray<zod.ZodObject<{
             role: zod.ZodEnum<["user", "assistant"]>;
@@ -346,6 +373,18 @@ declare class MCPClientManager {
                 type: zod.ZodLiteral<"image">;
                 data: zod.ZodString;
                 mimeType: zod.ZodString;
+            }, zod.ZodTypeAny, "passthrough">>, zod.ZodObject<{
+                type: zod.ZodLiteral<"audio">;
+                data: zod.ZodString;
+                mimeType: zod.ZodString;
+            }, "passthrough", zod.ZodTypeAny, zod.objectOutputType<{
+                type: zod.ZodLiteral<"audio">;
+                data: zod.ZodString;
+                mimeType: zod.ZodString;
+            }, zod.ZodTypeAny, "passthrough">, zod.objectInputType<{
+                type: zod.ZodLiteral<"audio">;
+                data: zod.ZodString;
+                mimeType: zod.ZodString;
             }, zod.ZodTypeAny, "passthrough">>, zod.ZodObject<{
                 type: zod.ZodLiteral<"resource">;
                 resource: zod.ZodUnion<[zod.ZodObject<zod.objectUtil.extendShape<{
@@ -469,6 +508,18 @@ declare class MCPClientManager {
                 type: zod.ZodLiteral<"image">;
                 data: zod.ZodString;
                 mimeType: zod.ZodString;
+            }, zod.ZodTypeAny, "passthrough">>, zod.ZodObject<{
+                type: zod.ZodLiteral<"audio">;
+                data: zod.ZodString;
+                mimeType: zod.ZodString;
+            }, "passthrough", zod.ZodTypeAny, zod.objectOutputType<{
+                type: zod.ZodLiteral<"audio">;
+                data: zod.ZodString;
+                mimeType: zod.ZodString;
+            }, zod.ZodTypeAny, "passthrough">, zod.objectInputType<{
+                type: zod.ZodLiteral<"audio">;
+                data: zod.ZodString;
+                mimeType: zod.ZodString;
             }, zod.ZodTypeAny, "passthrough">>, zod.ZodObject<{
                 type: zod.ZodLiteral<"resource">;
                 resource: zod.ZodUnion<[zod.ZodObject<zod.objectUtil.extendShape<{
@@ -592,6 +643,18 @@ declare class MCPClientManager {
                 type: zod.ZodLiteral<"image">;
                 data: zod.ZodString;
                 mimeType: zod.ZodString;
+            }, zod.ZodTypeAny, "passthrough">>, zod.ZodObject<{
+                type: zod.ZodLiteral<"audio">;
+                data: zod.ZodString;
+                mimeType: zod.ZodString;
+            }, "passthrough", zod.ZodTypeAny, zod.objectOutputType<{
+                type: zod.ZodLiteral<"audio">;
+                data: zod.ZodString;
+                mimeType: zod.ZodString;
+            }, zod.ZodTypeAny, "passthrough">, zod.objectInputType<{
+                type: zod.ZodLiteral<"audio">;
+                data: zod.ZodString;
+                mimeType: zod.ZodString;
             }, zod.ZodTypeAny, "passthrough">>, zod.ZodObject<{
                 type: zod.ZodLiteral<"resource">;
                 resource: zod.ZodUnion<[zod.ZodObject<zod.objectUtil.extendShape<{
@@ -693,7 +756,7 @@ declare class MCPClientManager {
                 }>, zod.ZodTypeAny, "passthrough">>]>;
             }, zod.ZodTypeAny, "passthrough">>]>;
         }, zod.ZodTypeAny, "passthrough">>, "many">;
-    }>, zod.ZodTypeAny, "passthrough">>;
+    }, zod.ZodTypeAny, "passthrough">>;
 }
 type NamespacedData = {
     tools: (Tool & {