@multiplayer-app/ai-agent-node 0.0.1 → 0.1.0-beta.2

package/dist/cjs/helpers/AIHelper.test.js

@@ -0,0 +1,334 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const vitest_1 = require("vitest");
+const AIHelper_js_1 = require("./AIHelper.js");
+const ModelStore_js_1 = require("../store/ModelStore.js");
+const ai_agent_types_1 = require("@multiplayer-app/ai-agent-types");
+(0, vitest_1.describe)('AIHelper.getAssistantResponse', () => {
+    let modelStore;
+    let configStore;
+    (0, vitest_1.beforeAll)(() => {
+        // Setup real ModelStore instance
+        modelStore = ModelStore_js_1.ModelStore.getInstance();
+        modelStore.setModels([
+            { id: 'openai/gpt-4o', label: 'GPT-4o Mini', provider: 'openrouter' }
+        ]);
+    });
+    (0, vitest_1.describe)('with tool approval responses', () => {
+        (0, vitest_1.it)('should handle approved tool with successful execution', async () => {
+            // Note: getAssistantResponse accepts Pick<AgentMessage, 'role' | 'content'>[]
+            // but convertMessages internally handles toolCalls, so we cast for testing
+            const messages = [
+                {
+                    role: ai_agent_types_1.MessageRole.User,
+                    content: 'Please execute the tool'
+                },
+                {
+                    role: ai_agent_types_1.MessageRole.Assistant,
+                    content: 'I will execute the tool for you.',
+                    toolCalls: [
+                        {
+                            id: 'tool-call-1',
+                            name: 'test-tool',
+                            input: { param: 'value' },
+                            status: ai_agent_types_1.AgentToolCallStatus.Succeeded,
+                            approved: true,
+                            output: { result: 'success', data: { executed: true } }
+                        }
+                    ]
+                }
+            ];
+            const response = await AIHelper_js_1.AIHelper.getAssistantResponse(messages);
+            // Verify we get a response (actual model call)
+            (0, vitest_1.expect)(response).toBeDefined();
+            (0, vitest_1.expect)(typeof response).toBe('string');
+            (0, vitest_1.expect)(response.length).toBeGreaterThan(0);
+            // The response should acknowledge the tool execution
+            // (exact text depends on model, so we just verify it's a valid response)
+        }, 30000);
+        (0, vitest_1.it)('should handle denied tool with execution-denied output', async () => {
+            const messages = [
+                {
+                    role: ai_agent_types_1.MessageRole.User,
+                    content: 'Please execute the tool'
+                },
+                {
+                    role: ai_agent_types_1.MessageRole.Assistant,
+                    content: 'I will execute the tool for you.',
+                    toolCalls: [
+                        {
+                            id: 'tool-call-1',
+                            name: 'test-tool',
+                            input: { param: 'value' },
+                            status: ai_agent_types_1.AgentToolCallStatus.Succeeded,
+                            approved: false,
+                            reason: 'User denied execution',
+                            output: {
+                                type: 'execution-denied',
+                                reason: 'User denied execution'
+                            }
+                        }
+                    ]
+                }
+            ];
+            const response = await AIHelper_js_1.AIHelper.getAssistantResponse(messages);
+            // Verify we get a response acknowledging the denial
+            (0, vitest_1.expect)(response).toBeDefined();
+            (0, vitest_1.expect)(typeof response).toBe('string');
+            (0, vitest_1.expect)(response.length).toBeGreaterThan(0);
+            // The model should respond to the execution-denied output
+        }, 30000);
+        (0, vitest_1.it)('should handle multiple tools with mixed approvals', async () => {
+            const messages = [
+                {
+                    role: ai_agent_types_1.MessageRole.User,
+                    content: 'Execute multiple tools'
+                },
+                {
+                    role: ai_agent_types_1.MessageRole.Assistant,
+                    content: 'I will execute the tools.',
+                    toolCalls: [
+                        {
+                            id: 'tool-call-1',
+                            name: 'approved-tool',
+                            input: { param: 'value1' },
+                            status: ai_agent_types_1.AgentToolCallStatus.Succeeded,
+                            approved: true,
+                            output: { result: 'approved-result' }
+                        },
+                        {
+                            id: 'tool-call-2',
+                            name: 'denied-tool',
+                            input: { param: 'value2' },
+                            status: ai_agent_types_1.AgentToolCallStatus.Succeeded,
+                            approved: false,
+                            reason: 'Not allowed',
+                            output: {
+                                type: 'execution-denied',
+                                reason: 'Not allowed'
+                            }
+                        }
+                    ]
+                }
+            ];
+            const response = await AIHelper_js_1.AIHelper.getAssistantResponse(messages);
+            // Verify we get a response handling both tool results
+            (0, vitest_1.expect)(response).toBeDefined();
+            (0, vitest_1.expect)(typeof response).toBe('string');
+            (0, vitest_1.expect)(response.length).toBeGreaterThan(0);
+            // The model should process both approved and denied tool results
+        }, 30000);
+        (0, vitest_1.it)('should handle tool execution failure', async () => {
+            const messages = [
+                {
+                    role: ai_agent_types_1.MessageRole.User,
+                    content: 'Execute the tool'
+                },
+                {
+                    role: ai_agent_types_1.MessageRole.Assistant,
+                    content: 'I will execute the tool.',
+                    toolCalls: [
+                        {
+                            id: 'tool-call-1',
+                            name: 'failing-tool',
+                            input: { param: 'value' },
+                            status: ai_agent_types_1.AgentToolCallStatus.Failed,
+                            approved: true,
+                            error: 'Tool execution failed: Network error'
+                        }
+                    ]
+                }
+            ];
+            const response = await AIHelper_js_1.AIHelper.getAssistantResponse(messages);
+            // Verify we get a response even when tool execution failed
+            (0, vitest_1.expect)(response).toBeDefined();
+            (0, vitest_1.expect)(typeof response).toBe('string');
+            (0, vitest_1.expect)(response.length).toBeGreaterThan(0);
+        }, 30000);
+        (0, vitest_1.it)('should handle tool with execution-denied due to tool not found', async () => {
+            const messages = [
+                {
+                    role: ai_agent_types_1.MessageRole.User,
+                    content: 'Execute the tool'
+                },
+                {
+                    role: ai_agent_types_1.MessageRole.Assistant,
+                    content: 'I will execute the tool.',
+                    toolCalls: [
+                        {
+                            id: 'tool-call-1',
+                            name: 'missing-tool',
+                            input: { param: 'value' },
+                            status: ai_agent_types_1.AgentToolCallStatus.Succeeded,
+                            approved: true,
+                            output: {
+                                type: 'execution-denied',
+                                reason: 'Tool was not found'
+                            }
+                        }
+                    ]
+                }
+            ];
+            const response = await AIHelper_js_1.AIHelper.getAssistantResponse(messages);
+            // Verify we get a response when tool was not found
+            (0, vitest_1.expect)(response).toBeDefined();
+            (0, vitest_1.expect)(typeof response).toBe('string');
+            (0, vitest_1.expect)(response.length).toBeGreaterThan(0);
+        }, 30000);
+        (0, vitest_1.it)('should handle tool with execution-denied due to tool execution error', async () => {
+            const messages = [
+                {
+                    role: ai_agent_types_1.MessageRole.User,
+                    content: 'Execute the tool'
+                },
+                {
+                    role: ai_agent_types_1.MessageRole.Assistant,
+                    content: 'I will execute the tool.',
+                    toolCalls: [
+                        {
+                            id: 'tool-call-1',
+                            name: 'error-tool',
+                            input: { param: 'value' },
+                            status: ai_agent_types_1.AgentToolCallStatus.Succeeded,
+                            approved: true,
+                            output: {
+                                type: 'execution-denied',
+                                reason: 'Tool failed with error'
+                            }
+                        }
+                    ]
+                }
+            ];
+            const response = await AIHelper_js_1.AIHelper.getAssistantResponse(messages);
+            // Verify we get a response when tool execution had an error
+            (0, vitest_1.expect)(response).toBeDefined();
+            (0, vitest_1.expect)(typeof response).toBe('string');
+            (0, vitest_1.expect)(response.length).toBeGreaterThan(0);
+        }, 30000);
+        (0, vitest_1.it)('should handle messages without tool calls', async () => {
+            const messages = [
+                {
+                    role: ai_agent_types_1.MessageRole.User,
+                    content: 'Hello, how are you?'
+                },
+                {
+                    role: ai_agent_types_1.MessageRole.Assistant,
+                    content: 'I am doing well, thank you!'
+                }
+            ];
+            const response = await AIHelper_js_1.AIHelper.getAssistantResponse(messages);
+            // Verify we get a response for normal conversation
+            (0, vitest_1.expect)(response).toBeDefined();
+            (0, vitest_1.expect)(typeof response).toBe('string');
+            (0, vitest_1.expect)(response.length).toBeGreaterThan(0);
+        }, 30000);
+        (0, vitest_1.it)('should handle tool calls without output (pending approval)', async () => {
+            const messages = [
+                {
+                    role: ai_agent_types_1.MessageRole.User,
+                    content: 'Execute the tool'
+                },
+                {
+                    role: ai_agent_types_1.MessageRole.Assistant,
+                    content: 'I need to execute a tool.',
+                    toolCalls: [
+                        {
+                            id: 'tool-call-1',
+                            name: 'pending-tool',
+                            input: { param: 'value' },
+                            status: ai_agent_types_1.AgentToolCallStatus.Pending,
+                            requiresConfirmation: true,
+                            approvalId: 'approval-123'
+                        }
+                    ]
+                }
+            ];
+            const response = await AIHelper_js_1.AIHelper.getAssistantResponse(messages);
+            // Verify we get a response even when tool is pending approval
+            // (tool calls without output should not include tool-result in messages)
+            (0, vitest_1.expect)(response).toBeDefined();
+            (0, vitest_1.expect)(typeof response).toBe('string');
+            (0, vitest_1.expect)(response.length).toBeGreaterThan(0);
+        }, 30000);
+        (0, vitest_1.it)('should handle sequential tool approvals', async () => {
+            const messages = [
+                {
+                    role: ai_agent_types_1.MessageRole.User,
+                    content: 'First request'
+                },
+                {
+                    role: ai_agent_types_1.MessageRole.Assistant,
+                    content: 'Executing first tool.',
+                    toolCalls: [
+                        {
+                            id: 'tool-call-1',
+                            name: 'first-tool',
+                            input: { step: 1 },
+                            status: ai_agent_types_1.AgentToolCallStatus.Succeeded,
+                            approved: true,
+                            output: { result: 'first-result' }
+                        }
+                    ]
+                },
+                {
+                    role: ai_agent_types_1.MessageRole.Assistant,
+                    content: 'Now executing second tool.',
+                    toolCalls: [
+                        {
+                            id: 'tool-call-2',
+                            name: 'second-tool',
+                            input: { step: 2 },
+                            status: ai_agent_types_1.AgentToolCallStatus.Succeeded,
+                            approved: false,
+                            reason: 'User denied',
+                            output: {
+                                type: 'execution-denied',
+                                reason: 'User denied'
+                            }
+                        }
+                    ]
+                }
+            ];
+            const response = await AIHelper_js_1.AIHelper.getAssistantResponse(messages);
+            // Verify we get a response for sequential tool calls
+            (0, vitest_1.expect)(response).toBeDefined();
+            (0, vitest_1.expect)(typeof response).toBe('string');
+            (0, vitest_1.expect)(response.length).toBeGreaterThan(0);
+            // The model should process both sequential tool results
+        }, 30000);
+        (0, vitest_1.it)('should pass through options correctly', async () => {
+            const messages = [
+                {
+                    role: ai_agent_types_1.MessageRole.User,
+                    content: 'Test message'
+                }
+            ];
+            const response = await AIHelper_js_1.AIHelper.getAssistantResponse(messages, undefined, {
+                temperature: 0.9,
+                maxOutputTokens: 1000
+            });
+            // Verify we get a response with custom options applied
+            (0, vitest_1.expect)(response).toBeDefined();
+            (0, vitest_1.expect)(typeof response).toBe('string');
+            (0, vitest_1.expect)(response.length).toBeGreaterThan(0);
+            // Options are passed through to the model
+        }, 30000);
+        (0, vitest_1.it)('should handle abort signal', async () => {
+            const messages = [
+                {
+                    role: ai_agent_types_1.MessageRole.User,
+                    content: 'Test message'
+                }
+            ];
+            const abortController = new AbortController();
+            const signal = abortController.signal;
+            const response = await AIHelper_js_1.AIHelper.getAssistantResponse(messages, signal);
+            // Verify we get a response with abort signal passed through
+            (0, vitest_1.expect)(response).toBeDefined();
+            (0, vitest_1.expect)(typeof response).toBe('string');
+            (0, vitest_1.expect)(response.length).toBeGreaterThan(0);
+            // Signal is passed through to the model for cancellation support
+        }, 30000);
+    });
+});
+//# sourceMappingURL=AIHelper.test.js.map

