@strav/brain 0.4.27 → 0.4.28

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@strav/brain",
-  "version": "0.4.27",
+  "version": "0.4.28",
   "type": "module",
   "description": "AI module for the Strav framework",
   "license": "MIT",
@@ -15,12 +15,16 @@
     "CHANGELOG.md"
   ],
   "peerDependencies": {
-    "@strav/kernel": "0.4.27"
+    "@strav/kernel": "0.4.28"
   },
   "dependencies": {
-    "@strav/workflow": "0.4.27",
+    "@strav/mcp": "0.4.28",
+    "@strav/workflow": "0.4.28",
     "zod": "^3.25 || ^4.0"
   },
+  "devDependencies": {
+    "@strav/http": "0.4.28"
+  },
   "scripts": {
     "test": "bun test tests/",
     "typecheck": "tsc --noEmit"

package/src/index.ts

@@ -5,6 +5,8 @@ export { default as BrainProvider } from './brain_provider.ts'
 export { brain, AgentRunner, Thread } from './helpers.ts'
 export { Agent } from './agent.ts'
 export { defineTool, defineToolbox } from './tool.ts'
+export { defineMcpToolbox } from './mcp_toolbox.ts'
+export type { McpToolboxOptions } from './mcp_toolbox.ts'
 export { Workflow } from './workflow.ts'
 export { AnthropicProvider } from './providers/anthropic_provider.ts'
 export { GoogleProvider } from './providers/google_provider.ts'

package/src/mcp_toolbox.ts

@@ -0,0 +1,62 @@
+import { McpClient } from '@strav/mcp'
+import type { McpClientOptions } from '@strav/mcp'
+import type { ToolDefinition, JsonSchema } from './types.ts'
+
+/** Options for {@link defineMcpToolbox}. */
+export interface McpToolboxOptions extends McpClientOptions {
+  /** Restrict the toolbox to these tool names. Default: every remote tool. */
+  only?: string[]
+}
+
+/**
+ * Build a brain toolbox backed by a remote MCP server.
+ *
+ * Connects an {@link McpClient}, lists the server's tools, and maps each one
+ * to a brain `ToolDefinition` whose `execute` performs the remote call — so a
+ * `brain` `Agent` calls remote MCP tools exactly as it calls native ones.
+ *
+ * This is async (it must `listTools()` first), so await it at setup time:
+ *
+ * @example
+ * const gateway = await defineMcpToolbox('gateway', {
+ *   url: 'https://gateway/mcp',
+ *   bearerToken: projectToken,
+ * })
+ *
+ * class BuilderAgent extends Agent {
+ *   tools = gateway
+ * }
+ */
+export async function defineMcpToolbox(
+  name: string,
+  options: McpToolboxOptions
+): Promise<ToolDefinition[]> {
+  const { only, ...clientOptions } = options
+
+  const client = new McpClient({
+    clientInfo: { name: `strav-brain:${name}`, version: '1.0.0' },
+    ...clientOptions,
+  })
+
+  const tools = await client.listTools()
+  const selected = only ? tools.filter(tool => only.includes(tool.name)) : tools
+
+  return selected.map(tool => ({
+    name: tool.name,
+    description: tool.description ?? '',
+    parameters: tool.inputSchema as JsonSchema,
+    execute: async (args: Record<string, unknown>) => {
+      const result = await client.callTool(tool.name, args)
+
+      // Prefer structured output; otherwise join the text content blocks.
+      if (result.structuredContent !== undefined) {
+        return result.isError ? { error: result.structuredContent } : result.structuredContent
+      }
+      const text = (result.content ?? [])
+        .map((block: any) => (block?.type === 'text' ? String(block.text ?? '') : ''))
+        .filter(Boolean)
+        .join('\n')
+      return result.isError ? { error: text } : text
+    },
+  }))
+}