Mcp (#175)

* chore: upgraded packages

* feat: mcp

* feat: set enviroment vars

* fix: various

* fix: various

* fix: updates

* fix: updates

Author: dosco

.cspell.json

@@ -10,6 +10,7 @@
   "ignorePaths": [
     "node_modules",
     "./src/docs/src/content/docs/03-apidocs",
-    "./src/ax/dsp/stopwords.ts"
+    "./src/ax/dsp/stopwords.ts",
+    "./src/examples/*.ts"
   ]
 }

.cspell/project-words.txt

@@ -42,6 +42,7 @@ Logprob
 logprobs
 Logprobs
 Macbook
+MCPHTTP
 minilm
 Mixtral
 modelinfo

README.md

@@ -12,6 +12,7 @@ Use Ax and get an end-to-end streaming, multi-modal DSPy framework with agents a
 - Prompts auto-generated from simple signatures
 - Full native end-to-end streaming
 - Build Agents that can call other agents
+- Built in MCP, Model Context Protocol support
 - Convert docs of any format to text
 - RAG, smart chunking, embedding, querying
 - Works with Vercel AI SDK
@@ -402,6 +403,43 @@ The load balancer is ideal for high availability while the router is perfect whe
 
 You can also use the balancer and the router together either the multiple balancers can be used with the router or the router can be used with the balancer.
 
+## Model Context Protocol (MCP)
+
+Ax provides seamless integration with the Model Context Protocol (MCP), allowing your agents to access external tools, and resources through a standardized interface.
+
+### Using AxMCPClient
+
+The `AxMCPClient` allows you to connect to any MCP-compatible server and use its capabilities within your Ax agents:
+
+```typescript
+import { AxMCPClient, AxMCPStdioTransport } from '@ax-llm/ax'
+
+// Initialize an MCP client with a transport
+const transport = new AxMCPStdioTransport({
+  command: 'npx',
+  args: ['-y', '@modelcontextprotocol/server-memory'],
+})
+
+// Create the client with optional debug mode
+const client = new AxMCPClient(transport, { debug: true })
+
+// Initialize the connection
+await client.init()
+
+// Use the client's functions in an agent
+const memoryAgent = new AxAgent({
+  name: 'MemoryAssistant',
+  description: 'An assistant with persistent memory',
+  signature: 'input, userId -> response',
+  functions: [client], // Pass the client as a function provider
+})
+
+// Or use the client with AxGen
+const memoryGen = new AxGen('input, userId -> response', {
+    functions: [client]
+})
+```
+
 ## Vercel AI SDK Integration
 
 Install the ax provider package
@@ -631,6 +669,7 @@ OPENAI_APIKEY=openai_key npm run tsx ./src/examples/marketing.ts
 | docker.ts           | Use the docker sandbox to find files by description     |
 | prime.ts            | Using field processors to process fields in a prompt    |
 | simple-classify.ts  | Use a simple classifier to classify stuff               |
+| mcp-client.ts       | Example of using an MCP server with Ax |
 
 ## Our Goal