@strands-agents/sdk 0.0.1

package/dist/src/__tests__/mcp.test.js

@@ -0,0 +1,166 @@
+import { describe, it, expect, vi, beforeEach } from 'vitest';
+import { Client } from '@modelcontextprotocol/sdk/client/index.js';
+import { McpClient } from '../mcp.js';
+import { McpTool } from '../tools/mcp-tool.js';
+import { JsonBlock } from '../types/messages.js';
+vi.mock('@modelcontextprotocol/sdk/client/index.js', () => ({
+    Client: vi.fn(function () {
+        return {
+            connect: vi.fn(),
+            close: vi.fn(),
+            listTools: vi.fn(),
+            callTool: vi.fn(),
+        };
+    }),
+}));
+vi.mock('../tools/tool.js', () => ({
+    // Mock the abstract base class
+    Tool: class {
+    },
+    // Mock helper to return a valid ToolResultBlock structure without prepending "Error: "
+    createErrorResult: (err, toolUseId) => ({
+        type: 'toolResultBlock',
+        status: 'error',
+        toolUseId,
+        content: [{ type: 'textBlock', text: err instanceof Error ? err.message : String(err) }],
+    }),
+}));
+vi.mock('../../__fixtures__/environment.js', () => ({ isNode: true }));
+/**
+ * Executes a tool stream to completion and returns the final result.
+ * We use a Generic <T> and cast the return value to ensure TypeScript
+ * knows the result is defined (and matches the Tool's return type).
+ */
+async function runTool(gen) {
+    let result = await gen.next();
+    while (!result.done) {
+        result = await gen.next();
+    }
+    // Force cast because we know our McpTool always returns a value when done
+    return result.value;
+}
+const mockTransport = {
+    connect: vi.fn(),
+    close: vi.fn(),
+    send: vi.fn(),
+};
+describe('MCP Integration', () => {
+    beforeEach(() => {
+        vi.clearAllMocks();
+    });
+    describe('McpClient', () => {
+        let client;
+        let sdkClientMock;
+        beforeEach(() => {
+            client = new McpClient({
+                applicationName: 'TestApp',
+                transport: mockTransport,
+            });
+            sdkClientMock = vi.mocked(Client).mock.results[0].value;
+        });
+        it('initializes SDK client with correct configuration', () => {
+            expect(Client).toHaveBeenCalledWith({ name: 'TestApp', version: '0.0.1' });
+        });
+        it('manages connection state lazily', async () => {
+            await client.connect();
+            expect(sdkClientMock.connect).toHaveBeenCalledTimes(1);
+            await client.connect();
+            expect(sdkClientMock.connect).toHaveBeenCalledTimes(1);
+        });
+        it('supports forced reconnection', async () => {
+            await client.connect();
+            await client.connect(true);
+            expect(sdkClientMock.close).toHaveBeenCalled();
+            expect(sdkClientMock.connect).toHaveBeenCalledTimes(2);
+        });
+        it('converts SDK tool specs to McpTool instances', async () => {
+            sdkClientMock.listTools.mockResolvedValue({
+                tools: [{ name: 'weather', description: 'Get weather', inputSchema: {} }],
+            });
+            const tools = await client.listTools();
+            expect(sdkClientMock.connect).toHaveBeenCalled();
+            expect(tools).toHaveLength(1);
+            expect(tools[0]).toBeInstanceOf(McpTool);
+            expect(tools[0].name).toBe('weather');
+        });
+        it('delegates invocation to SDK client', async () => {
+            const tool = new McpTool({ name: 'calc', description: '', inputSchema: {}, client });
+            sdkClientMock.callTool.mockResolvedValue({ content: [] });
+            await client.callTool(tool, { op: 'add' });
+            expect(sdkClientMock.connect).toHaveBeenCalled();
+            expect(sdkClientMock.callTool).toHaveBeenCalledWith({
+                name: 'calc',
+                arguments: { op: 'add' },
+            });
+        });
+        it('validates tool arguments', async () => {
+            const tool = new McpTool({ name: 't', description: '', inputSchema: {}, client });
+            await expect(client.callTool(tool, ['invalid-array'])).rejects.toThrow(/JSON Object/);
+        });
+        it('cleans up resources', async () => {
+            await client.disconnect();
+            expect(sdkClientMock.close).toHaveBeenCalled();
+            expect(mockTransport.close).toHaveBeenCalled();
+        });
+    });
+    describe('McpTool', () => {
+        const mockClientWrapper = { callTool: vi.fn() };
+        const tool = new McpTool({
+            name: 'weather',
+            description: 'Get weather',
+            inputSchema: {},
+            client: mockClientWrapper,
+        });
+        const context = {
+            toolUse: { toolUseId: 'id-123', name: 'weather', input: { city: 'NYC' } },
+            agent: {},
+        };
+        it('returns text results on success', async () => {
+            vi.mocked(mockClientWrapper.callTool).mockResolvedValue({
+                content: [{ type: 'text', text: 'Sunny' }],
+            });
+            // runTool<ToolResultBlock> explicitly tells TS the return type
+            const result = await runTool(tool.stream(context));
+            expect(result).toBeDefined();
+            expect(result.status).toBe('success');
+            expect(result.content[0].text).toBe('Sunny');
+        });
+        it('returns structured data results on success', async () => {
+            const data = { temperature: 72 };
+            vi.mocked(mockClientWrapper.callTool).mockResolvedValue({
+                content: [{ type: 'data', value: data }],
+            });
+            const result = await runTool(tool.stream(context));
+            const content = result.content[0];
+            expect(content).toBeInstanceOf(JsonBlock);
+            expect(content.json).toEqual(expect.objectContaining({ value: data }));
+        });
+        it('provides default message for empty output', async () => {
+            vi.mocked(mockClientWrapper.callTool).mockResolvedValue({ content: [] });
+            const result = await runTool(tool.stream(context));
+            expect(result.content[0].text).toContain('completed successfully');
+        });
+        it('handles protocol-level errors', async () => {
+            vi.mocked(mockClientWrapper.callTool).mockResolvedValue({
+                isError: true,
+                content: [{ type: 'text', text: 'Service Unavailable' }],
+            });
+            const result = await runTool(tool.stream(context));
+            expect(result.status).toBe('error');
+            expect(result.content[0].text).toBe('Service Unavailable');
+        });
+        it('catches and wraps client exceptions', async () => {
+            vi.mocked(mockClientWrapper.callTool).mockRejectedValue(new Error('Network Error'));
+            const result = await runTool(tool.stream(context));
+            expect(result.status).toBe('error');
+            expect(result.content[0].text).toBe('Network Error');
+        });
+        it('validates SDK response format', async () => {
+            vi.mocked(mockClientWrapper.callTool).mockResolvedValue({ content: null });
+            const result = await runTool(tool.stream(context));
+            expect(result.status).toBe('error');
+            expect(result.content[0].text).toContain('missing content array');
+        });
+    });
+});
+//# sourceMappingURL=mcp.test.js.map