package/dist/cjs/helpers/AIHelper.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"AIHelper.test.js","sourceRoot":"","sources":["../../../src/helpers/AIHelper.test.ts"],"names":[],"mappings":";;AAAA,mCAAyD;AACzD,+CAAyC;AACzC,0DAAoD;AAEpD,oEAAmF;AAGnF,IAAA,iBAAQ,EAAC,+BAA+B,EAAE,GAAG,EAAE;IAC7C,IAAI,UAAsB,CAAC;IAC3B,IAAI,WAAwB,CAAC;IAE7B,IAAA,kBAAS,EAAC,GAAG,EAAE;QACb,iCAAiC;QACjC,UAAU,GAAG,0BAAU,CAAC,WAAW,EAAE,CAAC;QACtC,UAAU,CAAC,SAAS,CAAC;YACnB,EAAE,EAAE,EAAE,eAAe,EAAE,KAAK,EAAE,aAAa,EAAE,QAAQ,EAAE,YAAY,EAAE;SACtE,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,IAAA,iBAAQ,EAAC,8BAA8B,EAAE,GAAG,EAAE;QAC5C,IAAA,WAAE,EAAC,uDAAuD,EAAE,KAAK,IAAI,EAAE;YACrE,8EAA8E;YAC9E,2EAA2E;YAC3E,MAAM,QAAQ,GAAG;gBACf;oBACE,IAAI,EAAE,4BAAW,CAAC,IAAI;oBACtB,OAAO,EAAE,yBAAyB;iBACnC;gBACD;oBACE,IAAI,EAAE,4BAAW,CAAC,SAAS;oBAC3B,OAAO,EAAE,kCAAkC;oBAC3C,SAAS,EAAE;wBACT;4BACE,EAAE,EAAE,aAAa;4BACjB,IAAI,EAAE,WAAW;4BACjB,KAAK,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE;4BACzB,MAAM,EAAE,oCAAmB,CAAC,SAAS;4BACrC,QAAQ,EAAE,IAAI;4BACd,MAAM,EAAE,EAAE,MAAM,EAAE,SAAS,EAAE,IAAI,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,EAAE;yBACxD;qBACF;iBACF;aACwD,CAAC;YAE5D,MAAM,QAAQ,GAAG,MAAM,sBAAQ,CAAC,oBAAoB,CAAC,QAAQ,CAAC,CAAC;YAE/D,+CAA+C;YAC/C,IAAA,eAAM,EAAC,QAAQ,CAAC,CAAC,WAAW,EAAE,CAAC;YAC/B,IAAA,eAAM,EAAC,OAAO,QAAQ,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YACvC,IAAA,eAAM,EAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;YAE3C,qDAAqD;YACrD,yEAAyE;QAC3E,CAAC,EAAE,KAAK,CAAC,CAAC;QAEV,IAAA,WAAE,EAAC,wDAAwD,EAAE,KAAK,IAAI,EAAE;YACtE,MAAM,QAAQ,GAAG;gBACf;oBACE,IAAI,EAAE,4BAAW,CAAC,IAAI;oBACtB,OAAO,EAAE,yBAAyB;iBACnC;gBACD;oBACE,IAAI,EAAE,4BAAW,CAAC,SAAS;oBAC3B,OAAO,EAAE,kCAAkC;oBAC3C,SAAS,EAAE;wBACT;4BACE,EAAE,EAAE,aAAa;4BACjB,IAAI,EAAE,WAAW;4BACjB,KAAK,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE;4BACzB,MAAM,EAAE,oCAAmB,CAAC,SAAS;4BACrC,QAAQ,EAAE,KAAK;4BACf,MAAM,EAAE,uBAAuB;4BAC/B,MAAM,EAAE;gCACN,IAAI,EAAE,kBAAkB;gCACxB,MAAM,EAAE,uBAAuB;6BAChC;yBACF;qBACF;iBACF;aACwD,CAAC;YAE5D,MAAM,QAAQ,GAAG,MAAM,sBAAQ,CAAC,oBAAoB,CAAC,QAAQ,CAAC,CAAC;YAE/D,oDAAoD;YACpD,IAAA,eAAM,EAAC,QAAQ,CAAC,CAAC,WAAW,EAAE,CAAC;YAC/B,IAAA,eAAM,EAAC,OAAO,QAAQ,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YACvC,IAAA,eAAM,EAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;YAE3C,0DAA0D;QAC5D,CAAC,EAAE,KAAK,CAAC,CAAC;QAEV,IAAA,WAAE,EAAC,mDAAmD,EAAE,KAAK,IAAI,EAAE;YACjE,MAAM,QAAQ,GAAG;gBACf;oBACE,IAAI,EAAE,4BAAW,CAAC,IAAI;oBACtB,OAAO,EAAE,wBAAwB;iBAClC;gBACD;oBACE,IAAI,EAAE,4BAAW,CAAC,SAAS;oBAC3B,OAAO,EAAE,2BAA2B;oBACpC,SAAS,EAAE;wBACT;4BACE,EAAE,EAAE,aAAa;4BACjB,IAAI,EAAE,eAAe;4BACrB,KAAK,EAAE,EAAE,KAAK,EAAE,QAAQ,EAAE;4BAC1B,MAAM,EAAE,oCAAmB,CAAC,SAAS;4BACrC,QAAQ,EAAE,IAAI;4BACd,MAAM,EAAE,EAAE,MAAM,EAAE,iBAAiB,EAAE;yBACtC;wBACD;4BACE,EAAE,EAAE,aAAa;4BACjB,IAAI,EAAE,aAAa;4BACnB,KAAK,EAAE,EAAE,KAAK,EAAE,QAAQ,EAAE;4BAC1B,MAAM,EAAE,oCAAmB,CAAC,SAAS;4BACrC,QAAQ,EAAE,KAAK;4BACf,MAAM,EAAE,aAAa;4BACrB,MAAM,EAAE;gCACN,IAAI,EAAE,kBAAkB;gCACxB,MAAM,EAAE,aAAa;6BACtB;yBACF;qBACF;iBACF;aACwD,CAAC;YAE5D,MAAM,QAAQ,GAAG,MAAM,sBAAQ,CAAC,oBAAoB,CAAC,QAAQ,CAAC,CAAC;YAE/D,sDAAsD;YACtD,IAAA,eAAM,EAAC,QAAQ,CAAC,CAAC,WAAW,EAAE,CAAC;YAC/B,IAAA,eAAM,EAAC,OAAO,QAAQ,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YACvC,IAAA,eAAM,EAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;YAE3C,iEAAiE;QACnE,CAAC,EAAE,KAAK,CAAC,CAAC;QAEV,IAAA,WAAE,EAAC,sCAAsC,EAAE,KAAK,IAAI,EAAE;YACpD,MAAM,QAAQ,GAAG;gBACf;oBACE,IAAI,EAAE,4BAAW,CAAC,IAAI;oBACtB,OAAO,EAAE,kBAAkB;iBAC5B;gBACD;oBACE,IAAI,EAAE,4BAAW,CAAC,SAAS;oBAC3B,OAAO,EAAE,0BAA0B;oBACnC,SAAS,EAAE;wBACT;4BACE,EAAE,EAAE,aAAa;4BACjB,IAAI,EAAE,cAAc;4BACpB,KAAK,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE;4BACzB,MAAM,EAAE,oCAAmB,CAAC,MAAM;4BAClC,QAAQ,EAAE,IAAI;4BACd,KAAK,EAAE,sCAAsC;yBAC9C;qBACF;iBACF;aACwD,CAAC;YAE5D,MAAM,QAAQ,GAAG,MAAM,sBAAQ,CAAC,oBAAoB,CAAC,QAAQ,CAAC,CAAC;YAE/D,2DAA2D;YAC3D,IAAA,eAAM,EAAC,QAAQ,CAAC,CAAC,WAAW,EAAE,CAAC;YAC/B,IAAA,eAAM,EAAC,OAAO,QAAQ,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YACvC,IAAA,eAAM,EAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;QAC7C,CAAC,EAAE,KAAK,CAAC,CAAC;QAEV,IAAA,WAAE,EAAC,gEAAgE,EAAE,KAAK,IAAI,EAAE;YAC9E,MAAM,QAAQ,GAAG;gBACf;oBACE,IAAI,EAAE,4BAAW,CAAC,IAAI;oBACtB,OAAO,EAAE,kBAAkB;iBAC5B;gBACD;oBACE,IAAI,EAAE,4BAAW,CAAC,SAAS;oBAC3B,OAAO,EAAE,0BAA0B;oBACnC,SAAS,EAAE;wBACT;4BACE,EAAE,EAAE,aAAa;4BACjB,IAAI,EAAE,cAAc;4BACpB,KAAK,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE;4BACzB,MAAM,EAAE,oCAAmB,CAAC,SAAS;4BACrC,QAAQ,EAAE,IAAI;4BACd,MAAM,EAAE;gCACN,IAAI,EAAE,kBAAkB;gCACxB,MAAM,EAAE,oBAAoB;6BAC7B;yBACF;qBACF;iBACF;aACwD,CAAC;YAE5D,MAAM,QAAQ,GAAG,MAAM,sBAAQ,CAAC,oBAAoB,CAAC,QAAQ,CAAC,CAAC;YAE/D,mDAAmD;YACnD,IAAA,eAAM,EAAC,QAAQ,CAAC,CAAC,WAAW,EAAE,CAAC;YAC/B,IAAA,eAAM,EAAC,OAAO,QAAQ,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YACvC,IAAA,eAAM,EAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;QAC7C,CAAC,EAAE,KAAK,CAAC,CAAC;QAEV,IAAA,WAAE,EAAC,sEAAsE,EAAE,KAAK,IAAI,EAAE;YACpF,MAAM,QAAQ,GAAG;gBACf;oBACE,IAAI,EAAE,4BAAW,CAAC,IAAI;oBACtB,OAAO,EAAE,kBAAkB;iBAC5B;gBACD;oBACE,IAAI,EAAE,4BAAW,CAAC,SAAS;oBAC3B,OAAO,EAAE,0BAA0B;oBACnC,SAAS,EAAE;wBACT;4BACE,EAAE,EAAE,aAAa;4BACjB,IAAI,EAAE,YAAY;4BAClB,KAAK,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE;4BACzB,MAAM,EAAE,oCAAmB,CAAC,SAAS;4BACrC,QAAQ,EAAE,IAAI;4BACd,MAAM,EAAE;gCACN,IAAI,EAAE,kBAAkB;gCACxB,MAAM,EAAE,wBAAwB;6BACjC;yBACF;qBACF;iBACF;aACwD,CAAC;YAE5D,MAAM,QAAQ,GAAG,MAAM,sBAAQ,CAAC,oBAAoB,CAAC,QAAQ,CAAC,CAAC;YAE/D,4DAA4D;YAC5D,IAAA,eAAM,EAAC,QAAQ,CAAC,CAAC,WAAW,EAAE,CAAC;YAC/B,IAAA,eAAM,EAAC,OAAO,QAAQ,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YACvC,IAAA,eAAM,EAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;QAC7C,CAAC,EAAE,KAAK,CAAC,CAAC;QAEV,IAAA,WAAE,EAAC,2CAA2C,EAAE,KAAK,IAAI,EAAE;YACzD,MAAM,QAAQ,GAA6C;gBACzD;oBACE,IAAI,EAAE,4BAAW,CAAC,IAAI;oBACtB,OAAO,EAAE,qBAAqB;iBAC/B;gBACD;oBACE,IAAI,EAAE,4BAAW,CAAC,SAAS;oBAC3B,OAAO,EAAE,6BAA6B;iBACvC;aACF,CAAC;YAEF,MAAM,QAAQ,GAAG,MAAM,sBAAQ,CAAC,oBAAoB,CAAC,QAAQ,CAAC,CAAC;YAE/D,mDAAmD;YACnD,IAAA,eAAM,EAAC,QAAQ,CAAC,CAAC,WAAW,EAAE,CAAC;YAC/B,IAAA,eAAM,EAAC,OAAO,QAAQ,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YACvC,IAAA,eAAM,EAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;QAC7C,CAAC,EAAE,KAAK,CAAC,CAAC;QAEV,IAAA,WAAE,EAAC,4DAA4D,EAAE,KAAK,IAAI,EAAE;YAC1E,MAAM,QAAQ,GAAG;gBACf;oBACE,IAAI,EAAE,4BAAW,CAAC,IAAI;oBACtB,OAAO,EAAE,kBAAkB;iBAC5B;gBACD;oBACE,IAAI,EAAE,4BAAW,CAAC,SAAS;oBAC3B,OAAO,EAAE,2BAA2B;oBACpC,SAAS,EAAE;wBACT;4BACE,EAAE,EAAE,aAAa;4BACjB,IAAI,EAAE,cAAc;4BACpB,KAAK,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE;4BACzB,MAAM,EAAE,oCAAmB,CAAC,OAAO;4BACnC,oBAAoB,EAAE,IAAI;4BAC1B,UAAU,EAAE,cAAc;yBAC3B;qBACF;iBACF;aACwD,CAAC;YAE5D,MAAM,QAAQ,GAAG,MAAM,sBAAQ,CAAC,oBAAoB,CAAC,QAAQ,CAAC,CAAC;YAE/D,8DAA8D;YAC9D,yEAAyE;YACzE,IAAA,eAAM,EAAC,QAAQ,CAAC,CAAC,WAAW,EAAE,CAAC;YAC/B,IAAA,eAAM,EAAC,OAAO,QAAQ,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YACvC,IAAA,eAAM,EAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;QAC7C,CAAC,EAAE,KAAK,CAAC,CAAC;QAEV,IAAA,WAAE,EAAC,yCAAyC,EAAE,KAAK,IAAI,EAAE;YACvD,MAAM,QAAQ,GAAG;gBACf;oBACE,IAAI,EAAE,4BAAW,CAAC,IAAI;oBACtB,OAAO,EAAE,eAAe;iBACzB;gBACD;oBACE,IAAI,EAAE,4BAAW,CAAC,SAAS;oBAC3B,OAAO,EAAE,uBAAuB;oBAChC,SAAS,EAAE;wBACT;4BACE,EAAE,EAAE,aAAa;4BACjB,IAAI,EAAE,YAAY;4BAClB,KAAK,EAAE,EAAE,IAAI,EAAE,CAAC,EAAE;4BAClB,MAAM,EAAE,oCAAmB,CAAC,SAAS;4BACrC,QAAQ,EAAE,IAAI;4BACd,MAAM,EAAE,EAAE,MAAM,EAAE,cAAc,EAAE;yBACnC;qBACF;iBACF;gBACD;oBACE,IAAI,EAAE,4BAAW,CAAC,SAAS;oBAC3B,OAAO,EAAE,4BAA4B;oBACrC,SAAS,EAAE;wBACT;4BACE,EAAE,EAAE,aAAa;4BACjB,IAAI,EAAE,aAAa;4BACnB,KAAK,EAAE,EAAE,IAAI,EAAE,CAAC,EAAE;4BAClB,MAAM,EAAE,oCAAmB,CAAC,SAAS;4BACrC,QAAQ,EAAE,KAAK;4BACf,MAAM,EAAE,aAAa;4BACrB,MAAM,EAAE;gCACN,IAAI,EAAE,kBAAkB;gCACxB,MAAM,EAAE,aAAa;6BACtB;yBACF;qBACF;iBACF;aACwD,CAAC;YAE5D,MAAM,QAAQ,GAAG,MAAM,sBAAQ,CAAC,oBAAoB,CAAC,QAAQ,CAAC,CAAC;YAE/D,qDAAqD;YACrD,IAAA,eAAM,EAAC,QAAQ,CAAC,CAAC,WAAW,EAAE,CAAC;YAC/B,IAAA,eAAM,EAAC,OAAO,QAAQ,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YACvC,IAAA,eAAM,EAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;YAE3C,wDAAwD;QAC1D,CAAC,EAAE,KAAK,CAAC,CAAC;QAEV,IAAA,WAAE,EAAC,uCAAuC,EAAE,KAAK,IAAI,EAAE;YACrD,MAAM,QAAQ,GAA6C;gBACzD;oBACE,IAAI,EAAE,4BAAW,CAAC,IAAI;oBACtB,OAAO,EAAE,cAAc;iBACxB;aACF,CAAC;YAEF,MAAM,QAAQ,GAAG,MAAM,sBAAQ,CAAC,oBAAoB,CAAC,QAAQ,EAAE,SAAS,EAAE;gBACxE,WAAW,EAAE,GAAG;gBAChB,eAAe,EAAE,IAAI;aACtB,CAAC,CAAC;YAEH,uDAAuD;YACvD,IAAA,eAAM,EAAC,QAAQ,CAAC,CAAC,WAAW,EAAE,CAAC;YAC/B,IAAA,eAAM,EAAC,OAAO,QAAQ,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YACvC,IAAA,eAAM,EAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;YAE3C,0CAA0C;QAC5C,CAAC,EAAE,KAAK,CAAC,CAAC;QAEV,IAAA,WAAE,EAAC,4BAA4B,EAAE,KAAK,IAAI,EAAE;YAC1C,MAAM,QAAQ,GAA6C;gBACzD;oBACE,IAAI,EAAE,4BAAW,CAAC,IAAI;oBACtB,OAAO,EAAE,cAAc;iBACxB;aACF,CAAC;YAEF,MAAM,eAAe,GAAG,IAAI,eAAe,EAAE,CAAC;YAC9C,MAAM,MAAM,GAAG,eAAe,CAAC,MAAM,CAAC;YAEtC,MAAM,QAAQ,GAAG,MAAM,sBAAQ,CAAC,oBAAoB,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;YAEvE,4DAA4D;YAC5D,IAAA,eAAM,EAAC,QAAQ,CAAC,CAAC,WAAW,EAAE,CAAC;YAC/B,IAAA,eAAM,EAAC,OAAO,QAAQ,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YACvC,IAAA,eAAM,EAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;YAE3C,iEAAiE;QACnE,CAAC,EAAE,KAAK,CAAC,CAAC;IACZ,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/cjs/helpers/ConfigHelper.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"ConfigHelper.d.ts","sourceRoot":"","sources":["../../../src/helpers/ConfigHelper.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,cAAc,EAAa,MAAM,iCAAiC,CAAC;AAIjF,qBAAa,YAAY;IAEvB,MAAM,CAAC,UAAU,CAAC,SAAS,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,GAAG,cAAc;IAcjE,OAAO,CAAC,MAAM,CAAC,eAAe;IAK9B;;;OAGG;IACH,OAAO,CAAC,MAAM,CAAC,uBAAuB;IAwBtC;;OAEG;IACH,OAAO,CAAC,MAAM,CAAC,WAAW;IA8B1B;;;OAGG;IACH,OAAO,CAAC,MAAM,CAAC,cAAc;CAsC9B"}

