support headers in mcpServers

Author: charleslien

.agents/types/util-types.ts

@@ -218,6 +218,7 @@ export const mcpConfigRemoteSchema = z.strictObject({
   type: z.enum(['http', 'sse']).default('http'),
   url: z.string(),
   params: z.record(z.string(), z.string()).default(() => ({})),
+  headers: z.record(z.string(), z.string()).default(() => ({})),
 })
 
 export const mcpConfigSchema = z.union([

common/src/mcp/client.ts

@@ -62,9 +62,17 @@ export async function getMCPClient(config: MCPConfig): Promise<string> {
       url.searchParams.set(key, value)
     }
     if (config.type === 'http') {
-      transport = new StreamableHTTPClientTransport(url)
+      transport = new StreamableHTTPClientTransport(url, {
+        requestInit: {
+          headers: config.headers,
+        },
+      })
     } else if (config.type === 'sse') {
-      transport = new SSEClientTransport(url)
+      transport = new SSEClientTransport(url, {
+        requestInit: {
+          headers: config.headers,
+        },
+      })
     } else {
       config.type satisfies never
       throw new Error(`Internal error: invalid MCP config type ${config.type}`)

common/src/templates/initial-agents-dir/types/util-types.ts

@@ -212,6 +212,7 @@ export const mcpConfigStdioSchema = z.strictObject({
     .array()
     .default(() => []),
   env: z.record(z.string(), z.string()).default(() => ({})),
+  headers: z.record(z.string(), z.string()).default(() => ({})),
 })
 
 export const mcpConfigRemoteSchema = z.strictObject({

common/src/types/mcp.ts

@@ -14,6 +14,7 @@ export const mcpConfigRemoteSchema = z.strictObject({
   type: z.enum(['http', 'sse']).default('http'),
   url: z.string(),
   params: z.record(z.string(), z.string()).default(() => ({})),
+  headers: z.record(z.string(), z.string()).default(() => ({})),
 })
 
 export const mcpConfigSchema = z.union([