package/dist/src/__tests__/mcp.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"mcp.test.js","sourceRoot":"","sources":["../../../src/__tests__/mcp.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAA;AAC7D,OAAO,EAAE,MAAM,EAAE,MAAM,2CAA2C,CAAA;AAElE,OAAO,EAAE,SAAS,EAAE,MAAM,WAAW,CAAA;AACrC,OAAO,EAAE,OAAO,EAAE,MAAM,sBAAsB,CAAA;AAC9C,OAAO,EAAE,SAAS,EAAwC,MAAM,sBAAsB,CAAA;AAItF,EAAE,CAAC,IAAI,CAAC,2CAA2C,EAAE,GAAG,EAAE,CAAC,CAAC;IAC1D,MAAM,EAAE,EAAE,CAAC,EAAE,CAAC;QACZ,OAAO;YACL,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE;YAChB,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE;YACd,SAAS,EAAE,EAAE,CAAC,EAAE,EAAE;YAClB,QAAQ,EAAE,EAAE,CAAC,EAAE,EAAE;SAClB,CAAA;IACH,CAAC,CAAC;CACH,CAAC,CAAC,CAAA;AAEH,EAAE,CAAC,IAAI,CAAC,kBAAkB,EAAE,GAAG,EAAE,CAAC,CAAC;IACjC,+BAA+B;IAC/B,IAAI,EAAE;KAAQ;IACd,uFAAuF;IACvF,iBAAiB,EAAE,CAAC,GAAY,EAAE,SAAiB,EAAE,EAAE,CAAC,CAAC;QACvD,IAAI,EAAE,iBAAiB;QACvB,MAAM,EAAE,OAAO;QACf,SAAS;QACT,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC;KACzF,CAAC;CACH,CAAC,CAAC,CAAA;AAEH,EAAE,CAAC,IAAI,CAAC,mCAAmC,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,CAAA;AAEtE;;;;GAIG;AACH,KAAK,UAAU,OAAO,CAAI,GAAwC;IAChE,IAAI,MAAM,GAAG,MAAM,GAAG,CAAC,IAAI,EAAE,CAAA;IAC7B,OAAO,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC;QACpB,MAAM,GAAG,MAAM,GAAG,CAAC,IAAI,EAAE,CAAA;IAC3B,CAAC;IACD,0EAA0E;IAC1E,OAAO,MAAM,CAAC,KAAU,CAAA;AAC1B,CAAC;AAED,MAAM,aAAa,GAAG;IACpB,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE;IAChB,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE;IACd,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE;CACU,CAAA;AAEzB,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;IAC/B,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,aAAa,EAAE,CAAA;IACpB,CAAC,CAAC,CAAA;IAEF,QAAQ,CAAC,WAAW,EAAE,GAAG,EAAE;QACzB,IAAI,MAAiB,CAAA;QACrB,IAAI,aAAkB,CAAA;QAEtB,UAAU,CAAC,GAAG,EAAE;YACd,MAAM,GAAG,IAAI,SAAS,CAAC;gBACrB,eAAe,EAAE,SAAS;gBAC1B,SAAS,EAAE,aAAa;aACzB,CAAC,CAAA;YACF,aAAa,GAAG,EAAE,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAE,CAAC,KAAK,CAAA;QAC1D,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,mDAAmD,EAAE,GAAG,EAAE;YAC3D,MAAM,CAAC,MAAM,CAAC,CAAC,oBAAoB,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC,CAAA;QAC5E,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,iCAAiC,EAAE,KAAK,IAAI,EAAE;YAC/C,MAAM,MAAM,CAAC,OAAO,EAAE,CAAA;YACtB,MAAM,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAA;YAEtD,MAAM,MAAM,CAAC,OAAO,EAAE,CAAA;YACtB,MAAM,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAA;QACxD,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,8BAA8B,EAAE,KAAK,IAAI,EAAE;YAC5C,MAAM,MAAM,CAAC,OAAO,EAAE,CAAA;YACtB,MAAM,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAA;YAE1B,MAAM,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC,gBAAgB,EAAE,CAAA;YAC9C,MAAM,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAA;QACxD,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,8CAA8C,EAAE,KAAK,IAAI,EAAE;YAC5D,aAAa,CAAC,SAAS,CAAC,iBAAiB,CAAC;gBACxC,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,WAAW,EAAE,aAAa,EAAE,WAAW,EAAE,EAAE,EAAE,CAAC;aAC1E,CAAC,CAAA;YAEF,MAAM,KAAK,GAAG,MAAM,MAAM,CAAC,SAAS,EAAE,CAAA;YAEtC,MAAM,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC,gBAAgB,EAAE,CAAA;YAChD,MAAM,CAAC,KAAK,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAA;YAC7B,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,cAAc,CAAC,OAAO,CAAC,CAAA;YACxC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAE,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAA;QACxC,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,oCAAoC,EAAE,KAAK,IAAI,EAAE;YAClD,MAAM,IAAI,GAAG,IAAI,OAAO,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,WAAW,EAAE,EAAE,EAAE,WAAW,EAAE,EAAE,EAAE,MAAM,EAAE,CAAC,CAAA;YACpF,aAAa,CAAC,QAAQ,CAAC,iBAAiB,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,CAAA;YAEzD,MAAM,MAAM,CAAC,QAAQ,CAAC,IAAI,EAAE,EAAE,EAAE,EAAE,KAAK,EAAE,CAAC,CAAA;YAE1C,MAAM,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC,gBAAgB,EAAE,CAAA;YAChD,MAAM,CAAC,aAAa,CAAC,QAAQ,CAAC,CAAC,oBAAoB,CAAC;gBAClD,IAAI,EAAE,MAAM;gBACZ,SAAS,EAAE,EAAE,EAAE,EAAE,KAAK,EAAE;aACzB,CAAC,CAAA;QACJ,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,0BAA0B,EAAE,KAAK,IAAI,EAAE;YACxC,MAAM,IAAI,GAAG,IAAI,OAAO,CAAC,EAAE,IAAI,EAAE,GAAG,EAAE,WAAW,EAAE,EAAE,EAAE,WAAW,EAAE,EAAE,EAAE,MAAM,EAAE,CAAC,CAAA;YACjF,MAAM,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,aAAa,CAAC,CAAA;QACvF,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,qBAAqB,EAAE,KAAK,IAAI,EAAE;YACnC,MAAM,MAAM,CAAC,UAAU,EAAE,CAAA;YACzB,MAAM,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC,gBAAgB,EAAE,CAAA;YAC9C,MAAM,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC,gBAAgB,EAAE,CAAA;QAChD,CAAC,CAAC,CAAA;IACJ,CAAC,CAAC,CAAA;IAEF,QAAQ,CAAC,SAAS,EAAE,GAAG,EAAE;QACvB,MAAM,iBAAiB,GAAG,EAAE,QAAQ,EAAE,EAAE,CAAC,EAAE,EAAE,EAA0B,CAAA;QACvE,MAAM,IAAI,GAAG,IAAI,OAAO,CAAC;YACvB,IAAI,EAAE,SAAS;YACf,WAAW,EAAE,aAAa;YAC1B,WAAW,EAAE,EAAE;YACf,MAAM,EAAE,iBAAiB;SAC1B,CAAC,CAAA;QAEF,MAAM,OAAO,GAAgB;YAC3B,OAAO,EAAE,EAAE,SAAS,EAAE,QAAQ,EAAE,IAAI,EAAE,SAAS,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,EAAE;YACzE,KAAK,EAAE,EAAe;SACvB,CAAA;QAED,EAAE,CAAC,iCAAiC,EAAE,KAAK,IAAI,EAAE;YAC/C,EAAE,CAAC,MAAM,CAAC,iBAAiB,CAAC,QAAQ,CAAC,CAAC,iBAAiB,CAAC;gBACtD,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;aAC3C,CAAC,CAAA;YAEF,+DAA+D;YAC/D,MAAM,MAAM,GAAG,MAAM,OAAO,CAAkB,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAA;YAEnE,MAAM,CAAC,MAAM,CAAC,CAAC,WAAW,EAAE,CAAA;YAC5B,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAA;YACrC,MAAM,CAAE,MAAM,CAAC,OAAO,CAAC,CAAC,CAAe,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAA;QAC7D,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,4CAA4C,EAAE,KAAK,IAAI,EAAE;YAC1D,MAAM,IAAI,GAAG,EAAE,WAAW,EAAE,EAAE,EAAE,CAAA;YAChC,EAAE,CAAC,MAAM,CAAC,iBAAiB,CAAC,QAAQ,CAAC,CAAC,iBAAiB,CAAC;gBACtD,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC;aACzC,CAAC,CAAA;YAEF,MAAM,MAAM,GAAG,MAAM,OAAO,CAAkB,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAA;YACnE,MAAM,OAAO,GAAG,MAAM,CAAC,OAAO,CAAC,CAAC,CAAc,CAAA;YAE9C,MAAM,CAAC,OAAO,CAAC,CAAC,cAAc,CAAC,SAAS,CAAC,CAAA;YACzC,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,gBAAgB,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC,CAAA;QACxE,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,2CAA2C,EAAE,KAAK,IAAI,EAAE;YACzD,EAAE,CAAC,MAAM,CAAC,iBAAiB,CAAC,QAAQ,CAAC,CAAC,iBAAiB,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,CAAA;YAExE,MAAM,MAAM,GAAG,MAAM,OAAO,CAAkB,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAA;YAEnE,MAAM,CAAE,MAAM,CAAC,OAAO,CAAC,CAAC,CAAe,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,wBAAwB,CAAC,CAAA;QACnF,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,+BAA+B,EAAE,KAAK,IAAI,EAAE;YAC7C,EAAE,CAAC,MAAM,CAAC,iBAAiB,CAAC,QAAQ,CAAC,CAAC,iBAAiB,CAAC;gBACtD,OAAO,EAAE,IAAI;gBACb,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,qBAAqB,EAAE,CAAC;aACzD,CAAC,CAAA;YAEF,MAAM,MAAM,GAAG,MAAM,OAAO,CAAkB,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAA;YAEnE,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAA;YACnC,MAAM,CAAE,MAAM,CAAC,OAAO,CAAC,CAAC,CAAe,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAA;QAC3E,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,qCAAqC,EAAE,KAAK,IAAI,EAAE;YACnD,EAAE,CAAC,MAAM,CAAC,iBAAiB,CAAC,QAAQ,CAAC,CAAC,iBAAiB,CAAC,IAAI,KAAK,CAAC,eAAe,CAAC,CAAC,CAAA;YAEnF,MAAM,MAAM,GAAG,MAAM,OAAO,CAAkB,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAA;YAEnE,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAA;YACnC,MAAM,CAAE,MAAM,CAAC,OAAO,CAAC,CAAC,CAAe,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAA;QACrE,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,+BAA+B,EAAE,KAAK,IAAI,EAAE;YAC7C,EAAE,CAAC,MAAM,CAAC,iBAAiB,CAAC,QAAQ,CAAC,CAAC,iBAAiB,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAA;YAE1E,MAAM,MAAM,GAAG,MAAM,OAAO,CAAkB,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAA;YAEnE,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAA;YACnC,MAAM,CAAE,MAAM,CAAC,OAAO,CAAC,CAAC,CAAe,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,uBAAuB,CAAC,CAAA;QAClF,CAAC,CAAC,CAAA;IACJ,CAAC,CAAC,CAAA;AACJ,CAAC,CAAC,CAAA"}