package/dist/cjs/helpers/ConfigHelper.js

@@ -0,0 +1,122 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ConfigHelper = void 0;
+const zod_1 = require("zod");
+const ai_agent_types_1 = require("@multiplayer-app/ai-agent-types");
+const index_js_1 = require("../libs/logger/index.js");
+class ConfigHelper {
+    static loadConfig(rawConfig) {
+        try {
+            // Convert to AgentAppConfig format
+            return ConfigHelper.convertToAgentAppConfig(rawConfig);
+        }
+        catch (error) {
+            if (error instanceof SyntaxError) {
+                index_js_1.logger.error(`Failed to parse config.json: ${error.message}`);
+                throw new Error(`Invalid JSON in config file: ${error.message}`);
+            }
+            index_js_1.logger.error(`Failed to load config: ${error}`);
+            throw error;
+        }
+    }
+    static jsonSchemaToZod(schema) {
+        if (!schema)
+            return null;
+        return zod_1.z.fromJSONSchema(schema);
+    }
+    /**
+     * Converts raw JSON config to AgentAppConfig format
+     * Validates structure and converts JSON schemas to Zod schemas
+     */
+    static convertToAgentAppConfig(rawConfig) {
+        // Convert agents, transforming JSON schemas to Zod schemas
+        const config = rawConfig.agents.reduce((config, agent) => {
+            agent.contextKeys.forEach((contextKey) => {
+                if (!config[contextKey]) {
+                    config[contextKey] = [];
+                }
+                config[contextKey].push({
+                    name: agent.name,
+                    description: agent.description,
+                    defaultModel: agent.defaultModel,
+                    systemPrompt: agent.systemPrompt,
+                    temperature: agent.temperature || 0.1,
+                    maxTokens: agent.maxTokens || 2000,
+                    tools: agent.tools.map((tool) => ConfigHelper.convertTool(tool)),
+                    outputSchema: agent.outputSchema ? ConfigHelper.jsonSchemaToZod(agent.outputSchema) : undefined
+                });
+            });
+            return config;
+        }, {});
+        return config;
+    }
+    /**
+     * Converts a tool from JSON format (with JSON schemas) to AgentTool format (with Zod schemas)
+     */
+    static convertTool(tool) {
+        var _a;
+        if (tool.type === ai_agent_types_1.AgentToolType.WEB_SEARCH) {
+            return {
+                type: ai_agent_types_1.AgentToolType.WEB_SEARCH,
+                data: {
+                    title: ai_agent_types_1.AgentToolType.WEB_SEARCH,
+                }
+            };
+        }
+        if (tool.type === ai_agent_types_1.AgentToolType.API_TOOL) {
+            return {
+                type: ai_agent_types_1.AgentToolType.API_TOOL,
+                data: {
+                    title: tool.data.title,
+                    description: tool.data.description,
+                    headersToPass: tool.data.headersToPass,
+                    method: tool.data.method,
+                    url: tool.data.url,
+                    body: tool.data.body ? ConfigHelper.jsonSchemaToZod(tool.data.body) : undefined,
+                    queryParams: tool.data.queryParams ? ConfigHelper.jsonSchemaToZod(tool.data.queryParams) : undefined,
+                    needsApproval: (_a = tool.data.needsApproval) !== null && _a !== void 0 ? _a : false,
+                }
+            };
+        }
+        throw new Error(`Unknown tool type: ${tool.type}`);
+    }
+    /**
+     * Validates the AgentAppConfig structure
+     * @throws Error if validation fails
+     */
+    static validateConfig(config) {
+        if (!config.agents || !Array.isArray(config.agents)) {
+            throw new Error('Config must have an "agents" array');
+        }
+        if (!config.contexts || typeof config.contexts !== 'object') {
+            throw new Error('Config must have a "contexts" object');
+        }
+        // Validate agents
+        const agentNames = new Set();
+        for (const agent of config.agents) {
+            if (!agent.name) {
+                throw new Error('All agents must have a "name" field');
+            }
+            if (agentNames.has(agent.name)) {
+                throw new Error(`Duplicate agent name found: ${agent.name}`);
+            }
+            agentNames.add(agent.name);
+            if (!agent.systemPrompt) {
+                throw new Error(`Agent "${agent.name}" must have a "systemPrompt" field`);
+            }
+        }
+        // Validate contexts reference valid agents
+        for (const [contextKey, contextAgentNames] of Object.entries(config.contexts)) {
+            if (!Array.isArray(contextAgentNames)) {
+                throw new Error(`Context "${contextKey}" must have an array of agent names`);
+            }
+            for (const agentName of contextAgentNames) {
+                if (!agentNames.has(agentName)) {
+                    throw new Error(`Context "${contextKey}" references unknown agent: ${agentName}`);
+                }
+            }
+        }
+    }
+}
+exports.ConfigHelper = ConfigHelper;
+//# sourceMappingURL=ConfigHelper.js.map

