Add LLMContextRouter

Author: cmsparks

.changeset/sharp-bears-appear.md

@@ -0,0 +1,5 @@
+---
+"agents": patch
+---
+
+Add ContextRouter to allow users to provide an LLM with a list of filtered tools and resources

packages/agents/package.json

@@ -68,6 +68,11 @@
       "types": "./dist/mcp/do-oauth-client-provider.d.ts",
       "require": "./dist/mcp/do-oauth-client-provider.js",
       "import": "./dist/mcp/do-oauth-client-provider.js"
+    },
+    "./mcp/context-router": {
+      "types": "./dist/mcp/context-router.d.ts",
+      "require": "./dist/mcp/context-router.js",
+      "import": "./dist/mcp/context-router.js"
     }
   },
   "keywords": [],

packages/agents/scripts/build.ts

@@ -9,6 +9,7 @@ async function main() {
       "src/mcp/index.ts",
       "src/mcp/client.ts",
       "src/mcp/do-oauth-client-provider.ts",
+      "src/mcp/context-router.ts",
     ],
     splitting: true,
     sourcemap: true,

packages/agents/src/mcp/client.ts

@@ -34,8 +34,10 @@ export class MCPClientManager {
   constructor(
     private _name: string,
     private _version: string,
-    public unstable_contextRouter = new BaseContextRouter(),
-  ) {}
+    public unstable_contextRouter: ContextRouter = new BaseContextRouter()
+  ) {
+    this.unstable_contextRouter.clientManager = this;
+  }
 
   /**
    * Connect to and register an MCP server
@@ -204,21 +206,21 @@ export class MCPClientManager {
    * @returns namespaced list of tools
    */
   listTools(): NamespacedData["tools"] {
-    return this.unstable_contextRouter.listTools(this)
+    return this.unstable_contextRouter.listTools();
   }
 
   /**
    * @returns a set of tools that you can use with the AI SDK
    */
   unstable_getAITools(): ToolSet {
-    return this.unstable_contextRouter.getAITools(this)
+    return this.unstable_contextRouter.getAITools();
   }
 
   /**
    * @returns namespaced list of tools
    */
   listResources(): NamespacedData["resources"] {
-    return this.unstable_contextRouter.listResources(this);
+    return this.unstable_contextRouter.listResources();
   }
 
   /**
@@ -288,7 +290,7 @@ export class MCPClientManager {
    * @param includeResources Whether to include resources in the prompt. This may be useful if you include a tool to fetch resources OR if the servers you connect to interact with resources.
    */
   unstable_systemPrompt(): string {
-    return this.unstable_contextRouter.systemPrompt(this)
+    return this.unstable_contextRouter.systemPrompt();
   }
 }