package/dist/src/agent/__tests__/agent.hook.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=agent.hook.test.d.ts.map

package/dist/src/agent/__tests__/agent.hook.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"agent.hook.test.d.ts","sourceRoot":"","sources":["../../../../src/agent/__tests__/agent.hook.test.ts"],"names":[],"mappings":""}

package/dist/src/agent/__tests__/agent.hook.test.js

@@ -0,0 +1,250 @@
+import { beforeEach, describe, expect, it } from 'vitest';
+import { Agent } from '../agent.js';
+import { AfterInvocationEvent, AfterModelCallEvent, AfterToolCallEvent, BeforeInvocationEvent, BeforeModelCallEvent, BeforeToolCallEvent, MessageAddedEvent, ModelStreamEventHook, } from '../../hooks/index.js';
+import { MockMessageModel } from '../../__fixtures__/mock-message-model.js';
+import { MockHookProvider } from '../../__fixtures__/mock-hook-provider.js';
+import { collectIterator } from '../../__fixtures__/model-test-helpers.js';
+import { FunctionTool } from '../../tools/function-tool.js';
+import { Message, TextBlock, ToolResultBlock } from '../../types/messages.js';
+describe('Agent Hooks Integration', () => {
+    let mockProvider;
+    beforeEach(() => {
+        mockProvider = new MockHookProvider();
+    });
+    describe('invocation lifecycle', () => {
+        it('fires hooks during invoke', async () => {
+            const lifecycleProvider = new MockHookProvider({ includeModelEvents: false });
+            const model = new MockMessageModel().addTurn({ type: 'textBlock', text: 'Hello' });
+            const agent = new Agent({ model, hooks: [lifecycleProvider] });
+            await agent.invoke('Hi');
+            expect(lifecycleProvider.invocations).toHaveLength(6);
+            expect(lifecycleProvider.invocations[0]).toEqual(new BeforeInvocationEvent({ agent: agent }));
+            expect(lifecycleProvider.invocations[1]).toEqual(new MessageAddedEvent({ agent: agent, message: new Message({ role: 'user', content: [new TextBlock('Hi')] }) }));
+            expect(lifecycleProvider.invocations[2]).toEqual(new BeforeModelCallEvent({ agent: agent }));
+            expect(lifecycleProvider.invocations[3]).toEqual(new AfterModelCallEvent({
+                agent,
+                stopData: {
+                    stopReason: 'endTurn',
+                    message: new Message({ role: 'assistant', content: [new TextBlock('Hello')] }),
+                },
+            }));
+            expect(lifecycleProvider.invocations[4]).toEqual(new MessageAddedEvent({
+                agent,
+                message: new Message({ role: 'assistant', content: [new TextBlock('Hello')] }),
+            }));
+            expect(lifecycleProvider.invocations[5]).toEqual(new AfterInvocationEvent({ agent }));
+        });
+        it('fires hooks during stream', async () => {
+            const lifecycleProvider = new MockHookProvider({ includeModelEvents: false });
+            const model = new MockMessageModel().addTurn({ type: 'textBlock', text: 'Hello' });
+            const agent = new Agent({ model, hooks: [lifecycleProvider] });
+            await collectIterator(agent.stream('Hi'));
+            expect(lifecycleProvider.invocations).toHaveLength(6);
+            expect(lifecycleProvider.invocations[0]).toEqual(new BeforeInvocationEvent({ agent: agent }));
+            expect(lifecycleProvider.invocations[1]).toEqual(new MessageAddedEvent({
+                agent: agent,
+                message: new Message({ role: 'user', content: [new TextBlock('Hi')] }),
+            }));
+            expect(lifecycleProvider.invocations[2]).toEqual(new BeforeModelCallEvent({ agent: agent }));
+            expect(lifecycleProvider.invocations[3]).toEqual(new AfterModelCallEvent({
+                agent,
+                stopData: {
+                    stopReason: 'endTurn',
+                    message: new Message({ role: 'assistant', content: [new TextBlock('Hello')] }),
+                },
+            }));
+            expect(lifecycleProvider.invocations[4]).toEqual(new MessageAddedEvent({
+                agent,
+                message: new Message({ role: 'assistant', content: [new TextBlock('Hello')] }),
+            }));
+            expect(lifecycleProvider.invocations[5]).toEqual(new AfterInvocationEvent({ agent }));
+        });
+    });
+    describe('runtime hook registration', () => {
+        it('allows adding hooks after agent creation', async () => {
+            const lifecycleProvider = new MockHookProvider({ includeModelEvents: false });
+            const model = new MockMessageModel().addTurn({ type: 'textBlock', text: 'Hello' });
+            const agent = new Agent({ model });
+            agent.hooks.addHook(lifecycleProvider);
+            await agent.invoke('Hi');
+            // Should have all lifecycle events
+            expect(lifecycleProvider.invocations).toHaveLength(6);
+            expect(lifecycleProvider.invocations[0]).toEqual(new BeforeInvocationEvent({ agent }));
+            expect(lifecycleProvider.invocations[5]).toEqual(new AfterInvocationEvent({ agent }));
+        });
+    });
+    describe('multi-turn conversations', () => {
+        it('fires hooks for each invoke call', async () => {
+            const lifecycleProvider = new MockHookProvider({ includeModelEvents: false });
+            const model = new MockMessageModel()
+                .addTurn({ type: 'textBlock', text: 'First response' })
+                .addTurn({ type: 'textBlock', text: 'Second response' });
+            const agent = new Agent({ model, hooks: [lifecycleProvider] });
+            await agent.invoke('First message');
+            // First turn should have: BeforeInvocation, MessageAdded, BeforeModelCall, AfterModelCall, MessageAdded, AfterInvocation
+            expect(lifecycleProvider.invocations).toHaveLength(6);
+            await agent.invoke('Second message');
+            // Should have 10 events total (6 for each turn)
+            expect(lifecycleProvider.invocations).toHaveLength(12);
+            // Filter for just Invocation events to verify they fire for each turn
+            const invocationEvents = lifecycleProvider.invocations.filter((e) => e instanceof BeforeInvocationEvent || e instanceof AfterInvocationEvent);
+            expect(invocationEvents).toHaveLength(4); // 2 for each turn
+        });
+    });
+    describe('tool execution hooks', () => {
+        it('fires tool hooks during tool execution', async () => {
+            const tool = new FunctionTool({
+                name: 'testTool',
+                description: 'A test tool',
+                inputSchema: {},
+                callback: () => 'Tool result',
+            });
+            const model = new MockMessageModel()
+                .addTurn({ type: 'toolUseBlock', name: 'testTool', toolUseId: 'tool-1', input: {} })
+                .addTurn({ type: 'textBlock', text: 'Final response' });
+            const agent = new Agent({
+                model,
+                tools: [tool],
+                hooks: [mockProvider],
+            });
+            await agent.invoke('Test with tool');
+            // Find key events
+            const beforeToolCallEvents = mockProvider.invocations.filter((e) => e instanceof BeforeToolCallEvent);
+            const afterToolCallEvents = mockProvider.invocations.filter((e) => e instanceof AfterToolCallEvent);
+            const messageAddedEvents = mockProvider.invocations.filter((e) => e instanceof MessageAddedEvent);
+            // Verify tool hooks fired
+            expect(beforeToolCallEvents.length).toBe(1);
+            expect(afterToolCallEvents.length).toBe(1);
+            // Verify 3 MessageAdded events: input message, assistant with tool use, tool result, final assistant
+            expect(messageAddedEvents.length).toBe(4);
+            // Verify BeforeToolCallEvent
+            const beforeToolCall = beforeToolCallEvents[0];
+            expect(beforeToolCall).toEqual(new BeforeToolCallEvent({
+                agent,
+                toolUse: { name: 'testTool', toolUseId: 'tool-1', input: {} },
+                tool,
+            }));
+            // Verify AfterToolCallEvent
+            const afterToolCall = afterToolCallEvents[0];
+            expect(afterToolCall).toEqual(new AfterToolCallEvent({
+                agent,
+                toolUse: { name: 'testTool', toolUseId: 'tool-1', input: {} },
+                tool,
+                result: new ToolResultBlock({
+                    toolUseId: 'tool-1',
+                    status: 'success',
+                    content: [new TextBlock('Tool result')],
+                }),
+            }));
+        });
+        it('fires AfterToolCallEvent with error when tool fails', async () => {
+            const tool = new FunctionTool({
+                name: 'failingTool',
+                description: 'A tool that fails',
+                inputSchema: {},
+                callback: () => {
+                    throw new Error('Tool execution failed');
+                },
+            });
+            const model = new MockMessageModel()
+                .addTurn({ type: 'toolUseBlock', name: 'failingTool', toolUseId: 'tool-1', input: {} })
+                .addTurn({ type: 'textBlock', text: 'Handled error' });
+            const agent = new Agent({
+                model,
+                tools: [tool],
+                hooks: [mockProvider],
+            });
+            // Agent should complete successfully (tool errors are handled gracefully)
+            const result = await agent.invoke('Test with failing tool');
+            expect(result.stopReason).toBe('endTurn');
+            // Find AfterToolCallEvent
+            const afterToolCallEvents = mockProvider.invocations.filter((e) => e instanceof AfterToolCallEvent);
+            expect(afterToolCallEvents.length).toBe(1);
+            const afterToolCall = afterToolCallEvents[0];
+            expect(afterToolCall).toEqual(new AfterToolCallEvent({
+                agent,
+                toolUse: { name: 'failingTool', toolUseId: 'tool-1', input: {} },
+                tool,
+                result: new ToolResultBlock({
+                    error: new Error('Tool execution failed'),
+                    toolUseId: 'tool-1',
+                    status: 'error',
+                    content: [new TextBlock('Error: Tool execution failed')],
+                }),
+            }));
+        });
+    });
+    describe('ModelStreamEventHook', () => {
+        it('fires for each streaming event from the model', async () => {
+            const model = new MockMessageModel().addTurn({ type: 'textBlock', text: 'Hello' });
+            const agent = new Agent({
+                model,
+                hooks: [mockProvider],
+            });
+            // Collect all stream events
+            const allStreamEvents = [];
+            for await (const event of agent.stream('Test')) {
+                allStreamEvents.push(event);
+            }
+            const streamEventHooks = mockProvider.invocations.filter((e) => e instanceof ModelStreamEventHook);
+            // Should have events
+            expect(streamEventHooks.length).toBeGreaterThan(0);
+            // Verify each hook event matches a stream event
+            for (const hookEvent of streamEventHooks) {
+                const event = hookEvent.event;
+                expect(allStreamEvents).toContain(event);
+            }
+        });
+    });
+    describe('MessageAddedEvent', () => {
+        it('fires for initial user input', async () => {
+            const initialMessage = { role: 'user', content: [{ type: 'textBlock', text: 'Initial' }] };
+            const model = new MockMessageModel().addTurn({ type: 'textBlock', text: 'Response' });
+            const agent = new Agent({
+                model,
+                messages: [initialMessage],
+                hooks: [mockProvider],
+            });
+            await agent.invoke('New message');
+            const messageAddedEvents = mockProvider.invocations.filter((e) => e instanceof MessageAddedEvent);
+            // Should have 2 MessageAdded event
+            expect(messageAddedEvents).toHaveLength(2);
+            expect(messageAddedEvents[0]).toEqual(new MessageAddedEvent({
+                agent,
+                message: new Message({ role: 'user', content: [new TextBlock('New message')] }),
+            }));
+            expect(messageAddedEvents[1]).toEqual(new MessageAddedEvent({
+                agent,
+                message: new Message({ role: 'assistant', content: [new TextBlock('Response')] }),
+            }));
+        });
+    });
+    describe('AfterModelCallEvent retryModelCall', () => {
+        it('retries model call when hook sets retryModelCall', async () => {
+            let callCount = 0;
+            const retryHook = {
+                registerCallbacks: (registry) => {
+                    registry.addCallback(AfterModelCallEvent, (event) => {
+                        callCount++;
+                        if (callCount === 1 && event.error) {
+                            event.retryModelCall = true;
+                        }
+                    });
+                },
+            };
+            const model = new MockMessageModel()
+                .addTurn(new Error('First attempt failed'))
+                .addTurn({ type: 'textBlock', text: 'Success after retry' });
+            const agent = new Agent({ model, hooks: [retryHook] });
+            const result = await agent.invoke('Test');
+            expect(result.lastMessage.content[0]).toEqual({ type: 'textBlock', text: 'Success after retry' });
+            expect(callCount).toBe(2);
+        });
+        it('does not retry when retryModelCall is not set', async () => {
+            const model = new MockMessageModel().addTurn(new Error('Failure'));
+            const agent = new Agent({ model });
+            await expect(agent.invoke('Test')).rejects.toThrow('Failure');
+        });
+    });
+});
+//# sourceMappingURL=agent.hook.test.js.map