package/dist/cjs/helpers/ConfigHelper.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ConfigHelper.js","sourceRoot":"","sources":["../../../src/helpers/ConfigHelper.ts"],"names":[],"mappings":";;;AAAA,6BAAwB;AAExB,oEAAgE;AAChE,sDAAiD;AAEjD,MAAa,YAAY;IAEvB,MAAM,CAAC,UAAU,CAAC,SAA8B;QAC9C,IAAI,CAAC;YACH,mCAAmC;YACnC,OAAO,YAAY,CAAC,uBAAuB,CAAC,SAAS,CAAC,CAAC;QACzD,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,KAAK,YAAY,WAAW,EAAE,CAAC;gBACjC,iBAAM,CAAC,KAAK,CAAC,gCAAgC,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC;gBAC9D,MAAM,IAAI,KAAK,CAAC,gCAAgC,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC;YACnE,CAAC;YACD,iBAAM,CAAC,KAAK,CAAC,0BAA0B,KAAK,EAAE,CAAC,CAAC;YAChD,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAEO,MAAM,CAAC,eAAe,CAAE,MAAW;QACzC,IAAI,CAAC,MAAM;YAAE,OAAO,IAAI,CAAC;QACzB,OAAO,OAAC,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC;IAClC,CAAC;IAED;;;OAGG;IACK,MAAM,CAAC,uBAAuB,CAAC,SAAc;QACnD,2DAA2D;QAC3D,MAAM,MAAM,GAAmB,SAAS,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,MAAsB,EAAE,KAAU,EAAE,EAAE;YAC5F,KAAK,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC,UAAkB,EAAE,EAAE;gBAC/C,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,EAAE,CAAC;oBACxB,MAAM,CAAC,UAAU,CAAC,GAAG,EAAE,CAAC;gBAC1B,CAAC;gBACD,MAAM,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC;oBACpB,IAAI,EAAE,KAAK,CAAC,IAAI;oBAChB,WAAW,EAAE,KAAK,CAAC,WAAW;oBAC9B,YAAY,EAAE,KAAK,CAAC,YAAY;oBAChC,YAAY,EAAE,KAAK,CAAC,YAAY;oBAChC,WAAW,EAAE,KAAK,CAAC,WAAW,IAAI,GAAG;oBACrC,SAAS,EAAE,KAAK,CAAC,SAAS,IAAI,IAAI;oBAClC,KAAK,EAAE,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAS,EAAE,EAAE,CAAC,YAAY,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;oBACrE,YAAY,EAAE,KAAK,CAAC,YAAY,CAAC,CAAC,CAAC,YAAY,CAAC,eAAe,CAAC,KAAK,CAAC,YAAY,CAAQ,CAAC,CAAC,CAAC,SAAS;iBACzG,CAAC,CAAC;YACL,CAAC,CAAC,CAAC;YACH,OAAO,MAAM,CAAC;QAChB,CAAC,EAAE,EAAoB,CAAC,CAAC;QAEzB,OAAO,MAAM,CAAA;IACf,CAAC;IAED;;OAEG;IACK,MAAM,CAAC,WAAW,CAAC,IAAS;;QAClC,IAAI,IAAI,CAAC,IAAI,KAAK,8BAAa,CAAC,UAAU,EAAE,CAAC;YAC3C,OAAO;gBACL,IAAI,EAAE,8BAAa,CAAC,UAAU;gBAC9B,IAAI,EAAE;oBACJ,KAAK,EAAE,8BAAa,CAAC,UAAU;iBAChC;aACF,CAAC;QACJ,CAAC;QAED,IAAI,IAAI,CAAC,IAAI,KAAK,8BAAa,CAAC,QAAQ,EAAE,CAAC;YACzC,OAAO;gBACL,IAAI,EAAE,8BAAa,CAAC,QAAQ;gBAC5B,IAAI,EAAE;oBACJ,KAAK,EAAE,IAAI,CAAC,IAAI,CAAC,KAAK;oBACtB,WAAW,EAAE,IAAI,CAAC,IAAI,CAAC,WAAW;oBAClC,aAAa,EAAE,IAAI,CAAC,IAAI,CAAC,aAAa;oBACtC,MAAM,EAAE,IAAI,CAAC,IAAI,CAAC,MAAM;oBACxB,GAAG,EAAE,IAAI,CAAC,IAAI,CAAC,GAAG;oBAClB,IAAI,EAAE,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,YAAY,CAAC,eAAe,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAQ,CAAC,CAAC,CAAC,SAAS;oBACtF,WAAW,EAAE,IAAI,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,YAAY,CAAC,eAAe,CAAC,IAAI,CAAC,IAAI,CAAC,WAAW,CAAQ,CAAC,CAAC,CAAC,SAAS;oBAC3G,aAAa,EAAE,MAAA,IAAI,CAAC,IAAI,CAAC,aAAa,mCAAI,KAAK;iBAChD;aACF,CAAC;QACJ,CAAC;QAED,MAAM,IAAI,KAAK,CAAC,sBAAsB,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;IACrD,CAAC;IAGD;;;OAGG;IACK,MAAM,CAAC,cAAc,CAAC,MAAsB;QAClD,IAAI,CAAC,MAAM,CAAC,MAAM,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC;YACpD,MAAM,IAAI,KAAK,CAAC,oCAAoC,CAAC,CAAC;QACxD,CAAC;QAED,IAAI,CAAC,MAAM,CAAC,QAAQ,IAAI,OAAO,MAAM,CAAC,QAAQ,KAAK,QAAQ,EAAE,CAAC;YAC5D,MAAM,IAAI,KAAK,CAAC,sCAAsC,CAAC,CAAC;QAC1D,CAAC;QAED,kBAAkB;QAClB,MAAM,UAAU,GAAG,IAAI,GAAG,EAAU,CAAC;QACrC,KAAK,MAAM,KAAK,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC;YAClC,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC;gBAChB,MAAM,IAAI,KAAK,CAAC,qCAAqC,CAAC,CAAC;YACzD,CAAC;YACD,IAAI,UAAU,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC;gBAC/B,MAAM,IAAI,KAAK,CAAC,+BAA+B,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC;YAC/D,CAAC;YACD,UAAU,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YAE3B,IAAI,CAAC,KAAK,CAAC,YAAY,EAAE,CAAC;gBACxB,MAAM,IAAI,KAAK,CAAC,UAAU,KAAK,CAAC,IAAI,oCAAoC,CAAC,CAAC;YAC5E,CAAC;QACH,CAAC;QAED,2CAA2C;QAC3C,KAAK,MAAM,CAAC,UAAU,EAAE,iBAAiB,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,EAAE,CAAC;YAC9E,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,iBAAiB,CAAC,EAAE,CAAC;gBACtC,MAAM,IAAI,KAAK,CAAC,YAAY,UAAU,qCAAqC,CAAC,CAAC;YAC/E,CAAC;YAED,KAAK,MAAM,SAAS,IAAI,iBAAiB,EAAE,CAAC;gBAC1C,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,CAAC;oBAC/B,MAAM,IAAI,KAAK,CAAC,YAAY,UAAU,+BAA+B,SAAS,EAAE,CAAC,CAAC;gBACpF,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;CACF;AA5HD,oCA4HC"}

