@multiplayer-app/ai-agent-node 0.1.0-beta.3 → 0.1.0-beta.30

package/dist/cjs/processors/ChatProcessor.d.ts

@@ -1,14 +1,31 @@
 import type { AgentChatRepository, AgentMessageRepository } from '@multiplayer-app/ai-agent-db';
 import { AgentArtifact, AgentChat, AgentChatResponse, AgentAttachment, AgentMessage, MessageRole, SendMessagePayload, SortOrder, SendMessagePayloadWithContent, AdditionalContext } from '@multiplayer-app/ai-agent-types';
-import { ArtifactStore } from '../store/index.js';
-import { GenerateTextOptions } from '../services/index.js';
+import { ArtifactStore } from '../store';
+import { GenerateTextOptions } from '../services';
 import { PassThrough } from 'stream';
+import { Config } from '../config';
+import { S3Lib } from '../libs/s3';
+import { SocketService } from '../services/SocketService';
+import { AgentStore } from '../store/AgentStore';
 export declare class ChatProcessor {
+    private aiHelper;
     private chatRepository;
     private messageRepository;
     private artifactStore;
-    constructor(chatRepository: AgentChatRepository, messageRepository: AgentMessageRepository, artifactStore: ArtifactStore);
+    private config;
+    private socketService;
+    private agentStore;
+    constructor(params: {
+        chatRepository: AgentChatRepository;
+        messageRepository: AgentMessageRepository;
+        artifactStore: ArtifactStore;
+        s3Lib: S3Lib;
+        config: Config;
+        socketService: SocketService;
+        agentStore: AgentStore;
+    });
     private getTemporaryTitle;
+    private isTemporaryTitle;
     listChats(params?: {
         contextKey?: string;
         limit?: number;
@@ -58,9 +75,8 @@ export declare class ChatProcessor {
     }, excludeSocketId?: string, onAgentStart?: () => void): Promise<void>;
     listArtifacts(chatId: string): AgentArtifact[];
     private eventToChunk;
-    getMessageStream(chatId: string): PassThrough;
-    createMessageStream(payload: SendMessagePayload & {
-        context?: AdditionalContext;
-    }, excludeSocketId?: string): Promise<PassThrough>;
+    getMessageStream(chatId: string, signal?: AbortSignal): PassThrough;
+    createMessageStream(payload: any, excludeSocketId: any, signal: any): Promise<PassThrough>;
+    private runMessageStream;
 }
 //# sourceMappingURL=ChatProcessor.d.ts.map

package/dist/cjs/processors/ChatProcessor.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"ChatProcessor.d.ts","sourceRoot":"","sources":["../../../src/processors/ChatProcessor.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAC,mBAAmB,EAAE,sBAAsB,EAAC,MAAM,8BAA8B,CAAC;AAC9F,OAAO,EACL,aAAa,EACb,SAAS,EACT,iBAAiB,EACjB,eAAe,EACf,YAAY,EAIZ,WAAW,EACX,kBAAkB,EAClB,SAAS,EAIT,6BAA6B,EAC7B,iBAAiB,EAClB,MAAM,iCAAiC,CAAC;AACzC,OAAO,EAAE,aAAa,EAAE,MAAM,mBAAmB,CAAC;AAKlD,OAAO,EAAE,mBAAmB,EAAE,MAAM,sBAAsB,CAAC;AAG3D,OAAO,EAAE,WAAW,EAAY,MAAM,QAAQ,CAAC;AAO/C,qBAAa,aAAa;IAEtB,OAAO,CAAC,cAAc;IACtB,OAAO,CAAC,iBAAiB;IACzB,OAAO,CAAC,aAAa;gBAFb,cAAc,EAAE,mBAAmB,EACnC,iBAAiB,EAAE,sBAAsB,EACzC,aAAa,EAAE,aAAa;IAGtC,OAAO,CAAC,iBAAiB;IAInB,SAAS,CAAC,MAAM,CAAC,EAAE;QACvB,UAAU,CAAC,EAAE,MAAM,CAAC;QACpB,KAAK,CAAC,EAAE,MAAM,CAAC;QACf,MAAM,CAAC,EAAE,MAAM,CAAC;QAChB,SAAS,CAAC,EAAE,MAAM,CAAC;QACnB,SAAS,CAAC,EAAE,SAAS,CAAC;KACvB,GAAG,OAAO,CAAC,iBAAiB,EAAE,CAAC;IA0B1B,OAAO,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,iBAAiB,CAAC;IAYnD,UAAU,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAQzC,gBAAgB,CAAC,OAAO,EAAE,kBAAkB,EAAE,eAAe,CAAC,EAAE,MAAM;IAyBtE,YAAY,CAAC,OAAO,EAAE;QAC1B,MAAM,CAAC,EAAE,MAAM,CAAC;QAChB,OAAO,CAAC,EAAE,MAAM,CAAC;QACjB,UAAU,EAAE,MAAM,CAAC;QACnB,MAAM,CAAC,EAAE,MAAM,CAAC;KACjB,GAAG,OAAO,CAAC,MAAM,CAAC;IAkBb,aAAa,CAAC,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE,WAAW,EAAE,WAAW,EAAE,IAAI,CAAC,YAAY,EAAE,SAAS,GAAG,WAAW,CAAC,EAAE,WAAW,CAAC,EAAE,eAAe,EAAE,EAAE,eAAe,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,YAAY,CAAC;IAgB7L,UAAU,CAAC,OAAO,EAAE,kBAAkB,EAAE,eAAe,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,SAAS,CAAC;IAuB3F;;;OAGG;IACG,oBAAoB,CAAC,MAAM,EAAE;QACjC,MAAM,EAAE,MAAM,CAAC;QACf,UAAU,EAAE,MAAM,CAAC;QACnB,MAAM,EAAE,MAAM,CAAC;QACf,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;QAC9B,aAAa,CAAC,EAAE,MAAM,CAAC;QACvB,eAAe,CAAC,EAAE,MAAM,CAAC;KAC1B,GAAG,OAAO,CAAC;QAAE,EAAE,EAAE,IAAI,CAAA;KAAE,CAAC;YAiEX,iBAAiB;IAmJ/B,OAAO,CAAC,iBAAiB;YAIX,sBAAsB;IAyD9B,qBAAqB,CAAC,IAAI,EAAE,SAAS,EAAE,OAAO,EAAE,6BAA6B,GAAG;QAAE,OAAO,CAAC,EAAE,iBAAiB,CAAA;KAAE,EAAE,YAAY,EAAE,YAAY,EAAE,EAAE,eAAe,CAAC,EAAE,MAAM;;;;;;;IAsBvK,aAAa,CAAC,IAAI,EAAE,SAAS,EAAE,OAAO,EAAE,kBAAkB,GAAG;QAAE,OAAO,CAAC,EAAE,iBAAiB,CAAA;KAAE,EAAE,eAAe,CAAC,EAAE,MAAM,EAAE,YAAY,CAAC,EAAE,MAAM,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC;IAiDvK,aAAa,CAAC,MAAM,EAAE,MAAM,GAAG,aAAa,EAAE;IAI9C,OAAO,CAAC,YAAY;IA6BpB,gBAAgB,CAAC,MAAM,EAAE,MAAM,GAAG,WAAW;IAgFvC,mBAAmB,CACrB,OAAO,EAAE,kBAAkB,GAAG;QAAE,OAAO,CAAC,EAAE,iBAAiB,CAAA;KAAE,EAC7D,eAAe,CAAC,EAAE,MAAM,GACzB,OAAO,CAAC,WAAW,CAAC;CAyDxB"}
+{"version":3,"file":"ChatProcessor.d.ts","sourceRoot":"","sources":["../../../src/processors/ChatProcessor.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAC,mBAAmB,EAAE,sBAAsB,EAAC,MAAM,8BAA8B,CAAC;AAC9F,OAAO,EACL,aAAa,EACb,SAAS,EACT,iBAAiB,EACjB,eAAe,EACf,YAAY,EAIZ,WAAW,EACX,kBAAkB,EAClB,SAAS,EAIT,6BAA6B,EAC7B,iBAAiB,EAClB,MAAM,iCAAiC,CAAC;AACzC,OAAO,EAAE,aAAa,EAAE,MAAM,UAAU,CAAC;AAGzC,OAAO,EAAE,mBAAmB,EAAE,MAAM,aAAa,CAAC;AAElD,OAAO,EAAE,WAAW,EAAE,MAAM,QAAQ,CAAC;AAErC,OAAO,EAAE,MAAM,EAAE,MAAM,WAAW,CAAC;AACnC,OAAO,EAAE,KAAK,EAAE,MAAM,YAAY,CAAC;AACnC,OAAO,EAAE,aAAa,EAAE,MAAM,2BAA2B,CAAC;AAC1D,OAAO,EAAE,UAAU,EAAE,MAAM,qBAAqB,CAAC;AAGjD,qBAAa,aAAa;IACxB,OAAO,CAAC,QAAQ,CAAW;IAC3B,OAAO,CAAC,cAAc,CAAsB;IAC5C,OAAO,CAAC,iBAAiB,CAAyB;IAClD,OAAO,CAAC,aAAa,CAAgB;IACrC,OAAO,CAAC,MAAM,CAAS;IACvB,OAAO,CAAC,aAAa,CAAgB;IACrC,OAAO,CAAC,UAAU,CAAa;gBAEnB,MAAM,EAAE;QAClB,cAAc,EAAE,mBAAmB,CAAC;QACpC,iBAAiB,EAAE,sBAAsB,CAAC;QAC1C,aAAa,EAAE,aAAa,CAAC;QAC7B,KAAK,EAAE,KAAK,CAAC;QACb,MAAM,EAAE,MAAM,CAAC;QACf,aAAa,EAAE,aAAa,CAAC;QAC7B,UAAU,EAAE,UAAU,CAAC;KACxB;IAYD,OAAO,CAAC,iBAAiB;IAIzB,OAAO,CAAC,gBAAgB;IASlB,SAAS,CAAC,MAAM,CAAC,EAAE;QACvB,UAAU,CAAC,EAAE,MAAM,CAAC;QACpB,KAAK,CAAC,EAAE,MAAM,CAAC;QACf,MAAM,CAAC,EAAE,MAAM,CAAC;QAChB,SAAS,CAAC,EAAE,MAAM,CAAC;QACnB,SAAS,CAAC,EAAE,SAAS,CAAC;KACvB,GAAG,OAAO,CAAC,iBAAiB,EAAE,CAAC;IA0B1B,OAAO,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,iBAAiB,CAAC;IAYnD,UAAU,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAQzC,gBAAgB,CAAC,OAAO,EAAE,kBAAkB,EAAE,eAAe,CAAC,EAAE,MAAM;IAyBtE,YAAY,CAAC,OAAO,EAAE;QAC1B,MAAM,CAAC,EAAE,MAAM,CAAC;QAChB,OAAO,CAAC,EAAE,MAAM,CAAC;QACjB,UAAU,EAAE,MAAM,CAAC;QACnB,MAAM,CAAC,EAAE,MAAM,CAAC;KACjB,GAAG,OAAO,CAAC,MAAM,CAAC;IAkBb,aAAa,CAAC,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE,WAAW,EAAE,WAAW,EAAE,IAAI,CAAC,YAAY,EAAE,SAAS,GAAG,WAAW,CAAC,EAAE,WAAW,CAAC,EAAE,eAAe,EAAE,EAAE,eAAe,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,YAAY,CAAC;IAgB7L,UAAU,CAAC,OAAO,EAAE,kBAAkB,EAAE,eAAe,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,SAAS,CAAC;IAuB3F;;;OAGG;IACG,oBAAoB,CAAC,MAAM,EAAE;QACjC,MAAM,EAAE,MAAM,CAAC;QACf,UAAU,EAAE,MAAM,CAAC;QACnB,MAAM,EAAE,MAAM,CAAC;QACf,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;QAC9B,aAAa,CAAC,EAAE,MAAM,CAAC;QACvB,eAAe,CAAC,EAAE,MAAM,CAAC;KAC1B,GAAG,OAAO,CAAC;QAAE,EAAE,EAAE,IAAI,CAAA;KAAE,CAAC;YAiEX,iBAAiB;IAuI/B,OAAO,CAAC,iBAAiB;YAIX,sBAAsB;IAyD9B,qBAAqB,CAAC,IAAI,EAAE,SAAS,EAAE,OAAO,EAAE,6BAA6B,GAAG;QAAE,OAAO,CAAC,EAAE,iBAAiB,CAAA;KAAE,EAAE,YAAY,EAAE,YAAY,EAAE,EAAE,eAAe,CAAC,EAAE,MAAM;;;;;;;IAsBvK,aAAa,CAAC,IAAI,EAAE,SAAS,EAAE,OAAO,EAAE,kBAAkB,GAAG;QAAE,OAAO,CAAC,EAAE,iBAAiB,CAAA;KAAE,EAAE,eAAe,CAAC,EAAE,MAAM,EAAE,YAAY,CAAC,EAAE,MAAM,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC;IAiDvK,aAAa,CAAC,MAAM,EAAE,MAAM,GAAG,aAAa,EAAE;IAI9C,OAAO,CAAC,YAAY;IA6BpB,gBAAgB,CAAC,MAAM,EAAE,MAAM,EAAE,MAAM,CAAC,EAAG,WAAW,GAAG,WAAW;IA8E9D,mBAAmB,CAAC,OAAO,KAAA,EAAE,eAAe,KAAA,EAAE,MAAM,KAAA;YAU5C,gBAAgB;CAsB/B"}

package/dist/cjs/processors/ChatProcessor.test.cjs

@@ -0,0 +1,276 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const vitest_1 = require("vitest");
+const ChatProcessor_1 = require("./ChatProcessor.cjs");
+const ai_agent_types_1 = require("@multiplayer-app/ai-agent-types");
+const ConfigStore_1 = require("../store/ConfigStore.cjs");
+const config_1 = require("../config.cjs");
+const s3_1 = require("../libs/s3/index.cjs");
+(0, vitest_1.describe)('ChatProcessor.streamMessage', () => {
+    let chatProcessor;
+    let mockChatRepository;
+    let mockMessageRepository;
+    let mockArtifactStore;
+    let mockAgentStore;
+    let mockSocketService;
+    let configStore;
+    let capturedOptions;
+    (0, vitest_1.beforeEach)(() => {
+        vitest_1.vi.clearAllMocks();
+        capturedOptions = undefined;
+        // Mock repositories
+        mockChatRepository = {
+            findById: vitest_1.vi.fn(),
+            create: vitest_1.vi.fn(),
+            update: vitest_1.vi.fn(),
+            delete: vitest_1.vi.fn(),
+            findWithMessages: vitest_1.vi.fn(),
+        };
+        mockMessageRepository = {
+            findByChatId: vitest_1.vi.fn().mockResolvedValue([]),
+            findById: vitest_1.vi.fn(),
+            create: vitest_1.vi.fn(),
+            update: vitest_1.vi.fn(),
+        };
+        // Mock artifact store
+        mockArtifactStore = {
+            listArtifacts: vitest_1.vi.fn().mockReturnValue([]),
+            deleteArtifacts: vitest_1.vi.fn(),
+        };
+        // Mock agent store
+        mockAgentStore = {
+            registerAgentProcess: vitest_1.vi.fn().mockReturnValue({
+                signal: new AbortController().signal,
+            }),
+            shareAgentProcessEvent: vitest_1.vi.fn(),
+            addListener: vitest_1.vi.fn(),
+            removeListener: vitest_1.vi.fn(),
+        };
+        // Mock socket service
+        mockSocketService = {
+            emitMessageUpdate: vitest_1.vi.fn(),
+            emitChatUpdate: vitest_1.vi.fn(),
+        };
+        // Setup config store with agent config
+        configStore = ConfigStore_1.ConfigStore.getInstance();
+        configStore.clear();
+        const agentConfig = {
+            name: 'test-agent',
+            description: 'Test agent for property verification',
+            defaultModel: 'openai/gpt-4o',
+            systemPrompt: 'You are a helpful assistant',
+            temperature: 0.8,
+            maxOutputTokens: 3000,
+            topP: 0.95,
+            topK: 50,
+            presencePenalty: 0.6,
+            frequencyPenalty: 0.4,
+            stopSequences: ['\n\n', '---', 'END'],
+            seed: 42,
+            tools: [],
+            toolChoice: 'auto',
+        };
+        // Use addAgent to bypass ConfigHelper conversion
+        configStore.addAgent(['test-context'], agentConfig);
+        const config = config_1.ConfigProvider.getInstance().getConfig();
+        const s3Lib = new s3_1.S3Lib(config.s3);
+        // Create ChatProcessor
+        chatProcessor = new ChatProcessor_1.ChatProcessor({
+            chatRepository: mockChatRepository,
+            messageRepository: mockMessageRepository,
+            artifactStore: mockArtifactStore,
+            s3Lib,
+            config,
+            socketService: mockSocketService,
+            agentStore: mockAgentStore,
+        });
+    });
+    (0, vitest_1.it)('should pass all agent properties correctly to streamAssistantResponse', async () => {
+        const chat = {
+            id: 'test-chat-id',
+            contextKey: 'test-context',
+            type: ai_agent_types_1.ChatType.Chat,
+            status: ai_agent_types_1.AgentStatus.Processing,
+            createdAt: new Date().toISOString(),
+            updatedAt: new Date().toISOString(),
+        };
+        const payload = {
+            chatId: chat.id,
+            content: 'Hello, test message',
+            contextKey: 'test-context',
+        };
+        // Mock chat repository
+        mockChatRepository.findById.mockResolvedValue(chat);
+        // Mock message creation
+        const userMessage = {
+            id: 'user-msg-1',
+            chat: chat.id,
+            role: ai_agent_types_1.MessageRole.User,
+            content: payload.content,
+            createdAt: new Date().toISOString(),
+        };
+        const assistantMessage = {
+            id: 'assistant-msg-1',
+            chat: chat.id,
+            role: ai_agent_types_1.MessageRole.Assistant,
+            content: '',
+            createdAt: new Date().toISOString(),
+            agentName: 'test-agent',
+        };
+        mockMessageRepository.create
+            .mockResolvedValueOnce(userMessage)
+            .mockResolvedValueOnce(assistantMessage);
+        // Mock stream to prevent actual API calls
+        const mockStream = {
+            fullStream: (async function* () {
+                yield { type: 'finish', finishReason: 'stop', text: 'Test response' };
+            })(),
+        };
+        // Capture options passed to streamAssistantResponse
+        const aiHelper = chatProcessor.aiHelper;
+        vitest_1.vi.spyOn(aiHelper, 'streamAssistantResponse').mockImplementation(async (messages, signal, options) => {
+            capturedOptions = options;
+            return mockStream;
+        });
+        // Call streamMessage
+        await chatProcessor.streamMessage(chat, payload);
+        // Verify options were captured
+        (0, vitest_1.expect)(capturedOptions).toBeDefined();
+        // Verify all agent properties are present in the options
+        if (capturedOptions) {
+            (0, vitest_1.expect)(capturedOptions).toHaveProperty('name', 'test-agent');
+            (0, vitest_1.expect)(capturedOptions).toHaveProperty('system', 'You are a helpful assistant');
+            (0, vitest_1.expect)(capturedOptions).toHaveProperty('temperature', 0.8);
+            (0, vitest_1.expect)(capturedOptions).toHaveProperty('maxOutputTokens', 3000);
+            (0, vitest_1.expect)(capturedOptions).toHaveProperty('topP', 0.95);
+            (0, vitest_1.expect)(capturedOptions).toHaveProperty('topK', 50);
+            (0, vitest_1.expect)(capturedOptions).toHaveProperty('presencePenalty', 0.6);
+            (0, vitest_1.expect)(capturedOptions).toHaveProperty('frequencyPenalty', 0.4);
+            (0, vitest_1.expect)(capturedOptions).toHaveProperty('stopSequences', ['\n\n', '---', 'END']);
+            (0, vitest_1.expect)(capturedOptions).toHaveProperty('seed', 42);
+            (0, vitest_1.expect)(capturedOptions).toHaveProperty('toolChoice', 'auto');
+            (0, vitest_1.expect)(capturedOptions).toHaveProperty('tools');
+        }
+    });
+    (0, vitest_1.it)('should handle optional agent properties when not provided', async () => {
+        configStore.clear();
+        const minimalAgentConfig = {
+            name: 'minimal-agent',
+            description: 'Minimal agent config',
+            systemPrompt: 'You are helpful',
+            tools: [],
+        };
+        configStore.addAgent(['test-context'], minimalAgentConfig);
+        const chat = {
+            id: 'test-chat-id-2',
+            contextKey: 'test-context',
+            type: ai_agent_types_1.ChatType.Chat,
+            status: ai_agent_types_1.AgentStatus.Processing,
+            createdAt: new Date().toISOString(),
+            updatedAt: new Date().toISOString(),
+        };
+        const payload = {
+            chatId: chat.id,
+            content: 'Test',
+            contextKey: 'test-context',
+        };
+        mockChatRepository.findById.mockResolvedValue(chat);
+        const userMessage = {
+            id: 'user-msg-2',
+            chat: chat.id,
+            role: ai_agent_types_1.MessageRole.User,
+            content: payload.content,
+            createdAt: new Date().toISOString(),
+        };
+        const assistantMessage = {
+            id: 'assistant-msg-2',
+            chat: chat.id,
+            role: ai_agent_types_1.MessageRole.Assistant,
+            content: '',
+            createdAt: new Date().toISOString(),
+            agentName: 'minimal-agent',
+        };
+        mockMessageRepository.create
+            .mockResolvedValueOnce(userMessage)
+            .mockResolvedValueOnce(assistantMessage);
+        const mockStream = {
+            fullStream: (async function* () {
+                yield { type: 'finish', finishReason: 'stop', text: 'Response' };
+            })(),
+        };
+        const aiHelper = chatProcessor.aiHelper;
+        vitest_1.vi.spyOn(aiHelper, 'streamAssistantResponse').mockImplementation(async (messages, signal, options) => {
+            capturedOptions = options;
+            return mockStream;
+        });
+        await chatProcessor.streamMessage(chat, payload);
+        // Verify required properties are present
+        (0, vitest_1.expect)(capturedOptions).toBeDefined();
+        if (capturedOptions) {
+            (0, vitest_1.expect)(capturedOptions).toHaveProperty('name', 'minimal-agent');
+            (0, vitest_1.expect)(capturedOptions).toHaveProperty('system', 'You are helpful');
+            (0, vitest_1.expect)(capturedOptions).toHaveProperty('tools');
+        }
+    });
+    (0, vitest_1.it)('should pass stopWhen condition when provided', async () => {
+        configStore.clear();
+        const stopWhenFn = vitest_1.vi.fn(() => false);
+        const agentConfigWithStopWhen = {
+            name: 'agent-with-stopwhen',
+            description: 'Agent with stopWhen',
+            systemPrompt: 'You are helpful',
+            tools: [],
+            stopWhen: stopWhenFn,
+        };
+        configStore.addAgent(['test-context'], agentConfigWithStopWhen);
+        const chat = {
+            id: 'test-chat-id-3',
+            contextKey: 'test-context',
+            type: ai_agent_types_1.ChatType.Chat,
+            status: ai_agent_types_1.AgentStatus.Processing,
+            createdAt: new Date().toISOString(),
+            updatedAt: new Date().toISOString(),
+        };
+        const payload = {
+            chatId: chat.id,
+            content: 'Test',
+            contextKey: 'test-context',
+        };
+        mockChatRepository.findById.mockResolvedValue(chat);
+        const userMessage = {
+            id: 'user-msg-3',
+            chat: chat.id,
+            role: ai_agent_types_1.MessageRole.User,
+            content: payload.content,
+            createdAt: new Date().toISOString(),
+        };
+        const assistantMessage = {
+            id: 'assistant-msg-3',
+            chat: chat.id,
+            role: ai_agent_types_1.MessageRole.Assistant,
+            content: '',
+            createdAt: new Date().toISOString(),
+            agentName: 'agent-with-stopwhen',
+        };
+        mockMessageRepository.create
+            .mockResolvedValueOnce(userMessage)
+            .mockResolvedValueOnce(assistantMessage);
+        const mockStream = {
+            fullStream: (async function* () {
+                yield { type: 'finish', finishReason: 'stop', text: 'Response' };
+            })(),
+        };
+        const aiHelper = chatProcessor.aiHelper;
+        vitest_1.vi.spyOn(aiHelper, 'streamAssistantResponse').mockImplementation(async (messages, signal, options) => {
+            capturedOptions = options;
+            return mockStream;
+        });
+        await chatProcessor.streamMessage(chat, payload);
+        // Verify stopWhen is passed
+        (0, vitest_1.expect)(capturedOptions).toBeDefined();
+        if (capturedOptions) {
+            (0, vitest_1.expect)(capturedOptions).toHaveProperty('stopWhen', stopWhenFn);
+        }
+    });
+});
+//# sourceMappingURL=ChatProcessor.test.js.map

package/dist/cjs/processors/ChatProcessor.test.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"ChatProcessor.test.js","sourceRoot":"","sources":["../../../src/processors/ChatProcessor.test.ts"],"names":[],"mappings":";;AAAA,mCAA8D;AAC9D,mDAAgD;AAQhD,oEAAqF;AAGrF,sDAAmD;AACnD,sCAA2C;AAC3C,mCAAmC;AAMnC,IAAA,iBAAQ,EAAC,6BAA6B,EAAE,GAAG,EAAE;IAC3C,IAAI,aAA4B,CAAC;IACjC,IAAI,kBAAuC,CAAC;IAC5C,IAAI,qBAA6C,CAAC;IAClD,IAAI,iBAAgC,CAAC;IACrC,IAAI,cAA0B,CAAC;IAC/B,IAAI,iBAAgC,CAAC;IACrC,IAAI,WAAwB,CAAC;IAC7B,IAAI,eAAkE,CAAC;IAEvE,IAAA,mBAAU,EAAC,GAAG,EAAE;QACd,WAAE,CAAC,aAAa,EAAE,CAAC;QACnB,eAAe,GAAG,SAAS,CAAC;QAE5B,oBAAoB;QACpB,kBAAkB,GAAG;YACnB,QAAQ,EAAE,WAAE,CAAC,EAAE,EAAE;YACjB,MAAM,EAAE,WAAE,CAAC,EAAE,EAAE;YACf,MAAM,EAAE,WAAE,CAAC,EAAE,EAAE;YACf,MAAM,EAAE,WAAE,CAAC,EAAE,EAAE;YACf,gBAAgB,EAAE,WAAE,CAAC,EAAE,EAAE;SACnB,CAAC;QAET,qBAAqB,GAAG;YACtB,YAAY,EAAE,WAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,CAAC;YAC3C,QAAQ,EAAE,WAAE,CAAC,EAAE,EAAE;YACjB,MAAM,EAAE,WAAE,CAAC,EAAE,EAAE;YACf,MAAM,EAAE,WAAE,CAAC,EAAE,EAAE;SACT,CAAC;QAET,sBAAsB;QACtB,iBAAiB,GAAG;YAClB,aAAa,EAAE,WAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,EAAE,CAAC;YAC1C,eAAe,EAAE,WAAE,CAAC,EAAE,EAAE;SAClB,CAAC;QAET,mBAAmB;QACnB,cAAc,GAAG;YACf,oBAAoB,EAAE,WAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC;gBAC5C,MAAM,EAAE,IAAI,eAAe,EAAE,CAAC,MAAM;aACrC,CAAC;YACF,sBAAsB,EAAE,WAAE,CAAC,EAAE,EAAE;YAC/B,WAAW,EAAE,WAAE,CAAC,EAAE,EAAE;YACpB,cAAc,EAAE,WAAE,CAAC,EAAE,EAAE;SACjB,CAAC;QAET,sBAAsB;QACtB,iBAAiB,GAAG;YAClB,iBAAiB,EAAE,WAAE,CAAC,EAAE,EAAE;YAC1B,cAAc,EAAE,WAAE,CAAC,EAAE,EAAE;SACjB,CAAC;QAET,uCAAuC;QACvC,WAAW,GAAG,yBAAW,CAAC,WAAW,EAAE,CAAC;QACxC,WAAW,CAAC,KAAK,EAAE,CAAC;QAEpB,MAAM,WAAW,GAAgB;YAC/B,IAAI,EAAE,YAAY;YAClB,WAAW,EAAE,sCAAsC;YACnD,YAAY,EAAE,eAAe;YAC7B,YAAY,EAAE,6BAA6B;YAC3C,WAAW,EAAE,GAAG;YAChB,eAAe,EAAE,IAAI;YACrB,IAAI,EAAE,IAAI;YACV,IAAI,EAAE,EAAE;YACR,eAAe,EAAE,GAAG;YACpB,gBAAgB,EAAE,GAAG;YACrB,aAAa,EAAE,CAAC,MAAM,EAAE,KAAK,EAAE,KAAK,CAAC;YACrC,IAAI,EAAE,EAAE;YACR,KAAK,EAAE,EAAE;YACT,UAAU,EAAE,MAAM;SACnB,CAAC;QAEF,iDAAiD;QACjD,WAAW,CAAC,QAAQ,CAAC,CAAC,cAAc,CAAC,EAAE,WAAW,CAAC,CAAC;QAEpD,MAAM,MAAM,GAAG,uBAAc,CAAC,WAAW,EAAE,CAAC,SAAS,EAAE,CAAC;QACxD,MAAM,KAAK,GAAG,IAAI,UAAK,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;QAEnC,uBAAuB;QACvB,aAAa,GAAG,IAAI,6BAAa,CAAC;YAChC,cAAc,EAAE,kBAAkB;YAClC,iBAAiB,EAAE,qBAAqB;YACxC,aAAa,EAAE,iBAAiB;YAChC,KAAK;YACL,MAAM;YACN,aAAa,EAAE,iBAAiB;YAChC,UAAU,EAAE,cAAc;SAC3B,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,IAAA,WAAE,EAAC,uEAAuE,EAAE,KAAK,IAAI,EAAE;QACrF,MAAM,IAAI,GAAc;YACtB,EAAE,EAAE,cAAc;YAClB,UAAU,EAAE,cAAc;YAC1B,IAAI,EAAE,yBAAQ,CAAC,IAAI;YACnB,MAAM,EAAE,4BAAW,CAAC,UAAU;YAC9B,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;YACnC,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;SACpC,CAAC;QAEF,MAAM,OAAO,GAAkC;YAC7C,MAAM,EAAE,IAAI,CAAC,EAAE;YACf,OAAO,EAAE,qBAAqB;YAC9B,UAAU,EAAE,cAAc;SAC3B,CAAC;QAEF,uBAAuB;QACtB,kBAAkB,CAAC,QAAgB,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC;QAE7D,wBAAwB;QACxB,MAAM,WAAW,GAAiB;YAChC,EAAE,EAAE,YAAY;YAChB,IAAI,EAAE,IAAI,CAAC,EAAE;YACb,IAAI,EAAE,4BAAW,CAAC,IAAI;YACtB,OAAO,EAAE,OAAO,CAAC,OAAO;YACxB,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;SACpC,CAAC;QAEF,MAAM,gBAAgB,GAAiB;YACrC,EAAE,EAAE,iBAAiB;YACrB,IAAI,EAAE,IAAI,CAAC,EAAE;YACb,IAAI,EAAE,4BAAW,CAAC,SAAS;YAC3B,OAAO,EAAE,EAAE;YACX,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;YACnC,SAAS,EAAE,YAAY;SACxB,CAAC;QAED,qBAAqB,CAAC,MAAc;aAClC,qBAAqB,CAAC,WAAW,CAAC;aAClC,qBAAqB,CAAC,gBAAgB,CAAC,CAAC;QAE3C,0CAA0C;QAC1C,MAAM,UAAU,GAAG;YACjB,UAAU,EAAE,CAAC,KAAK,SAAS,CAAC;gBAC1B,MAAM,EAAE,IAAI,EAAE,QAAQ,EAAE,YAAY,EAAE,MAAM,EAAE,IAAI,EAAE,eAAe,EAAE,CAAC;YACxE,CAAC,CAAC,EAAE;SACL,CAAC;QAEF,oDAAoD;QACpD,MAAM,QAAQ,GAAI,aAAqB,CAAC,QAAoB,CAAC;QAC7D,WAAE,CAAC,KAAK,CAAC,QAAQ,EAAE,yBAAyB,CAAC,CAAC,kBAAkB,CAC9D,KAAK,EAAE,QAAQ,EAAE,MAAM,EAAE,OAAO,EAAE,EAAE;YAClC,eAAe,GAAG,OAAO,CAAC;YAC1B,OAAO,UAAiB,CAAC;QAC3B,CAAC,CACF,CAAC;QAEF,qBAAqB;QACrB,MAAM,aAAa,CAAC,aAAa,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;QAEjD,+BAA+B;QAC/B,IAAA,eAAM,EAAC,eAAe,CAAC,CAAC,WAAW,EAAE,CAAC;QAEtC,yDAAyD;QACzD,IAAI,eAAe,EAAE,CAAC;YACpB,IAAA,eAAM,EAAC,eAAe,CAAC,CAAC,cAAc,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC;YAC7D,IAAA,eAAM,EAAC,eAAe,CAAC,CAAC,cAAc,CAAC,QAAQ,EAAE,6BAA6B,CAAC,CAAC;YAChF,IAAA,eAAM,EAAC,eAAe,CAAC,CAAC,cAAc,CAAC,aAAa,EAAE,GAAG,CAAC,CAAC;YAC3D,IAAA,eAAM,EAAC,eAAe,CAAC,CAAC,cAAc,CAAC,iBAAiB,EAAE,IAAI,CAAC,CAAC;YAChE,IAAA,eAAM,EAAC,eAAe,CAAC,CAAC,cAAc,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;YACrD,IAAA,eAAM,EAAC,eAAe,CAAC,CAAC,cAAc,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;YACnD,IAAA,eAAM,EAAC,eAAe,CAAC,CAAC,cAAc,CAAC,iBAAiB,EAAE,GAAG,CAAC,CAAC;YAC/D,IAAA,eAAM,EAAC,eAAe,CAAC,CAAC,cAAc,CAAC,kBAAkB,EAAE,GAAG,CAAC,CAAC;YAChE,IAAA,eAAM,EAAC,eAAe,CAAC,CAAC,cAAc,CAAC,eAAe,EAAE,CAAC,MAAM,EAAE,KAAK,EAAE,KAAK,CAAC,CAAC,CAAC;YAChF,IAAA,eAAM,EAAC,eAAe,CAAC,CAAC,cAAc,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;YACnD,IAAA,eAAM,EAAC,eAAe,CAAC,CAAC,cAAc,CAAC,YAAY,EAAE,MAAM,CAAC,CAAC;YAC7D,IAAA,eAAM,EAAC,eAAe,CAAC,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC;QAClD,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,IAAA,WAAE,EAAC,2DAA2D,EAAE,KAAK,IAAI,EAAE;QACzE,WAAW,CAAC,KAAK,EAAE,CAAC;QACpB,MAAM,kBAAkB,GAAgB;YACtC,IAAI,EAAE,eAAe;YACrB,WAAW,EAAE,sBAAsB;YACnC,YAAY,EAAE,iBAAiB;YAC/B,KAAK,EAAE,EAAE;SACV,CAAC;QAEF,WAAW,CAAC,QAAQ,CAAC,CAAC,cAAc,CAAC,EAAE,kBAAkB,CAAC,CAAC;QAE3D,MAAM,IAAI,GAAc;YACtB,EAAE,EAAE,gBAAgB;YACpB,UAAU,EAAE,cAAc;YAC1B,IAAI,EAAE,yBAAQ,CAAC,IAAI;YACnB,MAAM,EAAE,4BAAW,CAAC,UAAU;YAC9B,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;YACnC,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;SACpC,CAAC;QAEF,MAAM,OAAO,GAAkC;YAC7C,MAAM,EAAE,IAAI,CAAC,EAAE;YACf,OAAO,EAAE,MAAM;YACf,UAAU,EAAE,cAAc;SAC3B,CAAC;QAED,kBAAkB,CAAC,QAAgB,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC;QAE7D,MAAM,WAAW,GAAiB;YAChC,EAAE,EAAE,YAAY;YAChB,IAAI,EAAE,IAAI,CAAC,EAAE;YACb,IAAI,EAAE,4BAAW,CAAC,IAAI;YACtB,OAAO,EAAE,OAAO,CAAC,OAAO;YACxB,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;SACpC,CAAC;QAEF,MAAM,gBAAgB,GAAiB;YACrC,EAAE,EAAE,iBAAiB;YACrB,IAAI,EAAE,IAAI,CAAC,EAAE;YACb,IAAI,EAAE,4BAAW,CAAC,SAAS;YAC3B,OAAO,EAAE,EAAE;YACX,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;YACnC,SAAS,EAAE,eAAe;SAC3B,CAAC;QAED,qBAAqB,CAAC,MAAc;aAClC,qBAAqB,CAAC,WAAW,CAAC;aAClC,qBAAqB,CAAC,gBAAgB,CAAC,CAAC;QAE3C,MAAM,UAAU,GAAG;YACjB,UAAU,EAAE,CAAC,KAAK,SAAS,CAAC;gBAC1B,MAAM,EAAE,IAAI,EAAE,QAAQ,EAAE,YAAY,EAAE,MAAM,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC;YACnE,CAAC,CAAC,EAAE;SACL,CAAC;QAEF,MAAM,QAAQ,GAAI,aAAqB,CAAC,QAAoB,CAAC;QAC7D,WAAE,CAAC,KAAK,CAAC,QAAQ,EAAE,yBAAyB,CAAC,CAAC,kBAAkB,CAC9D,KAAK,EAAE,QAAQ,EAAE,MAAM,EAAE,OAAO,EAAE,EAAE;YAClC,eAAe,GAAG,OAAO,CAAC;YAC1B,OAAO,UAAiB,CAAC;QAC3B,CAAC,CACF,CAAC;QAEF,MAAM,aAAa,CAAC,aAAa,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;QAEjD,yCAAyC;QACzC,IAAA,eAAM,EAAC,eAAe,CAAC,CAAC,WAAW,EAAE,CAAC;QACtC,IAAI,eAAe,EAAE,CAAC;YACpB,IAAA,eAAM,EAAC,eAAe,CAAC,CAAC,cAAc,CAAC,MAAM,EAAE,eAAe,CAAC,CAAC;YAChE,IAAA,eAAM,EAAC,eAAe,CAAC,CAAC,cAAc,CAAC,QAAQ,EAAE,iBAAiB,CAAC,CAAC;YACpE,IAAA,eAAM,EAAC,eAAe,CAAC,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC;QAClD,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,IAAA,WAAE,EAAC,8CAA8C,EAAE,KAAK,IAAI,EAAE;QAC5D,WAAW,CAAC,KAAK,EAAE,CAAC;QACpB,MAAM,UAAU,GAAG,WAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,KAAK,CAAC,CAAC;QAEtC,MAAM,uBAAuB,GAAgB;YAC3C,IAAI,EAAE,qBAAqB;YAC3B,WAAW,EAAE,qBAAqB;YAClC,YAAY,EAAE,iBAAiB;YAC/B,KAAK,EAAE,EAAE;YACT,QAAQ,EAAE,UAAiB;SAC5B,CAAC;QAEF,WAAW,CAAC,QAAQ,CAAC,CAAC,cAAc,CAAC,EAAE,uBAAuB,CAAC,CAAC;QAEhE,MAAM,IAAI,GAAc;YACtB,EAAE,EAAE,gBAAgB;YACpB,UAAU,EAAE,cAAc;YAC1B,IAAI,EAAE,yBAAQ,CAAC,IAAI;YACnB,MAAM,EAAE,4BAAW,CAAC,UAAU;YAC9B,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;YACnC,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;SACpC,CAAC;QAEF,MAAM,OAAO,GAAkC;YAC7C,MAAM,EAAE,IAAI,CAAC,EAAE;YACf,OAAO,EAAE,MAAM;YACf,UAAU,EAAE,cAAc;SAC3B,CAAC;QAED,kBAAkB,CAAC,QAAgB,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC;QAE7D,MAAM,WAAW,GAAiB;YAChC,EAAE,EAAE,YAAY;YAChB,IAAI,EAAE,IAAI,CAAC,EAAE;YACb,IAAI,EAAE,4BAAW,CAAC,IAAI;YACtB,OAAO,EAAE,OAAO,CAAC,OAAO;YACxB,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;SACpC,CAAC;QAEF,MAAM,gBAAgB,GAAiB;YACrC,EAAE,EAAE,iBAAiB;YACrB,IAAI,EAAE,IAAI,CAAC,EAAE;YACb,IAAI,EAAE,4BAAW,CAAC,SAAS;YAC3B,OAAO,EAAE,EAAE;YACX,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;YACnC,SAAS,EAAE,qBAAqB;SACjC,CAAC;QAED,qBAAqB,CAAC,MAAc;aAClC,qBAAqB,CAAC,WAAW,CAAC;aAClC,qBAAqB,CAAC,gBAAgB,CAAC,CAAC;QAE3C,MAAM,UAAU,GAAG;YACjB,UAAU,EAAE,CAAC,KAAK,SAAS,CAAC;gBAC1B,MAAM,EAAE,IAAI,EAAE,QAAQ,EAAE,YAAY,EAAE,MAAM,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC;YACnE,CAAC,CAAC,EAAE;SACL,CAAC;QAEF,MAAM,QAAQ,GAAI,aAAqB,CAAC,QAAoB,CAAC;QAC7D,WAAE,CAAC,KAAK,CAAC,QAAQ,EAAE,yBAAyB,CAAC,CAAC,kBAAkB,CAC9D,KAAK,EAAE,QAAQ,EAAE,MAAM,EAAE,OAAO,EAAE,EAAE;YAClC,eAAe,GAAG,OAAO,CAAC;YAC1B,OAAO,UAAiB,CAAC;QAC3B,CAAC,CACF,CAAC;QAEF,MAAM,aAAa,CAAC,aAAa,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;QAEjD,4BAA4B;QAC5B,IAAA,eAAM,EAAC,eAAe,CAAC,CAAC,WAAW,EAAE,CAAC;QACtC,IAAI,eAAe,EAAE,CAAC;YACpB,IAAA,eAAM,EAAC,eAAe,CAAC,CAAC,cAAc,CAAC,UAAU,EAAE,UAAU,CAAC,CAAC;QACjE,CAAC;IACH,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/cjs/processors/ChatProcessor.test.d.ts

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

package/dist/cjs/processors/ChatProcessor.test.d.ts.map

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

package/dist/cjs/processors/{ModelsProcessor.js → ModelsProcessor.cjs}

@@ -1,7 +1,7 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.ModelsProcessor = void 0;
-const index_js_1 = require("../store/index.js");
+const store_1 = require("../store/index.cjs");
 const providerNames = {
     openai: 'OpenAI',
     anthropic: 'Anthropic',
@@ -10,7 +10,7 @@ const providerNames = {
 };
 class ModelsProcessor {
     async getModels() {
-        const modelStore = index_js_1.ModelStore.getInstance();
+        const modelStore = store_1.ModelStore.getInstance();
         if (!modelStore.hasModels()) {
             // Return empty if models haven't been loaded yet
             return {

package/dist/cjs/processors/{ModelsProcessor.js.map → ModelsProcessor.cjs.map}

@@ -1 +1 @@
-{"version":3,"file":"ModelsProcessor.js","sourceRoot":"","sources":["../../../src/processors/ModelsProcessor.ts"],"names":[],"mappings":";;;AAAA,gDAA+C;AAE/C,MAAM,aAAa,GAA2B;IAC1C,MAAM,EAAE,QAAQ;IAChB,SAAS,EAAE,WAAW;IACtB,MAAM,EAAE,QAAQ;IAChB,UAAU,EAAE,YAAY;CAC3B,CAAC;AACF,MAAa,eAAe;IAEzB,KAAK,CAAC,SAAS;QACd,MAAM,UAAU,GAAG,qBAAU,CAAC,WAAW,EAAE,CAAC;QAE5C,IAAI,CAAC,UAAU,CAAC,SAAS,EAAE,EAAE,CAAC;YAC5B,iDAAiD;YACjD,OAAO;gBACL,MAAM,EAAE,EAAE;gBACV,SAAS,EAAE,EAAE;aACd,CAAC;QACJ,CAAC;QAED,MAAM,MAAM,GAAG,UAAU,CAAC,SAAS,EAAE,CAAC;QACtC,MAAM,SAAS,GAAG,UAAU,CAAC,YAAY,EAAE,CAAC;QAE5C,OAAO;YACL,MAAM;YACN,SAAS,EAAE,SAAS,CAAC,GAAG,CAAC,CAAC,UAAU,EAAE,EAAE,CAAC,CAAC;gBACxC,EAAE,EAAE,UAAU;gBACd,IAAI,EAAE,aAAa,CAAC,UAAU,CAAC,IAAI,UAAU;gBAC7C,MAAM,EAAE,UAAU,CAAC,mBAAmB,CAAC,UAAU,CAAC;aACnD,CAAC,CAAC;SACJ,CAAC;IACJ,CAAC;CACF;AAzBD,0CAyBC"}
+{"version":3,"file":"ModelsProcessor.js","sourceRoot":"","sources":["../../../src/processors/ModelsProcessor.ts"],"names":[],"mappings":";;;AAAA,oCAAsC;AAEtC,MAAM,aAAa,GAA2B;IAC1C,MAAM,EAAE,QAAQ;IAChB,SAAS,EAAE,WAAW;IACtB,MAAM,EAAE,QAAQ;IAChB,UAAU,EAAE,YAAY;CAC3B,CAAC;AACF,MAAa,eAAe;IAEzB,KAAK,CAAC,SAAS;QACd,MAAM,UAAU,GAAG,kBAAU,CAAC,WAAW,EAAE,CAAC;QAE5C,IAAI,CAAC,UAAU,CAAC,SAAS,EAAE,EAAE,CAAC;YAC5B,iDAAiD;YACjD,OAAO;gBACL,MAAM,EAAE,EAAE;gBACV,SAAS,EAAE,EAAE;aACd,CAAC;QACJ,CAAC;QAED,MAAM,MAAM,GAAG,UAAU,CAAC,SAAS,EAAE,CAAC;QACtC,MAAM,SAAS,GAAG,UAAU,CAAC,YAAY,EAAE,CAAC;QAE5C,OAAO;YACL,MAAM;YACN,SAAS,EAAE,SAAS,CAAC,GAAG,CAAC,CAAC,UAAU,EAAE,EAAE,CAAC,CAAC;gBACxC,EAAE,EAAE,UAAU;gBACd,IAAI,EAAE,aAAa,CAAC,UAAU,CAAC,IAAI,UAAU;gBAC7C,MAAM,EAAE,UAAU,CAAC,mBAAmB,CAAC,UAAU,CAAC;aACnD,CAAC,CAAC;SACJ,CAAC;IACJ,CAAC;CACF;AAzBD,0CAyBC"}

package/dist/cjs/processors/{index.js → index.cjs}

@@ -14,6 +14,6 @@ var __exportStar = (this && this.__exportStar) || function(m, exports) {
     for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-__exportStar(require("./ChatProcessor.js"), exports);
-__exportStar(require("./ModelsProcessor.js"), exports);
+__exportStar(require("./ChatProcessor.cjs"), exports);
+__exportStar(require("./ModelsProcessor.cjs"), exports);
 //# sourceMappingURL=index.js.map

package/dist/cjs/processors/{index.js.map → index.cjs.map}

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/processors/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;AAAA,qDAAmC;AACnC,uDAAqC"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/processors/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;AAAA,kDAAgC;AAChC,oDAAkC"}

package/dist/cjs/processors/index.d.ts

@@ -1,3 +1,3 @@
-export * from './ChatProcessor.js';
-export * from './ModelsProcessor.js';
+export * from './ChatProcessor';
+export * from './ModelsProcessor';
 //# sourceMappingURL=index.d.ts.map

package/dist/cjs/processors/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/processors/index.ts"],"names":[],"mappings":"AAAA,cAAc,oBAAoB,CAAC;AACnC,cAAc,sBAAsB,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/processors/index.ts"],"names":[],"mappings":"AAAA,cAAc,iBAAiB,CAAC;AAChC,cAAc,mBAAmB,CAAC"}

package/dist/cjs/services/{AIService.js → AIService.cjs}

@@ -8,14 +8,14 @@ const zod_1 = require("zod");
 const ai_1 = require("ai");
 const ai_agent_types_1 = require("@multiplayer-app/ai-agent-types");
 const ai_sdk_provider_1 = require("@openrouter/ai-sdk-provider");
-const config_js_1 = require("../config.js");
-const index_js_1 = require("../libs/logger/index.js");
+const logger_1 = require("../libs/logger/index.cjs");
 class AIService {
-    constructor(modelStore) {
+    constructor(modelStore, config) {
         this.modelStore = modelStore;
+        this.config = config;
     }
     getLanguageModel(modelId, websearchEnabled) {
-        const selectedModel = modelId || config_js_1.config.ai.defaultModel || this.modelStore.getModels()[0].id;
+        const selectedModel = modelId || this.config.defaultModel || this.modelStore.getModels()[0].id;
         const model = selectedModel ?
             this.modelStore.getModelById(selectedModel) :
             this.modelStore.getModels()[0]; //get first model if no model is selected
@@ -31,21 +31,21 @@ class AIService {
         const provider = model.provider || 'openai'; // Default to openai if provider not specified
         switch (provider) {
             case 'openai':
-                return (0, openai_1.openai)(model.id);
+                return (0, openai_1.createOpenAI)({ apiKey: this.config.openaiApiKey })(model.id);
             case 'anthropic':
-                return (0, anthropic_1.anthropic)(model.id);
+                return (0, anthropic_1.createAnthropic)({ apiKey: this.config.anthropicApiKey })(model.id);
             case 'google':
-                return (0, google_1.google)(model.id);
+                return (0, google_1.createGoogleGenerativeAI)({ apiKey: this.config.googleApiKey })(model.id);
             case 'openrouter':
                 return (0, ai_sdk_provider_1.createOpenRouter)({
-                    apiKey: config_js_1.config.ai.openrouterApiKey,
+                    apiKey: this.config.openrouterApiKey,
                 })(model.id, { plugins: websearchEnabled ? [{ id: "web", engine: undefined, max_results: 2 }] : [] });
             default:
                 throw new Error(`Unsupported provider: ${provider}`);
         }
     }
     getWebSearchTool(modelId) {
-        const selectedModel = modelId || config_js_1.config.ai.defaultModel || this.modelStore.getModels()[0].id;
+        const selectedModel = modelId || this.config.defaultModel || this.modelStore.getModels()[0].id;
         const model = selectedModel ?
             this.modelStore.getModelById(selectedModel) :
             this.modelStore.getModels()[0]; //get first model if no model is selected
@@ -59,7 +59,7 @@ class AIService {
                 return anthropic_1.anthropic.tools.webSearch_20250305();
             case 'google':
                 return google_1.google.tools.googleSearch({});
-            case 'openrouter': // is set through plugins
+            case 'openrouter':
                 return undefined;
             default:
                 return undefined;
@@ -88,8 +88,7 @@ class AIService {
                         }
                         if (data.headersToPass) {
                             params.headers = data.headersToPass.reduce((acc, headerName) => {
-                                var _a, _b;
-                                acc[headerName] = (_b = (_a = context === null || context === void 0 ? void 0 : context.headers) === null || _a === void 0 ? void 0 : _a[headerName]) !== null && _b !== void 0 ? _b : '';
+                                acc[headerName] = context?.headers?.[headerName] ?? '';
                                 return acc;
                             }, {});
                         }
@@ -137,12 +136,15 @@ class AIService {
     //   });
     // }
     getAgentOptions(config, modelId, context) {
+        let websearchEnabled = false;
         const tools = config.tools.reduce((acc, toolConfig) => {
-            var _a;
             const tool = this.getTool(toolConfig, modelId, context);
-            if (tool && ((_a = toolConfig.data) === null || _a === void 0 ? void 0 : _a.title)) {
+            if (tool && toolConfig.data?.title) {
                 acc[toolConfig.data.title] = tool;
             }
+            if (toolConfig.type === ai_agent_types_1.AgentToolType.WEB_SEARCH) {
+                websearchEnabled = true;
+            }
             return acc;
         }, {});
         return {
@@ -151,44 +153,54 @@ class AIService {
             system: config.systemPrompt,
             toolChoice: config.toolChoice,
             name: config.name,
+            stopWhen: config.stopWhen,
+            maxOutputTokens: config.maxOutputTokens,
+            temperature: config.temperature,
+            topP: config.topP,
+            topK: config.topK,
+            presencePenalty: config.presencePenalty,
+            frequencyPenalty: config.frequencyPenalty,
+            stopSequences: config.stopSequences,
+            seed: config.seed,
+            websearchEnabled
         };
     }
     async generateText(options) {
-        var _a;
-        const websearchEnabled = !!((_a = options.tools) === null || _a === void 0 ? void 0 : _a[ai_agent_types_1.AgentToolType.WEB_SEARCH]);
         try {
+            const { stopWhen, model, websearchEnabled, ...restOptions } = options;
             const { text } = await (0, ai_1.generateText)({
-                ...options,
-                model: this.getLanguageModel(options.model, websearchEnabled),
+                ...restOptions,
+                model: this.getLanguageModel(model, !!websearchEnabled),
+                ...(stopWhen !== undefined ? { stopWhen } : {}),
             });
             return text;
         }
         catch (error) {
-            index_js_1.logger.error(error);
+            logger_1.logger.error(error);
             throw new Error(`Failed to generate text: ${error instanceof Error ? error.message : 'Unknown error'}`);
         }
     }
     streamText(options) {
-        var _a;
-        const { onError, onChunk, onFinish, onStepFinish } = options;
-        const websearchEnabled = !!((_a = options.tools) === null || _a === void 0 ? void 0 : _a[ai_agent_types_1.AgentToolType.WEB_SEARCH]);
+        const { onError, onChunk, onFinish, onStepFinish, onAbort, signal, websearchEnabled, model, ...restOptions } = options;
         try {
             return (0, ai_1.streamText)({
-                ...options,
-                model: this.getLanguageModel(options.model, websearchEnabled),
+                ...restOptions,
+                abortSignal: signal,
+                model: this.getLanguageModel(model, !!websearchEnabled),
                 onError({ error }) {
-                    index_js_1.logger.error(error);
-                    onError === null || onError === void 0 ? void 0 : onError(error);
+                    logger_1.logger.error(error);
+                    onError?.(error);
                 },
                 onChunk(chunk) {
-                    onChunk === null || onChunk === void 0 ? void 0 : onChunk(chunk);
+                    onChunk?.(chunk);
                 },
                 onFinish({ text, finishReason, usage, response, steps, totalUsage }) {
-                    onFinish === null || onFinish === void 0 ? void 0 : onFinish({ text, finishReason, usage, response, steps, totalUsage });
+                    onFinish?.({ text, finishReason, usage, response, steps, totalUsage });
                 },
                 onStepFinish(step) {
-                    onStepFinish === null || onStepFinish === void 0 ? void 0 : onStepFinish(step);
-                }
+                    onStepFinish?.(step);
+                },
+                ...(onAbort ? { onAbort } : {}),
             });
         }
         catch (error) {

package/dist/cjs/services/AIService.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"AIService.js","sourceRoot":"","sources":["../../../src/services/AIService.ts"],"names":[],"mappings":";;;AAAA,2CAAoD;AACpD,iDAA6D;AAC7D,2CAAgE;AAChE,6BAAsB;AACtB,2BAUY;AACZ,oEASyC;AAEzC,iEAA6D;AAC7D,2CAAsC;AAkCtC,MAAa,SAAS;IAGpB,YAAY,UAAsB,EAAE,MAAoB;QACtD,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC;QAC7B,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACvB,CAAC;IAED,gBAAgB,CAAC,OAA2B,EAAE,gBAAyB;QACrE,MAAM,aAAa,GAAG,OAAO,IAAI,IAAI,CAAC,MAAM,CAAC,YAAY,IAAI,IAAI,CAAC,UAAU,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;QAC/F,MAAM,KAAK,GAAG,aAAa,CAAC,CAAC;YAC3B,IAAI,CAAC,UAAU,CAAC,YAAY,CAAC,aAAa,CAAC,CAAC,CAAC;YAC7C,IAAI,CAAC,UAAU,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,yCAAyC;QAC3E,IAAI,CAAC,KAAK,EAAE,CAAC;YACX,MAAM,IAAI,KAAK,CAAC,kBAAkB,aAAa,mFAAmF,CAAC,CAAC;QACtI,CAAC;QACD,OAAO,IAAI,CAAC,mBAAmB,CAAC,KAAK,EAAE,gBAAgB,CAAC,CAAC;IAC3D,CAAC;IAED;;OAEG;IACK,mBAAmB,CAAC,KAAkB,EAAE,gBAAgB,GAAG,KAAK;QACtE,MAAM,QAAQ,GAAG,KAAK,CAAC,QAAQ,IAAI,QAAQ,CAAC,CAAC,8CAA8C;QAC3F,QAAQ,QAAQ,EAAE,CAAC;YACjB,KAAK,QAAQ;gBACX,OAAO,IAAA,qBAAY,EAAC,EAAC,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,YAAY,EAAC,CAAC,CAAC,KAAK,CAAC,EAAE,CAAC,CAAA;YACnE,KAAK,WAAW;gBACd,OAAO,IAAA,2BAAe,EAAC,EAAC,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,eAAe,EAAC,CAAC,CAAC,KAAK,CAAC,EAAE,CAAC,CAAA;YACzE,KAAK,QAAQ;gBACX,OAAO,IAAA,iCAAwB,EAAC,EAAC,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,YAAY,EAAC,CAAC,CAAC,KAAK,CAAC,EAAE,CAAC,CAAA;YAC/E,KAAK,YAAY;gBACf,OAAO,IAAA,kCAAgB,EAAC;oBACtB,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,gBAAgB;iBACrC,CAAC,CAAC,KAAK,CAAC,EAAE,EAAE,EAAE,OAAO,EAAE,gBAAgB,CAAC,CAAC,CAAC,CAAC,EAAE,EAAE,EAAE,KAAK,EAAE,MAAM,EAAE,SAAS,EAAE,WAAW,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,EAAE,CAA6B,CAAC;YACpI;gBACE,MAAM,IAAI,KAAK,CAAC,yBAAyB,QAAQ,EAAE,CAAC,CAAC;QACzD,CAAC;IACH,CAAC;IAEO,gBAAgB,CAAC,OAAgB;QACvC,MAAM,aAAa,GAAG,OAAO,IAAI,IAAI,CAAC,MAAM,CAAC,YAAY,IAAI,IAAI,CAAC,UAAU,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;QAC/F,MAAM,KAAK,GAAG,aAAa,CAAC,CAAC;YAC3B,IAAI,CAAC,UAAU,CAAC,YAAY,CAAC,aAAa,CAAC,CAAC,CAAC;YAC7C,IAAI,CAAC,UAAU,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,yCAAyC;QAC3E,IAAI,CAAC,KAAK,EAAE,CAAC;YACX,MAAM,IAAI,KAAK,CAAC,kBAAkB,aAAa,mFAAmF,CAAC,CAAC;QACtI,CAAC;QACD,QAAQ,KAAK,CAAC,QAAQ,EAAE,CAAC;YACvB,KAAK,QAAQ;gBACX,OAAO,eAAM,CAAC,KAAK,CAAC,SAAS,EAAU,CAAA;YACzC,KAAK,WAAW;gBACd,OAAO,qBAAS,CAAC,KAAK,CAAC,kBAAkB,EAAU,CAAA;YACrD,KAAK,QAAQ;gBACX,OAAO,eAAM,CAAC,KAAK,CAAC,YAAY,CAAC,EAAE,CAAS,CAAA;YAC9C,KAAK,YAAY;gBACf,OAAO,SAAS,CAAA;YAClB;gBACE,OAAO,SAAS,CAAC;QACrB,CAAC;IACH,CAAC;IAEO,OAAO,CAAC,UAAwC,EAAE,OAAgB,EAAE,OAA2B;QACrG,IAAI,UAAU,CAAC,IAAI,KAAK,8BAAa,CAAC,QAAQ,EAAE,CAAC;YAC/C,MAAM,IAAI,GAAG,UAAU,CAAC,IAAoB,CAAC;YAC7C,OAAO,IAAA,SAAI,EAAuE;gBAChF,KAAK,EAAE,IAAI,CAAC,KAAK;gBACjB,WAAW,EAAE,IAAI,CAAC,WAAW;gBAC7B,aAAa,EAAE,IAAI,CAAC,aAAa;gBACjC,WAAW,EAAE,OAAC,CAAC,MAAM,CAAC;oBACpB,IAAI,EAAE,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,OAAC,CAAC,GAAG,EAAE,CAAC,QAAQ,EAAE;oBAChD,WAAW,EAAE,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,OAAC,CAAC,GAAG,EAAE,CAAC,QAAQ,EAAE;iBACtE,CAAC;gBACF,OAAO,EAAE,KAAK,EAAE,KAAsE,EAAE,EAAE;oBACxF,IAAI,CAAC;wBACH,IAAI,GAAG,GAAG,IAAI,CAAC,GAAG,CAAC;wBACnB,MAAM,MAAM,GAAgB,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,CAAA;wBACnD,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;4BACd,MAAM,CAAC,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;wBAC3C,CAAC;wBACD,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;4BACrB,GAAG,GAAG,GAAG,GAAG,GAAG,GAAG,IAAI,eAAe,CAAC,KAAK,CAAC,WAAqC,CAAC,CAAC,QAAQ,EAAE,CAAC;wBAChG,CAAC;wBACD,IAAI,IAAI,CAAC,aAAa,EAAE,CAAC;4BACvB,MAAM,CAAC,OAAO,GAAG,IAAI,CAAC,aAAa,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,UAAU,EAAE,EAAE;gCAC7D,GAAG,CAAC,UAAU,CAAC,GAAG,OAAO,EAAE,OAAO,EAAE,CAAC,UAAU,CAAC,IAAI,EAAE,CAAC;gCACvD,OAAO,GAAG,CAAC;4BACb,CAAC,EAAE,EAAE,CAAC,CAAA;wBACR,CAAC;wBAEC,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;wBAC1C,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;wBACrC,OAAO,MAAM,CAAC;oBAChB,CAAC;oBAAC,OAAO,KAAK,EAAE,CAAC;wBACf,OAAO,KAAK,CAAC;oBACf,CAAC;gBACH,CAAC;aACJ,CAAC,CAAA;QACJ,CAAC;QAED,IAAI,UAAU,CAAC,IAAI,KAAK,8BAAa,CAAC,UAAU,EAAE,CAAC;YACjD,OAAO,IAAI,CAAC,gBAAgB,CAAC,OAAO,CAAC,CAAC;QACxC,CAAC;QACD,IAAI,UAAU,CAAC,IAAI,KAAK,8BAAa,CAAC,cAAc,EAAE,CAAC;YACrD,MAAM,IAAI,GAAG,UAAU,CAAC,IAAoC,CAAC;YAC7D,iFAAiF;YACjF,IAAI,IAAI,IAAI,OAAO,IAAI,CAAC,OAAO,KAAK,UAAU,EAAE,CAAC;gBAE/C,OAAO,IAAA,SAAI,EAAC;oBACV,GAAG,IAAI;oBACP,OAAO,EAAE,KAAK,EAAE,KAAmB,EAAE,OAA8B,EAAE,EAAE;wBACrE,OAAO,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,OAAO,EAAE,OAAO,CAAC,CAAA;oBAC9C,CAAC;iBACF,CAAC,CAAC;YACL,CAAC;YACD,OAAO,IAAA,SAAI,EAAC,IAAY,CAAC,CAAC;QAC5B,CAAC;QACD,OAAO,SAAS,CAAC;IACnB,CAAC;IAGD,0FAA0F;IAC1F,sEAAsE;IACtE,sDAAsD;IACtD,2CAA2C;IAC3C,yCAAyC;IACzC,QAAQ;IACR,kBAAkB;IAClB,uBAAuB;IAEvB,uBAAuB;IACvB,oEAAoE;IACpE,mCAAmC;IACnC,aAAa;IACb,+FAA+F;IAC/F,QAAQ;IACR,IAAI;IAEJ,eAAe,CAAC,MAAsC,EAAE,OAAgB,EAAE,OAA2B;QACnG,IAAI,gBAAgB,GAAG,KAAK,CAAC;QAC7B,MAAM,KAAK,GAAY,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,UAAU,EAAE,EAAE;YAC7D,MAAM,IAAI,GAAG,IAAI,CAAC,OAAO,CAAC,UAAU,EAAE,OAAO,EAAE,OAAO,CAAC,CAAC;YACxD,IAAI,IAAI,IAAI,UAAU,CAAC,IAAI,EAAE,KAAK,EAAE,CAAC;gBACnC,GAAG,CAAC,UAAU,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,IAAI,CAAA;YACnC,CAAC;YACD,IAAI,UAAU,CAAC,IAAI,KAAK,8BAAa,CAAC,UAAU,EAAE,CAAC;gBACjD,gBAAgB,GAAG,IAAI,CAAA;YACzB,CAAC;YACD,OAAO,GAAG,CAAC;QACb,CAAC,EAAE,EAAa,CAAC,CAAC;QAClB,OAAO;YACL,KAAK,EAAE,OAAO,IAAI,MAAM,CAAC,YAAY;YACrC,KAAK;YACL,MAAM,EAAE,MAAM,CAAC,YAAY;YAC3B,UAAU,EAAE,MAAM,CAAC,UAAU;YAC7B,IAAI,EAAE,MAAM,CAAC,IAAI;YACjB,QAAQ,EAAE,MAAM,CAAC,QAAQ;YACzB,eAAe,EAAE,MAAM,CAAC,eAAe;YACvC,WAAW,EAAE,MAAM,CAAC,WAAW;YAC/B,IAAI,EAAE,MAAM,CAAC,IAAI;YACjB,IAAI,EAAE,MAAM,CAAC,IAAI;YACjB,eAAe,EAAE,MAAM,CAAC,eAAe;YACvC,gBAAgB,EAAE,MAAM,CAAC,gBAAgB;YACzC,aAAa,EAAE,MAAM,CAAC,aAAa;YACnC,IAAI,EAAE,MAAM,CAAC,IAAI;YACjB,gBAAgB;SACjB,CAAC;IACJ,CAAC;IAGD,KAAK,CAAC,YAAY,CAAC,OAA4B;QAC7C,IAAI,CAAC;YACH,MAAM,EAAE,QAAQ,EAAE,KAAK,EAAE,gBAAgB,EAAE,GAAG,WAAW,EAAE,GAAG,OAAO,CAAC;YACtE,MAAM,EAAE,IAAI,EAAE,GAAG,MAAM,IAAA,iBAAY,EAAC;gBAClC,GAAG,WAAW;gBACd,KAAK,EAAE,IAAI,CAAC,gBAAgB,CAAC,KAAK,EAAE,CAAC,CAAC,gBAAgB,CAAC;gBACvD,GAAG,CAAC,QAAQ,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;aAChD,CAAC,CAAC;YAEH,OAAO,IAAI,CAAC;QACd,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,eAAM,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;YACpB,MAAM,IAAI,KAAK,CAAC,4BAA4B,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,eAAe,EAAE,CAAC,CAAC;QAC1G,CAAC;IACH,CAAC;IAED,UAAU,CAAC,OAA0B;QACnC,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,QAAQ,EAAE,YAAY,EAAE,OAAO,EAAE,MAAM,EAAE,gBAAgB,EAAE,KAAK,EAAE,GAAG,WAAW,EAAE,GAAG,OAAO,CAAC;QAEvH,IAAI,CAAC;YACH,OAAO,IAAA,eAAU,EAAC;gBAChB,GAAG,WAAW;gBACd,WAAW,EAAE,MAAM;gBACnB,KAAK,EAAE,IAAI,CAAC,gBAAgB,CAAC,KAAK,EAAE,CAAC,CAAC,gBAAgB,CAAC;gBACvD,OAAO,CAAC,EAAE,KAAK,EAAE;oBACf,eAAM,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;oBACpB,OAAO,EAAE,CAAC,KAAc,CAAC,CAAC;gBAC5B,CAAC;gBACD,OAAO,CAAC,KAAK;oBACX,OAAO,EAAE,CAAC,KAAK,CAAC,CAAC;gBACnB,CAAC;gBACD,QAAQ,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,KAAK,EAAE,QAAQ,EAAE,KAAK,EAAE,UAAU,EAAE;oBACjE,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,KAAK,EAAE,QAAQ,EAAE,KAAK,EAAE,UAAU,EAAE,CAAC,CAAC;gBACzE,CAAC;gBACD,YAAY,CAAC,IAAI;oBACf,YAAY,EAAE,CAAC,IAAI,CAAC,CAAC;gBACvB,CAAC;gBACD,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;aAChC,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,IAAI,KAAK,CAAC,0BAA0B,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,eAAe,EAAE,CAAC,CAAC;QACxG,CAAC;IACH,CAAC;CACF;AArND,8BAqNC"}

package/dist/cjs/services/AIService.d.ts

@@ -1,16 +1,24 @@
-import { streamText, type LanguageModel, ModelMessage, ToolSet, Tool } from 'ai';
-import type { AgentConfig, ToolChoice, AdditionalContext } from '@multiplayer-app/ai-agent-types';
-import { ModelStore } from '../store/ModelStore.js';
+import { type LanguageModel, ModelMessage, streamText, Tool, ToolSet, type StopCondition } from 'ai';
+import { AdditionalContext, AgentConfig, ToolChoice } from '@multiplayer-app/ai-agent-types';
+import { ModelStore } from '../store/ModelStore';
+import { Config } from '../config';
 export interface GenerateTextOptions {
     messages: ModelMessage[];
     system?: string;
     temperature?: number;
     maxOutputTokens?: number;
     topP?: number;
+    topK?: number;
+    presencePenalty?: number;
+    frequencyPenalty?: number;
+    stopSequences?: string[];
+    seed?: number;
     toolChoice?: ToolChoice<ToolSet>;
     tools?: ToolSet;
     model?: string;
     signal?: AbortSignal;
+    websearchEnabled?: boolean;
+    stopWhen?: StopCondition<ToolSet> | Array<StopCondition<ToolSet>>;
 }
 export interface StreamTextOptions extends GenerateTextOptions {
     onChunk?: (chunk: any) => void;
@@ -31,7 +39,8 @@ export interface LocalFunctionTool extends Omit<Tool, 'title'> {
 }
 export declare class AIService {
     private modelStore;
-    constructor(modelStore: ModelStore);
+    private config;
+    constructor(modelStore: ModelStore, config: Config["ai"]);
     getLanguageModel(modelId: string | undefined, websearchEnabled: boolean): LanguageModel;
     /**
      * Create a language model instance based on the provider

package/dist/cjs/services/AIService.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"AIService.d.ts","sourceRoot":"","sources":["../../../src/services/AIService.ts"],"names":[],"mappings":"AAIA,OAAO,EAEL,UAAU,EACV,KAAK,aAAa,EAClB,YAAY,EACZ,OAAO,EAEP,IAAI,EAEL,MAAM,IAAI,CAAC;AACZ,OAAO,KAAK,EAAwC,WAAW,EAAE,UAAU,EAA0B,iBAAiB,EAAC,MAAM,iCAAiC,CAAC;AAE/J,OAAO,EAAE,UAAU,EAAE,MAAM,wBAAwB,CAAC;AAIpD,MAAM,WAAW,mBAAmB;IAClC,QAAQ,EAAE,YAAY,EAAE,CAAC;IACzB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,UAAU,CAAC,EAAE,UAAU,CAAC,OAAO,CAAC,CAAC;IACjC,KAAK,CAAC,EAAE,OAAO,CAAC;IAChB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,MAAM,CAAC,EAAE,WAAW,CAAC;CACtB;AAED,MAAM,WAAW,iBAAkB,SAAQ,mBAAmB;IAC5D,OAAO,CAAC,EAAE,CAAC,KAAK,EAAE,GAAG,KAAK,IAAI,CAAC;IAC/B,OAAO,CAAC,EAAE,CAAC,KAAK,EAAE,KAAK,KAAK,IAAI,CAAC;IACjC,QAAQ,CAAC,EAAE,CAAC,IAAI,EAAE;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,YAAY,EAAE,MAAM,CAAC;QAAC,KAAK,EAAE,GAAG,CAAC;QAAC,QAAQ,EAAE,GAAG,CAAC;QAAC,KAAK,EAAE,GAAG,CAAC;QAAC,UAAU,EAAE,GAAG,CAAA;KAAE,KAAK,IAAI,CAAC;IAC1H,OAAO,CAAC,EAAE,CAAC,IAAI,EAAE,GAAG,KAAK,IAAI,CAAC;IAC9B,YAAY,CAAC,EAAE,CAAC,IAAI,EAAE,GAAG,KAAK,IAAI,CAAC;CACpC;AAED,MAAM,WAAW,iBAAkB,SAAQ,IAAI,CAAC,IAAI,EAAE,OAAO,CAAC;IAC5D,KAAK,EAAE,MAAM,CAAC;CACf;AAED,qBAAa,SAAS;IACpB,OAAO,CAAC,UAAU,CAAa;gBAEnB,UAAU,EAAE,UAAU;IAIlC,gBAAgB,CAAC,OAAO,EAAE,MAAM,GAAG,SAAS,EAAE,gBAAgB,EAAE,OAAO,GAAG,aAAa;IAWvF;;OAEG;IACH,OAAO,CAAC,mBAAmB;IAkB3B,OAAO,CAAC,gBAAgB;IAsBxB,OAAO,CAAC,OAAO;IA4Ef,eAAe,CAAC,MAAM,EAAE,WAAW,CAAC,iBAAiB,CAAC,EAAE,OAAO,CAAC,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,iBAAiB,GAAG,IAAI,CAAC,mBAAmB,EAAE,UAAU,CAAC,GAAG;QAAC,IAAI,EAAE,MAAM,CAAA;KAAC;IAkBxJ,YAAY,CAAC,OAAO,EAAE,mBAAmB,GAAG,OAAO,CAAC,MAAM,CAAC;IAejE,UAAU,CAAC,OAAO,EAAE,iBAAiB,GAAG,UAAU,CAAC,OAAO,UAAU,CAAC;CA0BtE"}
+{"version":3,"file":"AIService.d.ts","sourceRoot":"","sources":["../../../src/services/AIService.ts"],"names":[],"mappings":"AAIA,OAAO,EAEL,KAAK,aAAa,EAClB,YAAY,EACZ,UAAU,EAEV,IAAI,EAEJ,OAAO,EACP,KAAK,aAAa,EACnB,MAAM,IAAI,CAAC;AACZ,OAAO,EACL,iBAAiB,EAEjB,WAAW,EAKX,UAAU,EACX,MAAM,iCAAiC,CAAC;AACzC,OAAO,EAAC,UAAU,EAAC,MAAM,qBAAqB,CAAC;AAG/C,OAAO,EAAC,MAAM,EAAC,MAAM,WAAW,CAAC;AAEjC,MAAM,WAAW,mBAAmB;IAClC,QAAQ,EAAE,YAAY,EAAE,CAAC;IACzB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,aAAa,CAAC,EAAE,MAAM,EAAE,CAAC;IACzB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,UAAU,CAAC,EAAE,UAAU,CAAC,OAAO,CAAC,CAAC;IACjC,KAAK,CAAC,EAAE,OAAO,CAAC;IAChB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,MAAM,CAAC,EAAE,WAAW,CAAC;IACrB,gBAAgB,CAAC,EAAE,OAAO,CAAC;IAC3B,QAAQ,CAAC,EAAE,aAAa,CAAC,OAAO,CAAC,GAAG,KAAK,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC,CAAC;CACnE;AAED,MAAM,WAAW,iBAAkB,SAAQ,mBAAmB;IAC5D,OAAO,CAAC,EAAE,CAAC,KAAK,EAAE,GAAG,KAAK,IAAI,CAAC;IAC/B,OAAO,CAAC,EAAE,CAAC,KAAK,EAAE,KAAK,KAAK,IAAI,CAAC;IACjC,QAAQ,CAAC,EAAE,CAAC,IAAI,EAAE;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,YAAY,EAAE,MAAM,CAAC;QAAC,KAAK,EAAE,GAAG,CAAC;QAAC,QAAQ,EAAE,GAAG,CAAC;QAAC,KAAK,EAAE,GAAG,CAAC;QAAC,UAAU,EAAE,GAAG,CAAA;KAAE,KAAK,IAAI,CAAC;IAC1H,OAAO,CAAC,EAAE,CAAC,IAAI,EAAE,GAAG,KAAK,IAAI,CAAC;IAC9B,YAAY,CAAC,EAAE,CAAC,IAAI,EAAE,GAAG,KAAK,IAAI,CAAC;CACpC;AAED,MAAM,WAAW,iBAAkB,SAAQ,IAAI,CAAC,IAAI,EAAE,OAAO,CAAC;IAC5D,KAAK,EAAE,MAAM,CAAC;CACf;AAED,qBAAa,SAAS;IACpB,OAAO,CAAC,UAAU,CAAa;IAC/B,OAAO,CAAC,MAAM,CAAe;gBACjB,UAAU,EAAE,UAAU,EAAE,MAAM,EAAE,MAAM,CAAC,IAAI,CAAC;IAKxD,gBAAgB,CAAC,OAAO,EAAE,MAAM,GAAG,SAAS,EAAE,gBAAgB,EAAE,OAAO,GAAG,aAAa;IAWvF;;OAEG;IACH,OAAO,CAAC,mBAAmB;IAkB3B,OAAO,CAAC,gBAAgB;IAsBxB,OAAO,CAAC,OAAO;IA4Ef,eAAe,CAAC,MAAM,EAAE,WAAW,CAAC,iBAAiB,CAAC,EAAE,OAAO,CAAC,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,iBAAiB,GAAG,IAAI,CAAC,mBAAmB,EAAE,UAAU,CAAC,GAAG;QAAC,IAAI,EAAE,MAAM,CAAA;KAAC;IAgCxJ,YAAY,CAAC,OAAO,EAAE,mBAAmB,GAAG,OAAO,CAAC,MAAM,CAAC;IAgBjE,UAAU,CAAC,OAAO,EAAE,iBAAiB,GAAG,UAAU,CAAC,OAAO,UAAU,CAAC;CA2BtE"}