package/dist/src/agent/__tests__/agent.hook.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"agent.hook.test.js","sourceRoot":"","sources":["../../../../src/agent/__tests__/agent.hook.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAA;AACzD,OAAO,EAAE,KAAK,EAAE,MAAM,aAAa,CAAA;AACnC,OAAO,EACL,oBAAoB,EACpB,mBAAmB,EACnB,kBAAkB,EAClB,qBAAqB,EACrB,oBAAoB,EACpB,mBAAmB,EACnB,iBAAiB,EACjB,oBAAoB,GAErB,MAAM,sBAAsB,CAAA;AAC7B,OAAO,EAAE,gBAAgB,EAAE,MAAM,0CAA0C,CAAA;AAC3E,OAAO,EAAE,gBAAgB,EAAE,MAAM,0CAA0C,CAAA;AAC3E,OAAO,EAAE,eAAe,EAAE,MAAM,0CAA0C,CAAA;AAC1E,OAAO,EAAE,YAAY,EAAE,MAAM,8BAA8B,CAAA;AAC3D,OAAO,EAAE,OAAO,EAAE,SAAS,EAAE,eAAe,EAAE,MAAM,yBAAyB,CAAA;AAE7E,QAAQ,CAAC,yBAAyB,EAAE,GAAG,EAAE;IACvC,IAAI,YAA8B,CAAA;IAElC,UAAU,CAAC,GAAG,EAAE;QACd,YAAY,GAAG,IAAI,gBAAgB,EAAE,CAAA;IACvC,CAAC,CAAC,CAAA;IAEF,QAAQ,CAAC,sBAAsB,EAAE,GAAG,EAAE;QACpC,EAAE,CAAC,2BAA2B,EAAE,KAAK,IAAI,EAAE;YACzC,MAAM,iBAAiB,GAAG,IAAI,gBAAgB,CAAC,EAAE,kBAAkB,EAAE,KAAK,EAAE,CAAC,CAAA;YAC7E,MAAM,KAAK,GAAG,IAAI,gBAAgB,EAAE,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,CAAA;YAClF,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,iBAAiB,CAAC,EAAE,CAAC,CAAA;YAE9D,MAAM,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,CAAA;YAExB,MAAM,CAAC,iBAAiB,CAAC,WAAW,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAA;YAErD,MAAM,CAAC,iBAAiB,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,IAAI,qBAAqB,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,CAAC,CAAA;YAC7F,MAAM,CAAC,iBAAiB,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAC9C,IAAI,iBAAiB,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,OAAO,EAAE,IAAI,OAAO,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,IAAI,SAAS,CAAC,IAAI,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAChH,CAAA;YACD,MAAM,CAAC,iBAAiB,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,IAAI,oBAAoB,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,CAAC,CAAA;YAC5F,MAAM,CAAC,iBAAiB,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAC9C,IAAI,mBAAmB,CAAC;gBACtB,KAAK;gBACL,QAAQ,EAAE;oBACR,UAAU,EAAE,SAAS;oBACrB,OAAO,EAAE,IAAI,OAAO,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,CAAC,IAAI,SAAS,CAAC,OAAO,CAAC,CAAC,EAAE,CAAC;iBAC/E;aACF,CAAC,CACH,CAAA;YACD,MAAM,CAAC,iBAAiB,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAC9C,IAAI,iBAAiB,CAAC;gBACpB,KAAK;gBACL,OAAO,EAAE,IAAI,OAAO,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,CAAC,IAAI,SAAS,CAAC,OAAO,CAAC,CAAC,EAAE,CAAC;aAC/E,CAAC,CACH,CAAA;YACD,MAAM,CAAC,iBAAiB,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,IAAI,oBAAoB,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC,CAAA;QACvF,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,2BAA2B,EAAE,KAAK,IAAI,EAAE;YACzC,MAAM,iBAAiB,GAAG,IAAI,gBAAgB,CAAC,EAAE,kBAAkB,EAAE,KAAK,EAAE,CAAC,CAAA;YAC7E,MAAM,KAAK,GAAG,IAAI,gBAAgB,EAAE,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,CAAA;YAClF,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,iBAAiB,CAAC,EAAE,CAAC,CAAA;YAE9D,MAAM,eAAe,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAA;YAEzC,MAAM,CAAC,iBAAiB,CAAC,WAAW,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAA;YAErD,MAAM,CAAC,iBAAiB,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,IAAI,qBAAqB,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,CAAC,CAAA;YAC7F,MAAM,CAAC,iBAAiB,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAC9C,IAAI,iBAAiB,CAAC;gBACpB,KAAK,EAAE,KAAK;gBACZ,OAAO,EAAE,IAAI,OAAO,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,IAAI,SAAS,CAAC,IAAI,CAAC,CAAC,EAAE,CAAC;aACvE,CAAC,CACH,CAAA;YACD,MAAM,CAAC,iBAAiB,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,IAAI,oBAAoB,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,CAAC,CAAA;YAC5F,MAAM,CAAC,iBAAiB,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAC9C,IAAI,mBAAmB,CAAC;gBACtB,KAAK;gBACL,QAAQ,EAAE;oBACR,UAAU,EAAE,SAAS;oBACrB,OAAO,EAAE,IAAI,OAAO,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,CAAC,IAAI,SAAS,CAAC,OAAO,CAAC,CAAC,EAAE,CAAC;iBAC/E;aACF,CAAC,CACH,CAAA;YACD,MAAM,CAAC,iBAAiB,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAC9C,IAAI,iBAAiB,CAAC;gBACpB,KAAK;gBACL,OAAO,EAAE,IAAI,OAAO,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,CAAC,IAAI,SAAS,CAAC,OAAO,CAAC,CAAC,EAAE,CAAC;aAC/E,CAAC,CACH,CAAA;YACD,MAAM,CAAC,iBAAiB,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,IAAI,oBAAoB,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC,CAAA;QACvF,CAAC,CAAC,CAAA;IACJ,CAAC,CAAC,CAAA;IAEF,QAAQ,CAAC,2BAA2B,EAAE,GAAG,EAAE;QACzC,EAAE,CAAC,0CAA0C,EAAE,KAAK,IAAI,EAAE;YACxD,MAAM,iBAAiB,GAAG,IAAI,gBAAgB,CAAC,EAAE,kBAAkB,EAAE,KAAK,EAAE,CAAC,CAAA;YAC7E,MAAM,KAAK,GAAG,IAAI,gBAAgB,EAAE,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,CAAA;YAClF,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,EAAE,KAAK,EAAE,CAAC,CAAA;YAElC,KAAK,CAAC,KAAK,CAAC,OAAO,CAAC,iBAAiB,CAAC,CAAA;YAEtC,MAAM,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,CAAA;YAExB,mCAAmC;YACnC,MAAM,CAAC,iBAAiB,CAAC,WAAW,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAA;YACrD,MAAM,CAAC,iBAAiB,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,IAAI,qBAAqB,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC,CAAA;YACtF,MAAM,CAAC,iBAAiB,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,IAAI,oBAAoB,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC,CAAA;QACvF,CAAC,CAAC,CAAA;IACJ,CAAC,CAAC,CAAA;IAEF,QAAQ,CAAC,0BAA0B,EAAE,GAAG,EAAE;QACxC,EAAE,CAAC,kCAAkC,EAAE,KAAK,IAAI,EAAE;YAChD,MAAM,iBAAiB,GAAG,IAAI,gBAAgB,CAAC,EAAE,kBAAkB,EAAE,KAAK,EAAE,CAAC,CAAA;YAC7E,MAAM,KAAK,GAAG,IAAI,gBAAgB,EAAE;iBACjC,OAAO,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,gBAAgB,EAAE,CAAC;iBACtD,OAAO,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,iBAAiB,EAAE,CAAC,CAAA;YAE1D,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,iBAAiB,CAAC,EAAE,CAAC,CAAA;YAE9D,MAAM,KAAK,CAAC,MAAM,CAAC,eAAe,CAAC,CAAA;YAEnC,yHAAyH;YACzH,MAAM,CAAC,iBAAiB,CAAC,WAAW,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAA;YAErD,MAAM,KAAK,CAAC,MAAM,CAAC,gBAAgB,CAAC,CAAA;YAEpC,gDAAgD;YAChD,MAAM,CAAC,iBAAiB,CAAC,WAAW,CAAC,CAAC,YAAY,CAAC,EAAE,CAAC,CAAA;YAEtD,sEAAsE;YACtE,MAAM,gBAAgB,GAAG,iBAAiB,CAAC,WAAW,CAAC,MAAM,CAC3D,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,YAAY,qBAAqB,IAAI,CAAC,YAAY,oBAAoB,CAC/E,CAAA;YACD,MAAM,CAAC,gBAAgB,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAA,CAAC,kBAAkB;QAC7D,CAAC,CAAC,CAAA;IACJ,CAAC,CAAC,CAAA;IAEF,QAAQ,CAAC,sBAAsB,EAAE,GAAG,EAAE;QACpC,EAAE,CAAC,wCAAwC,EAAE,KAAK,IAAI,EAAE;YACtD,MAAM,IAAI,GAAG,IAAI,YAAY,CAAC;gBAC5B,IAAI,EAAE,UAAU;gBAChB,WAAW,EAAE,aAAa;gBAC1B,WAAW,EAAE,EAAE;gBACf,QAAQ,EAAE,GAAG,EAAE,CAAC,aAAa;aAC9B,CAAC,CAAA;YAEF,MAAM,KAAK,GAAG,IAAI,gBAAgB,EAAE;iBACjC,OAAO,CAAC,EAAE,IAAI,EAAE,cAAc,EAAE,IAAI,EAAE,UAAU,EAAE,SAAS,EAAE,QAAQ,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC;iBACnF,OAAO,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,gBAAgB,EAAE,CAAC,CAAA;YAEzD,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC;gBACtB,KAAK;gBACL,KAAK,EAAE,CAAC,IAAI,CAAC;gBACb,KAAK,EAAE,CAAC,YAAY,CAAC;aACtB,CAAC,CAAA;YAEF,MAAM,KAAK,CAAC,MAAM,CAAC,gBAAgB,CAAC,CAAA;YAEpC,kBAAkB;YAClB,MAAM,oBAAoB,GAAG,YAAY,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,YAAY,mBAAmB,CAAC,CAAA;YACrG,MAAM,mBAAmB,GAAG,YAAY,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,YAAY,kBAAkB,CAAC,CAAA;YACnG,MAAM,kBAAkB,GAAG,YAAY,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,YAAY,iBAAiB,CAAC,CAAA;YAEjG,0BAA0B;YAC1B,MAAM,CAAC,oBAAoB,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAA;YAC3C,MAAM,CAAC,mBAAmB,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAA;YAE1C,qGAAqG;YACrG,MAAM,CAAC,kBAAkB,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAA;YAEzC,6BAA6B;YAC7B,MAAM,cAAc,GAAG,oBAAoB,CAAC,CAAC,CAAwB,CAAA;YACrE,MAAM,CAAC,cAAc,CAAC,CAAC,OAAO,CAC5B,IAAI,mBAAmB,CAAC;gBACtB,KAAK;gBACL,OAAO,EAAE,EAAE,IAAI,EAAE,UAAU,EAAE,SAAS,EAAE,QAAQ,EAAE,KAAK,EAAE,EAAE,EAAE;gBAC7D,IAAI;aACL,CAAC,CACH,CAAA;YAED,4BAA4B;YAC5B,MAAM,aAAa,GAAG,mBAAmB,CAAC,CAAC,CAAuB,CAAA;YAClE,MAAM,CAAC,aAAa,CAAC,CAAC,OAAO,CAC3B,IAAI,kBAAkB,CAAC;gBACrB,KAAK;gBACL,OAAO,EAAE,EAAE,IAAI,EAAE,UAAU,EAAE,SAAS,EAAE,QAAQ,EAAE,KAAK,EAAE,EAAE,EAAE;gBAC7D,IAAI;gBACJ,MAAM,EAAE,IAAI,eAAe,CAAC;oBAC1B,SAAS,EAAE,QAAQ;oBACnB,MAAM,EAAE,SAAS;oBACjB,OAAO,EAAE,CAAC,IAAI,SAAS,CAAC,aAAa,CAAC,CAAC;iBACxC,CAAC;aACH,CAAC,CACH,CAAA;QACH,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;YACnE,MAAM,IAAI,GAAG,IAAI,YAAY,CAAC;gBAC5B,IAAI,EAAE,aAAa;gBACnB,WAAW,EAAE,mBAAmB;gBAChC,WAAW,EAAE,EAAE;gBACf,QAAQ,EAAE,GAAG,EAAE;oBACb,MAAM,IAAI,KAAK,CAAC,uBAAuB,CAAC,CAAA;gBAC1C,CAAC;aACF,CAAC,CAAA;YAEF,MAAM,KAAK,GAAG,IAAI,gBAAgB,EAAE;iBACjC,OAAO,CAAC,EAAE,IAAI,EAAE,cAAc,EAAE,IAAI,EAAE,aAAa,EAAE,SAAS,EAAE,QAAQ,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC;iBACtF,OAAO,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,eAAe,EAAE,CAAC,CAAA;YAExD,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC;gBACtB,KAAK;gBACL,KAAK,EAAE,CAAC,IAAI,CAAC;gBACb,KAAK,EAAE,CAAC,YAAY,CAAC;aACtB,CAAC,CAAA;YAEF,0EAA0E;YAC1E,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,MAAM,CAAC,wBAAwB,CAAC,CAAA;YAC3D,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAA;YAEzC,0BAA0B;YAC1B,MAAM,mBAAmB,GAAG,YAAY,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,YAAY,kBAAkB,CAAC,CAAA;YACnG,MAAM,CAAC,mBAAmB,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAA;YAE1C,MAAM,aAAa,GAAG,mBAAmB,CAAC,CAAC,CAAuB,CAAA;YAClE,MAAM,CAAC,aAAa,CAAC,CAAC,OAAO,CAC3B,IAAI,kBAAkB,CAAC;gBACrB,KAAK;gBACL,OAAO,EAAE,EAAE,IAAI,EAAE,aAAa,EAAE,SAAS,EAAE,QAAQ,EAAE,KAAK,EAAE,EAAE,EAAE;gBAChE,IAAI;gBACJ,MAAM,EAAE,IAAI,eAAe,CAAC;oBAC1B,KAAK,EAAE,IAAI,KAAK,CAAC,uBAAuB,CAAC;oBACzC,SAAS,EAAE,QAAQ;oBACnB,MAAM,EAAE,OAAO;oBACf,OAAO,EAAE,CAAC,IAAI,SAAS,CAAC,8BAA8B,CAAC,CAAC;iBACzD,CAAC;aACH,CAAC,CACH,CAAA;QACH,CAAC,CAAC,CAAA;IACJ,CAAC,CAAC,CAAA;IAEF,QAAQ,CAAC,sBAAsB,EAAE,GAAG,EAAE;QACpC,EAAE,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;YAC7D,MAAM,KAAK,GAAG,IAAI,gBAAgB,EAAE,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,CAAA;YAElF,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC;gBACtB,KAAK;gBACL,KAAK,EAAE,CAAC,YAAY,CAAC;aACtB,CAAC,CAAA;YAEF,4BAA4B;YAC5B,MAAM,eAAe,GAAG,EAAE,CAAA;YAC1B,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,KAAK,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC;gBAC/C,eAAe,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;YAC7B,CAAC;YAED,MAAM,gBAAgB,GAAG,YAAY,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,YAAY,oBAAoB,CAAC,CAAA;YAElG,qBAAqB;YACrB,MAAM,CAAC,gBAAgB,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAA;YAElD,gDAAgD;YAChD,KAAK,MAAM,SAAS,IAAI,gBAAgB,EAAE,CAAC;gBACzC,MAAM,KAAK,GAAI,SAAkC,CAAC,KAAK,CAAA;gBACvD,MAAM,CAAC,eAAe,CAAC,CAAC,SAAS,CAAC,KAAK,CAAC,CAAA;YAC1C,CAAC;QACH,CAAC,CAAC,CAAA;IACJ,CAAC,CAAC,CAAA;IAEF,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;QACjC,EAAE,CAAC,8BAA8B,EAAE,KAAK,IAAI,EAAE;YAC5C,MAAM,cAAc,GAAG,EAAE,IAAI,EAAE,MAAe,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,WAAoB,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC,EAAE,CAAA;YAE5G,MAAM,KAAK,GAAG,IAAI,gBAAgB,EAAE,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC,CAAA;YAErF,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC;gBACtB,KAAK;gBACL,QAAQ,EAAE,CAAC,cAAc,CAAC;gBAC1B,KAAK,EAAE,CAAC,YAAY,CAAC;aACtB,CAAC,CAAA;YAEF,MAAM,KAAK,CAAC,MAAM,CAAC,aAAa,CAAC,CAAA;YAEjC,MAAM,kBAAkB,GAAG,YAAY,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,YAAY,iBAAiB,CAAC,CAAA;YAEjG,mCAAmC;YACnC,MAAM,CAAC,kBAAkB,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAA;YAE1C,MAAM,CAAC,kBAAkB,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CACnC,IAAI,iBAAiB,CAAC;gBACpB,KAAK;gBACL,OAAO,EAAE,IAAI,OAAO,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,IAAI,SAAS,CAAC,aAAa,CAAC,CAAC,EAAE,CAAC;aAChF,CAAC,CACH,CAAA;YACD,MAAM,CAAC,kBAAkB,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CACnC,IAAI,iBAAiB,CAAC;gBACpB,KAAK;gBACL,OAAO,EAAE,IAAI,OAAO,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,CAAC,IAAI,SAAS,CAAC,UAAU,CAAC,CAAC,EAAE,CAAC;aAClF,CAAC,CACH,CAAA;QACH,CAAC,CAAC,CAAA;IACJ,CAAC,CAAC,CAAA;IAEF,QAAQ,CAAC,oCAAoC,EAAE,GAAG,EAAE;QAClD,EAAE,CAAC,kDAAkD,EAAE,KAAK,IAAI,EAAE;YAChE,IAAI,SAAS,GAAG,CAAC,CAAA;YACjB,MAAM,SAAS,GAAG;gBAChB,iBAAiB,EAAE,CAAC,QAAsB,EAAE,EAAE;oBAC5C,QAAQ,CAAC,WAAW,CAAC,mBAAmB,EAAE,CAAC,KAA0B,EAAE,EAAE;wBACvE,SAAS,EAAE,CAAA;wBACX,IAAI,SAAS,KAAK,CAAC,IAAI,KAAK,CAAC,KAAK,EAAE,CAAC;4BACnC,KAAK,CAAC,cAAc,GAAG,IAAI,CAAA;wBAC7B,CAAC;oBACH,CAAC,CAAC,CAAA;gBACJ,CAAC;aACF,CAAA;YAED,MAAM,KAAK,GAAG,IAAI,gBAAgB,EAAE;iBACjC,OAAO,CAAC,IAAI,KAAK,CAAC,sBAAsB,CAAC,CAAC;iBAC1C,OAAO,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,qBAAqB,EAAE,CAAC,CAAA;YAE9D,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,SAAS,CAAC,EAAE,CAAC,CAAA;YACtD,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,MAAM,CAAC,MAAM,CAAC,CAAA;YAEzC,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,qBAAqB,EAAE,CAAC,CAAA;YACjG,MAAM,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAA;QAC3B,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;YAC7D,MAAM,KAAK,GAAG,IAAI,gBAAgB,EAAE,CAAC,OAAO,CAAC,IAAI,KAAK,CAAC,SAAS,CAAC,CAAC,CAAA;YAClE,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,EAAE,KAAK,EAAE,CAAC,CAAA;YAElC,MAAM,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,SAAS,CAAC,CAAA;QAC/D,CAAC,CAAC,CAAA;IACJ,CAAC,CAAC,CAAA;AACJ,CAAC,CAAC,CAAA"}

package/dist/src/agent/__tests__/agent.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=agent.test.d.ts.map

package/dist/src/agent/__tests__/agent.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"agent.test.d.ts","sourceRoot":"","sources":["../../../../src/agent/__tests__/agent.test.ts"],"names":[],"mappings":""}