package/dist/cjs/helpers/ContextLimiter.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"ContextLimiter.d.ts","sourceRoot":"","sources":["../../../src/helpers/ContextLimiter.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAe,MAAM,iCAAiC,CAAC;AAE5E;;GAEG;AACH,MAAM,WAAW,mBAAmB;IAClC;;;OAGG;IACH,WAAW,CAAC,EAAE,MAAM,CAAC;IAErB;;;OAGG;IACH,SAAS,CAAC,EAAE,MAAM,CAAC;IAEnB;;;OAGG;IACH,oBAAoB,CAAC,EAAE,OAAO,CAAC;IAE/B;;;OAGG;IACH,kBAAkB,CAAC,EAAE,OAAO,CAAC;CAC9B;AAED;;;;;;;;;;;;;;;;;;;;;;;;GAwBG;AACH,qBAAa,cAAc;IACzB;;;;;;;OAOG;IACH,MAAM,CAAC,YAAY,CACjB,QAAQ,EAAE,YAAY,EAAE,EACxB,OAAO,GAAE,mBAAwB,GAChC,YAAY,EAAE;IA0EjB;;;;;;;;OAQG;IACH,MAAM,CAAC,cAAc,CAAC,OAAO,EAAE,YAAY,GAAG,MAAM;IAwBpD;;;;;;;OAOG;IACH,MAAM,CAAC,aAAa,CAClB,QAAQ,EAAE,YAAY,EAAE,EACxB,SAAS,EAAE,MAAM,GAChB,YAAY,EAAE;CAwClB"}

