@amodalai/runtime 0.1.26 → 0.2.0

package/dist/src/tools/mcp-tool-adapter.test.js

@@ -0,0 +1,227 @@
+/**
+ * @license
+ * Copyright 2026 Amodal Labs, Inc.
+ * SPDX-License-Identifier: MIT
+ */
+/**
+ * MCP Tool Adapter Tests
+ *
+ * Tests the conversion of MCP discovered tools to ToolDefinition objects:
+ * 1. Tool creation with JSON Schema passthrough
+ * 2. Tool execution via McpManager.callTool()
+ * 3. Registration on ToolRegistry
+ * 4. Error classification (connection, auth, generic)
+ */
+import { describe, it, expect, vi, beforeEach } from 'vitest';
+import { createMcpToolDefinition, registerMcpTools } from './mcp-tool-adapter.js';
+import { createToolRegistry } from './registry.js';
+import { ConnectionError } from '../errors.js';
+// ---------------------------------------------------------------------------
+// Helpers
+// ---------------------------------------------------------------------------
+function makeMockLogger() {
+    return {
+        trace: vi.fn(),
+        debug: vi.fn(),
+        info: vi.fn(),
+        warn: vi.fn(),
+        error: vi.fn(),
+        fatal: vi.fn(),
+        child: vi.fn().mockReturnThis(),
+    };
+}
+function makeMockContext(overrides) {
+    return {
+        request: vi.fn(),
+        store: vi.fn(),
+        env: vi.fn(),
+        log: vi.fn(),
+        user: { roles: [] },
+        signal: AbortSignal.timeout(10000),
+        sessionId: 'test-session',
+        tenantId: 'test-tenant',
+        ...overrides,
+    };
+}
+function makeMockMcpManager(overrides) {
+    return {
+        getDiscoveredTools: overrides?.getDiscoveredTools ?? (() => []),
+        callTool: overrides?.callTool ?? vi.fn().mockResolvedValue({
+            content: [{ type: 'text', text: 'mock result' }],
+        }),
+        isMcpTool: vi.fn(),
+        startServers: vi.fn(),
+        getServerInfo: vi.fn(),
+        shutdown: vi.fn(),
+    };
+}
+// ---------------------------------------------------------------------------
+// 1. Tool creation with JSON Schema passthrough
+// ---------------------------------------------------------------------------
+describe('createMcpToolDefinition', () => {
+    let logger;
+    beforeEach(() => {
+        logger = makeMockLogger();
+    });
+    it('creates a ToolDefinition with correct metadata', () => {
+        const tool = {
+            name: 'github__search_repos',
+            originalName: 'search_repos',
+            serverName: 'github',
+            description: 'Search GitHub repositories',
+            parameters: { type: 'object', properties: { query: { type: 'string' } }, required: ['query'] },
+        };
+        const mcpManager = makeMockMcpManager();
+        const def = createMcpToolDefinition(tool, mcpManager, logger);
+        expect(def.description).toBe('Search GitHub repositories');
+        expect(def.readOnly).toBe(false);
+        expect(def.metadata).toEqual({
+            category: 'mcp',
+            connection: 'github',
+            originalName: 'search_repos',
+        });
+        // Parameters should be defined (jsonSchema passthrough)
+        expect(def.parameters).toBeDefined();
+    });
+    it('execute calls mcpManager.callTool with correct args', async () => {
+        const callTool = vi.fn().mockResolvedValue({
+            content: [{ type: 'text', text: '{"repos": [{"name": "amodal"}]}' }],
+        });
+        const mcpManager = makeMockMcpManager({ callTool });
+        const tool = {
+            name: 'github__search',
+            originalName: 'search',
+            serverName: 'github',
+            description: 'Search',
+            parameters: { type: 'object', properties: { q: { type: 'string' } } },
+        };
+        const def = createMcpToolDefinition(tool, mcpManager, logger);
+        const ctx = makeMockContext();
+        const result = await def.execute({ q: 'test' }, ctx);
+        expect(callTool).toHaveBeenCalledWith('github__search', { q: 'test' });
+        expect(result).toEqual({ output: '{"repos": [{"name": "amodal"}]}' });
+        // Verify structured logging
+        expect(logger.debug).toHaveBeenCalledWith('mcp_tool_call_start', expect.objectContaining({
+            tool: 'github__search',
+            server: 'github',
+        }));
+        expect(logger.info).toHaveBeenCalledWith('mcp_tool_call_complete', expect.objectContaining({
+            tool: 'github__search',
+            isError: false,
+            durationMs: expect.any(Number),
+        }));
+    });
+    it('returns error object when MCP tool signals an error', async () => {
+        const mcpManager = makeMockMcpManager({
+            callTool: vi.fn().mockResolvedValue({
+                content: [{ type: 'text', text: 'Permission denied' }],
+                isError: true,
+            }),
+        });
+        const tool = {
+            name: 'server__action',
+            originalName: 'action',
+            serverName: 'server',
+            description: 'Do action',
+            parameters: { type: 'object' },
+        };
+        const def = createMcpToolDefinition(tool, mcpManager, logger);
+        const result = await def.execute({}, makeMockContext());
+        expect(result).toEqual({ error: 'Permission denied' });
+        expect(logger.info).toHaveBeenCalledWith('mcp_tool_call_complete', expect.objectContaining({
+            isError: true,
+        }));
+    });
+    it('throws ConnectionError on ECONNREFUSED', async () => {
+        const mcpManager = makeMockMcpManager({
+            callTool: vi.fn().mockRejectedValue(new Error('connect ECONNREFUSED 127.0.0.1:3000')),
+        });
+        const tool = {
+            name: 'broken__tool',
+            originalName: 'tool',
+            serverName: 'broken',
+            description: 'Broken',
+            parameters: { type: 'object' },
+        };
+        const def = createMcpToolDefinition(tool, mcpManager, logger);
+        await expect(def.execute({}, makeMockContext())).rejects.toThrow(ConnectionError);
+        await expect(def.execute({}, makeMockContext())).rejects.toThrow(/unreachable/);
+        expect(logger.error).toHaveBeenCalledWith('mcp_tool_call_failed', expect.objectContaining({
+            tool: 'broken__tool',
+        }));
+    });
+    it('throws ConnectionError on auth failure', async () => {
+        const mcpManager = makeMockMcpManager({
+            callTool: vi.fn().mockRejectedValue(new Error('401 Unauthorized')),
+        });
+        const tool = {
+            name: 'secure__tool',
+            originalName: 'tool',
+            serverName: 'secure',
+            description: 'Secure',
+            parameters: { type: 'object' },
+        };
+        const def = createMcpToolDefinition(tool, mcpManager, logger);
+        await expect(def.execute({}, makeMockContext())).rejects.toThrow(ConnectionError);
+        await expect(def.execute({}, makeMockContext())).rejects.toThrow(/authentication failed/i);
+    });
+    it('throws ConnectionError when ctx.signal is already aborted (timeout)', async () => {
+        const mcpManager = makeMockMcpManager({
+            callTool: vi.fn().mockImplementation(() => new Promise(() => { })), // never resolves
+        });
+        const tool = {
+            name: 'slow__tool',
+            originalName: 'tool',
+            serverName: 'slow',
+            description: 'Slow',
+            parameters: { type: 'object' },
+        };
+        const def = createMcpToolDefinition(tool, mcpManager, logger);
+        const abortController = new AbortController();
+        abortController.abort();
+        const ctx = makeMockContext({ signal: abortController.signal });
+        await expect(def.execute({}, ctx)).rejects.toThrow(ConnectionError);
+        await expect(def.execute({}, ctx)).rejects.toThrow(/timed out/);
+    });
+});
+// ---------------------------------------------------------------------------
+// 3. Registration on ToolRegistry
+// ---------------------------------------------------------------------------
+describe('registerMcpTools', () => {
+    it('registers all discovered tools on the registry', () => {
+        const logger = makeMockLogger();
+        const mcpManager = makeMockMcpManager({
+            getDiscoveredTools: () => [
+                { name: 'github__search', originalName: 'search', serverName: 'github', description: 'Search repos', parameters: { type: 'object', properties: { q: { type: 'string' } } } },
+                { name: 'github__get_repo', originalName: 'get_repo', serverName: 'github', description: 'Get repo', parameters: { type: 'object', properties: { owner: { type: 'string' }, repo: { type: 'string' } } } },
+                { name: 'slack__send', originalName: 'send', serverName: 'slack', description: 'Send message', parameters: { type: 'object', properties: { channel: { type: 'string' }, text: { type: 'string' } } } },
+            ],
+        });
+        const registry = createToolRegistry();
+        const count = registerMcpTools(registry, mcpManager, logger);
+        expect(count).toBe(3);
+        expect(registry.names()).toContain('github__search');
+        expect(registry.names()).toContain('github__get_repo');
+        expect(registry.names()).toContain('slack__send');
+        expect(registry.size).toBe(3);
+        // Each tool should have correct metadata
+        const searchTool = registry.get('github__search');
+        expect(searchTool).toBeDefined();
+        expect(searchTool.metadata?.category).toBe('mcp');
+        expect(searchTool.metadata?.connection).toBe('github');
+        // Log event
+        expect(logger.info).toHaveBeenCalledWith('mcp_tools_registered', expect.objectContaining({
+            count: 3,
+        }));
+    });
+    it('returns 0 and does not log when no tools discovered', () => {
+        const logger = makeMockLogger();
+        const mcpManager = makeMockMcpManager({ getDiscoveredTools: () => [] });
+        const registry = createToolRegistry();
+        const count = registerMcpTools(registry, mcpManager, logger);
+        expect(count).toBe(0);
+        expect(registry.size).toBe(0);
+        expect(logger.info).not.toHaveBeenCalledWith('mcp_tools_registered', expect.anything());
+    });
+});
+//# sourceMappingURL=mcp-tool-adapter.test.js.map

