Skip to main content
Instead of injecting all tools at once, you can let the LLM discover and load tools on-demand using BM25 or semantic search.

​
ToolRouter

import { ToolRouter } from '@mcp-ts/sdk/shared';
import { embed } from 'ai';
import { openai } from '@ai-sdk/openai';

const router = new ToolRouter(client: MCPClient | MultiSessionClient, {
  // 'all' (default), 'search' (exposes meta-tools only), or 'groups'
  strategy: 'search',
  
  // Max tools to return from a search or group (default: 40)
  maxTools: 5,
  
  // Optional embedding function for semantic search
  embedFn: async (texts) => {
    const { embeddings } = await embed({
      model: openai.embedding('text-embedding-3-small'),
      values: texts,
    });
    return embeddings;
  },
  
  // Weight between keyword (BM25) and embedding search (0 to 1, default: 0.4)
  keywordWeight: 0.4,
});
Methods:
  • getFilteredTools() - Get tools based on current strategy
  • searchTools(query, topK?) - Search via BM25 + embeddings
  • searchToolsRegex(pattern, topK?) - Search via regex pattern
  • refresh() - Re-index tools from all connected clients
  • setStrategy(strategy) - Change tool selection strategy at runtime
Use the ToolRouter with adapters like AIAdapter: 
const adapter = new AIAdapter(client, {
  toolRouter: router
});
const tools = await adapter.getTools();

​
ToolIndex

Lightweight in-memory search index used internally by ToolRouter, or directly for specific custom discovery flows. 
import { ToolIndex } from '@mcp-ts/sdk/shared';

const index = new ToolIndex({
  embedFn: async (texts) => [/* ... */],
  keywordWeight: 0.4
});

await index.buildIndex(tools);

// Returns ToolSummary[] using BM25 and explicit/optional embeddings
const results = await index.search("query", 5);

// Regex search
const regexResults = index.searchRegex("get_.*_data", 5);

​
SchemaCompressor

Utility for reducing tool schema token overhead by yielding compact representations (name + description + inline parameterHint). 
import { SchemaCompressor } from '@mcp-ts/sdk/shared';

// Get a compact schema omitting full `inputSchema`
const compact = SchemaCompressor.toCompact(tool);

// Estimate token savings across all tools
const stats = SchemaCompressor.estimateSavings(tools);
console.log(stats.savingsPercent); // e.g., "82.3%"