package/dist/cjs/helpers/ContextLimiter.js

@@ -0,0 +1,169 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ContextLimiter = void 0;
+const ai_agent_types_1 = require("@multiplayer-app/ai-agent-types");
+/**
+ * Common practices for limiting AI context:
+ *
+ * 1. **Message Count Limiting** (implemented here)
+ *    - Keep only the last N messages
+ *    - Most common and simplest approach
+ *    - Typical values: 20-100 messages depending on use case
+ *
+ * 2. **Token-Based Limiting** (future enhancement)
+ *    - Count tokens and keep messages until hitting token budget
+ *    - More accurate for cost control
+ *    - Requires token counting library (e.g., tiktoken, @anthropic-ai/tokenizer)
+ *
+ * 3. **Hybrid Approach** (future enhancement)
+ *    - Combine message count with token budget as safety check
+ *    - Best of both worlds
+ *
+ * 4. **Sliding Window with Summary** (advanced)
+ *    - Keep recent messages + summarize older ones
+ *    - Requires summarization step
+ *
+ * 5. **Priority-Based** (implemented via keepFirstUserMessage)
+ *    - Always keep system messages, first user message, and recent messages
+ *    - Preserves important context while limiting size
+ */
+class ContextLimiter {
+    /**
+     * Limits the context by keeping only the most recent messages
+     * while preserving important messages (first user message, system messages).
+     *
+     * @param messages - All messages from the chat (assumed to be in chronological order)
+     * @param options - Limiting options
+     * @returns Limited array of messages in original order
+     */
+    static limitContext(messages, options = {}) {
+        const { maxMessages = 5, keepFirstUserMessage = true, keepSystemMessages = true, } = options;
+        // If we have fewer messages than the limit, return all
+        if (messages.length <= maxMessages) {
+            return messages;
+        }
+        // Separate system messages from the rest
+        const systemMessages = [];
+        const nonSystemMessages = [];
+        messages.forEach((msg) => {
+            if (msg.role === ai_agent_types_1.MessageRole.System && keepSystemMessages) {
+                systemMessages.push(msg);
+            }
+            else {
+                nonSystemMessages.push(msg);
+            }
+        });
+        // Find the first user message (if we need to keep it)
+        const firstUserMessageIndex = keepFirstUserMessage
+            ? nonSystemMessages.findIndex((msg) => msg.role === ai_agent_types_1.MessageRole.User)
+            : -1;
+        const firstUserMessage = firstUserMessageIndex >= 0
+            ? nonSystemMessages[firstUserMessageIndex]
+            : undefined;
+        // Calculate how many recent messages we can keep
+        // Reserve slots for: system messages
+        let availableSlots = maxMessages - systemMessages.length;
+        const shouldKeepFirstUser = keepFirstUserMessage && firstUserMessage !== undefined;
+        // Get the most recent messages (from the end)
+        // We'll check if first user is in recent messages after slicing
+        const recentCount = Math.max(0, availableSlots);
+        const recentStartIndex = Math.max(0, nonSystemMessages.length - recentCount);
+        const recentMessages = nonSystemMessages.slice(recentStartIndex);
+        // Check if first user message is already in recent messages
+        const firstUserInRecent = shouldKeepFirstUser && firstUserMessage
+            ? recentMessages.some((msg) => msg.id === firstUserMessage.id)
+            : false;
+        // If first user is not in recent and we need to keep it, we need to make room
+        if (shouldKeepFirstUser && firstUserMessage && !firstUserInRecent) {
+            // Remove the oldest recent message to make room for first user message
+            if (recentMessages.length > 0) {
+                recentMessages.shift();
+            }
+        }
+        // Build the set of messages to keep
+        const messagesToKeep = new Set();
+        // Add system messages
+        systemMessages.forEach((msg) => messagesToKeep.add(msg.id));
+        // Add first user message if needed
+        if (shouldKeepFirstUser && firstUserMessage) {
+            messagesToKeep.add(firstUserMessage.id);
+        }
+        // Add recent messages
+        recentMessages.forEach((msg) => messagesToKeep.add(msg.id));
+        // Reconstruct in original order
+        return messages.filter((msg) => messagesToKeep.has(msg.id));
+    }
+    /**
+     * Estimates token count for a message (rough approximation).
+     * For accurate token counting, use a proper tokenizer library.
+     *
+     * This is a simple approximation: ~4 characters per token for English text.
+     *
+     * @param message - Message to estimate tokens for
+     * @returns Estimated token count
+     */
+    static estimateTokens(message) {
+        var _a;
+        const content = typeof message.content === 'string'
+            ? message.content
+            : JSON.stringify(message.content);
+        // Rough approximation: 1 token ≈ 4 characters for English
+        // This is a simplified estimate - actual tokenization varies by model
+        const baseTokens = Math.ceil(content.length / 4);
+        // Add overhead for message structure (role, formatting, etc.)
+        const overhead = 10;
+        // Add tokens for tool calls if present
+        const toolCallTokens = ((_a = message.toolCalls) === null || _a === void 0 ? void 0 : _a.reduce((acc, toolCall) => {
+            const inputTokens = Math.ceil(JSON.stringify(toolCall.input).length / 4);
+            const outputTokens = toolCall.output
+                ? Math.ceil(JSON.stringify(toolCall.output).length / 4)
+                : 0;
+            return acc + inputTokens + outputTokens + 20; // 20 for tool call structure
+        }, 0)) || 0;
+        return baseTokens + overhead + toolCallTokens;
+    }
+    /**
+     * Limits context by token count (future enhancement).
+     * Currently not implemented - requires proper tokenizer.
+     *
+     * @param messages - All messages from the chat
+     * @param maxTokens - Maximum token count
+     * @returns Limited array of messages
+     */
+    static limitByTokens(messages, maxTokens) {
+        // This is a placeholder for token-based limiting
+        // To implement properly, you would:
+        // 1. Install a tokenizer (e.g., tiktoken for OpenAI, @anthropic-ai/tokenizer for Claude)
+        // 2. Count tokens for each message
+        // 3. Keep messages from the end until hitting the token limit
+        // 4. Always keep system messages and optionally first user message
+        let totalTokens = 0;
+        const limitedMessages = [];
+        // Keep system messages first
+        const systemMessages = messages.filter((msg) => msg.role === ai_agent_types_1.MessageRole.System);
+        systemMessages.forEach((msg) => {
+            const tokens = this.estimateTokens(msg);
+            if (totalTokens + tokens <= maxTokens) {
+                limitedMessages.push(msg);
+                totalTokens += tokens;
+            }
+        });
+        // Then keep messages from the end
+        const nonSystemMessages = messages
+            .filter((msg) => msg.role !== ai_agent_types_1.MessageRole.System)
+            .reverse();
+        for (const message of nonSystemMessages) {
+            const tokens = this.estimateTokens(message);
+            if (totalTokens + tokens <= maxTokens) {
+                limitedMessages.unshift(message); // Add to beginning to maintain order
+                totalTokens += tokens;
+            }
+            else {
+                break;
+            }
+        }
+        return limitedMessages;
+    }
+}
+exports.ContextLimiter = ContextLimiter;
+//# sourceMappingURL=ContextLimiter.js.map