package/dist/src/tools/mcp-tool-adapter.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"mcp-tool-adapter.test.js","sourceRoot":"","sources":["../../../src/tools/mcp-tool-adapter.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH;;;;;;;;GAQG;AAEH,OAAO,EAAC,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAC,MAAM,QAAQ,CAAC;AAE5D,OAAO,EAAC,uBAAuB,EAAE,gBAAgB,EAAC,MAAM,uBAAuB,CAAC;AAChF,OAAO,EAAC,kBAAkB,EAAC,MAAM,eAAe,CAAC;AACjD,OAAO,EAAC,eAAe,EAAC,MAAM,cAAc,CAAC;AAG7C,8EAA8E;AAC9E,UAAU;AACV,8EAA8E;AAE9E,SAAS,cAAc;IACrB,OAAO;QACL,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE;QACd,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE;QACd,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE;QACb,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE;QACb,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE;QACd,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE;QACd,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,cAAc,EAAE;KAChC,CAAC;AACJ,CAAC;AAED,SAAS,eAAe,CAAC,SAAgC;IACvD,OAAO;QACL,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE;QAChB,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE;QACd,GAAG,EAAE,EAAE,CAAC,EAAE,EAAE;QACZ,GAAG,EAAE,EAAE,CAAC,EAAE,EAAE;QACZ,IAAI,EAAE,EAAC,KAAK,EAAE,EAAE,EAAC;QACjB,MAAM,EAAE,WAAW,CAAC,OAAO,CAAC,KAAK,CAAC;QAClC,SAAS,EAAE,cAAc;QACzB,QAAQ,EAAE,aAAa;QACvB,GAAG,SAAS;KACb,CAAC;AACJ,CAAC;AAED,SAAS,kBAAkB,CAAC,SAG3B;IACC,OAAO;QACL,kBAAkB,EAAE,SAAS,EAAE,kBAAkB,IAAI,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC;QAC/D,QAAQ,EAAE,SAAS,EAAE,QAAQ,IAAI,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC;YACzD,OAAO,EAAE,CAAC,EAAC,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,aAAa,EAAC,CAAC;SAC/C,CAAC;QACF,SAAS,EAAE,EAAE,CAAC,EAAE,EAAE;QAClB,YAAY,EAAE,EAAE,CAAC,EAAE,EAAE;QACrB,aAAa,EAAE,EAAE,CAAC,EAAE,EAAE;QACtB,QAAQ,EAAE,EAAE,CAAC,EAAE,EAAE;KACO,CAAC;AAC7B,CAAC;AAED,8EAA8E;AAC9E,gDAAgD;AAChD,8EAA8E;AAE9E,QAAQ,CAAC,yBAAyB,EAAE,GAAG,EAAE;IACvC,IAAI,MAAyC,CAAC;IAE9C,UAAU,CAAC,GAAG,EAAE;QACd,MAAM,GAAG,cAAc,EAAE,CAAC;IAC5B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gDAAgD,EAAE,GAAG,EAAE;QACxD,MAAM,IAAI,GAAsB;YAC9B,IAAI,EAAE,sBAAsB;YAC5B,YAAY,EAAE,cAAc;YAC5B,UAAU,EAAE,QAAQ;YACpB,WAAW,EAAE,4BAA4B;YACzC,UAAU,EAAE,EAAC,IAAI,EAAE,QAAQ,EAAE,UAAU,EAAE,EAAC,KAAK,EAAE,EAAC,IAAI,EAAE,QAAQ,EAAC,EAAC,EAAE,QAAQ,EAAE,CAAC,OAAO,CAAC,EAAC;SACzF,CAAC;QAEF,MAAM,UAAU,GAAG,kBAAkB,EAAE,CAAC;QACxC,MAAM,GAAG,GAAG,uBAAuB,CAAC,IAAI,EAAE,UAAU,EAAE,MAAM,CAAC,CAAC;QAE9D,MAAM,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,4BAA4B,CAAC,CAAC;QAC3D,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACjC,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC;YAC3B,QAAQ,EAAE,KAAK;YACf,UAAU,EAAE,QAAQ;YACpB,YAAY,EAAE,cAAc;SAC7B,CAAC,CAAC;QACH,wDAAwD;QACxD,MAAM,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC,WAAW,EAAE,CAAC;IACvC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;QACnE,MAAM,QAAQ,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC;YACzC,OAAO,EAAE,CAAC,EAAC,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,iCAAiC,EAAC,CAAC;SACnE,CAAC,CAAC;QAEH,MAAM,UAAU,GAAG,kBAAkB,CAAC,EAAC,QAAQ,EAAC,CAAC,CAAC;QAElD,MAAM,IAAI,GAAsB;YAC9B,IAAI,EAAE,gBAAgB;YACtB,YAAY,EAAE,QAAQ;YACtB,UAAU,EAAE,QAAQ;YACpB,WAAW,EAAE,QAAQ;YACrB,UAAU,EAAE,EAAC,IAAI,EAAE,QAAQ,EAAE,UAAU,EAAE,EAAC,CAAC,EAAE,EAAC,IAAI,EAAE,QAAQ,EAAC,EAAC,EAAC;SAChE,CAAC;QAEF,MAAM,GAAG,GAAG,uBAAuB,CAAC,IAAI,EAAE,UAAU,EAAE,MAAM,CAAC,CAAC;QAC9D,MAAM,GAAG,GAAG,eAAe,EAAE,CAAC;QAC9B,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,OAAO,CAAC,EAAC,CAAC,EAAE,MAAM,EAAC,EAAE,GAAG,CAAC,CAAC;QAEnD,MAAM,CAAC,QAAQ,CAAC,CAAC,oBAAoB,CAAC,gBAAgB,EAAE,EAAC,CAAC,EAAE,MAAM,EAAC,CAAC,CAAC;QACrE,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAC,MAAM,EAAE,iCAAiC,EAAC,CAAC,CAAC;QAEpE,4BAA4B;QAC5B,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,oBAAoB,CAAC,qBAAqB,EAAE,MAAM,CAAC,gBAAgB,CAAC;YACvF,IAAI,EAAE,gBAAgB;YACtB,MAAM,EAAE,QAAQ;SACjB,CAAC,CAAC,CAAC;QACJ,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,oBAAoB,CAAC,wBAAwB,EAAE,MAAM,CAAC,gBAAgB,CAAC;YACzF,IAAI,EAAE,gBAAgB;YACtB,OAAO,EAAE,KAAK;YACd,UAAU,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC;SAC/B,CAAC,CAAC,CAAC;IACN,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;QACnE,MAAM,UAAU,GAAG,kBAAkB,CAAC;YACpC,QAAQ,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC;gBAClC,OAAO,EAAE,CAAC,EAAC,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,mBAAmB,EAAC,CAAC;gBACpD,OAAO,EAAE,IAAI;aACd,CAAC;SACH,CAAC,CAAC;QAEH,MAAM,IAAI,GAAsB;YAC9B,IAAI,EAAE,gBAAgB;YACtB,YAAY,EAAE,QAAQ;YACtB,UAAU,EAAE,QAAQ;YACpB,WAAW,EAAE,WAAW;YACxB,UAAU,EAAE,EAAC,IAAI,EAAE,QAAQ,EAAC;SAC7B,CAAC;QAEF,MAAM,GAAG,GAAG,uBAAuB,CAAC,IAAI,EAAE,UAAU,EAAE,MAAM,CAAC,CAAC;QAC9D,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,OAAO,CAAC,EAAE,EAAE,eAAe,EAAE,CAAC,CAAC;QAExD,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAC,KAAK,EAAE,mBAAmB,EAAC,CAAC,CAAC;QACrD,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,oBAAoB,CAAC,wBAAwB,EAAE,MAAM,CAAC,gBAAgB,CAAC;YACzF,OAAO,EAAE,IAAI;SACd,CAAC,CAAC,CAAC;IACN,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wCAAwC,EAAE,KAAK,IAAI,EAAE;QACtD,MAAM,UAAU,GAAG,kBAAkB,CAAC;YACpC,QAAQ,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,IAAI,KAAK,CAAC,qCAAqC,CAAC,CAAC;SACtF,CAAC,CAAC;QAEH,MAAM,IAAI,GAAsB;YAC9B,IAAI,EAAE,cAAc;YACpB,YAAY,EAAE,MAAM;YACpB,UAAU,EAAE,QAAQ;YACpB,WAAW,EAAE,QAAQ;YACrB,UAAU,EAAE,EAAC,IAAI,EAAE,QAAQ,EAAC;SAC7B,CAAC;QAEF,MAAM,GAAG,GAAG,uBAAuB,CAAC,IAAI,EAAE,UAAU,EAAE,MAAM,CAAC,CAAC;QAC9D,MAAM,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,EAAE,eAAe,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC;QAClF,MAAM,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,EAAE,eAAe,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC;QAEhF,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,oBAAoB,CAAC,sBAAsB,EAAE,MAAM,CAAC,gBAAgB,CAAC;YACxF,IAAI,EAAE,cAAc;SACrB,CAAC,CAAC,CAAC;IACN,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wCAAwC,EAAE,KAAK,IAAI,EAAE;QACtD,MAAM,UAAU,GAAG,kBAAkB,CAAC;YACpC,QAAQ,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,IAAI,KAAK,CAAC,kBAAkB,CAAC,CAAC;SACnE,CAAC,CAAC;QAEH,MAAM,IAAI,GAAsB;YAC9B,IAAI,EAAE,cAAc;YACpB,YAAY,EAAE,MAAM;YACpB,UAAU,EAAE,QAAQ;YACpB,WAAW,EAAE,QAAQ;YACrB,UAAU,EAAE,EAAC,IAAI,EAAE,QAAQ,EAAC;SAC7B,CAAC;QAEF,MAAM,GAAG,GAAG,uBAAuB,CAAC,IAAI,EAAE,UAAU,EAAE,MAAM,CAAC,CAAC;QAC9D,MAAM,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,EAAE,eAAe,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC;QAClF,MAAM,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,EAAE,eAAe,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,wBAAwB,CAAC,CAAC;IAC7F,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qEAAqE,EAAE,KAAK,IAAI,EAAE;QACnF,MAAM,UAAU,GAAG,kBAAkB,CAAC;YACpC,QAAQ,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,kBAAkB,CAAC,GAAG,EAAE,CAAC,IAAI,OAAO,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC,EAAE,iBAAiB;SACrF,CAAC,CAAC;QAEH,MAAM,IAAI,GAAsB;YAC9B,IAAI,EAAE,YAAY;YAClB,YAAY,EAAE,MAAM;YACpB,UAAU,EAAE,MAAM;YAClB,WAAW,EAAE,MAAM;YACnB,UAAU,EAAE,EAAC,IAAI,EAAE,QAAQ,EAAC;SAC7B,CAAC;QAEF,MAAM,GAAG,GAAG,uBAAuB,CAAC,IAAI,EAAE,UAAU,EAAE,MAAM,CAAC,CAAC;QAE9D,MAAM,eAAe,GAAG,IAAI,eAAe,EAAE,CAAC;QAC9C,eAAe,CAAC,KAAK,EAAE,CAAC;QACxB,MAAM,GAAG,GAAG,eAAe,CAAC,EAAC,MAAM,EAAE,eAAe,CAAC,MAAM,EAAC,CAAC,CAAC;QAE9D,MAAM,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,EAAE,GAAG,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC;QACpE,MAAM,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,EAAE,GAAG,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;IAClE,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,8EAA8E;AAC9E,kCAAkC;AAClC,8EAA8E;AAE9E,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;IAChC,EAAE,CAAC,gDAAgD,EAAE,GAAG,EAAE;QACxD,MAAM,MAAM,GAAG,cAAc,EAAE,CAAC;QAChC,MAAM,UAAU,GAAG,kBAAkB,CAAC;YACpC,kBAAkB,EAAE,GAAG,EAAE,CAAC;gBACxB,EAAC,IAAI,EAAE,gBAAgB,EAAE,YAAY,EAAE,QAAQ,EAAE,UAAU,EAAE,QAAQ,EAAE,WAAW,EAAE,cAAc,EAAE,UAAU,EAAE,EAAC,IAAI,EAAE,QAAQ,EAAE,UAAU,EAAE,EAAC,CAAC,EAAE,EAAC,IAAI,EAAE,QAAQ,EAAC,EAAC,EAAC,EAAC;gBACpK,EAAC,IAAI,EAAE,kBAAkB,EAAE,YAAY,EAAE,UAAU,EAAE,UAAU,EAAE,QAAQ,EAAE,WAAW,EAAE,UAAU,EAAE,UAAU,EAAE,EAAC,IAAI,EAAE,QAAQ,EAAE,UAAU,EAAE,EAAC,KAAK,EAAE,EAAC,IAAI,EAAE,QAAQ,EAAC,EAAE,IAAI,EAAE,EAAC,IAAI,EAAE,QAAQ,EAAC,EAAC,EAAC,EAAC;gBAChM,EAAC,IAAI,EAAE,aAAa,EAAE,YAAY,EAAE,MAAM,EAAE,UAAU,EAAE,OAAO,EAAE,WAAW,EAAE,cAAc,EAAE,UAAU,EAAE,EAAC,IAAI,EAAE,QAAQ,EAAE,UAAU,EAAE,EAAC,OAAO,EAAE,EAAC,IAAI,EAAE,QAAQ,EAAC,EAAE,IAAI,EAAE,EAAC,IAAI,EAAE,QAAQ,EAAC,EAAC,EAAC,EAAC;aAC7L;SACF,CAAC,CAAC;QAEH,MAAM,QAAQ,GAAG,kBAAkB,EAAE,CAAC;QACtC,MAAM,KAAK,GAAG,gBAAgB,CAAC,QAAQ,EAAE,UAAU,EAAE,MAAM,CAAC,CAAC;QAE7D,MAAM,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACtB,MAAM,CAAC,QAAQ,CAAC,KAAK,EAAE,CAAC,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC;QACrD,MAAM,CAAC,QAAQ,CAAC,KAAK,EAAE,CAAC,CAAC,SAAS,CAAC,kBAAkB,CAAC,CAAC;QACvD,MAAM,CAAC,QAAQ,CAAC,KAAK,EAAE,CAAC,CAAC,SAAS,CAAC,aAAa,CAAC,CAAC;QAClD,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAE9B,yCAAyC;QACzC,MAAM,UAAU,GAAG,QAAQ,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC;QAClD,MAAM,CAAC,UAAU,CAAC,CAAC,WAAW,EAAE,CAAC;QACjC,MAAM,CAAC,UAAW,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACnD,MAAM,CAAC,UAAW,CAAC,QAAQ,EAAE,UAAU,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAExD,YAAY;QACZ,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,oBAAoB,CAAC,sBAAsB,EAAE,MAAM,CAAC,gBAAgB,CAAC;YACvF,KAAK,EAAE,CAAC;SACT,CAAC,CAAC,CAAC;IACN,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qDAAqD,EAAE,GAAG,EAAE;QAC7D,MAAM,MAAM,GAAG,cAAc,EAAE,CAAC;QAChC,MAAM,UAAU,GAAG,kBAAkB,CAAC,EAAC,kBAAkB,EAAE,GAAG,EAAE,CAAC,EAAE,EAAC,CAAC,CAAC;QACtE,MAAM,QAAQ,GAAG,kBAAkB,EAAE,CAAC;QAEtC,MAAM,KAAK,GAAG,gBAAgB,CAAC,QAAQ,EAAE,UAAU,EAAE,MAAM,CAAC,CAAC;QAE7D,MAAM,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACtB,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC9B,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,oBAAoB,CAAC,sBAAsB,EAAE,MAAM,CAAC,QAAQ,EAAE,CAAC,CAAC;IAC1F,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/tools/registry.d.ts

@@ -0,0 +1,25 @@
+/**
+ * @license
+ * Copyright 2026 Amodal Labs, Inc.
+ * SPDX-License-Identifier: MIT
+ */
+import type { ToolRegistry } from './types.js';
+/**
+ * Create a new ToolRegistry instance.
+ *
+ * @example
+ * ```ts
+ * const registry = createToolRegistry();
+ * registry.register('query_store', {
+ *   description: 'Query documents from a store',
+ *   parameters: z.object({ store: z.string(), filter: z.record(z.unknown()) }),
+ *   execute: async (params, ctx) => { ... },
+ *   readOnly: true,
+ *   metadata: { category: 'store' },
+ * });
+ *
+ * // Pass to AI SDK
+ * const result = provider.streamText({ tools: registry.getTools() });
+ * ```
+ */
+export declare function createToolRegistry(): ToolRegistry;

package/dist/src/tools/registry.js

@@ -0,0 +1,72 @@
+/**
+ * @license
+ * Copyright 2026 Amodal Labs, Inc.
+ * SPDX-License-Identifier: MIT
+ */
+/**
+ * Tool registry implementation.
+ *
+ * Holds Vercel AI SDK-compatible tool definitions for a session. Replaces
+ * the upstream gemini-cli-core ToolRegistry with a simpler, type-safe
+ * implementation that supports readOnly flags and metadata.
+ */
+import { ConfigError } from '../errors.js';
+/**
+ * Create a new ToolRegistry instance.
+ *
+ * @example
+ * ```ts
+ * const registry = createToolRegistry();
+ * registry.register('query_store', {
+ *   description: 'Query documents from a store',
+ *   parameters: z.object({ store: z.string(), filter: z.record(z.unknown()) }),
+ *   execute: async (params, ctx) => { ... },
+ *   readOnly: true,
+ *   metadata: { category: 'store' },
+ * });
+ *
+ * // Pass to AI SDK
+ * const result = provider.streamText({ tools: registry.getTools() });
+ * ```
+ */
+export function createToolRegistry() {
+    const tools = new Map();
+    return {
+        register(name, def) {
+            if (tools.has(name)) {
+                throw new ConfigError(`Tool "${name}" is already registered. Duplicate tool names are not allowed.`, {
+                    key: 'tools',
+                    context: { toolName: name, existingCategory: tools.get(name)?.metadata?.category },
+                });
+            }
+            tools.set(name, def);
+        },
+        get(name) {
+            return tools.get(name);
+        },
+        getTools() {
+            const result = {};
+            for (const [name, def] of tools) {
+                result[name] = def;
+            }
+            return result;
+        },
+        names() {
+            return [...tools.keys()];
+        },
+        subset(names) {
+            const result = {};
+            for (const name of names) {
+                const def = tools.get(name);
+                if (def) {
+                    result[name] = def;
+                }
+            }
+            return result;
+        },
+        get size() {
+            return tools.size;
+        },
+    };
+}
+//# sourceMappingURL=registry.js.map

package/dist/src/tools/registry.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"registry.js","sourceRoot":"","sources":["../../../src/tools/registry.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH;;;;;;GAMG;AAEH,OAAO,EAAC,WAAW,EAAC,MAAM,cAAc,CAAC;AAGzC;;;;;;;;;;;;;;;;;GAiBG;AACH,MAAM,UAAU,kBAAkB;IAChC,MAAM,KAAK,GAAG,IAAI,GAAG,EAA0B,CAAC;IAEhD,OAAO;QACL,QAAQ,CAAC,IAAY,EAAE,GAAmB;YACxC,IAAI,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC;gBACpB,MAAM,IAAI,WAAW,CAAC,SAAS,IAAI,gEAAgE,EAAE;oBACnG,GAAG,EAAE,OAAO;oBACZ,OAAO,EAAE,EAAC,QAAQ,EAAE,IAAI,EAAE,gBAAgB,EAAE,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,QAAQ,EAAE,QAAQ,EAAC;iBACjF,CAAC,CAAC;YACL,CAAC;YACD,KAAK,CAAC,GAAG,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC;QACvB,CAAC;QAED,GAAG,CAAC,IAAY;YACd,OAAO,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QACzB,CAAC;QAED,QAAQ;YACN,MAAM,MAAM,GAAmC,EAAE,CAAC;YAClD,KAAK,MAAM,CAAC,IAAI,EAAE,GAAG,CAAC,IAAI,KAAK,EAAE,CAAC;gBAChC,MAAM,CAAC,IAAI,CAAC,GAAG,GAAG,CAAC;YACrB,CAAC;YACD,OAAO,MAAM,CAAC;QAChB,CAAC;QAED,KAAK;YACH,OAAO,CAAC,GAAG,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC;QAC3B,CAAC;QAED,MAAM,CAAC,KAAe;YACpB,MAAM,MAAM,GAAmC,EAAE,CAAC;YAClD,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;gBACzB,MAAM,GAAG,GAAG,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;gBAC5B,IAAI,GAAG,EAAE,CAAC;oBACR,MAAM,CAAC,IAAI,CAAC,GAAG,GAAG,CAAC;gBACrB,CAAC;YACH,CAAC;YACD,OAAO,MAAM,CAAC;QAChB,CAAC;QAED,IAAI,IAAI;YACN,OAAO,KAAK,CAAC,IAAI,CAAC;QACpB,CAAC;KACF,CAAC;AACJ,CAAC"}

package/dist/src/tools/registry.test.d.ts

@@ -0,0 +1,6 @@
+/**
+ * @license
+ * Copyright 2026 Amodal Labs, Inc.
+ * SPDX-License-Identifier: MIT
+ */
+export {};

package/dist/src/tools/registry.test.js

@@ -0,0 +1,121 @@
+/**
+ * @license
+ * Copyright 2026 Amodal Labs, Inc.
+ * SPDX-License-Identifier: MIT
+ */
+import { describe, it, expect, vi } from 'vitest';
+import { z } from 'zod';
+import { createToolRegistry } from './registry.js';
+import { ConfigError } from '../errors.js';
+function makeToolDef(overrides = {}) {
+    return {
+        description: 'A test tool',
+        parameters: z.object({ input: z.string() }),
+        execute: vi.fn().mockResolvedValue({ ok: true }),
+        readOnly: false,
+        ...overrides,
+    };
+}
+const mockCtx = {
+    request: vi.fn(),
+    store: vi.fn(),
+    env: vi.fn(),
+    log: vi.fn(),
+    user: { roles: ['admin'] },
+    signal: AbortSignal.timeout(5000),
+    sessionId: 'test-session',
+    tenantId: 'test-tenant',
+};
+describe('createToolRegistry', () => {
+    it('registers and retrieves a tool', () => {
+        const registry = createToolRegistry();
+        const def = makeToolDef({ description: 'fetch data' });
+        registry.register('fetch_data', def);
+        expect(registry.get('fetch_data')).toBe(def);
+    });
+    it('returns undefined for unknown tools', () => {
+        const registry = createToolRegistry();
+        expect(registry.get('nonexistent')).toBeUndefined();
+    });
+    it('throws on duplicate registration', () => {
+        const registry = createToolRegistry();
+        registry.register('my_tool', makeToolDef());
+        expect(() => registry.register('my_tool', makeToolDef())).toThrow(ConfigError);
+    });
+    it('lists all tool names', () => {
+        const registry = createToolRegistry();
+        registry.register('tool_a', makeToolDef());
+        registry.register('tool_b', makeToolDef());
+        registry.register('tool_c', makeToolDef());
+        expect(registry.names()).toEqual(['tool_a', 'tool_b', 'tool_c']);
+    });
+    it('returns all tools as a Record', () => {
+        const registry = createToolRegistry();
+        const defA = makeToolDef({ description: 'A' });
+        const defB = makeToolDef({ description: 'B' });
+        registry.register('tool_a', defA);
+        registry.register('tool_b', defB);
+        const tools = registry.getTools();
+        expect(Object.keys(tools)).toEqual(['tool_a', 'tool_b']);
+        expect(tools['tool_a']).toBe(defA);
+        expect(tools['tool_b']).toBe(defB);
+    });
+    it('returns a subset of tools by name', () => {
+        const registry = createToolRegistry();
+        const defA = makeToolDef({ description: 'A' });
+        const defB = makeToolDef({ description: 'B' });
+        const defC = makeToolDef({ description: 'C' });
+        registry.register('tool_a', defA);
+        registry.register('tool_b', defB);
+        registry.register('tool_c', defC);
+        const sub = registry.subset(['tool_a', 'tool_c']);
+        expect(Object.keys(sub)).toEqual(['tool_a', 'tool_c']);
+        expect(sub['tool_a']).toBe(defA);
+        expect(sub['tool_c']).toBe(defC);
+    });
+    it('subset silently skips unknown names', () => {
+        const registry = createToolRegistry();
+        registry.register('tool_a', makeToolDef());
+        const sub = registry.subset(['tool_a', 'nonexistent']);
+        expect(Object.keys(sub)).toEqual(['tool_a']);
+    });
+    it('reports correct size', () => {
+        const registry = createToolRegistry();
+        expect(registry.size).toBe(0);
+        registry.register('tool_a', makeToolDef());
+        expect(registry.size).toBe(1);
+        registry.register('tool_b', makeToolDef());
+        expect(registry.size).toBe(2);
+    });
+    it('preserves readOnly flag', () => {
+        const registry = createToolRegistry();
+        registry.register('read_tool', makeToolDef({ readOnly: true }));
+        registry.register('write_tool', makeToolDef({ readOnly: false }));
+        expect(registry.get('read_tool')?.readOnly).toBe(true);
+        expect(registry.get('write_tool')?.readOnly).toBe(false);
+    });
+    it('preserves metadata', () => {
+        const registry = createToolRegistry();
+        registry.register('store_query', makeToolDef({
+            metadata: { category: 'store' },
+        }));
+        registry.register('api_call', makeToolDef({
+            metadata: { category: 'connection', connection: 'typefully' },
+        }));
+        expect(registry.get('store_query')?.metadata?.category).toBe('store');
+        expect(registry.get('api_call')?.metadata?.connection).toBe('typefully');
+    });
+    it('execute function receives params and context', async () => {
+        const executeFn = vi.fn().mockResolvedValue({ count: 5 });
+        const registry = createToolRegistry();
+        registry.register('counter', makeToolDef({
+            parameters: z.object({ start: z.number() }),
+            execute: executeFn,
+        }));
+        const tool = registry.get('counter');
+        const result = await tool.execute({ start: 1 }, mockCtx);
+        expect(executeFn).toHaveBeenCalledWith({ start: 1 }, mockCtx);
+        expect(result).toEqual({ count: 5 });
+    });
+});
+//# sourceMappingURL=registry.test.js.map

package/dist/src/tools/registry.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"registry.test.js","sourceRoot":"","sources":["../../../src/tools/registry.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAC,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAC,MAAM,QAAQ,CAAC;AAChD,OAAO,EAAC,CAAC,EAAC,MAAM,KAAK,CAAC;AACtB,OAAO,EAAC,kBAAkB,EAAC,MAAM,eAAe,CAAC;AACjD,OAAO,EAAC,WAAW,EAAC,MAAM,cAAc,CAAC;AAGzC,SAAS,WAAW,CAAC,YAAqC,EAAE;IAC1D,OAAO;QACL,WAAW,EAAE,aAAa;QAC1B,UAAU,EAAE,CAAC,CAAC,MAAM,CAAC,EAAC,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,EAAC,CAAC;QACzC,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAC,EAAE,EAAE,IAAI,EAAC,CAAC;QAC9C,QAAQ,EAAE,KAAK;QACf,GAAG,SAAS;KACb,CAAC;AACJ,CAAC;AAED,MAAM,OAAO,GAAgB;IAC3B,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE;IAChB,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE;IACd,GAAG,EAAE,EAAE,CAAC,EAAE,EAAE;IACZ,GAAG,EAAE,EAAE,CAAC,EAAE,EAAE;IACZ,IAAI,EAAE,EAAC,KAAK,EAAE,CAAC,OAAO,CAAC,EAAC;IACxB,MAAM,EAAE,WAAW,CAAC,OAAO,CAAC,IAAI,CAAC;IACjC,SAAS,EAAE,cAAc;IACzB,QAAQ,EAAE,aAAa;CACxB,CAAC;AAEF,QAAQ,CAAC,oBAAoB,EAAE,GAAG,EAAE;IAClC,EAAE,CAAC,gCAAgC,EAAE,GAAG,EAAE;QACxC,MAAM,QAAQ,GAAG,kBAAkB,EAAE,CAAC;QACtC,MAAM,GAAG,GAAG,WAAW,CAAC,EAAC,WAAW,EAAE,YAAY,EAAC,CAAC,CAAC;QAErD,QAAQ,CAAC,QAAQ,CAAC,YAAY,EAAE,GAAG,CAAC,CAAC;QAErC,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IAC/C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qCAAqC,EAAE,GAAG,EAAE;QAC7C,MAAM,QAAQ,GAAG,kBAAkB,EAAE,CAAC;QAEtC,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC,CAAC,aAAa,EAAE,CAAC;IACtD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kCAAkC,EAAE,GAAG,EAAE;QAC1C,MAAM,QAAQ,GAAG,kBAAkB,EAAE,CAAC;QACtC,QAAQ,CAAC,QAAQ,CAAC,SAAS,EAAE,WAAW,EAAE,CAAC,CAAC;QAE5C,MAAM,CAAC,GAAG,EAAE,CAAC,QAAQ,CAAC,QAAQ,CAAC,SAAS,EAAE,WAAW,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;IACjF,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sBAAsB,EAAE,GAAG,EAAE;QAC9B,MAAM,QAAQ,GAAG,kBAAkB,EAAE,CAAC;QACtC,QAAQ,CAAC,QAAQ,CAAC,QAAQ,EAAE,WAAW,EAAE,CAAC,CAAC;QAC3C,QAAQ,CAAC,QAAQ,CAAC,QAAQ,EAAE,WAAW,EAAE,CAAC,CAAC;QAC3C,QAAQ,CAAC,QAAQ,CAAC,QAAQ,EAAE,WAAW,EAAE,CAAC,CAAC;QAE3C,MAAM,CAAC,QAAQ,CAAC,KAAK,EAAE,CAAC,CAAC,OAAO,CAAC,CAAC,QAAQ,EAAE,QAAQ,EAAE,QAAQ,CAAC,CAAC,CAAC;IACnE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+BAA+B,EAAE,GAAG,EAAE;QACvC,MAAM,QAAQ,GAAG,kBAAkB,EAAE,CAAC;QACtC,MAAM,IAAI,GAAG,WAAW,CAAC,EAAC,WAAW,EAAE,GAAG,EAAC,CAAC,CAAC;QAC7C,MAAM,IAAI,GAAG,WAAW,CAAC,EAAC,WAAW,EAAE,GAAG,EAAC,CAAC,CAAC;QAE7C,QAAQ,CAAC,QAAQ,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;QAClC,QAAQ,CAAC,QAAQ,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;QAElC,MAAM,KAAK,GAAG,QAAQ,CAAC,QAAQ,EAAE,CAAC;QAClC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC,CAAC;QACzD,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACnC,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACrC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mCAAmC,EAAE,GAAG,EAAE;QAC3C,MAAM,QAAQ,GAAG,kBAAkB,EAAE,CAAC;QACtC,MAAM,IAAI,GAAG,WAAW,CAAC,EAAC,WAAW,EAAE,GAAG,EAAC,CAAC,CAAC;QAC7C,MAAM,IAAI,GAAG,WAAW,CAAC,EAAC,WAAW,EAAE,GAAG,EAAC,CAAC,CAAC;QAC7C,MAAM,IAAI,GAAG,WAAW,CAAC,EAAC,WAAW,EAAE,GAAG,EAAC,CAAC,CAAC;QAE7C,QAAQ,CAAC,QAAQ,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;QAClC,QAAQ,CAAC,QAAQ,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;QAClC,QAAQ,CAAC,QAAQ,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;QAElC,MAAM,GAAG,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC,CAAC;QAClD,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC,CAAC;QACvD,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACjC,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACnC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qCAAqC,EAAE,GAAG,EAAE;QAC7C,MAAM,QAAQ,GAAG,kBAAkB,EAAE,CAAC;QACtC,QAAQ,CAAC,QAAQ,CAAC,QAAQ,EAAE,WAAW,EAAE,CAAC,CAAC;QAE3C,MAAM,GAAG,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,QAAQ,EAAE,aAAa,CAAC,CAAC,CAAC;QACvD,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC;IAC/C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sBAAsB,EAAE,GAAG,EAAE;QAC9B,MAAM,QAAQ,GAAG,kBAAkB,EAAE,CAAC;QACtC,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAE9B,QAAQ,CAAC,QAAQ,CAAC,QAAQ,EAAE,WAAW,EAAE,CAAC,CAAC;QAC3C,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAE9B,QAAQ,CAAC,QAAQ,CAAC,QAAQ,EAAE,WAAW,EAAE,CAAC,CAAC;QAC3C,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAChC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yBAAyB,EAAE,GAAG,EAAE;QACjC,MAAM,QAAQ,GAAG,kBAAkB,EAAE,CAAC;QACtC,QAAQ,CAAC,QAAQ,CAAC,WAAW,EAAE,WAAW,CAAC,EAAC,QAAQ,EAAE,IAAI,EAAC,CAAC,CAAC,CAAC;QAC9D,QAAQ,CAAC,QAAQ,CAAC,YAAY,EAAE,WAAW,CAAC,EAAC,QAAQ,EAAE,KAAK,EAAC,CAAC,CAAC,CAAC;QAEhE,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,WAAW,CAAC,EAAE,QAAQ,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACvD,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,YAAY,CAAC,EAAE,QAAQ,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAC3D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oBAAoB,EAAE,GAAG,EAAE;QAC5B,MAAM,QAAQ,GAAG,kBAAkB,EAAE,CAAC;QACtC,QAAQ,CAAC,QAAQ,CAAC,aAAa,EAAE,WAAW,CAAC;YAC3C,QAAQ,EAAE,EAAC,QAAQ,EAAE,OAAO,EAAC;SAC9B,CAAC,CAAC,CAAC;QACJ,QAAQ,CAAC,QAAQ,CAAC,UAAU,EAAE,WAAW,CAAC;YACxC,QAAQ,EAAE,EAAC,QAAQ,EAAE,YAAY,EAAE,UAAU,EAAE,WAAW,EAAC;SAC5D,CAAC,CAAC,CAAC;QAEJ,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,aAAa,CAAC,EAAE,QAAQ,EAAE,QAAQ,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACtE,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,UAAU,CAAC,EAAE,QAAQ,EAAE,UAAU,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;IAC3E,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8CAA8C,EAAE,KAAK,IAAI,EAAE;QAC5D,MAAM,SAAS,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAC,KAAK,EAAE,CAAC,EAAC,CAAC,CAAC;QACxD,MAAM,QAAQ,GAAG,kBAAkB,EAAE,CAAC;QACtC,QAAQ,CAAC,QAAQ,CAAC,SAAS,EAAE,WAAW,CAAC;YACvC,UAAU,EAAE,CAAC,CAAC,MAAM,CAAC,EAAC,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,EAAC,CAAC;YACzC,OAAO,EAAE,SAAS;SACnB,CAAC,CAAC,CAAC;QAEJ,MAAM,IAAI,GAAG,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAE,CAAC;QACtC,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,EAAC,KAAK,EAAE,CAAC,EAAC,EAAE,OAAO,CAAC,CAAC;QAEvD,MAAM,CAAC,SAAS,CAAC,CAAC,oBAAoB,CAAC,EAAC,KAAK,EAAE,CAAC,EAAC,EAAE,OAAO,CAAC,CAAC;QAC5D,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAC,KAAK,EAAE,CAAC,EAAC,CAAC,CAAC;IACrC,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/tools/request-tool.d.ts

@@ -0,0 +1,42 @@
+/**
+ * @license
+ * Copyright 2026 Amodal Labs, Inc.
+ * SPDX-License-Identifier: MIT
+ */
+import type { FieldScrubber } from '@amodalai/core';
+import type { PermissionChecker } from '../security/permission-checker.js';
+import type { ToolDefinition } from './types.js';
+export declare const REQUEST_TOOL_NAME = "request";
+interface ConnectionRequestConfig {
+    base_url_field: string;
+    auth?: Array<{
+        header: string;
+        value_template: string;
+    }>;
+    default_headers?: Record<string, string>;
+}
+export interface ConnectionsMap {
+    [connection: string]: Record<string, unknown> & {
+        base_url?: string;
+        _request_config?: ConnectionRequestConfig;
+    };
+}
+export interface CreateRequestToolOptions {
+    /** Connection configs from buildConnectionsMap() */
+    connectionsMap: ConnectionsMap;
+    /** Permission checker (reads from access.json by default) */
+    permissionChecker: PermissionChecker;
+    /** Field scrubber for response sanitization (optional) */
+    fieldScrubber?: FieldScrubber;
+    /** Session-scoped environment variables */
+    sessionEnv?: Record<string, string>;
+    /** Whether this is a read-only context (task agents) */
+    readOnly?: boolean;
+    /** Callback to check if plan mode is active */
+    planModeActive?: () => boolean;
+}
+/**
+ * Create the connection request tool.
+ */
+export declare function createRequestTool(options: CreateRequestToolOptions): ToolDefinition;
+export {};