@multiplayer-app/ai-agent-node 0.1.0-beta.3 → 0.1.0-beta.30
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/dist/cjs/config.cjs +98 -0
- package/dist/cjs/config.cjs.map +1 -0
- package/dist/cjs/config.d.ts +56 -23
- package/dist/cjs/config.d.ts.map +1 -1
- package/dist/cjs/helpers/AIHelper.cjs +386 -0
- package/dist/cjs/helpers/AIHelper.cjs.map +1 -0
- package/dist/cjs/helpers/AIHelper.d.ts +20 -13
- package/dist/cjs/helpers/AIHelper.d.ts.map +1 -1
- package/dist/cjs/helpers/{AIHelper.test.js → AIHelper.test.cjs} +22 -15
- package/dist/cjs/helpers/AIHelper.test.cjs.map +1 -0
- package/dist/cjs/helpers/{ConfigHelper.js → ConfigHelper.cjs} +15 -7
- package/dist/cjs/helpers/ConfigHelper.cjs.map +1 -0
- package/dist/cjs/helpers/ConfigHelper.d.ts.map +1 -1
- package/dist/cjs/helpers/{ContextLimiter.js → ContextLimiter.cjs} +2 -3
- package/dist/cjs/helpers/{ContextLimiter.js.map → ContextLimiter.cjs.map} +1 -1
- package/dist/cjs/helpers/FileHelper.cjs +193 -0
- package/dist/cjs/helpers/FileHelper.cjs.map +1 -0
- package/dist/cjs/helpers/FileHelper.d.ts +19 -25
- package/dist/cjs/helpers/FileHelper.d.ts.map +1 -1
- package/dist/cjs/{store/index.js → helpers/index.cjs} +4 -4
- package/dist/cjs/helpers/{index.js.map → index.cjs.map} +1 -1
- package/dist/cjs/helpers/index.d.ts +4 -5
- package/dist/cjs/helpers/index.d.ts.map +1 -1
- package/dist/cjs/index.cjs +114 -0
- package/dist/cjs/index.cjs.map +1 -0
- package/dist/cjs/index.d.ts +39 -13
- package/dist/cjs/index.d.ts.map +1 -1
- package/dist/cjs/libs/{kafka/index.js → index.cjs} +2 -2
- package/dist/cjs/libs/{index.js.map → index.cjs.map} +1 -1
- package/dist/cjs/libs/index.d.ts +2 -3
- package/dist/cjs/libs/index.d.ts.map +1 -1
- package/dist/cjs/libs/logger/{index.js → index.cjs} +5 -5
- package/dist/cjs/libs/logger/{index.js.map → index.cjs.map} +1 -1
- package/dist/cjs/libs/logger/index.d.ts +2 -2
- package/dist/cjs/libs/logger/index.d.ts.map +1 -1
- package/dist/cjs/libs/logger/{kafkajs-logger-creator.js → kafkajs-logger-creator.cjs} +2 -2
- package/dist/cjs/libs/logger/{kafkajs-logger-creator.js.map → kafkajs-logger-creator.cjs.map} +1 -1
- package/dist/cjs/libs/s3/index.cjs +8 -0
- package/dist/cjs/libs/s3/index.cjs.map +1 -0
- package/dist/cjs/libs/s3/index.d.ts +1 -2
- package/dist/cjs/libs/s3/index.d.ts.map +1 -1
- package/dist/cjs/libs/s3/s3.lib.cjs +207 -0
- package/dist/cjs/libs/s3/s3.lib.cjs.map +1 -0
- package/dist/cjs/libs/s3/s3.lib.d.ts +29 -22
- package/dist/cjs/libs/s3/s3.lib.d.ts.map +1 -1
- package/dist/cjs/processors/{ChatProcessor.js → ChatProcessor.cjs} +111 -149
- package/dist/cjs/processors/ChatProcessor.cjs.map +1 -0
- package/dist/cjs/processors/ChatProcessor.d.ts +23 -7
- package/dist/cjs/processors/ChatProcessor.d.ts.map +1 -1
- package/dist/cjs/processors/ChatProcessor.test.cjs +276 -0
- package/dist/cjs/processors/ChatProcessor.test.cjs.map +1 -0
- package/dist/cjs/processors/ChatProcessor.test.d.ts +2 -0
- package/dist/cjs/processors/ChatProcessor.test.d.ts.map +1 -0
- package/dist/cjs/processors/{ModelsProcessor.js → ModelsProcessor.cjs} +2 -2
- package/dist/cjs/processors/{ModelsProcessor.js.map → ModelsProcessor.cjs.map} +1 -1
- package/dist/cjs/processors/{index.js → index.cjs} +2 -2
- package/dist/cjs/processors/{index.js.map → index.cjs.map} +1 -1
- package/dist/cjs/processors/index.d.ts +2 -2
- package/dist/cjs/processors/index.d.ts.map +1 -1
- package/dist/cjs/services/{AIService.js → AIService.cjs} +42 -30
- package/dist/cjs/services/AIService.cjs.map +1 -0
- package/dist/cjs/services/AIService.d.ts +13 -4
- package/dist/cjs/services/AIService.d.ts.map +1 -1
- package/dist/cjs/services/{InternalEventsHandler.js → InternalEventsHandler.cjs} +5 -5
- package/dist/cjs/services/InternalEventsHandler.cjs.map +1 -0
- package/dist/cjs/services/InternalEventsHandler.d.ts +3 -1
- package/dist/cjs/services/InternalEventsHandler.d.ts.map +1 -1
- package/dist/cjs/services/{ModelFetcher.js → ModelFetcher.cjs} +9 -15
- package/dist/cjs/services/ModelFetcher.cjs.map +1 -0
- package/dist/cjs/services/ModelFetcher.d.ts +2 -7
- package/dist/cjs/services/ModelFetcher.d.ts.map +1 -1
- package/dist/cjs/services/{RedisService.js → RedisService.cjs} +35 -32
- package/dist/cjs/services/RedisService.cjs.map +1 -0
- package/dist/cjs/services/RedisService.d.ts +5 -2
- package/dist/cjs/services/RedisService.d.ts.map +1 -1
- package/dist/cjs/services/{SocketService.js → SocketService.cjs} +19 -20
- package/dist/cjs/services/SocketService.cjs.map +1 -0
- package/dist/cjs/services/SocketService.d.ts +7 -4
- package/dist/cjs/services/SocketService.d.ts.map +1 -1
- package/dist/cjs/{helpers/index.js → services/index.cjs} +5 -5
- package/dist/cjs/services/{index.js.map → index.cjs.map} +1 -1
- package/dist/cjs/services/index.d.ts +5 -6
- package/dist/cjs/services/index.d.ts.map +1 -1
- package/dist/cjs/store/{AgentStore.js → AgentStore.cjs} +4 -5
- package/dist/cjs/store/AgentStore.cjs.map +1 -0
- package/dist/cjs/store/AgentStore.d.ts +2 -1
- package/dist/cjs/store/AgentStore.d.ts.map +1 -1
- package/dist/cjs/store/{ArtifactStore.js → ArtifactStore.cjs} +2 -4
- package/dist/cjs/store/ArtifactStore.cjs.map +1 -0
- package/dist/cjs/store/{ConfigStore.js → ConfigStore.cjs} +8 -11
- package/dist/cjs/store/ConfigStore.cjs.map +1 -0
- package/dist/cjs/store/ConfigStore.d.ts.map +1 -1
- package/dist/cjs/store/{ConfigStore.test.js → ConfigStore.test.cjs} +2 -2
- package/dist/cjs/store/{ConfigStore.test.js.map → ConfigStore.test.cjs.map} +1 -1
- package/dist/cjs/store/{ModelStore.js → ModelStore.cjs} +4 -4
- package/dist/cjs/store/ModelStore.cjs.map +1 -0
- package/dist/cjs/store/{ModelStore.test.js → ModelStore.test.cjs} +17 -17
- package/dist/cjs/store/ModelStore.test.cjs.map +1 -0
- package/dist/cjs/{libs/index.js → store/index.cjs} +4 -3
- package/dist/cjs/store/{index.js.map → index.cjs.map} +1 -1
- package/dist/cjs/store/index.d.ts +4 -4
- package/dist/cjs/store/index.d.ts.map +1 -1
- package/dist/cjs/tools/proposeFormValuesTool.d.ts +2 -2
- package/dist/cjs/tsconfig.cjs.tsbuildinfo +1 -0
- package/dist/esm/config.d.ts +56 -23
- package/dist/esm/config.d.ts.map +1 -1
- package/dist/esm/config.js +88 -36
- package/dist/esm/config.js.map +1 -1
- package/dist/esm/helpers/AIHelper.d.ts +20 -13
- package/dist/esm/helpers/AIHelper.d.ts.map +1 -1
- package/dist/esm/helpers/AIHelper.js +152 -94
- package/dist/esm/helpers/AIHelper.js.map +1 -1
- package/dist/esm/helpers/AIHelper.test.js +21 -14
- package/dist/esm/helpers/AIHelper.test.js.map +1 -1
- package/dist/esm/helpers/ConfigHelper.d.ts.map +1 -1
- package/dist/esm/helpers/ConfigHelper.js +13 -5
- package/dist/esm/helpers/ConfigHelper.js.map +1 -1
- package/dist/esm/helpers/ContextLimiter.js +2 -3
- package/dist/esm/helpers/ContextLimiter.js.map +1 -1
- package/dist/esm/helpers/FileHelper.d.ts +19 -25
- package/dist/esm/helpers/FileHelper.d.ts.map +1 -1
- package/dist/esm/helpers/FileHelper.js +132 -149
- package/dist/esm/helpers/FileHelper.js.map +1 -1
- package/dist/esm/helpers/index.d.ts +4 -5
- package/dist/esm/helpers/index.d.ts.map +1 -1
- package/dist/esm/helpers/index.js +4 -5
- package/dist/esm/helpers/index.js.map +1 -1
- package/dist/esm/index.d.ts +39 -13
- package/dist/esm/index.d.ts.map +1 -1
- package/dist/esm/index.js +79 -12
- package/dist/esm/index.js.map +1 -1
- package/dist/esm/libs/index.d.ts +2 -3
- package/dist/esm/libs/index.d.ts.map +1 -1
- package/dist/esm/libs/index.js +2 -3
- package/dist/esm/libs/index.js.map +1 -1
- package/dist/esm/libs/logger/index.d.ts +2 -2
- package/dist/esm/libs/logger/index.d.ts.map +1 -1
- package/dist/esm/libs/logger/index.js +2 -2
- package/dist/esm/libs/logger/index.js.map +1 -1
- package/dist/esm/libs/logger/kafkajs-logger-creator.js +1 -1
- package/dist/esm/libs/logger/kafkajs-logger-creator.js.map +1 -1
- package/dist/esm/libs/s3/index.d.ts +1 -2
- package/dist/esm/libs/s3/index.d.ts.map +1 -1
- package/dist/esm/libs/s3/index.js +1 -2
- package/dist/esm/libs/s3/index.js.map +1 -1
- package/dist/esm/libs/s3/s3.lib.d.ts +29 -22
- package/dist/esm/libs/s3/s3.lib.d.ts.map +1 -1
- package/dist/esm/libs/s3/s3.lib.js +178 -177
- package/dist/esm/libs/s3/s3.lib.js.map +1 -1
- package/dist/esm/processors/ChatProcessor.d.ts +23 -7
- package/dist/esm/processors/ChatProcessor.d.ts.map +1 -1
- package/dist/esm/processors/ChatProcessor.js +107 -138
- package/dist/esm/processors/ChatProcessor.js.map +1 -1
- package/dist/esm/processors/ChatProcessor.test.d.ts +2 -0
- package/dist/esm/processors/ChatProcessor.test.d.ts.map +1 -0
- package/dist/esm/processors/ChatProcessor.test.js +274 -0
- package/dist/esm/processors/ChatProcessor.test.js.map +1 -0
- package/dist/esm/processors/ModelsProcessor.js +1 -1
- package/dist/esm/processors/ModelsProcessor.js.map +1 -1
- package/dist/esm/processors/index.d.ts +2 -2
- package/dist/esm/processors/index.d.ts.map +1 -1
- package/dist/esm/processors/index.js +2 -2
- package/dist/esm/processors/index.js.map +1 -1
- package/dist/esm/services/AIService.d.ts +13 -4
- package/dist/esm/services/AIService.d.ts.map +1 -1
- package/dist/esm/services/AIService.js +45 -31
- package/dist/esm/services/AIService.js.map +1 -1
- package/dist/esm/services/InternalEventsHandler.d.ts +3 -1
- package/dist/esm/services/InternalEventsHandler.d.ts.map +1 -1
- package/dist/esm/services/InternalEventsHandler.js +10 -8
- package/dist/esm/services/InternalEventsHandler.js.map +1 -1
- package/dist/esm/services/ModelFetcher.d.ts +2 -7
- package/dist/esm/services/ModelFetcher.d.ts.map +1 -1
- package/dist/esm/services/ModelFetcher.js +4 -9
- package/dist/esm/services/ModelFetcher.js.map +1 -1
- package/dist/esm/services/RedisService.d.ts +5 -2
- package/dist/esm/services/RedisService.d.ts.map +1 -1
- package/dist/esm/services/RedisService.js +25 -21
- package/dist/esm/services/RedisService.js.map +1 -1
- package/dist/esm/services/SocketService.d.ts +7 -4
- package/dist/esm/services/SocketService.d.ts.map +1 -1
- package/dist/esm/services/SocketService.js +15 -13
- package/dist/esm/services/SocketService.js.map +1 -1
- package/dist/esm/services/index.d.ts +5 -6
- package/dist/esm/services/index.d.ts.map +1 -1
- package/dist/esm/services/index.js +5 -6
- package/dist/esm/services/index.js.map +1 -1
- package/dist/esm/store/AgentStore.d.ts +2 -1
- package/dist/esm/store/AgentStore.d.ts.map +1 -1
- package/dist/esm/store/AgentStore.js +5 -5
- package/dist/esm/store/AgentStore.js.map +1 -1
- package/dist/esm/store/ArtifactStore.js +3 -7
- package/dist/esm/store/ArtifactStore.js.map +1 -1
- package/dist/esm/store/ConfigStore.d.ts.map +1 -1
- package/dist/esm/store/ConfigStore.js +16 -19
- package/dist/esm/store/ConfigStore.js.map +1 -1
- package/dist/esm/store/ConfigStore.test.js +1 -1
- package/dist/esm/store/ConfigStore.test.js.map +1 -1
- package/dist/esm/store/ModelStore.js +6 -6
- package/dist/esm/store/ModelStore.js.map +1 -1
- package/dist/esm/store/ModelStore.test.js +12 -12
- package/dist/esm/store/ModelStore.test.js.map +1 -1
- package/dist/esm/store/index.d.ts +4 -4
- package/dist/esm/store/index.d.ts.map +1 -1
- package/dist/esm/store/index.js +4 -4
- package/dist/esm/store/index.js.map +1 -1
- package/dist/esm/tools/proposeFormValuesTool.d.ts +2 -2
- package/dist/esm/tsconfig.esm.tsbuildinfo +1 -0
- package/package.json +13 -12
- package/dist/cjs/config.js +0 -48
- package/dist/cjs/config.js.map +0 -1
- package/dist/cjs/helpers/AIHelper.js +0 -330
- package/dist/cjs/helpers/AIHelper.js.map +0 -1
- package/dist/cjs/helpers/AIHelper.test.js.map +0 -1
- package/dist/cjs/helpers/ConfigHelper.js.map +0 -1
- package/dist/cjs/helpers/FileHelper.js +0 -215
- package/dist/cjs/helpers/FileHelper.js.map +0 -1
- package/dist/cjs/helpers/SetupHelper.d.ts +0 -5
- package/dist/cjs/helpers/SetupHelper.d.ts.map +0 -1
- package/dist/cjs/helpers/SetupHelper.js +0 -37
- package/dist/cjs/helpers/SetupHelper.js.map +0 -1
- package/dist/cjs/index.js +0 -34
- package/dist/cjs/index.js.map +0 -1
- package/dist/cjs/libs/kafka/config.d.ts +0 -5
- package/dist/cjs/libs/kafka/config.d.ts.map +0 -1
- package/dist/cjs/libs/kafka/config.js +0 -8
- package/dist/cjs/libs/kafka/config.js.map +0 -1
- package/dist/cjs/libs/kafka/consumer.d.ts +0 -16
- package/dist/cjs/libs/kafka/consumer.d.ts.map +0 -1
- package/dist/cjs/libs/kafka/consumer.js +0 -133
- package/dist/cjs/libs/kafka/consumer.js.map +0 -1
- package/dist/cjs/libs/kafka/index.d.ts +0 -3
- package/dist/cjs/libs/kafka/index.d.ts.map +0 -1
- package/dist/cjs/libs/kafka/index.js.map +0 -1
- package/dist/cjs/libs/kafka/kafka.d.ts +0 -3
- package/dist/cjs/libs/kafka/kafka.d.ts.map +0 -1
- package/dist/cjs/libs/kafka/kafka.js +0 -27
- package/dist/cjs/libs/kafka/kafka.js.map +0 -1
- package/dist/cjs/libs/kafka/producer.d.ts +0 -11
- package/dist/cjs/libs/kafka/producer.d.ts.map +0 -1
- package/dist/cjs/libs/kafka/producer.js +0 -48
- package/dist/cjs/libs/kafka/producer.js.map +0 -1
- package/dist/cjs/libs/logger/config.d.ts +0 -5
- package/dist/cjs/libs/logger/config.d.ts.map +0 -1
- package/dist/cjs/libs/logger/config.js +0 -9
- package/dist/cjs/libs/logger/config.js.map +0 -1
- package/dist/cjs/libs/s3/config.d.ts +0 -7
- package/dist/cjs/libs/s3/config.d.ts.map +0 -1
- package/dist/cjs/libs/s3/config.js +0 -10
- package/dist/cjs/libs/s3/config.js.map +0 -1
- package/dist/cjs/libs/s3/index.js +0 -44
- package/dist/cjs/libs/s3/index.js.map +0 -1
- package/dist/cjs/libs/s3/s3.lib.js +0 -224
- package/dist/cjs/libs/s3/s3.lib.js.map +0 -1
- package/dist/cjs/processors/ChatProcessor.js.map +0 -1
- package/dist/cjs/services/AIService.js.map +0 -1
- package/dist/cjs/services/InternalEventsHandler.js.map +0 -1
- package/dist/cjs/services/KafkaService.d.ts +0 -35
- package/dist/cjs/services/KafkaService.d.ts.map +0 -1
- package/dist/cjs/services/KafkaService.js +0 -123
- package/dist/cjs/services/KafkaService.js.map +0 -1
- package/dist/cjs/services/ModelFetcher.js.map +0 -1
- package/dist/cjs/services/RedisService.js.map +0 -1
- package/dist/cjs/services/SocketService.js.map +0 -1
- package/dist/cjs/services/index.js +0 -23
- package/dist/cjs/store/AgentStore.js.map +0 -1
- package/dist/cjs/store/ArtifactStore.js.map +0 -1
- package/dist/cjs/store/ConfigStore.js.map +0 -1
- package/dist/cjs/store/ModelStore.js.map +0 -1
- package/dist/cjs/store/ModelStore.test.js.map +0 -1
- package/dist/esm/helpers/SetupHelper.d.ts +0 -5
- package/dist/esm/helpers/SetupHelper.d.ts.map +0 -1
- package/dist/esm/helpers/SetupHelper.js +0 -32
- package/dist/esm/helpers/SetupHelper.js.map +0 -1
- package/dist/esm/libs/kafka/config.d.ts +0 -5
- package/dist/esm/libs/kafka/config.d.ts.map +0 -1
- package/dist/esm/libs/kafka/config.js +0 -5
- package/dist/esm/libs/kafka/config.js.map +0 -1
- package/dist/esm/libs/kafka/consumer.d.ts +0 -16
- package/dist/esm/libs/kafka/consumer.d.ts.map +0 -1
- package/dist/esm/libs/kafka/consumer.js +0 -126
- package/dist/esm/libs/kafka/consumer.js.map +0 -1
- package/dist/esm/libs/kafka/index.d.ts +0 -3
- package/dist/esm/libs/kafka/index.d.ts.map +0 -1
- package/dist/esm/libs/kafka/index.js +0 -3
- package/dist/esm/libs/kafka/index.js.map +0 -1
- package/dist/esm/libs/kafka/kafka.d.ts +0 -3
- package/dist/esm/libs/kafka/kafka.d.ts.map +0 -1
- package/dist/esm/libs/kafka/kafka.js +0 -24
- package/dist/esm/libs/kafka/kafka.js.map +0 -1
- package/dist/esm/libs/kafka/producer.d.ts +0 -11
- package/dist/esm/libs/kafka/producer.d.ts.map +0 -1
- package/dist/esm/libs/kafka/producer.js +0 -44
- package/dist/esm/libs/kafka/producer.js.map +0 -1
- package/dist/esm/libs/logger/config.d.ts +0 -5
- package/dist/esm/libs/logger/config.d.ts.map +0 -1
- package/dist/esm/libs/logger/config.js +0 -6
- package/dist/esm/libs/logger/config.js.map +0 -1
- package/dist/esm/libs/s3/config.d.ts +0 -7
- package/dist/esm/libs/s3/config.d.ts.map +0 -1
- package/dist/esm/libs/s3/config.js +0 -7
- package/dist/esm/libs/s3/config.js.map +0 -1
- package/dist/esm/services/KafkaService.d.ts +0 -35
- package/dist/esm/services/KafkaService.d.ts.map +0 -1
- package/dist/esm/services/KafkaService.js +0 -120
- package/dist/esm/services/KafkaService.js.map +0 -1
- package/dist/tsconfig.cjs.tsbuildinfo +0 -1
- package/dist/tsconfig.esm.tsbuildinfo +0 -1
- /package/dist/cjs/libs/logger/{logger.js → logger.cjs} +0 -0
- /package/dist/cjs/libs/logger/{logger.js.map → logger.cjs.map} +0 -0
- /package/dist/cjs/tools/{generateChartTool.js → generateChartTool.cjs} +0 -0
- /package/dist/cjs/tools/{generateChartTool.js.map → generateChartTool.cjs.map} +0 -0
- /package/dist/cjs/tools/{proposeFormValuesTool.js → proposeFormValuesTool.cjs} +0 -0
- /package/dist/cjs/tools/{proposeFormValuesTool.js.map → proposeFormValuesTool.cjs.map} +0 -0
|
@@ -0,0 +1,274 @@
|
|
|
1
|
+
import { describe, it, expect, beforeEach, vi } from 'vitest';
|
|
2
|
+
import { ChatProcessor } from './ChatProcessor';
|
|
3
|
+
import { MessageRole, AgentStatus, ChatType } from '@multiplayer-app/ai-agent-types';
|
|
4
|
+
import { ConfigStore } from '../store/ConfigStore';
|
|
5
|
+
import { ConfigProvider } from '../config';
|
|
6
|
+
import { S3Lib } from '../libs/s3';
|
|
7
|
+
describe('ChatProcessor.streamMessage', () => {
|
|
8
|
+
let chatProcessor;
|
|
9
|
+
let mockChatRepository;
|
|
10
|
+
let mockMessageRepository;
|
|
11
|
+
let mockArtifactStore;
|
|
12
|
+
let mockAgentStore;
|
|
13
|
+
let mockSocketService;
|
|
14
|
+
let configStore;
|
|
15
|
+
let capturedOptions;
|
|
16
|
+
beforeEach(() => {
|
|
17
|
+
vi.clearAllMocks();
|
|
18
|
+
capturedOptions = undefined;
|
|
19
|
+
// Mock repositories
|
|
20
|
+
mockChatRepository = {
|
|
21
|
+
findById: vi.fn(),
|
|
22
|
+
create: vi.fn(),
|
|
23
|
+
update: vi.fn(),
|
|
24
|
+
delete: vi.fn(),
|
|
25
|
+
findWithMessages: vi.fn(),
|
|
26
|
+
};
|
|
27
|
+
mockMessageRepository = {
|
|
28
|
+
findByChatId: vi.fn().mockResolvedValue([]),
|
|
29
|
+
findById: vi.fn(),
|
|
30
|
+
create: vi.fn(),
|
|
31
|
+
update: vi.fn(),
|
|
32
|
+
};
|
|
33
|
+
// Mock artifact store
|
|
34
|
+
mockArtifactStore = {
|
|
35
|
+
listArtifacts: vi.fn().mockReturnValue([]),
|
|
36
|
+
deleteArtifacts: vi.fn(),
|
|
37
|
+
};
|
|
38
|
+
// Mock agent store
|
|
39
|
+
mockAgentStore = {
|
|
40
|
+
registerAgentProcess: vi.fn().mockReturnValue({
|
|
41
|
+
signal: new AbortController().signal,
|
|
42
|
+
}),
|
|
43
|
+
shareAgentProcessEvent: vi.fn(),
|
|
44
|
+
addListener: vi.fn(),
|
|
45
|
+
removeListener: vi.fn(),
|
|
46
|
+
};
|
|
47
|
+
// Mock socket service
|
|
48
|
+
mockSocketService = {
|
|
49
|
+
emitMessageUpdate: vi.fn(),
|
|
50
|
+
emitChatUpdate: vi.fn(),
|
|
51
|
+
};
|
|
52
|
+
// Setup config store with agent config
|
|
53
|
+
configStore = ConfigStore.getInstance();
|
|
54
|
+
configStore.clear();
|
|
55
|
+
const agentConfig = {
|
|
56
|
+
name: 'test-agent',
|
|
57
|
+
description: 'Test agent for property verification',
|
|
58
|
+
defaultModel: 'openai/gpt-4o',
|
|
59
|
+
systemPrompt: 'You are a helpful assistant',
|
|
60
|
+
temperature: 0.8,
|
|
61
|
+
maxOutputTokens: 3000,
|
|
62
|
+
topP: 0.95,
|
|
63
|
+
topK: 50,
|
|
64
|
+
presencePenalty: 0.6,
|
|
65
|
+
frequencyPenalty: 0.4,
|
|
66
|
+
stopSequences: ['\n\n', '---', 'END'],
|
|
67
|
+
seed: 42,
|
|
68
|
+
tools: [],
|
|
69
|
+
toolChoice: 'auto',
|
|
70
|
+
};
|
|
71
|
+
// Use addAgent to bypass ConfigHelper conversion
|
|
72
|
+
configStore.addAgent(['test-context'], agentConfig);
|
|
73
|
+
const config = ConfigProvider.getInstance().getConfig();
|
|
74
|
+
const s3Lib = new S3Lib(config.s3);
|
|
75
|
+
// Create ChatProcessor
|
|
76
|
+
chatProcessor = new ChatProcessor({
|
|
77
|
+
chatRepository: mockChatRepository,
|
|
78
|
+
messageRepository: mockMessageRepository,
|
|
79
|
+
artifactStore: mockArtifactStore,
|
|
80
|
+
s3Lib,
|
|
81
|
+
config,
|
|
82
|
+
socketService: mockSocketService,
|
|
83
|
+
agentStore: mockAgentStore,
|
|
84
|
+
});
|
|
85
|
+
});
|
|
86
|
+
it('should pass all agent properties correctly to streamAssistantResponse', async () => {
|
|
87
|
+
const chat = {
|
|
88
|
+
id: 'test-chat-id',
|
|
89
|
+
contextKey: 'test-context',
|
|
90
|
+
type: ChatType.Chat,
|
|
91
|
+
status: AgentStatus.Processing,
|
|
92
|
+
createdAt: new Date().toISOString(),
|
|
93
|
+
updatedAt: new Date().toISOString(),
|
|
94
|
+
};
|
|
95
|
+
const payload = {
|
|
96
|
+
chatId: chat.id,
|
|
97
|
+
content: 'Hello, test message',
|
|
98
|
+
contextKey: 'test-context',
|
|
99
|
+
};
|
|
100
|
+
// Mock chat repository
|
|
101
|
+
mockChatRepository.findById.mockResolvedValue(chat);
|
|
102
|
+
// Mock message creation
|
|
103
|
+
const userMessage = {
|
|
104
|
+
id: 'user-msg-1',
|
|
105
|
+
chat: chat.id,
|
|
106
|
+
role: MessageRole.User,
|
|
107
|
+
content: payload.content,
|
|
108
|
+
createdAt: new Date().toISOString(),
|
|
109
|
+
};
|
|
110
|
+
const assistantMessage = {
|
|
111
|
+
id: 'assistant-msg-1',
|
|
112
|
+
chat: chat.id,
|
|
113
|
+
role: MessageRole.Assistant,
|
|
114
|
+
content: '',
|
|
115
|
+
createdAt: new Date().toISOString(),
|
|
116
|
+
agentName: 'test-agent',
|
|
117
|
+
};
|
|
118
|
+
mockMessageRepository.create
|
|
119
|
+
.mockResolvedValueOnce(userMessage)
|
|
120
|
+
.mockResolvedValueOnce(assistantMessage);
|
|
121
|
+
// Mock stream to prevent actual API calls
|
|
122
|
+
const mockStream = {
|
|
123
|
+
fullStream: (async function* () {
|
|
124
|
+
yield { type: 'finish', finishReason: 'stop', text: 'Test response' };
|
|
125
|
+
})(),
|
|
126
|
+
};
|
|
127
|
+
// Capture options passed to streamAssistantResponse
|
|
128
|
+
const aiHelper = chatProcessor.aiHelper;
|
|
129
|
+
vi.spyOn(aiHelper, 'streamAssistantResponse').mockImplementation(async (messages, signal, options) => {
|
|
130
|
+
capturedOptions = options;
|
|
131
|
+
return mockStream;
|
|
132
|
+
});
|
|
133
|
+
// Call streamMessage
|
|
134
|
+
await chatProcessor.streamMessage(chat, payload);
|
|
135
|
+
// Verify options were captured
|
|
136
|
+
expect(capturedOptions).toBeDefined();
|
|
137
|
+
// Verify all agent properties are present in the options
|
|
138
|
+
if (capturedOptions) {
|
|
139
|
+
expect(capturedOptions).toHaveProperty('name', 'test-agent');
|
|
140
|
+
expect(capturedOptions).toHaveProperty('system', 'You are a helpful assistant');
|
|
141
|
+
expect(capturedOptions).toHaveProperty('temperature', 0.8);
|
|
142
|
+
expect(capturedOptions).toHaveProperty('maxOutputTokens', 3000);
|
|
143
|
+
expect(capturedOptions).toHaveProperty('topP', 0.95);
|
|
144
|
+
expect(capturedOptions).toHaveProperty('topK', 50);
|
|
145
|
+
expect(capturedOptions).toHaveProperty('presencePenalty', 0.6);
|
|
146
|
+
expect(capturedOptions).toHaveProperty('frequencyPenalty', 0.4);
|
|
147
|
+
expect(capturedOptions).toHaveProperty('stopSequences', ['\n\n', '---', 'END']);
|
|
148
|
+
expect(capturedOptions).toHaveProperty('seed', 42);
|
|
149
|
+
expect(capturedOptions).toHaveProperty('toolChoice', 'auto');
|
|
150
|
+
expect(capturedOptions).toHaveProperty('tools');
|
|
151
|
+
}
|
|
152
|
+
});
|
|
153
|
+
it('should handle optional agent properties when not provided', async () => {
|
|
154
|
+
configStore.clear();
|
|
155
|
+
const minimalAgentConfig = {
|
|
156
|
+
name: 'minimal-agent',
|
|
157
|
+
description: 'Minimal agent config',
|
|
158
|
+
systemPrompt: 'You are helpful',
|
|
159
|
+
tools: [],
|
|
160
|
+
};
|
|
161
|
+
configStore.addAgent(['test-context'], minimalAgentConfig);
|
|
162
|
+
const chat = {
|
|
163
|
+
id: 'test-chat-id-2',
|
|
164
|
+
contextKey: 'test-context',
|
|
165
|
+
type: ChatType.Chat,
|
|
166
|
+
status: AgentStatus.Processing,
|
|
167
|
+
createdAt: new Date().toISOString(),
|
|
168
|
+
updatedAt: new Date().toISOString(),
|
|
169
|
+
};
|
|
170
|
+
const payload = {
|
|
171
|
+
chatId: chat.id,
|
|
172
|
+
content: 'Test',
|
|
173
|
+
contextKey: 'test-context',
|
|
174
|
+
};
|
|
175
|
+
mockChatRepository.findById.mockResolvedValue(chat);
|
|
176
|
+
const userMessage = {
|
|
177
|
+
id: 'user-msg-2',
|
|
178
|
+
chat: chat.id,
|
|
179
|
+
role: MessageRole.User,
|
|
180
|
+
content: payload.content,
|
|
181
|
+
createdAt: new Date().toISOString(),
|
|
182
|
+
};
|
|
183
|
+
const assistantMessage = {
|
|
184
|
+
id: 'assistant-msg-2',
|
|
185
|
+
chat: chat.id,
|
|
186
|
+
role: MessageRole.Assistant,
|
|
187
|
+
content: '',
|
|
188
|
+
createdAt: new Date().toISOString(),
|
|
189
|
+
agentName: 'minimal-agent',
|
|
190
|
+
};
|
|
191
|
+
mockMessageRepository.create
|
|
192
|
+
.mockResolvedValueOnce(userMessage)
|
|
193
|
+
.mockResolvedValueOnce(assistantMessage);
|
|
194
|
+
const mockStream = {
|
|
195
|
+
fullStream: (async function* () {
|
|
196
|
+
yield { type: 'finish', finishReason: 'stop', text: 'Response' };
|
|
197
|
+
})(),
|
|
198
|
+
};
|
|
199
|
+
const aiHelper = chatProcessor.aiHelper;
|
|
200
|
+
vi.spyOn(aiHelper, 'streamAssistantResponse').mockImplementation(async (messages, signal, options) => {
|
|
201
|
+
capturedOptions = options;
|
|
202
|
+
return mockStream;
|
|
203
|
+
});
|
|
204
|
+
await chatProcessor.streamMessage(chat, payload);
|
|
205
|
+
// Verify required properties are present
|
|
206
|
+
expect(capturedOptions).toBeDefined();
|
|
207
|
+
if (capturedOptions) {
|
|
208
|
+
expect(capturedOptions).toHaveProperty('name', 'minimal-agent');
|
|
209
|
+
expect(capturedOptions).toHaveProperty('system', 'You are helpful');
|
|
210
|
+
expect(capturedOptions).toHaveProperty('tools');
|
|
211
|
+
}
|
|
212
|
+
});
|
|
213
|
+
it('should pass stopWhen condition when provided', async () => {
|
|
214
|
+
configStore.clear();
|
|
215
|
+
const stopWhenFn = vi.fn(() => false);
|
|
216
|
+
const agentConfigWithStopWhen = {
|
|
217
|
+
name: 'agent-with-stopwhen',
|
|
218
|
+
description: 'Agent with stopWhen',
|
|
219
|
+
systemPrompt: 'You are helpful',
|
|
220
|
+
tools: [],
|
|
221
|
+
stopWhen: stopWhenFn,
|
|
222
|
+
};
|
|
223
|
+
configStore.addAgent(['test-context'], agentConfigWithStopWhen);
|
|
224
|
+
const chat = {
|
|
225
|
+
id: 'test-chat-id-3',
|
|
226
|
+
contextKey: 'test-context',
|
|
227
|
+
type: ChatType.Chat,
|
|
228
|
+
status: AgentStatus.Processing,
|
|
229
|
+
createdAt: new Date().toISOString(),
|
|
230
|
+
updatedAt: new Date().toISOString(),
|
|
231
|
+
};
|
|
232
|
+
const payload = {
|
|
233
|
+
chatId: chat.id,
|
|
234
|
+
content: 'Test',
|
|
235
|
+
contextKey: 'test-context',
|
|
236
|
+
};
|
|
237
|
+
mockChatRepository.findById.mockResolvedValue(chat);
|
|
238
|
+
const userMessage = {
|
|
239
|
+
id: 'user-msg-3',
|
|
240
|
+
chat: chat.id,
|
|
241
|
+
role: MessageRole.User,
|
|
242
|
+
content: payload.content,
|
|
243
|
+
createdAt: new Date().toISOString(),
|
|
244
|
+
};
|
|
245
|
+
const assistantMessage = {
|
|
246
|
+
id: 'assistant-msg-3',
|
|
247
|
+
chat: chat.id,
|
|
248
|
+
role: MessageRole.Assistant,
|
|
249
|
+
content: '',
|
|
250
|
+
createdAt: new Date().toISOString(),
|
|
251
|
+
agentName: 'agent-with-stopwhen',
|
|
252
|
+
};
|
|
253
|
+
mockMessageRepository.create
|
|
254
|
+
.mockResolvedValueOnce(userMessage)
|
|
255
|
+
.mockResolvedValueOnce(assistantMessage);
|
|
256
|
+
const mockStream = {
|
|
257
|
+
fullStream: (async function* () {
|
|
258
|
+
yield { type: 'finish', finishReason: 'stop', text: 'Response' };
|
|
259
|
+
})(),
|
|
260
|
+
};
|
|
261
|
+
const aiHelper = chatProcessor.aiHelper;
|
|
262
|
+
vi.spyOn(aiHelper, 'streamAssistantResponse').mockImplementation(async (messages, signal, options) => {
|
|
263
|
+
capturedOptions = options;
|
|
264
|
+
return mockStream;
|
|
265
|
+
});
|
|
266
|
+
await chatProcessor.streamMessage(chat, payload);
|
|
267
|
+
// Verify stopWhen is passed
|
|
268
|
+
expect(capturedOptions).toBeDefined();
|
|
269
|
+
if (capturedOptions) {
|
|
270
|
+
expect(capturedOptions).toHaveProperty('stopWhen', stopWhenFn);
|
|
271
|
+
}
|
|
272
|
+
});
|
|
273
|
+
});
|
|
274
|
+
//# sourceMappingURL=ChatProcessor.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"ChatProcessor.test.js","sourceRoot":"","sources":["../../../src/processors/ChatProcessor.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAC9D,OAAO,EAAE,aAAa,EAAE,MAAM,iBAAiB,CAAC;AAQhD,OAAO,EAAE,WAAW,EAAE,WAAW,EAAE,QAAQ,EAAE,MAAM,iCAAiC,CAAC;AAGrF,OAAO,EAAE,WAAW,EAAE,MAAM,sBAAsB,CAAC;AACnD,OAAO,EAAE,cAAc,EAAE,MAAM,WAAW,CAAC;AAC3C,OAAO,EAAE,KAAK,EAAE,MAAM,YAAY,CAAC;AAMnC,QAAQ,CAAC,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,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,aAAa,EAAE,CAAC;QACnB,eAAe,GAAG,SAAS,CAAC;QAE5B,oBAAoB;QACpB,kBAAkB,GAAG;YACnB,QAAQ,EAAE,EAAE,CAAC,EAAE,EAAE;YACjB,MAAM,EAAE,EAAE,CAAC,EAAE,EAAE;YACf,MAAM,EAAE,EAAE,CAAC,EAAE,EAAE;YACf,MAAM,EAAE,EAAE,CAAC,EAAE,EAAE;YACf,gBAAgB,EAAE,EAAE,CAAC,EAAE,EAAE;SACnB,CAAC;QAET,qBAAqB,GAAG;YACtB,YAAY,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,CAAC;YAC3C,QAAQ,EAAE,EAAE,CAAC,EAAE,EAAE;YACjB,MAAM,EAAE,EAAE,CAAC,EAAE,EAAE;YACf,MAAM,EAAE,EAAE,CAAC,EAAE,EAAE;SACT,CAAC;QAET,sBAAsB;QACtB,iBAAiB,GAAG;YAClB,aAAa,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,EAAE,CAAC;YAC1C,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE;SAClB,CAAC;QAET,mBAAmB;QACnB,cAAc,GAAG;YACf,oBAAoB,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC;gBAC5C,MAAM,EAAE,IAAI,eAAe,EAAE,CAAC,MAAM;aACrC,CAAC;YACF,sBAAsB,EAAE,EAAE,CAAC,EAAE,EAAE;YAC/B,WAAW,EAAE,EAAE,CAAC,EAAE,EAAE;YACpB,cAAc,EAAE,EAAE,CAAC,EAAE,EAAE;SACjB,CAAC;QAET,sBAAsB;QACtB,iBAAiB,GAAG;YAClB,iBAAiB,EAAE,EAAE,CAAC,EAAE,EAAE;YAC1B,cAAc,EAAE,EAAE,CAAC,EAAE,EAAE;SACjB,CAAC;QAET,uCAAuC;QACvC,WAAW,GAAG,WAAW,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,cAAc,CAAC,WAAW,EAAE,CAAC,SAAS,EAAE,CAAC;QACxD,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;QAEnC,uBAAuB;QACvB,aAAa,GAAG,IAAI,aAAa,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,EAAE,CAAC,uEAAuE,EAAE,KAAK,IAAI,EAAE;QACrF,MAAM,IAAI,GAAc;YACtB,EAAE,EAAE,cAAc;YAClB,UAAU,EAAE,cAAc;YAC1B,IAAI,EAAE,QAAQ,CAAC,IAAI;YACnB,MAAM,EAAE,WAAW,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,WAAW,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,WAAW,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,EAAE,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,MAAM,CAAC,eAAe,CAAC,CAAC,WAAW,EAAE,CAAC;QAEtC,yDAAyD;QACzD,IAAI,eAAe,EAAE,CAAC;YACpB,MAAM,CAAC,eAAe,CAAC,CAAC,cAAc,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC;YAC7D,MAAM,CAAC,eAAe,CAAC,CAAC,cAAc,CAAC,QAAQ,EAAE,6BAA6B,CAAC,CAAC;YAChF,MAAM,CAAC,eAAe,CAAC,CAAC,cAAc,CAAC,aAAa,EAAE,GAAG,CAAC,CAAC;YAC3D,MAAM,CAAC,eAAe,CAAC,CAAC,cAAc,CAAC,iBAAiB,EAAE,IAAI,CAAC,CAAC;YAChE,MAAM,CAAC,eAAe,CAAC,CAAC,cAAc,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;YACrD,MAAM,CAAC,eAAe,CAAC,CAAC,cAAc,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;YACnD,MAAM,CAAC,eAAe,CAAC,CAAC,cAAc,CAAC,iBAAiB,EAAE,GAAG,CAAC,CAAC;YAC/D,MAAM,CAAC,eAAe,CAAC,CAAC,cAAc,CAAC,kBAAkB,EAAE,GAAG,CAAC,CAAC;YAChE,MAAM,CAAC,eAAe,CAAC,CAAC,cAAc,CAAC,eAAe,EAAE,CAAC,MAAM,EAAE,KAAK,EAAE,KAAK,CAAC,CAAC,CAAC;YAChF,MAAM,CAAC,eAAe,CAAC,CAAC,cAAc,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;YACnD,MAAM,CAAC,eAAe,CAAC,CAAC,cAAc,CAAC,YAAY,EAAE,MAAM,CAAC,CAAC;YAC7D,MAAM,CAAC,eAAe,CAAC,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC;QAClD,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,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,QAAQ,CAAC,IAAI;YACnB,MAAM,EAAE,WAAW,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,WAAW,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,WAAW,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,EAAE,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,MAAM,CAAC,eAAe,CAAC,CAAC,WAAW,EAAE,CAAC;QACtC,IAAI,eAAe,EAAE,CAAC;YACpB,MAAM,CAAC,eAAe,CAAC,CAAC,cAAc,CAAC,MAAM,EAAE,eAAe,CAAC,CAAC;YAChE,MAAM,CAAC,eAAe,CAAC,CAAC,cAAc,CAAC,QAAQ,EAAE,iBAAiB,CAAC,CAAC;YACpE,MAAM,CAAC,eAAe,CAAC,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC;QAClD,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8CAA8C,EAAE,KAAK,IAAI,EAAE;QAC5D,WAAW,CAAC,KAAK,EAAE,CAAC;QACpB,MAAM,UAAU,GAAG,EAAE,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,QAAQ,CAAC,IAAI;YACnB,MAAM,EAAE,WAAW,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,WAAW,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,WAAW,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,EAAE,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,MAAM,CAAC,eAAe,CAAC,CAAC,WAAW,EAAE,CAAC;QACtC,IAAI,eAAe,EAAE,CAAC;YACpB,MAAM,CAAC,eAAe,CAAC,CAAC,cAAc,CAAC,UAAU,EAAE,UAAU,CAAC,CAAC;QACjE,CAAC;IACH,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"ModelsProcessor.js","sourceRoot":"","sources":["../../../src/processors/ModelsProcessor.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,
|
|
1
|
+
{"version":3,"file":"ModelsProcessor.js","sourceRoot":"","sources":["../../../src/processors/ModelsProcessor.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,UAAU,CAAC;AAEtC,MAAM,aAAa,GAA2B;IAC1C,MAAM,EAAE,QAAQ;IAChB,SAAS,EAAE,WAAW;IACtB,MAAM,EAAE,QAAQ;IAChB,UAAU,EAAE,YAAY;CAC3B,CAAC;AACF,MAAM,OAAO,eAAe;IAEzB,KAAK,CAAC,SAAS;QACd,MAAM,UAAU,GAAG,UAAU,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"}
|
|
@@ -1,3 +1,3 @@
|
|
|
1
|
-
export * from './ChatProcessor
|
|
2
|
-
export * from './ModelsProcessor
|
|
1
|
+
export * from './ChatProcessor';
|
|
2
|
+
export * from './ModelsProcessor';
|
|
3
3
|
//# sourceMappingURL=index.d.ts.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/processors/index.ts"],"names":[],"mappings":"AAAA,cAAc,
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/processors/index.ts"],"names":[],"mappings":"AAAA,cAAc,iBAAiB,CAAC;AAChC,cAAc,mBAAmB,CAAC"}
|
|
@@ -1,3 +1,3 @@
|
|
|
1
|
-
export * from './ChatProcessor
|
|
2
|
-
export * from './ModelsProcessor
|
|
1
|
+
export * from './ChatProcessor';
|
|
2
|
+
export * from './ModelsProcessor';
|
|
3
3
|
//# sourceMappingURL=index.js.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/processors/index.ts"],"names":[],"mappings":"AAAA,cAAc,
|
|
1
|
+
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/processors/index.ts"],"names":[],"mappings":"AAAA,cAAc,iBAAiB,CAAC;AAChC,cAAc,mBAAmB,CAAC"}
|
|
@@ -1,16 +1,24 @@
|
|
|
1
|
-
import {
|
|
2
|
-
import
|
|
3
|
-
import { ModelStore } from '../store/ModelStore
|
|
1
|
+
import { type LanguageModel, ModelMessage, streamText, Tool, ToolSet, type StopCondition } from 'ai';
|
|
2
|
+
import { AdditionalContext, AgentConfig, ToolChoice } from '@multiplayer-app/ai-agent-types';
|
|
3
|
+
import { ModelStore } from '../store/ModelStore';
|
|
4
|
+
import { Config } from '../config';
|
|
4
5
|
export interface GenerateTextOptions {
|
|
5
6
|
messages: ModelMessage[];
|
|
6
7
|
system?: string;
|
|
7
8
|
temperature?: number;
|
|
8
9
|
maxOutputTokens?: number;
|
|
9
10
|
topP?: number;
|
|
11
|
+
topK?: number;
|
|
12
|
+
presencePenalty?: number;
|
|
13
|
+
frequencyPenalty?: number;
|
|
14
|
+
stopSequences?: string[];
|
|
15
|
+
seed?: number;
|
|
10
16
|
toolChoice?: ToolChoice<ToolSet>;
|
|
11
17
|
tools?: ToolSet;
|
|
12
18
|
model?: string;
|
|
13
19
|
signal?: AbortSignal;
|
|
20
|
+
websearchEnabled?: boolean;
|
|
21
|
+
stopWhen?: StopCondition<ToolSet> | Array<StopCondition<ToolSet>>;
|
|
14
22
|
}
|
|
15
23
|
export interface StreamTextOptions extends GenerateTextOptions {
|
|
16
24
|
onChunk?: (chunk: any) => void;
|
|
@@ -31,7 +39,8 @@ export interface LocalFunctionTool extends Omit<Tool, 'title'> {
|
|
|
31
39
|
}
|
|
32
40
|
export declare class AIService {
|
|
33
41
|
private modelStore;
|
|
34
|
-
|
|
42
|
+
private config;
|
|
43
|
+
constructor(modelStore: ModelStore, config: Config["ai"]);
|
|
35
44
|
getLanguageModel(modelId: string | undefined, websearchEnabled: boolean): LanguageModel;
|
|
36
45
|
/**
|
|
37
46
|
* Create a language model instance based on the provider
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"AIService.d.ts","sourceRoot":"","sources":["../../../src/services/AIService.ts"],"names":[],"mappings":"AAIA,OAAO,EAEL,
|
|
1
|
+
{"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"}
|
|
@@ -1,18 +1,20 @@
|
|
|
1
|
-
import { openai } from '@ai-sdk/openai';
|
|
2
|
-
import { anthropic } from '@ai-sdk/anthropic';
|
|
3
|
-
import { google } from '@ai-sdk/google';
|
|
1
|
+
import { createOpenAI, openai } from '@ai-sdk/openai';
|
|
2
|
+
import { anthropic, createAnthropic } from '@ai-sdk/anthropic';
|
|
3
|
+
import { createGoogleGenerativeAI, google } from '@ai-sdk/google';
|
|
4
4
|
import { z } from 'zod';
|
|
5
5
|
import { generateText, streamText, tool, } from 'ai';
|
|
6
6
|
import { AgentToolType } from '@multiplayer-app/ai-agent-types';
|
|
7
7
|
import { createOpenRouter } from '@openrouter/ai-sdk-provider';
|
|
8
|
-
import {
|
|
9
|
-
import { logger } from '../libs/logger/index.js';
|
|
8
|
+
import { logger } from '../libs/logger';
|
|
10
9
|
export class AIService {
|
|
11
|
-
|
|
10
|
+
modelStore;
|
|
11
|
+
config;
|
|
12
|
+
constructor(modelStore, config) {
|
|
12
13
|
this.modelStore = modelStore;
|
|
14
|
+
this.config = config;
|
|
13
15
|
}
|
|
14
16
|
getLanguageModel(modelId, websearchEnabled) {
|
|
15
|
-
const selectedModel = modelId || config.
|
|
17
|
+
const selectedModel = modelId || this.config.defaultModel || this.modelStore.getModels()[0].id;
|
|
16
18
|
const model = selectedModel ?
|
|
17
19
|
this.modelStore.getModelById(selectedModel) :
|
|
18
20
|
this.modelStore.getModels()[0]; //get first model if no model is selected
|
|
@@ -28,21 +30,21 @@ export class AIService {
|
|
|
28
30
|
const provider = model.provider || 'openai'; // Default to openai if provider not specified
|
|
29
31
|
switch (provider) {
|
|
30
32
|
case 'openai':
|
|
31
|
-
return
|
|
33
|
+
return createOpenAI({ apiKey: this.config.openaiApiKey })(model.id);
|
|
32
34
|
case 'anthropic':
|
|
33
|
-
return
|
|
35
|
+
return createAnthropic({ apiKey: this.config.anthropicApiKey })(model.id);
|
|
34
36
|
case 'google':
|
|
35
|
-
return
|
|
37
|
+
return createGoogleGenerativeAI({ apiKey: this.config.googleApiKey })(model.id);
|
|
36
38
|
case 'openrouter':
|
|
37
39
|
return createOpenRouter({
|
|
38
|
-
apiKey: config.
|
|
40
|
+
apiKey: this.config.openrouterApiKey,
|
|
39
41
|
})(model.id, { plugins: websearchEnabled ? [{ id: "web", engine: undefined, max_results: 2 }] : [] });
|
|
40
42
|
default:
|
|
41
43
|
throw new Error(`Unsupported provider: ${provider}`);
|
|
42
44
|
}
|
|
43
45
|
}
|
|
44
46
|
getWebSearchTool(modelId) {
|
|
45
|
-
const selectedModel = modelId || config.
|
|
47
|
+
const selectedModel = modelId || this.config.defaultModel || this.modelStore.getModels()[0].id;
|
|
46
48
|
const model = selectedModel ?
|
|
47
49
|
this.modelStore.getModelById(selectedModel) :
|
|
48
50
|
this.modelStore.getModels()[0]; //get first model if no model is selected
|
|
@@ -56,7 +58,7 @@ export class AIService {
|
|
|
56
58
|
return anthropic.tools.webSearch_20250305();
|
|
57
59
|
case 'google':
|
|
58
60
|
return google.tools.googleSearch({});
|
|
59
|
-
case 'openrouter':
|
|
61
|
+
case 'openrouter':
|
|
60
62
|
return undefined;
|
|
61
63
|
default:
|
|
62
64
|
return undefined;
|
|
@@ -85,8 +87,7 @@ export class AIService {
|
|
|
85
87
|
}
|
|
86
88
|
if (data.headersToPass) {
|
|
87
89
|
params.headers = data.headersToPass.reduce((acc, headerName) => {
|
|
88
|
-
|
|
89
|
-
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 : '';
|
|
90
|
+
acc[headerName] = context?.headers?.[headerName] ?? '';
|
|
90
91
|
return acc;
|
|
91
92
|
}, {});
|
|
92
93
|
}
|
|
@@ -134,12 +135,15 @@ export class AIService {
|
|
|
134
135
|
// });
|
|
135
136
|
// }
|
|
136
137
|
getAgentOptions(config, modelId, context) {
|
|
138
|
+
let websearchEnabled = false;
|
|
137
139
|
const tools = config.tools.reduce((acc, toolConfig) => {
|
|
138
|
-
var _a;
|
|
139
140
|
const tool = this.getTool(toolConfig, modelId, context);
|
|
140
|
-
if (tool &&
|
|
141
|
+
if (tool && toolConfig.data?.title) {
|
|
141
142
|
acc[toolConfig.data.title] = tool;
|
|
142
143
|
}
|
|
144
|
+
if (toolConfig.type === AgentToolType.WEB_SEARCH) {
|
|
145
|
+
websearchEnabled = true;
|
|
146
|
+
}
|
|
143
147
|
return acc;
|
|
144
148
|
}, {});
|
|
145
149
|
return {
|
|
@@ -148,15 +152,25 @@ export class AIService {
|
|
|
148
152
|
system: config.systemPrompt,
|
|
149
153
|
toolChoice: config.toolChoice,
|
|
150
154
|
name: config.name,
|
|
155
|
+
stopWhen: config.stopWhen,
|
|
156
|
+
maxOutputTokens: config.maxOutputTokens,
|
|
157
|
+
temperature: config.temperature,
|
|
158
|
+
topP: config.topP,
|
|
159
|
+
topK: config.topK,
|
|
160
|
+
presencePenalty: config.presencePenalty,
|
|
161
|
+
frequencyPenalty: config.frequencyPenalty,
|
|
162
|
+
stopSequences: config.stopSequences,
|
|
163
|
+
seed: config.seed,
|
|
164
|
+
websearchEnabled
|
|
151
165
|
};
|
|
152
166
|
}
|
|
153
167
|
async generateText(options) {
|
|
154
|
-
var _a;
|
|
155
|
-
const websearchEnabled = !!((_a = options.tools) === null || _a === void 0 ? void 0 : _a[AgentToolType.WEB_SEARCH]);
|
|
156
168
|
try {
|
|
169
|
+
const { stopWhen, model, websearchEnabled, ...restOptions } = options;
|
|
157
170
|
const { text } = await generateText({
|
|
158
|
-
...
|
|
159
|
-
model: this.getLanguageModel(
|
|
171
|
+
...restOptions,
|
|
172
|
+
model: this.getLanguageModel(model, !!websearchEnabled),
|
|
173
|
+
...(stopWhen !== undefined ? { stopWhen } : {}),
|
|
160
174
|
});
|
|
161
175
|
return text;
|
|
162
176
|
}
|
|
@@ -166,26 +180,26 @@ export class AIService {
|
|
|
166
180
|
}
|
|
167
181
|
}
|
|
168
182
|
streamText(options) {
|
|
169
|
-
|
|
170
|
-
const { onError, onChunk, onFinish, onStepFinish } = options;
|
|
171
|
-
const websearchEnabled = !!((_a = options.tools) === null || _a === void 0 ? void 0 : _a[AgentToolType.WEB_SEARCH]);
|
|
183
|
+
const { onError, onChunk, onFinish, onStepFinish, onAbort, signal, websearchEnabled, model, ...restOptions } = options;
|
|
172
184
|
try {
|
|
173
185
|
return streamText({
|
|
174
|
-
...
|
|
175
|
-
|
|
186
|
+
...restOptions,
|
|
187
|
+
abortSignal: signal,
|
|
188
|
+
model: this.getLanguageModel(model, !!websearchEnabled),
|
|
176
189
|
onError({ error }) {
|
|
177
190
|
logger.error(error);
|
|
178
|
-
onError
|
|
191
|
+
onError?.(error);
|
|
179
192
|
},
|
|
180
193
|
onChunk(chunk) {
|
|
181
|
-
onChunk
|
|
194
|
+
onChunk?.(chunk);
|
|
182
195
|
},
|
|
183
196
|
onFinish({ text, finishReason, usage, response, steps, totalUsage }) {
|
|
184
|
-
onFinish
|
|
197
|
+
onFinish?.({ text, finishReason, usage, response, steps, totalUsage });
|
|
185
198
|
},
|
|
186
199
|
onStepFinish(step) {
|
|
187
|
-
onStepFinish
|
|
188
|
-
}
|
|
200
|
+
onStepFinish?.(step);
|
|
201
|
+
},
|
|
202
|
+
...(onAbort ? { onAbort } : {}),
|
|
189
203
|
});
|
|
190
204
|
}
|
|
191
205
|
catch (error) {
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"AIService.js","sourceRoot":"","sources":["../../../src/services/AIService.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,
|
|
1
|
+
{"version":3,"file":"AIService.js","sourceRoot":"","sources":["../../../src/services/AIService.ts"],"names":[],"mappings":"AAAA,OAAO,EAAC,YAAY,EAAE,MAAM,EAAC,MAAM,gBAAgB,CAAC;AACpD,OAAO,EAAC,SAAS,EAAE,eAAe,EAAC,MAAM,mBAAmB,CAAC;AAC7D,OAAO,EAAC,wBAAwB,EAAE,MAAM,EAAC,MAAM,gBAAgB,CAAC;AAChE,OAAO,EAAC,CAAC,EAAC,MAAM,KAAK,CAAC;AACtB,OAAO,EACL,YAAY,EAGZ,UAAU,EACV,IAAI,GAKL,MAAM,IAAI,CAAC;AACZ,OAAO,EAML,aAAa,EAGd,MAAM,iCAAiC,CAAC;AAEzC,OAAO,EAAC,gBAAgB,EAAC,MAAM,6BAA6B,CAAC;AAC7D,OAAO,EAAC,MAAM,EAAC,MAAM,gBAAgB,CAAC;AAkCtC,MAAM,OAAO,SAAS;IACZ,UAAU,CAAa;IACvB,MAAM,CAAe;IAC7B,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,YAAY,CAAC,EAAC,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,YAAY,EAAC,CAAC,CAAC,KAAK,CAAC,EAAE,CAAC,CAAA;YACnE,KAAK,WAAW;gBACd,OAAO,eAAe,CAAC,EAAC,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,eAAe,EAAC,CAAC,CAAC,KAAK,CAAC,EAAE,CAAC,CAAA;YACzE,KAAK,QAAQ;gBACX,OAAO,wBAAwB,CAAC,EAAC,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,YAAY,EAAC,CAAC,CAAC,KAAK,CAAC,EAAE,CAAC,CAAA;YAC/E,KAAK,YAAY;gBACf,OAAO,gBAAgB,CAAC;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,MAAM,CAAC,KAAK,CAAC,SAAS,EAAU,CAAA;YACzC,KAAK,WAAW;gBACd,OAAO,SAAS,CAAC,KAAK,CAAC,kBAAkB,EAAU,CAAA;YACrD,KAAK,QAAQ;gBACX,OAAO,MAAM,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,aAAa,CAAC,QAAQ,EAAE,CAAC;YAC/C,MAAM,IAAI,GAAG,UAAU,CAAC,IAAoB,CAAC;YAC7C,OAAO,IAAI,CAAuE;gBAChF,KAAK,EAAE,IAAI,CAAC,KAAK;gBACjB,WAAW,EAAE,IAAI,CAAC,WAAW;gBAC7B,aAAa,EAAE,IAAI,CAAC,aAAa;gBACjC,WAAW,EAAE,CAAC,CAAC,MAAM,CAAC;oBACpB,IAAI,EAAE,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,EAAE,CAAC,QAAQ,EAAE;oBAChD,WAAW,EAAE,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,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,aAAa,CAAC,UAAU,EAAE,CAAC;YACjD,OAAO,IAAI,CAAC,gBAAgB,CAAC,OAAO,CAAC,CAAC;QACxC,CAAC;QACD,IAAI,UAAU,CAAC,IAAI,KAAK,aAAa,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,IAAI,CAAC;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,IAAI,CAAC,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,aAAa,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,YAAY,CAAC;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,MAAM,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,UAAU,CAAC;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,MAAM,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"}
|
|
@@ -1,9 +1,11 @@
|
|
|
1
|
+
import { RedisService } from './RedisService';
|
|
1
2
|
/**
|
|
2
3
|
* Lightweight Redis pub/sub helper to share events across service replicas.
|
|
3
4
|
*/
|
|
4
5
|
export declare class InternalEventsHandler<T = unknown> {
|
|
5
6
|
private readonly channel;
|
|
6
|
-
|
|
7
|
+
private readonly redisService;
|
|
8
|
+
constructor(channel: string, redisService: RedisService);
|
|
7
9
|
private pubClient;
|
|
8
10
|
private subClient;
|
|
9
11
|
private initialized;
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"InternalEventsHandler.d.ts","sourceRoot":"","sources":["../../../src/services/InternalEventsHandler.ts"],"names":[],"mappings":"
|
|
1
|
+
{"version":3,"file":"InternalEventsHandler.d.ts","sourceRoot":"","sources":["../../../src/services/InternalEventsHandler.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAqB,MAAM,gBAAgB,CAAC;AAGjE;;GAEG;AACH,qBAAa,qBAAqB,CAAC,CAAC,GAAG,OAAO;IAE1C,OAAO,CAAC,QAAQ,CAAC,OAAO;IACxB,OAAO,CAAC,QAAQ,CAAC,YAAY;gBADZ,OAAO,EAAE,MAAM,EACf,YAAY,EAAE,YAAY;IAG7C,OAAO,CAAC,SAAS,CAAkC;IACnD,OAAO,CAAC,SAAS,CAAkC;IACnD,OAAO,CAAC,WAAW,CAAS;IAC5B,OAAO,CAAC,UAAU,CAAS;YAEb,aAAa;IAS3B;;OAEG;IACG,OAAO,CAAC,CAAC,EAAE,OAAO,EAAE,CAAC,GAAG,OAAO,CAAC,MAAM,CAAC;IAS7C;;OAEG;IACG,MAAM,CAAC,CAAC,EAAE,OAAO,EAAE,CAAC,OAAO,EAAE,CAAC,KAAK,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,GAAG,OAAO,CAAC,IAAI,CAAC;CAoB9E"}
|
|
@@ -1,20 +1,22 @@
|
|
|
1
|
-
import {
|
|
2
|
-
import { logger } from '../libs/logger/index.js';
|
|
1
|
+
import { logger } from '../libs/logger';
|
|
3
2
|
/**
|
|
4
3
|
* Lightweight Redis pub/sub helper to share events across service replicas.
|
|
5
4
|
*/
|
|
6
5
|
export class InternalEventsHandler {
|
|
7
|
-
|
|
6
|
+
channel;
|
|
7
|
+
redisService;
|
|
8
|
+
constructor(channel, redisService) {
|
|
8
9
|
this.channel = channel;
|
|
9
|
-
this.
|
|
10
|
-
this.subClient = null;
|
|
11
|
-
this.initialized = false;
|
|
12
|
-
this.subscribed = false;
|
|
10
|
+
this.redisService = redisService;
|
|
13
11
|
}
|
|
12
|
+
pubClient = null;
|
|
13
|
+
subClient = null;
|
|
14
|
+
initialized = false;
|
|
15
|
+
subscribed = false;
|
|
14
16
|
async ensureClients() {
|
|
15
17
|
if (this.initialized)
|
|
16
18
|
return;
|
|
17
|
-
const { pubClient, subClient } = await redisService.getPubSubClients(this.channel);
|
|
19
|
+
const { pubClient, subClient } = await this.redisService.getPubSubClients(this.channel);
|
|
18
20
|
this.pubClient = pubClient;
|
|
19
21
|
this.subClient = subClient;
|
|
20
22
|
this.initialized = true;
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"InternalEventsHandler.js","sourceRoot":"","sources":["../../../src/services/InternalEventsHandler.ts"],"names":[],"mappings":"
|
|
1
|
+
{"version":3,"file":"InternalEventsHandler.js","sourceRoot":"","sources":["../../../src/services/InternalEventsHandler.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,MAAM,EAAE,MAAM,gBAAgB,CAAC;AAExC;;GAEG;AACH,MAAM,OAAO,qBAAqB;IAEb;IACA;IAFnB,YACmB,OAAe,EACf,YAA0B;QAD1B,YAAO,GAAP,OAAO,CAAQ;QACf,iBAAY,GAAZ,YAAY,CAAc;IAC1C,CAAC;IAEI,SAAS,GAA6B,IAAI,CAAC;IAC3C,SAAS,GAA6B,IAAI,CAAC;IAC3C,WAAW,GAAG,KAAK,CAAC;IACpB,UAAU,GAAG,KAAK,CAAC;IAEnB,KAAK,CAAC,aAAa;QACzB,IAAI,IAAI,CAAC,WAAW;YAAE,OAAO;QAE7B,MAAM,EAAE,SAAS,EAAE,SAAS,EAAE,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,gBAAgB,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACxF,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC3B,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC3B,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC;IAC1B,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,OAAO,CAAI,OAAU;QACzB,MAAM,IAAI,CAAC,aAAa,EAAE,CAAC;QAC3B,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;YACpB,MAAM,IAAI,KAAK,CAAC,gCAAgC,CAAC,CAAC;QACpD,CAAC;QACD,MAAM,OAAO,GAAG,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;QACxC,OAAO,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;IACvD,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,MAAM,CAAI,OAA6C;QAC3D,MAAM,IAAI,CAAC,aAAa,EAAE,CAAC;QAC3B,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;YACpB,MAAM,IAAI,KAAK,CAAC,gCAAgC,CAAC,CAAC;QACpD,CAAC;QAED,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;YACpB,OAAO;QACT,CAAC;QACD,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC;QAEvB,MAAM,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC,IAAI,CAAC,OAAO,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE;YAC7D,IAAI,CAAC;gBACH,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAM,CAAC;gBACxC,MAAM,OAAO,CAAC,MAAM,CAAC,CAAC;YACxB,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,MAAM,CAAC,KAAK,CAAC,sDAAsD,IAAI,CAAC,OAAO,GAAG,EAAE,KAAK,CAAC,CAAC;YAC7F,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC;CACF"}
|