@multiplayer-app/ai-agent-node 0.1.0-beta.5 → 0.1.0-beta.50
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/README.md +2 -2
- package/dist/cjs/config.cjs +88 -37
- package/dist/cjs/config.cjs.map +1 -1
- package/dist/cjs/config.d.ts +57 -23
- package/dist/cjs/config.d.ts.map +1 -1
- package/dist/cjs/helpers/AIHelper.cjs +127 -65
- package/dist/cjs/helpers/AIHelper.cjs.map +1 -1
- package/dist/cjs/helpers/AIHelper.d.ts +22 -16
- package/dist/cjs/helpers/AIHelper.d.ts.map +1 -1
- package/dist/cjs/helpers/AIHelper.test.cjs +22 -15
- package/dist/cjs/helpers/AIHelper.test.cjs.map +1 -1
- package/dist/cjs/helpers/ConfigHelper.cjs +15 -6
- package/dist/cjs/helpers/ConfigHelper.cjs.map +1 -1
- package/dist/cjs/helpers/ConfigHelper.d.ts.map +1 -1
- package/dist/cjs/helpers/FileHelper.cjs +131 -151
- package/dist/cjs/helpers/FileHelper.cjs.map +1 -1
- package/dist/cjs/helpers/FileHelper.d.ts +19 -25
- package/dist/cjs/helpers/FileHelper.d.ts.map +1 -1
- package/dist/cjs/helpers/index.cjs +0 -1
- package/dist/cjs/helpers/index.cjs.map +1 -1
- package/dist/cjs/helpers/index.d.ts +0 -1
- package/dist/cjs/helpers/index.d.ts.map +1 -1
- package/dist/cjs/index.cjs +120 -28
- package/dist/cjs/index.cjs.map +1 -1
- package/dist/cjs/index.d.ts +43 -11
- package/dist/cjs/index.d.ts.map +1 -1
- package/dist/cjs/libs/index.cjs +0 -1
- package/dist/cjs/libs/index.cjs.map +1 -1
- package/dist/cjs/libs/index.d.ts +0 -1
- package/dist/cjs/libs/index.d.ts.map +1 -1
- package/dist/cjs/libs/s3/index.cjs +3 -39
- package/dist/cjs/libs/s3/index.cjs.map +1 -1
- 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 +173 -186
- package/dist/cjs/libs/s3/s3.lib.cjs.map +1 -1
- 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/ActivityProcessor.cjs +39 -0
- package/dist/cjs/processors/ActivityProcessor.cjs.map +1 -0
- package/dist/cjs/processors/ActivityProcessor.d.ts +32 -0
- package/dist/cjs/processors/ActivityProcessor.d.ts.map +1 -0
- package/dist/cjs/processors/ActivityProcessor.test.cjs +84 -0
- package/dist/cjs/processors/ActivityProcessor.test.cjs.map +1 -0
- package/dist/cjs/processors/ActivityProcessor.test.d.ts +2 -0
- package/dist/cjs/processors/ActivityProcessor.test.d.ts.map +1 -0
- package/dist/cjs/processors/AgentProcessor.cjs +46 -0
- package/dist/cjs/processors/AgentProcessor.cjs.map +1 -0
- package/dist/cjs/processors/AgentProcessor.d.ts +25 -0
- package/dist/cjs/processors/AgentProcessor.d.ts.map +1 -0
- package/dist/cjs/processors/AgentProcessor.test.cjs +103 -0
- package/dist/cjs/processors/AgentProcessor.test.cjs.map +1 -0
- package/dist/cjs/processors/AgentProcessor.test.d.ts +2 -0
- package/dist/cjs/processors/AgentProcessor.test.d.ts.map +1 -0
- package/dist/cjs/processors/ChatProcessor.cjs +355 -122
- package/dist/cjs/processors/ChatProcessor.cjs.map +1 -1
- package/dist/cjs/processors/ChatProcessor.d.ts +71 -11
- package/dist/cjs/processors/ChatProcessor.d.ts.map +1 -1
- package/dist/cjs/processors/ChatProcessor.test.cjs +762 -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/index.cjs +2 -0
- package/dist/cjs/processors/index.cjs.map +1 -1
- package/dist/cjs/processors/index.d.ts +2 -0
- package/dist/cjs/processors/index.d.ts.map +1 -1
- package/dist/cjs/services/AIService.cjs +87 -21
- package/dist/cjs/services/AIService.cjs.map +1 -1
- package/dist/cjs/services/AIService.d.ts +19 -7
- package/dist/cjs/services/AIService.d.ts.map +1 -1
- package/dist/cjs/services/InternalEventsHandler.cjs +3 -3
- package/dist/cjs/services/InternalEventsHandler.cjs.map +1 -1
- package/dist/cjs/services/InternalEventsHandler.d.ts +3 -1
- package/dist/cjs/services/InternalEventsHandler.d.ts.map +1 -1
- package/dist/cjs/services/ModelFetcher.cjs +2 -8
- package/dist/cjs/services/ModelFetcher.cjs.map +1 -1
- package/dist/cjs/services/ModelFetcher.d.ts +2 -7
- package/dist/cjs/services/ModelFetcher.d.ts.map +1 -1
- package/dist/cjs/services/RedisService.cjs +20 -16
- package/dist/cjs/services/RedisService.cjs.map +1 -1
- package/dist/cjs/services/RedisService.d.ts +5 -2
- package/dist/cjs/services/RedisService.d.ts.map +1 -1
- package/dist/cjs/services/SocketService.cjs +8 -8
- package/dist/cjs/services/SocketService.cjs.map +1 -1
- package/dist/cjs/services/SocketService.d.ts +9 -6
- package/dist/cjs/services/SocketService.d.ts.map +1 -1
- package/dist/cjs/services/index.cjs +0 -1
- package/dist/cjs/services/index.cjs.map +1 -1
- package/dist/cjs/services/index.d.ts +0 -1
- package/dist/cjs/services/index.d.ts.map +1 -1
- package/dist/cjs/store/AgentStore.cjs +3 -4
- package/dist/cjs/store/AgentStore.cjs.map +1 -1
- package/dist/cjs/store/AgentStore.d.ts +2 -1
- package/dist/cjs/store/AgentStore.d.ts.map +1 -1
- package/dist/cjs/store/ConfigStore.cjs +7 -3
- package/dist/cjs/store/ConfigStore.cjs.map +1 -1
- package/dist/cjs/store/ConfigStore.d.ts +1 -0
- package/dist/cjs/store/ConfigStore.d.ts.map +1 -1
- package/dist/cjs/tools/generateChartTool.d.ts +2 -2
- package/dist/cjs/tools/proposeFormValuesTool.d.ts +2 -2
- package/dist/cjs/tsconfig.cjs.tsbuildinfo +1 -1
- package/dist/esm/config.d.ts +57 -23
- package/dist/esm/config.d.ts.map +1 -1
- package/dist/esm/config.js +88 -35
- package/dist/esm/config.js.map +1 -1
- package/dist/esm/helpers/AIHelper.d.ts +22 -16
- package/dist/esm/helpers/AIHelper.d.ts.map +1 -1
- package/dist/esm/helpers/AIHelper.js +134 -70
- package/dist/esm/helpers/AIHelper.js.map +1 -1
- package/dist/esm/helpers/AIHelper.test.js +22 -15
- 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 +15 -6
- package/dist/esm/helpers/ConfigHelper.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 +131 -146
- package/dist/esm/helpers/FileHelper.js.map +1 -1
- package/dist/esm/helpers/index.d.ts +0 -1
- package/dist/esm/helpers/index.d.ts.map +1 -1
- package/dist/esm/helpers/index.js +0 -1
- package/dist/esm/helpers/index.js.map +1 -1
- package/dist/esm/index.d.ts +43 -11
- package/dist/esm/index.d.ts.map +1 -1
- package/dist/esm/index.js +92 -11
- package/dist/esm/index.js.map +1 -1
- package/dist/esm/libs/index.d.ts +0 -1
- package/dist/esm/libs/index.d.ts.map +1 -1
- package/dist/esm/libs/index.js +0 -1
- package/dist/esm/libs/index.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 +177 -172
- package/dist/esm/libs/s3/s3.lib.js.map +1 -1
- package/dist/esm/processors/ActivityProcessor.d.ts +32 -0
- package/dist/esm/processors/ActivityProcessor.d.ts.map +1 -0
- package/dist/esm/processors/ActivityProcessor.js +36 -0
- package/dist/esm/processors/ActivityProcessor.js.map +1 -0
- package/dist/esm/processors/ActivityProcessor.test.d.ts +2 -0
- package/dist/esm/processors/ActivityProcessor.test.d.ts.map +1 -0
- package/dist/esm/processors/ActivityProcessor.test.js +82 -0
- package/dist/esm/processors/ActivityProcessor.test.js.map +1 -0
- package/dist/esm/processors/AgentProcessor.d.ts +25 -0
- package/dist/esm/processors/AgentProcessor.d.ts.map +1 -0
- package/dist/esm/processors/AgentProcessor.js +43 -0
- package/dist/esm/processors/AgentProcessor.js.map +1 -0
- package/dist/esm/processors/AgentProcessor.test.d.ts +2 -0
- package/dist/esm/processors/AgentProcessor.test.d.ts.map +1 -0
- package/dist/esm/processors/AgentProcessor.test.js +101 -0
- package/dist/esm/processors/AgentProcessor.test.js.map +1 -0
- package/dist/esm/processors/ChatProcessor.d.ts +71 -11
- package/dist/esm/processors/ChatProcessor.d.ts.map +1 -1
- package/dist/esm/processors/ChatProcessor.js +366 -126
- 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 +760 -0
- package/dist/esm/processors/ChatProcessor.test.js.map +1 -0
- package/dist/esm/processors/index.d.ts +2 -0
- package/dist/esm/processors/index.d.ts.map +1 -1
- package/dist/esm/processors/index.js +2 -0
- package/dist/esm/processors/index.js.map +1 -1
- package/dist/esm/services/AIService.d.ts +19 -7
- package/dist/esm/services/AIService.d.ts.map +1 -1
- package/dist/esm/services/AIService.js +91 -24
- 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 +4 -3
- 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 +2 -8
- 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 +21 -14
- package/dist/esm/services/RedisService.js.map +1 -1
- package/dist/esm/services/SocketService.d.ts +9 -6
- package/dist/esm/services/SocketService.d.ts.map +1 -1
- package/dist/esm/services/SocketService.js +10 -6
- package/dist/esm/services/SocketService.js.map +1 -1
- package/dist/esm/services/index.d.ts +0 -1
- package/dist/esm/services/index.d.ts.map +1 -1
- package/dist/esm/services/index.js +0 -1
- 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 +4 -2
- package/dist/esm/store/AgentStore.js.map +1 -1
- package/dist/esm/store/ConfigStore.d.ts +1 -0
- package/dist/esm/store/ConfigStore.d.ts.map +1 -1
- package/dist/esm/store/ConfigStore.js +7 -3
- package/dist/esm/store/ConfigStore.js.map +1 -1
- package/dist/esm/tools/generateChartTool.d.ts +2 -2
- package/dist/esm/tools/proposeFormValuesTool.d.ts +2 -2
- package/dist/esm/tsconfig.esm.tsbuildinfo +1 -1
- package/package.json +12 -12
- package/dist/cjs/helpers/SetupHelper.cjs +0 -37
- package/dist/cjs/helpers/SetupHelper.cjs.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/libs/kafka/config.cjs +0 -8
- package/dist/cjs/libs/kafka/config.cjs.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/consumer.cjs +0 -131
- package/dist/cjs/libs/kafka/consumer.cjs.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/index.cjs +0 -19
- package/dist/cjs/libs/kafka/index.cjs.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/kafka.cjs +0 -27
- package/dist/cjs/libs/kafka/kafka.cjs.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/producer.cjs +0 -48
- package/dist/cjs/libs/kafka/producer.cjs.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/logger/config.cjs +0 -8
- package/dist/cjs/libs/logger/config.cjs.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/s3/config.cjs +0 -10
- package/dist/cjs/libs/s3/config.cjs.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/services/KafkaService.cjs +0 -122
- package/dist/cjs/services/KafkaService.cjs.map +0 -1
- package/dist/cjs/services/KafkaService.d.ts +0 -35
- package/dist/cjs/services/KafkaService.d.ts.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 -125
- 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 -45
- 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 -5
- 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 -123
- package/dist/esm/services/KafkaService.js.map +0 -1
|
@@ -1,35 +0,0 @@
|
|
|
1
|
-
import type { AgentChatRepository, AgentMessageRepository } from '@multiplayer-app/ai-agent-db';
|
|
2
|
-
declare enum BackgroundChatProcessingEvent {
|
|
3
|
-
Start = "start"
|
|
4
|
-
}
|
|
5
|
-
declare class KafkaService {
|
|
6
|
-
private kafkaConsumer;
|
|
7
|
-
private kafkaProducer;
|
|
8
|
-
private chatRepository?;
|
|
9
|
-
private messageRepository?;
|
|
10
|
-
private readonly MAX_RETRIES;
|
|
11
|
-
private readonly BASE_DELAY_MS;
|
|
12
|
-
private readonly MAX_DELAY_MS;
|
|
13
|
-
constructor(chatRepository?: AgentChatRepository, messageRepository?: AgentMessageRepository);
|
|
14
|
-
setRepositories(chatRepository: AgentChatRepository, messageRepository: AgentMessageRepository): void;
|
|
15
|
-
start(): Promise<void>;
|
|
16
|
-
stop(): Promise<void>;
|
|
17
|
-
sendStartBackgroundAgentEvent(chatId: string): Promise<void>;
|
|
18
|
-
sendChatTitleGenerationEvent(chatId: string, retryCount?: number): Promise<void>;
|
|
19
|
-
private sleep;
|
|
20
|
-
private calculateBackoffDelay;
|
|
21
|
-
onBackgroundAgentEvent(key: any, message: {
|
|
22
|
-
event: BackgroundChatProcessingEvent;
|
|
23
|
-
chatId: string;
|
|
24
|
-
}): Promise<void>;
|
|
25
|
-
onChatTitleGenerationEvent(key: any, message: {
|
|
26
|
-
chatId: string;
|
|
27
|
-
retryCount?: number;
|
|
28
|
-
}): Promise<void>;
|
|
29
|
-
onStartBackgroundAgentEvent(chatId: string): Promise<void>;
|
|
30
|
-
connect(): Promise<void>;
|
|
31
|
-
disconnect(): Promise<void>;
|
|
32
|
-
}
|
|
33
|
-
export declare const kafkaService: KafkaService;
|
|
34
|
-
export {};
|
|
35
|
-
//# sourceMappingURL=KafkaService.d.ts.map
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"KafkaService.d.ts","sourceRoot":"","sources":["../../../src/services/KafkaService.ts"],"names":[],"mappings":"AAMA,OAAO,KAAK,EAAE,mBAAmB,EAAE,sBAAsB,EAAE,MAAM,8BAA8B,CAAC;AAEhG,aAAK,6BAA6B;IAChC,KAAK,UAAU;CAChB;AAED,cAAM,YAAY;IAChB,OAAO,CAAC,aAAa,CAAe;IACpC,OAAO,CAAC,aAAa,CAAe;IACpC,OAAO,CAAC,cAAc,CAAC,CAAqB;IAC5C,OAAO,CAAC,iBAAiB,CAAC,CAAwB;IAElD,OAAO,CAAC,QAAQ,CAAC,WAAW,CAAI;IAChC,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAO;IACrC,OAAO,CAAC,QAAQ,CAAC,YAAY,CAAQ;gBAEzB,cAAc,CAAC,EAAE,mBAAmB,EAAE,iBAAiB,CAAC,EAAE,sBAAsB;IAO5F,eAAe,CAAC,cAAc,EAAE,mBAAmB,EAAE,iBAAiB,EAAE,sBAAsB;IAKxF,KAAK;IAIL,IAAI;IAIJ,6BAA6B,CAAC,MAAM,EAAE,MAAM;IAI5C,4BAA4B,CAAC,MAAM,EAAE,MAAM,EAAE,UAAU,GAAE,MAAU;YAI3D,KAAK;IAInB,OAAO,CAAC,qBAAqB;IAUvB,sBAAsB,CAAC,GAAG,KAAA,EAAE,OAAO,EAAE;QAAE,KAAK,EAAE,6BAA6B,CAAC;QAAC,MAAM,EAAE,MAAM,CAAA;KAAE;IAQ7F,0BAA0B,CAAC,GAAG,KAAA,EAAE,OAAO,EAAE;QAAE,MAAM,EAAE,MAAM,CAAC;QAAC,UAAU,CAAC,EAAE,MAAM,CAAA;KAAE;IAuChF,2BAA2B,CAAC,MAAM,EAAE,MAAM;IAkB1C,OAAO;IAOP,UAAU;CAGjB;AACD,eAAO,MAAM,YAAY,cAAqB,CAAA"}
|
|
@@ -1,5 +0,0 @@
|
|
|
1
|
-
import type { AgentChatRepository, AgentMessageRepository } from '@multiplayer-app/ai-agent-db';
|
|
2
|
-
export declare function fetchModels(): Promise<void>;
|
|
3
|
-
export declare function startServices(agentChatRepository: AgentChatRepository, agentMessageRepository: AgentMessageRepository): Promise<[void, void, void, void]>;
|
|
4
|
-
export declare function stopServices(): Promise<[void, void]>;
|
|
5
|
-
//# sourceMappingURL=SetupHelper.d.ts.map
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"SetupHelper.d.ts","sourceRoot":"","sources":["../../../src/helpers/SetupHelper.ts"],"names":[],"mappings":"AAKA,OAAO,KAAK,EAAE,mBAAmB,EAAE,sBAAsB,EAAE,MAAM,8BAA8B,CAAC;AAEhG,wBAAsB,WAAW,kBAShC;AAED,wBAAgB,aAAa,CAAC,mBAAmB,EAAE,mBAAmB,EAAE,sBAAsB,EAAE,sBAAsB,qCASrH;AAED,wBAAgB,YAAY,0BAK3B"}
|
|
@@ -1,32 +0,0 @@
|
|
|
1
|
-
import { logger } from '../libs/logger';
|
|
2
|
-
import { agentStore, ModelStore } from '../store';
|
|
3
|
-
import { redisService, kafkaService, ModelFetcher } from '../services';
|
|
4
|
-
import { s3 } from "../libs/s3";
|
|
5
|
-
import { config } from "../config";
|
|
6
|
-
export async function fetchModels() {
|
|
7
|
-
try {
|
|
8
|
-
const modelFetcher = new ModelFetcher();
|
|
9
|
-
const models = await modelFetcher.fetchAllModels();
|
|
10
|
-
ModelStore.getInstance().setModels(models);
|
|
11
|
-
}
|
|
12
|
-
catch (error) {
|
|
13
|
-
logger.error('Failed to fetch models on startup:', error);
|
|
14
|
-
throw error;
|
|
15
|
-
}
|
|
16
|
-
}
|
|
17
|
-
export function startServices(agentChatRepository, agentMessageRepository) {
|
|
18
|
-
kafkaService.setRepositories(agentChatRepository, agentMessageRepository);
|
|
19
|
-
return Promise.all([
|
|
20
|
-
kafkaService.start(),
|
|
21
|
-
agentStore.initialize(),
|
|
22
|
-
fetchModels(),
|
|
23
|
-
s3.ensureBucketExists(config.s3.bucket),
|
|
24
|
-
]);
|
|
25
|
-
}
|
|
26
|
-
export function stopServices() {
|
|
27
|
-
return Promise.all([
|
|
28
|
-
redisService.disconnect(),
|
|
29
|
-
kafkaService.stop()
|
|
30
|
-
]);
|
|
31
|
-
}
|
|
32
|
-
//# sourceMappingURL=SetupHelper.js.map
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"SetupHelper.js","sourceRoot":"","sources":["../../../src/helpers/SetupHelper.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,gBAAgB,CAAC;AACxC,OAAO,EAAE,UAAU,EAAE,UAAU,EAAE,MAAM,UAAU,CAAC;AAClD,OAAO,EAAE,YAAY,EAAE,YAAY,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AACvE,OAAO,EAAE,EAAE,EAAE,MAAM,YAAY,CAAC;AAChC,OAAO,EAAE,MAAM,EAAE,MAAM,WAAW,CAAC;AAGnC,MAAM,CAAC,KAAK,UAAU,WAAW;IAC7B,IAAI,CAAC;QACH,MAAM,YAAY,GAAG,IAAI,YAAY,EAAE,CAAC;QACxC,MAAM,MAAM,GAAG,MAAM,YAAY,CAAC,cAAc,EAAE,CAAC;QACnD,UAAU,CAAC,WAAW,EAAE,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;IAC7C,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,CAAC,KAAK,CAAC,oCAAoC,EAAE,KAAK,CAAC,CAAC;QAC1D,MAAM,KAAK,CAAC;IACd,CAAC;AACL,CAAC;AAED,MAAM,UAAU,aAAa,CAAC,mBAAwC,EAAE,sBAA8C;IACpH,YAAY,CAAC,eAAe,CAAC,mBAAmB,EAAE,sBAAsB,CAAC,CAAC;IAE1E,OAAO,OAAO,CAAC,GAAG,CAAC;QACjB,YAAY,CAAC,KAAK,EAAE;QACpB,UAAU,CAAC,UAAU,EAAE;QACvB,WAAW,EAAE;QACb,EAAE,CAAC,kBAAkB,CAAC,MAAM,CAAC,EAAE,CAAC,MAAM,CAAC;KACxC,CAAC,CAAA;AACJ,CAAC;AAED,MAAM,UAAU,YAAY;IAC1B,OAAO,OAAO,CAAC,GAAG,CAAC;QACjB,YAAY,CAAC,UAAU,EAAE;QACzB,YAAY,CAAC,IAAI,EAAE;KACpB,CAAC,CAAA;AACJ,CAAC"}
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"config.d.ts","sourceRoot":"","sources":["../../../../src/libs/kafka/config.ts"],"names":[],"mappings":"AAAA,eAAO,MAAM,YAAY,oBAA+B,CAAA;AACxD,eAAO,MAAM,SAAS,UAAmE,CAAA;AACzF,eAAO,MAAM,eAAe,oBAAwD,CAAA;AACpF,eAAO,MAAM,qBAAqB,QAAgE,CAAA"}
|
|
@@ -1,5 +0,0 @@
|
|
|
1
|
-
export const SERVICE_NAME = process.env.npm_package_name;
|
|
2
|
-
export const KAFKA_URI = (process.env.KAFKA_URI || 'localhost:9092').split(',');
|
|
3
|
-
export const KAFKA_CLIENT_ID = process.env.KAFKA_CLIENT_ID || SERVICE_NAME;
|
|
4
|
-
export const KAFKA_SESSION_TIMEOUT = Number.parseInt(process.env.KAFKA_SESSION_TIMEOUT || '30000');
|
|
5
|
-
//# sourceMappingURL=config.js.map
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"config.js","sourceRoot":"","sources":["../../../../src/libs/kafka/config.ts"],"names":[],"mappings":"AAAA,MAAM,CAAC,MAAM,YAAY,GAAG,OAAO,CAAC,GAAG,CAAC,gBAAgB,CAAA;AACxD,MAAM,CAAC,MAAM,SAAS,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,SAAmB,IAAI,gBAAgB,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC,CAAA;AACzF,MAAM,CAAC,MAAM,eAAe,GAAG,OAAO,CAAC,GAAG,CAAC,eAAyB,IAAI,YAAY,CAAA;AACpF,MAAM,CAAC,MAAM,qBAAqB,GAAG,MAAM,CAAC,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,qBAAqB,IAAI,OAAO,CAAC,CAAA"}
|
|
@@ -1,16 +0,0 @@
|
|
|
1
|
-
import { ConsumerRunConfig, IHeaders } from 'kafkajs';
|
|
2
|
-
export type KafkaConsumerListener = (key: any, value: any, headers?: IHeaders) => void | Promise<void>;
|
|
3
|
-
export declare class KafkaConsumer {
|
|
4
|
-
private consumer;
|
|
5
|
-
private listeners;
|
|
6
|
-
private lastHeartbeat;
|
|
7
|
-
constructor(groupId: string);
|
|
8
|
-
isConnected(): Promise<boolean>;
|
|
9
|
-
connect(): Promise<void>;
|
|
10
|
-
disconnect(): Promise<void>;
|
|
11
|
-
subscribe(topic: string, listener: KafkaConsumerListener): Promise<void>;
|
|
12
|
-
private listenFnWrapper;
|
|
13
|
-
listen(options?: Omit<ConsumerRunConfig, 'eachMessage' | 'eachBatch'>): Promise<void>;
|
|
14
|
-
listenBatch(options: Omit<ConsumerRunConfig, 'eachMessage' | 'eachBatch'>): Promise<void>;
|
|
15
|
-
}
|
|
16
|
-
//# sourceMappingURL=consumer.d.ts.map
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"consumer.d.ts","sourceRoot":"","sources":["../../../../src/libs/kafka/consumer.ts"],"names":[],"mappings":"AAAA,OAAO,EAAY,iBAAiB,EAAE,QAAQ,EAAW,MAAM,SAAS,CAAA;AAcxE,MAAM,MAAM,qBAAqB,GAAG,CAAC,GAAG,KAAA,EAAE,KAAK,KAAA,EAAE,OAAO,CAAC,EAAE,QAAQ,KAAK,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,CAAA;AAC5F,qBAAa,aAAa;IACxB,OAAO,CAAC,QAAQ,CAAU;IAC1B,OAAO,CAAC,SAAS,CAA8C;IAC/D,OAAO,CAAC,aAAa,CAAY;gBAErB,OAAO,EAAE,MAAM;IAgBd,WAAW;IAYX,OAAO;IAIP,UAAU;IAIV,SAAS,CAAC,KAAK,EAAE,MAAM,EAAE,QAAQ,EAAE,qBAAqB;YASvD,eAAe;IAwChB,MAAM,CAAC,OAAO,GAAE,IAAI,CAAC,iBAAiB,EAAE,aAAa,GAAG,WAAW,CAE/E;IAeY,WAAW,CAAC,OAAO,EAAE,IAAI,CAAC,iBAAiB,EAAE,aAAa,GAAG,WAAW,CAAC;CAsCvF"}
|
|
@@ -1,125 +0,0 @@
|
|
|
1
|
-
import logger from '../logger';
|
|
2
|
-
import { kafka } from './kafka';
|
|
3
|
-
import { KAFKA_SESSION_TIMEOUT } from './config';
|
|
4
|
-
const startTimer = () => {
|
|
5
|
-
const startTime = process.hrtime();
|
|
6
|
-
return startTime;
|
|
7
|
-
};
|
|
8
|
-
const getDuration = (startTime) => {
|
|
9
|
-
const diff = process.hrtime(startTime);
|
|
10
|
-
return diff[0] * 1e3 + diff[1] * 1e-6;
|
|
11
|
-
};
|
|
12
|
-
export class KafkaConsumer {
|
|
13
|
-
consumer;
|
|
14
|
-
listeners = {};
|
|
15
|
-
lastHeartbeat = 0;
|
|
16
|
-
constructor(groupId) {
|
|
17
|
-
this.consumer = kafka.consumer({
|
|
18
|
-
groupId,
|
|
19
|
-
sessionTimeout: KAFKA_SESSION_TIMEOUT,
|
|
20
|
-
});
|
|
21
|
-
this.consumer.on(this.consumer.events.HEARTBEAT, ({ timestamp }) => {
|
|
22
|
-
this.lastHeartbeat = timestamp;
|
|
23
|
-
});
|
|
24
|
-
this.consumer.on(this.consumer.events.REQUEST_TIMEOUT, ({ timestamp }) => {
|
|
25
|
-
this.lastHeartbeat = 0;
|
|
26
|
-
});
|
|
27
|
-
this.consumer.on(this.consumer.events.CRASH, ({ timestamp }) => {
|
|
28
|
-
this.lastHeartbeat = 0;
|
|
29
|
-
});
|
|
30
|
-
}
|
|
31
|
-
async isConnected() {
|
|
32
|
-
if (Date.now() - this.lastHeartbeat < KAFKA_SESSION_TIMEOUT) {
|
|
33
|
-
return true;
|
|
34
|
-
}
|
|
35
|
-
try {
|
|
36
|
-
const { state } = await this.consumer.describeGroup();
|
|
37
|
-
return ['CompletingRebalance', 'PreparingRebalance', 'Stable'].includes(state);
|
|
38
|
-
}
|
|
39
|
-
catch (err) {
|
|
40
|
-
return false;
|
|
41
|
-
}
|
|
42
|
-
}
|
|
43
|
-
async connect() {
|
|
44
|
-
await this.consumer.connect();
|
|
45
|
-
}
|
|
46
|
-
async disconnect() {
|
|
47
|
-
await this.consumer.disconnect();
|
|
48
|
-
}
|
|
49
|
-
async subscribe(topic, listener) {
|
|
50
|
-
logger.info({ topic }, '[KAFKAJS] Subscribing for topic');
|
|
51
|
-
if (!this.listeners[topic]) {
|
|
52
|
-
this.listeners[topic] = [];
|
|
53
|
-
}
|
|
54
|
-
this.listeners[topic].push(listener);
|
|
55
|
-
}
|
|
56
|
-
async listenFnWrapper(topic, message) {
|
|
57
|
-
const parseStartTime = startTimer();
|
|
58
|
-
try {
|
|
59
|
-
if (!this.listeners[topic]?.length) {
|
|
60
|
-
logger.error(`[KAFKAJS] Missing handler for topic ${topic}`);
|
|
61
|
-
}
|
|
62
|
-
if (!message?.value) {
|
|
63
|
-
return;
|
|
64
|
-
}
|
|
65
|
-
const key = message?.key?.toString('utf8');
|
|
66
|
-
const value = JSON.parse(message.value.toString('utf8'));
|
|
67
|
-
const headers = message.headers;
|
|
68
|
-
logger.debug({
|
|
69
|
-
topic,
|
|
70
|
-
key,
|
|
71
|
-
value,
|
|
72
|
-
}, `[KAFKAJS] Received message from topic ${topic}`);
|
|
73
|
-
await Promise.all(this.listeners[topic].map(func => func(key, value, headers)));
|
|
74
|
-
logger.debug({
|
|
75
|
-
key,
|
|
76
|
-
topic,
|
|
77
|
-
duration: getDuration(parseStartTime),
|
|
78
|
-
}, '[KAFKAJS] Done with processing data from kafka');
|
|
79
|
-
}
|
|
80
|
-
catch (err) {
|
|
81
|
-
logger.error(err, '[KAFKAJS] Error on handling incoming message');
|
|
82
|
-
}
|
|
83
|
-
}
|
|
84
|
-
async listen(options = {
|
|
85
|
-
partitionsConsumedConcurrently: 3, autoCommit: true,
|
|
86
|
-
}) {
|
|
87
|
-
const topics = Object.keys(this.listeners);
|
|
88
|
-
await this.consumer.subscribe({
|
|
89
|
-
topics,
|
|
90
|
-
fromBeginning: true,
|
|
91
|
-
});
|
|
92
|
-
await this.consumer.run({
|
|
93
|
-
autoCommit: true,
|
|
94
|
-
...options,
|
|
95
|
-
eachMessage: async ({ topic, message }) => this.listenFnWrapper(topic, message),
|
|
96
|
-
});
|
|
97
|
-
}
|
|
98
|
-
async listenBatch(options) {
|
|
99
|
-
const topics = Object.keys(this.listeners);
|
|
100
|
-
await this.consumer.subscribe({
|
|
101
|
-
topics,
|
|
102
|
-
fromBeginning: true,
|
|
103
|
-
});
|
|
104
|
-
await this.consumer.run({
|
|
105
|
-
eachBatchAutoResolve: true,
|
|
106
|
-
...options,
|
|
107
|
-
eachBatch: async ({ batch, resolveOffset, heartbeat, commitOffsetsIfNecessary, uncommittedOffsets, isRunning, isStale, pause, }) => {
|
|
108
|
-
for (const message of batch.messages) {
|
|
109
|
-
if (!this.listeners[batch.topic]?.length) {
|
|
110
|
-
logger.error(`[KAFKAJS] Missing handler for topic ${batch.topic}`);
|
|
111
|
-
}
|
|
112
|
-
try {
|
|
113
|
-
await Promise.all(batch.messages.map(message => this.listenFnWrapper(batch.topic, message)));
|
|
114
|
-
}
|
|
115
|
-
catch (batchError) {
|
|
116
|
-
logger.error(batchError, '[KAFKAJS] Batch fn error');
|
|
117
|
-
}
|
|
118
|
-
resolveOffset(message.offset);
|
|
119
|
-
await heartbeat();
|
|
120
|
-
}
|
|
121
|
-
},
|
|
122
|
-
});
|
|
123
|
-
}
|
|
124
|
-
}
|
|
125
|
-
//# sourceMappingURL=consumer.js.map
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"consumer.js","sourceRoot":"","sources":["../../../../src/libs/kafka/consumer.ts"],"names":[],"mappings":"AACA,OAAO,MAAM,MAAM,WAAW,CAAA;AAC9B,OAAO,EAAE,KAAK,EAAE,MAAM,SAAS,CAAA;AAC/B,OAAO,EAAE,qBAAqB,EAAE,MAAM,UAAU,CAAA;AAEhD,MAAM,UAAU,GAAG,GAAG,EAAE;IACtB,MAAM,SAAS,GAAG,OAAO,CAAC,MAAM,EAAE,CAAA;IAClC,OAAO,SAAS,CAAA;AAClB,CAAC,CAAA;AACD,MAAM,WAAW,GAAG,CAAC,SAAS,EAAE,EAAE;IAChC,MAAM,IAAI,GAAG,OAAO,CAAC,MAAM,CAAC,SAAS,CAAC,CAAA;IACtC,OAAO,IAAI,CAAC,CAAC,CAAC,GAAG,GAAG,GAAG,IAAI,CAAC,CAAC,CAAC,GAAG,IAAI,CAAA;AACvC,CAAC,CAAA;AAGD,MAAM,OAAO,aAAa;IAChB,QAAQ,CAAU;IAClB,SAAS,GAA4C,EAAE,CAAA;IACvD,aAAa,GAAW,CAAC,CAAA;IAEjC,YAAY,OAAe;QACzB,IAAI,CAAC,QAAQ,GAAG,KAAK,CAAC,QAAQ,CAAC;YAC7B,OAAO;YACP,cAAc,EAAE,qBAAqB;SACtC,CAAC,CAAA;QACF,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,SAAS,EAAE,CAAC,EAAE,SAAS,EAAE,EAAE,EAAE;YACjE,IAAI,CAAC,aAAa,GAAG,SAAS,CAAA;QAChC,CAAC,CAAC,CAAA;QACF,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,eAAe,EAAE,CAAC,EAAE,SAAS,EAAE,EAAE,EAAE;YACvE,IAAI,CAAC,aAAa,GAAG,CAAC,CAAA;QACxB,CAAC,CAAC,CAAA;QACF,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC,EAAE,SAAS,EAAE,EAAE,EAAE;YAC7D,IAAI,CAAC,aAAa,GAAG,CAAC,CAAA;QACxB,CAAC,CAAC,CAAA;IACJ,CAAC;IAEM,KAAK,CAAC,WAAW;QACtB,IAAI,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,aAAa,GAAG,qBAAqB,EAAE,CAAC;YAC5D,OAAO,IAAI,CAAA;QACb,CAAC;QACD,IAAI,CAAC;YACH,MAAM,EAAE,KAAK,EAAE,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,aAAa,EAAE,CAAA;YACrD,OAAO,CAAC,qBAAqB,EAAE,oBAAoB,EAAE,QAAQ,CAAC,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAA;QAChF,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,OAAO,KAAK,CAAA;QACd,CAAC;IACH,CAAC;IAEM,KAAK,CAAC,OAAO;QAClB,MAAM,IAAI,CAAC,QAAQ,CAAC,OAAO,EAAE,CAAA;IAC/B,CAAC;IAEM,KAAK,CAAC,UAAU;QACrB,MAAM,IAAI,CAAC,QAAQ,CAAC,UAAU,EAAE,CAAA;IAClC,CAAC;IAEM,KAAK,CAAC,SAAS,CAAC,KAAa,EAAE,QAA+B;QACnE,MAAM,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,EAAE,iCAAiC,CAAC,CAAA;QAEzD,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,EAAE,CAAC;YAC3B,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,GAAG,EAAE,CAAA;QAC5B,CAAC;QACD,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAA;IACtC,CAAC;IAEO,KAAK,CAAC,eAAe,CAC3B,KAAa,EACb,OAAgB;QAGhB,MAAM,cAAc,GAAG,UAAU,EAAE,CAAA;QACnC,IAAI,CAAC;YACH,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,EAAE,MAAM,EAAE,CAAC;gBACnC,MAAM,CAAC,KAAK,CAAC,uCAAuC,KAAK,EAAE,CAAC,CAAA;YAC9D,CAAC;YAED,IAAI,CAAC,OAAO,EAAE,KAAK,EAAE,CAAC;gBACpB,OAAM;YACR,CAAC;YAED,MAAM,GAAG,GAAG,OAAO,EAAE,GAAG,EAAE,QAAQ,CAAC,MAAM,CAAC,CAAA;YAC1C,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAA;YACxD,MAAM,OAAO,GAAG,OAAO,CAAC,OAAO,CAAA;YAE/B,MAAM,CAAC,KAAK,CACV;gBACE,KAAK;gBACL,GAAG;gBACH,KAAK;aACN,EACD,yCAAyC,KAAK,EAAE,CACjD,CAAA;YAED,MAAM,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,GAAG,EAAE,KAAK,EAAE,OAAO,CAAC,CAAC,CAAC,CAAA;YAE/E,MAAM,CAAC,KAAK,CAAC;gBACX,GAAG;gBACH,KAAK;gBACL,QAAQ,EAAE,WAAW,CAAC,cAAc,CAAC;aACtC,EAAE,gDAAgD,CAAC,CAAA;QACtD,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,MAAM,CAAC,KAAK,CAAC,GAAG,EAAE,8CAA8C,CAAC,CAAA;QACnE,CAAC;IACH,CAAC;IAEM,KAAK,CAAC,MAAM,CAAC,UAAgE;QAClF,8BAA8B,EAAE,CAAC,EAAE,UAAU,EAAE,IAAI;KACpD;QACC,MAAM,MAAM,GAAG,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,CAAA;QAE1C,MAAM,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC;YAC5B,MAAM;YACN,aAAa,EAAE,IAAI;SACpB,CAAC,CAAA;QAEF,MAAM,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC;YACtB,UAAU,EAAE,IAAI;YAChB,GAAG,OAAO;YACV,WAAW,EAAE,KAAK,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,IAAI,CAAC,eAAe,CAAC,KAAK,EAAE,OAAO,CAAC;SAChF,CAAC,CAAA;IACJ,CAAC;IAEM,KAAK,CAAC,WAAW,CAAC,OAA6D;QACpF,MAAM,MAAM,GAAG,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,CAAA;QAE1C,MAAM,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC;YAC5B,MAAM;YACN,aAAa,EAAE,IAAI;SACpB,CAAC,CAAA;QAEF,MAAM,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC;YACtB,oBAAoB,EAAE,IAAI;YAC1B,GAAG,OAAO;YACV,SAAS,EAAE,KAAK,EAAE,EAChB,KAAK,EACL,aAAa,EACb,SAAS,EACT,wBAAwB,EACxB,kBAAkB,EAClB,SAAS,EACT,OAAO,EACP,KAAK,GACN,EAAE,EAAE;gBACH,KAAK,MAAM,OAAO,IAAI,KAAK,CAAC,QAAQ,EAAE,CAAC;oBACrC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,KAAK,CAAC,EAAE,MAAM,EAAE,CAAC;wBACzC,MAAM,CAAC,KAAK,CAAC,uCAAuC,KAAK,CAAC,KAAK,EAAE,CAAC,CAAA;oBACpE,CAAC;oBAED,IAAI,CAAC;wBACH,MAAM,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,QAAQ,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC,IAAI,CAAC,eAAe,CAAC,KAAK,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC,CAAC,CAAA;oBAC9F,CAAC;oBAAC,OAAO,UAAU,EAAE,CAAC;wBACpB,MAAM,CAAC,KAAK,CAAC,UAAU,EAAE,0BAA0B,CAAC,CAAA;oBACtD,CAAC;oBAED,aAAa,CAAC,OAAO,CAAC,MAAM,CAAC,CAAA;oBAC7B,MAAM,SAAS,EAAE,CAAA;gBACnB,CAAC;YACH,CAAC;SACF,CAAC,CAAA;IACJ,CAAC;CACF"}
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/libs/kafka/index.ts"],"names":[],"mappings":"AAAA,cAAc,YAAY,CAAA;AAC1B,cAAc,YAAY,CAAA"}
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../src/libs/kafka/index.ts"],"names":[],"mappings":"AAAA,cAAc,YAAY,CAAA;AAC1B,cAAc,YAAY,CAAA"}
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"kafka.d.ts","sourceRoot":"","sources":["../../../../src/libs/kafka/kafka.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,EAAY,MAAM,SAAS,CAAA;AAsBzC,eAAO,MAAM,KAAK,OAIhB,CAAA"}
|
|
@@ -1,24 +0,0 @@
|
|
|
1
|
-
import { Kafka, logLevel } from 'kafkajs';
|
|
2
|
-
import { KAFKA_CLIENT_ID, KAFKA_URI } from './config';
|
|
3
|
-
import { KafkaJsLogCreator } from '../logger';
|
|
4
|
-
const toBunyanLogLevel = level => {
|
|
5
|
-
switch (level) {
|
|
6
|
-
case logLevel.ERROR:
|
|
7
|
-
case logLevel.NOTHING:
|
|
8
|
-
return 'error';
|
|
9
|
-
case logLevel.WARN:
|
|
10
|
-
return 'warn';
|
|
11
|
-
case logLevel.INFO:
|
|
12
|
-
return 'info';
|
|
13
|
-
case logLevel.DEBUG:
|
|
14
|
-
return 'debug';
|
|
15
|
-
default:
|
|
16
|
-
return 'info';
|
|
17
|
-
}
|
|
18
|
-
};
|
|
19
|
-
export const kafka = new Kafka({
|
|
20
|
-
clientId: KAFKA_CLIENT_ID,
|
|
21
|
-
brokers: KAFKA_URI,
|
|
22
|
-
logCreator: KafkaJsLogCreator(toBunyanLogLevel),
|
|
23
|
-
});
|
|
24
|
-
//# sourceMappingURL=kafka.js.map
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"kafka.js","sourceRoot":"","sources":["../../../../src/libs/kafka/kafka.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,EAAE,QAAQ,EAAE,MAAM,SAAS,CAAA;AACzC,OAAO,EAAE,eAAe,EAAE,SAAS,EAAE,MAAM,UAAU,CAAA;AACrD,OAAO,EAAE,iBAAiB,EAAE,MAAM,WAAW,CAAA;AAG7C,MAAM,gBAAgB,GAAG,KAAK,CAAC,EAAE;IAC/B,QAAQ,KAAK,EAAE,CAAC;QACd,KAAK,QAAQ,CAAC,KAAK,CAAC;QACpB,KAAK,QAAQ,CAAC,OAAO;YACnB,OAAO,OAAO,CAAA;QAChB,KAAK,QAAQ,CAAC,IAAI;YAChB,OAAO,MAAM,CAAA;QACf,KAAK,QAAQ,CAAC,IAAI;YAChB,OAAO,MAAM,CAAA;QACf,KAAK,QAAQ,CAAC,KAAK;YACjB,OAAO,OAAO,CAAA;QAChB;YACE,OAAO,MAAM,CAAA;IACjB,CAAC;AACH,CAAC,CAAA;AAGD,MAAM,CAAC,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC;IAC7B,QAAQ,EAAE,eAAe;IACzB,OAAO,EAAE,SAAS;IAClB,UAAU,EAAE,iBAAiB,CAAC,gBAAgB,CAAC;CAChD,CAAC,CAAA"}
|
|
@@ -1,11 +0,0 @@
|
|
|
1
|
-
import { IHeaders, ProducerConfig } from 'kafkajs';
|
|
2
|
-
export declare class KafkaProducer {
|
|
3
|
-
private producer;
|
|
4
|
-
private _isConnected;
|
|
5
|
-
constructor(config?: ProducerConfig);
|
|
6
|
-
connect(): Promise<void>;
|
|
7
|
-
isConnected(): boolean;
|
|
8
|
-
disconnect(): Promise<void>;
|
|
9
|
-
send(topic: string, value: Record<string, any>, key?: string, headers?: IHeaders): Promise<void>;
|
|
10
|
-
}
|
|
11
|
-
//# sourceMappingURL=producer.d.ts.map
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"producer.d.ts","sourceRoot":"","sources":["../../../../src/libs/kafka/producer.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAY,cAAc,EAAE,MAAM,SAAS,CAAA;AAG5D,qBAAa,aAAa;IACxB,OAAO,CAAC,QAAQ,CAAU;IAC1B,OAAO,CAAC,YAAY,CAAQ;gBAChB,MAAM,GAAE,cAAmB;IAI1B,OAAO;IAUb,WAAW;IAIL,UAAU;IAKV,IAAI,CAAC,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,EAAE,GAAG,CAAC,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,QAAQ;CAkB9F"}
|
|
@@ -1,45 +0,0 @@
|
|
|
1
|
-
import { kafka } from './kafka';
|
|
2
|
-
export class KafkaProducer {
|
|
3
|
-
producer;
|
|
4
|
-
_isConnected = false;
|
|
5
|
-
constructor(config = {}) {
|
|
6
|
-
this.producer = kafka.producer(config);
|
|
7
|
-
}
|
|
8
|
-
async connect() {
|
|
9
|
-
try {
|
|
10
|
-
await this.producer.connect();
|
|
11
|
-
this._isConnected = true;
|
|
12
|
-
}
|
|
13
|
-
catch (err) {
|
|
14
|
-
this._isConnected = false;
|
|
15
|
-
throw err;
|
|
16
|
-
}
|
|
17
|
-
}
|
|
18
|
-
isConnected() {
|
|
19
|
-
return this._isConnected;
|
|
20
|
-
}
|
|
21
|
-
async disconnect() {
|
|
22
|
-
await this.producer.disconnect();
|
|
23
|
-
this._isConnected = false;
|
|
24
|
-
}
|
|
25
|
-
async send(topic, value, key, headers) {
|
|
26
|
-
try {
|
|
27
|
-
await this.producer.send({
|
|
28
|
-
topic,
|
|
29
|
-
messages: [
|
|
30
|
-
{
|
|
31
|
-
key,
|
|
32
|
-
value: JSON.stringify(value),
|
|
33
|
-
headers: headers
|
|
34
|
-
},
|
|
35
|
-
],
|
|
36
|
-
});
|
|
37
|
-
this._isConnected = true;
|
|
38
|
-
}
|
|
39
|
-
catch (err) {
|
|
40
|
-
this._isConnected = false;
|
|
41
|
-
throw err;
|
|
42
|
-
}
|
|
43
|
-
}
|
|
44
|
-
}
|
|
45
|
-
//# sourceMappingURL=producer.js.map
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"producer.js","sourceRoot":"","sources":["../../../../src/libs/kafka/producer.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,KAAK,EAAE,MAAM,SAAS,CAAA;AAE/B,MAAM,OAAO,aAAa;IAChB,QAAQ,CAAU;IAClB,YAAY,GAAG,KAAK,CAAA;IAC5B,YAAY,SAAyB,EAAE;QACrC,IAAI,CAAC,QAAQ,GAAG,KAAK,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAA;IACxC,CAAC;IAEM,KAAK,CAAC,OAAO;QAClB,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,QAAQ,CAAC,OAAO,EAAE,CAAA;YAC7B,IAAI,CAAC,YAAY,GAAG,IAAI,CAAA;QAC1B,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,IAAI,CAAC,YAAY,GAAG,KAAK,CAAA;YACzB,MAAM,GAAG,CAAA;QACX,CAAC;IACH,CAAC;IAEM,WAAW;QAChB,OAAO,IAAI,CAAC,YAAY,CAAA;IAC1B,CAAC;IAEM,KAAK,CAAC,UAAU;QACrB,MAAM,IAAI,CAAC,QAAQ,CAAC,UAAU,EAAE,CAAA;QAChC,IAAI,CAAC,YAAY,GAAG,KAAK,CAAA;IAC3B,CAAC;IAEM,KAAK,CAAC,IAAI,CAAC,KAAa,EAAE,KAA0B,EAAE,GAAY,EAAE,OAAkB;QAC3F,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC;gBACvB,KAAK;gBACL,QAAQ,EAAE;oBACR;wBACE,GAAG;wBACH,KAAK,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC;wBAC5B,OAAO,EAAE,OAAO;qBACjB;iBACF;aACF,CAAC,CAAA;YACF,IAAI,CAAC,YAAY,GAAG,IAAI,CAAA;QAC1B,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,IAAI,CAAC,YAAY,GAAG,KAAK,CAAA;YACzB,MAAM,GAAG,CAAA;QACX,CAAC;IACH,CAAC;CACF"}
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"config.d.ts","sourceRoot":"","sources":["../../../../src/libs/logger/config.ts"],"names":[],"mappings":"AAAA,eAAO,MAAM,QAAQ,QAAwC,CAAA;AAC7D,eAAO,MAAM,YAAY,SAA4B,CAAA;AACrD,eAAO,MAAM,QAAQ,QAAsE,CAAA;AAC3F,eAAO,MAAM,SAAS,QAA6D,CAAA"}
|
|
@@ -1,5 +0,0 @@
|
|
|
1
|
-
export const NODE_ENV = process.env.NODE_ENV || 'development';
|
|
2
|
-
export const isProduction = NODE_ENV === 'production';
|
|
3
|
-
export const APP_NAME = process.env.npm_package_name?.split('/').pop() || 'tests';
|
|
4
|
-
export const LOG_LEVEL = process.env.LOG_LEVEL || (isProduction ? 'info' : 'debug');
|
|
5
|
-
//# sourceMappingURL=config.js.map
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"config.js","sourceRoot":"","sources":["../../../../src/libs/logger/config.ts"],"names":[],"mappings":"AAAA,MAAM,CAAC,MAAM,QAAQ,GAAG,OAAO,CAAC,GAAG,CAAC,QAAQ,IAAI,aAAa,CAAA;AAC7D,MAAM,CAAC,MAAM,YAAY,GAAG,QAAQ,KAAK,YAAY,CAAA;AACrD,MAAM,CAAC,MAAM,QAAQ,GAAG,OAAO,CAAC,GAAG,CAAC,gBAAgB,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,EAAY,IAAI,OAAO,CAAA;AAC3F,MAAM,CAAC,MAAM,SAAS,GAAG,OAAO,CAAC,GAAG,CAAC,SAAS,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC,CAAA"}
|
|
@@ -1,7 +0,0 @@
|
|
|
1
|
-
export declare const S3_HOST: string;
|
|
2
|
-
export declare const S3_EXPORT_HOST: string;
|
|
3
|
-
export declare const S3_PRESIGNED_URL_EXPIRES: number;
|
|
4
|
-
export declare const AWS_REGION: string;
|
|
5
|
-
export declare const AWS_ACCESS_KEY_ID: string;
|
|
6
|
-
export declare const AWS_SECRET_ACCESS_KEY: string;
|
|
7
|
-
//# sourceMappingURL=config.d.ts.map
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"config.d.ts","sourceRoot":"","sources":["../../../../src/libs/s3/config.ts"],"names":[],"mappings":"AAAA,eAAO,MAAM,OAAO,QAAoD,CAAA;AACxE,eAAO,MAAM,cAAc,QAAwC,CAAA;AACnE,eAAO,MAAM,wBAAwB,QAAsD,CAAA;AAE3F,eAAO,MAAM,UAAU,EAA6B,MAAM,CAAA;AAC1D,eAAO,MAAM,iBAAiB,EAAoC,MAAM,CAAA;AACxE,eAAO,MAAM,qBAAqB,EAAwC,MAAM,CAAA"}
|
|
@@ -1,7 +0,0 @@
|
|
|
1
|
-
export const S3_HOST = process.env.S3_HOST || 'https://s3.amazonaws.com';
|
|
2
|
-
export const S3_EXPORT_HOST = process.env.S3_EXPORT_HOST || S3_HOST;
|
|
3
|
-
export const S3_PRESIGNED_URL_EXPIRES = Number(process.env.S3_PRESIGNED_URL_EXPIRES) || 120;
|
|
4
|
-
export const AWS_REGION = process.env.AWS_REGION;
|
|
5
|
-
export const AWS_ACCESS_KEY_ID = process.env.AWS_ACCESS_KEY_ID;
|
|
6
|
-
export const AWS_SECRET_ACCESS_KEY = process.env.AWS_SECRET_ACCESS_KEY;
|
|
7
|
-
//# sourceMappingURL=config.js.map
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"config.js","sourceRoot":"","sources":["../../../../src/libs/s3/config.ts"],"names":[],"mappings":"AAAA,MAAM,CAAC,MAAM,OAAO,GAAG,OAAO,CAAC,GAAG,CAAC,OAAO,IAAI,0BAA0B,CAAA;AACxE,MAAM,CAAC,MAAM,cAAc,GAAG,OAAO,CAAC,GAAG,CAAC,cAAc,IAAI,OAAO,CAAA;AACnE,MAAM,CAAC,MAAM,wBAAwB,GAAG,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,wBAAwB,CAAC,IAAI,GAAG,CAAA;AAE3F,MAAM,CAAC,MAAM,UAAU,GAAG,OAAO,CAAC,GAAG,CAAC,UAAoB,CAAA;AAC1D,MAAM,CAAC,MAAM,iBAAiB,GAAG,OAAO,CAAC,GAAG,CAAC,iBAA2B,CAAA;AACxE,MAAM,CAAC,MAAM,qBAAqB,GAAG,OAAO,CAAC,GAAG,CAAC,qBAA+B,CAAA"}
|
|
@@ -1,35 +0,0 @@
|
|
|
1
|
-
import type { AgentChatRepository, AgentMessageRepository } from '@multiplayer-app/ai-agent-db';
|
|
2
|
-
declare enum BackgroundChatProcessingEvent {
|
|
3
|
-
Start = "start"
|
|
4
|
-
}
|
|
5
|
-
declare class KafkaService {
|
|
6
|
-
private kafkaConsumer;
|
|
7
|
-
private kafkaProducer;
|
|
8
|
-
private chatRepository?;
|
|
9
|
-
private messageRepository?;
|
|
10
|
-
private readonly MAX_RETRIES;
|
|
11
|
-
private readonly BASE_DELAY_MS;
|
|
12
|
-
private readonly MAX_DELAY_MS;
|
|
13
|
-
constructor(chatRepository?: AgentChatRepository, messageRepository?: AgentMessageRepository);
|
|
14
|
-
setRepositories(chatRepository: AgentChatRepository, messageRepository: AgentMessageRepository): void;
|
|
15
|
-
start(): Promise<void>;
|
|
16
|
-
stop(): Promise<void>;
|
|
17
|
-
sendStartBackgroundAgentEvent(chatId: string): Promise<void>;
|
|
18
|
-
sendChatTitleGenerationEvent(chatId: string, retryCount?: number): Promise<void>;
|
|
19
|
-
private sleep;
|
|
20
|
-
private calculateBackoffDelay;
|
|
21
|
-
onBackgroundAgentEvent(key: any, message: {
|
|
22
|
-
event: BackgroundChatProcessingEvent;
|
|
23
|
-
chatId: string;
|
|
24
|
-
}): Promise<void>;
|
|
25
|
-
onChatTitleGenerationEvent(key: any, message: {
|
|
26
|
-
chatId: string;
|
|
27
|
-
retryCount?: number;
|
|
28
|
-
}): Promise<void>;
|
|
29
|
-
onStartBackgroundAgentEvent(chatId: string): Promise<void>;
|
|
30
|
-
connect(): Promise<void>;
|
|
31
|
-
disconnect(): Promise<void>;
|
|
32
|
-
}
|
|
33
|
-
export declare const kafkaService: KafkaService;
|
|
34
|
-
export {};
|
|
35
|
-
//# sourceMappingURL=KafkaService.d.ts.map
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"KafkaService.d.ts","sourceRoot":"","sources":["../../../src/services/KafkaService.ts"],"names":[],"mappings":"AAMA,OAAO,KAAK,EAAE,mBAAmB,EAAE,sBAAsB,EAAE,MAAM,8BAA8B,CAAC;AAEhG,aAAK,6BAA6B;IAChC,KAAK,UAAU;CAChB;AAED,cAAM,YAAY;IAChB,OAAO,CAAC,aAAa,CAAe;IACpC,OAAO,CAAC,aAAa,CAAe;IACpC,OAAO,CAAC,cAAc,CAAC,CAAqB;IAC5C,OAAO,CAAC,iBAAiB,CAAC,CAAwB;IAElD,OAAO,CAAC,QAAQ,CAAC,WAAW,CAAI;IAChC,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAO;IACrC,OAAO,CAAC,QAAQ,CAAC,YAAY,CAAQ;gBAEzB,cAAc,CAAC,EAAE,mBAAmB,EAAE,iBAAiB,CAAC,EAAE,sBAAsB;IAO5F,eAAe,CAAC,cAAc,EAAE,mBAAmB,EAAE,iBAAiB,EAAE,sBAAsB;IAKxF,KAAK;IAIL,IAAI;IAIJ,6BAA6B,CAAC,MAAM,EAAE,MAAM;IAI5C,4BAA4B,CAAC,MAAM,EAAE,MAAM,EAAE,UAAU,GAAE,MAAU;YAI3D,KAAK;IAInB,OAAO,CAAC,qBAAqB;IAUvB,sBAAsB,CAAC,GAAG,KAAA,EAAE,OAAO,EAAE;QAAE,KAAK,EAAE,6BAA6B,CAAC;QAAC,MAAM,EAAE,MAAM,CAAA;KAAE;IAQ7F,0BAA0B,CAAC,GAAG,KAAA,EAAE,OAAO,EAAE;QAAE,MAAM,EAAE,MAAM,CAAC;QAAC,UAAU,CAAC,EAAE,MAAM,CAAA;KAAE;IAuChF,2BAA2B,CAAC,MAAM,EAAE,MAAM;IAkB1C,OAAO;IAOP,UAAU;CAGjB;AACD,eAAO,MAAM,YAAY,cAAqB,CAAA"}
|
|
@@ -1,123 +0,0 @@
|
|
|
1
|
-
import { KafkaConsumer, KafkaProducer } from '../libs/kafka';
|
|
2
|
-
import { config } from '../config';
|
|
3
|
-
import { AIHelper } from '../helpers/AIHelper';
|
|
4
|
-
import { logger } from "../libs/logger";
|
|
5
|
-
import { socketService } from "./SocketService";
|
|
6
|
-
var BackgroundChatProcessingEvent;
|
|
7
|
-
(function (BackgroundChatProcessingEvent) {
|
|
8
|
-
BackgroundChatProcessingEvent["Start"] = "start";
|
|
9
|
-
})(BackgroundChatProcessingEvent || (BackgroundChatProcessingEvent = {}));
|
|
10
|
-
class KafkaService {
|
|
11
|
-
kafkaConsumer;
|
|
12
|
-
kafkaProducer;
|
|
13
|
-
chatRepository;
|
|
14
|
-
messageRepository;
|
|
15
|
-
MAX_RETRIES = 5;
|
|
16
|
-
BASE_DELAY_MS = 1000; // 1 second base delay
|
|
17
|
-
MAX_DELAY_MS = 60000; // 60 seconds max delay
|
|
18
|
-
constructor(chatRepository, messageRepository) {
|
|
19
|
-
this.kafkaConsumer = new KafkaConsumer(config.kafka.groupId);
|
|
20
|
-
this.kafkaProducer = new KafkaProducer();
|
|
21
|
-
this.chatRepository = chatRepository;
|
|
22
|
-
this.messageRepository = messageRepository;
|
|
23
|
-
}
|
|
24
|
-
setRepositories(chatRepository, messageRepository) {
|
|
25
|
-
this.chatRepository = chatRepository;
|
|
26
|
-
this.messageRepository = messageRepository;
|
|
27
|
-
}
|
|
28
|
-
async start() {
|
|
29
|
-
await Promise.all([this.connect(), this.kafkaProducer.connect()]);
|
|
30
|
-
}
|
|
31
|
-
async stop() {
|
|
32
|
-
await Promise.all([this.kafkaConsumer.disconnect(), this.kafkaProducer.disconnect()]);
|
|
33
|
-
}
|
|
34
|
-
async sendStartBackgroundAgentEvent(chatId) {
|
|
35
|
-
await this.kafkaProducer.send(config.kafka.backgroundChatProcessingTopic, { event: BackgroundChatProcessingEvent.Start, chatId });
|
|
36
|
-
}
|
|
37
|
-
async sendChatTitleGenerationEvent(chatId, retryCount = 0) {
|
|
38
|
-
await this.kafkaProducer.send(config.kafka.chatTitleGenerationTopic, { chatId, retryCount });
|
|
39
|
-
}
|
|
40
|
-
async sleep(ms) {
|
|
41
|
-
return new Promise(resolve => setTimeout(resolve, ms));
|
|
42
|
-
}
|
|
43
|
-
calculateBackoffDelay(retryCount) {
|
|
44
|
-
// Exponential backoff: baseDelay * 2^retryCount with jitter
|
|
45
|
-
const exponentialDelay = this.BASE_DELAY_MS * Math.pow(2, retryCount);
|
|
46
|
-
// Add jitter (random 0-25% of delay) to prevent thundering herd
|
|
47
|
-
const jitter = Math.random() * 0.25 * exponentialDelay;
|
|
48
|
-
const delay = exponentialDelay + jitter;
|
|
49
|
-
// Cap at max delay
|
|
50
|
-
return Math.min(delay, this.MAX_DELAY_MS);
|
|
51
|
-
}
|
|
52
|
-
async onBackgroundAgentEvent(key, message) {
|
|
53
|
-
switch (message.event) {
|
|
54
|
-
case BackgroundChatProcessingEvent.Start:
|
|
55
|
-
await this.onStartBackgroundAgentEvent(message.chatId);
|
|
56
|
-
break;
|
|
57
|
-
}
|
|
58
|
-
}
|
|
59
|
-
async onChatTitleGenerationEvent(key, message) {
|
|
60
|
-
if (!this.chatRepository || !this.messageRepository) {
|
|
61
|
-
logger.warn('KafkaService: Repositories not set, skipping chat title generation');
|
|
62
|
-
return;
|
|
63
|
-
}
|
|
64
|
-
const retryCount = message.retryCount ?? 0;
|
|
65
|
-
try {
|
|
66
|
-
const chat = await this.chatRepository.findById(message.chatId);
|
|
67
|
-
if (!chat) {
|
|
68
|
-
return;
|
|
69
|
-
}
|
|
70
|
-
const messages = await this.messageRepository.findByChatId(message.chatId);
|
|
71
|
-
if (messages.length === 0) {
|
|
72
|
-
return;
|
|
73
|
-
}
|
|
74
|
-
const title = await AIHelper.generateTitleForMessage(chat.contextKey, messages);
|
|
75
|
-
await this.chatRepository.update(chat.id, { title });
|
|
76
|
-
// Update the chat object with the new title before emitting
|
|
77
|
-
const updatedChat = {
|
|
78
|
-
...chat,
|
|
79
|
-
title
|
|
80
|
-
};
|
|
81
|
-
if (chat.userId)
|
|
82
|
-
socketService.emitChatUpdate(chat.userId, { ...updatedChat, messages });
|
|
83
|
-
}
|
|
84
|
-
catch (error) {
|
|
85
|
-
logger.error(`KafkaService: Error generating chat title for chatId ${message.chatId}, retryCount: ${retryCount}`, error);
|
|
86
|
-
if (retryCount < this.MAX_RETRIES) {
|
|
87
|
-
const delay = this.calculateBackoffDelay(retryCount);
|
|
88
|
-
logger.info(`KafkaService: Retrying chat title generation for chatId ${message.chatId} after ${Math.round(delay)}ms (attempt ${retryCount + 1}/${this.MAX_RETRIES})`);
|
|
89
|
-
await this.sleep(delay);
|
|
90
|
-
await this.sendChatTitleGenerationEvent(message.chatId, retryCount + 1);
|
|
91
|
-
}
|
|
92
|
-
else {
|
|
93
|
-
logger.error(`KafkaService: Max retries (${this.MAX_RETRIES}) exceeded for chat title generation, chatId: ${message.chatId}`);
|
|
94
|
-
}
|
|
95
|
-
}
|
|
96
|
-
}
|
|
97
|
-
async onStartBackgroundAgentEvent(chatId) {
|
|
98
|
-
if (!this.chatRepository || !this.messageRepository) {
|
|
99
|
-
logger.warn('KafkaService: Repositories not set, skipping background agent event');
|
|
100
|
-
return;
|
|
101
|
-
}
|
|
102
|
-
const chat = await this.chatRepository.findById(chatId);
|
|
103
|
-
if (!chat) {
|
|
104
|
-
return;
|
|
105
|
-
}
|
|
106
|
-
const messages = await this.messageRepository.findByChatId(chatId);
|
|
107
|
-
if (messages.length === 0) {
|
|
108
|
-
return;
|
|
109
|
-
}
|
|
110
|
-
// TODO: Implement background agent processing
|
|
111
|
-
}
|
|
112
|
-
async connect() {
|
|
113
|
-
await this.kafkaConsumer.connect();
|
|
114
|
-
await this.kafkaConsumer.subscribe(config.kafka.chatTitleGenerationTopic, this.onChatTitleGenerationEvent.bind(this));
|
|
115
|
-
await this.kafkaConsumer.subscribe(config.kafka.backgroundChatProcessingTopic, this.onBackgroundAgentEvent.bind(this));
|
|
116
|
-
await this.kafkaConsumer.listen();
|
|
117
|
-
}
|
|
118
|
-
async disconnect() {
|
|
119
|
-
await this.kafkaConsumer.disconnect();
|
|
120
|
-
}
|
|
121
|
-
}
|
|
122
|
-
export const kafkaService = new KafkaService();
|
|
123
|
-
//# sourceMappingURL=KafkaService.js.map
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"KafkaService.js","sourceRoot":"","sources":["../../../src/services/KafkaService.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,aAAa,EAAE,aAAa,EAAE,MAAM,eAAe,CAAA;AAC5D,OAAO,EAAE,MAAM,EAAE,MAAM,WAAW,CAAA;AAElC,OAAO,EAAE,QAAQ,EAAE,MAAM,qBAAqB,CAAA;AAC9C,OAAO,EAAC,MAAM,EAAC,MAAM,gBAAgB,CAAC;AACtC,OAAO,EAAC,aAAa,EAAC,MAAM,iBAAiB,CAAC;AAG9C,IAAK,6BAEJ;AAFD,WAAK,6BAA6B;IAChC,gDAAe,CAAA;AACjB,CAAC,EAFI,6BAA6B,KAA7B,6BAA6B,QAEjC;AAED,MAAM,YAAY;IACR,aAAa,CAAe;IAC5B,aAAa,CAAe;IAC5B,cAAc,CAAsB;IACpC,iBAAiB,CAAyB;IAEjC,WAAW,GAAG,CAAC,CAAA;IACf,aAAa,GAAG,IAAI,CAAA,CAAC,sBAAsB;IAC3C,YAAY,GAAG,KAAK,CAAA,CAAC,uBAAuB;IAE7D,YAAY,cAAoC,EAAE,iBAA0C;QAC1F,IAAI,CAAC,aAAa,GAAG,IAAI,aAAa,CAAC,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAA;QAC5D,IAAI,CAAC,aAAa,GAAG,IAAI,aAAa,EAAE,CAAA;QACxC,IAAI,CAAC,cAAc,GAAG,cAAc,CAAA;QACpC,IAAI,CAAC,iBAAiB,GAAG,iBAAiB,CAAA;IAC5C,CAAC;IAED,eAAe,CAAC,cAAmC,EAAE,iBAAyC;QAC5F,IAAI,CAAC,cAAc,GAAG,cAAc,CAAA;QACpC,IAAI,CAAC,iBAAiB,GAAG,iBAAiB,CAAA;IAC5C,CAAC;IAED,KAAK,CAAC,KAAK;QACT,MAAM,OAAO,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,OAAO,EAAE,EAAE,IAAI,CAAC,aAAa,CAAC,OAAO,EAAE,CAAC,CAAC,CAAA;IACnE,CAAC;IAED,KAAK,CAAC,IAAI;QACR,MAAM,OAAO,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,UAAU,EAAE,EAAE,IAAI,CAAC,aAAa,CAAC,UAAU,EAAE,CAAC,CAAC,CAAA;IACvF,CAAC;IAED,KAAK,CAAC,6BAA6B,CAAC,MAAc;QAChD,MAAM,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,6BAA6B,EAAE,EAAE,KAAK,EAAE,6BAA6B,CAAC,KAAK,EAAE,MAAM,EAAE,CAAC,CAAA;IACnI,CAAC;IAED,KAAK,CAAC,4BAA4B,CAAC,MAAc,EAAE,aAAqB,CAAC;QACvE,MAAM,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,wBAAwB,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,CAAC,CAAA;IAC9F,CAAC;IAEO,KAAK,CAAC,KAAK,CAAC,EAAU;QAC5B,OAAO,IAAI,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,CAAA;IACxD,CAAC;IAEO,qBAAqB,CAAC,UAAkB;QAC9C,4DAA4D;QAC5D,MAAM,gBAAgB,GAAG,IAAI,CAAC,aAAa,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,UAAU,CAAC,CAAA;QACrE,gEAAgE;QAChE,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,EAAE,GAAG,IAAI,GAAG,gBAAgB,CAAA;QACtD,MAAM,KAAK,GAAG,gBAAgB,GAAG,MAAM,CAAA;QACvC,mBAAmB;QACnB,OAAO,IAAI,CAAC,GAAG,CAAC,KAAK,EAAE,IAAI,CAAC,YAAY,CAAC,CAAA;IAC3C,CAAC;IAED,KAAK,CAAC,sBAAsB,CAAC,GAAG,EAAE,OAAiE;QACjG,QAAQ,OAAO,CAAC,KAAK,EAAE,CAAC;YACtB,KAAK,6BAA6B,CAAC,KAAK;gBACtC,MAAM,IAAI,CAAC,2BAA2B,CAAC,OAAO,CAAC,MAAM,CAAC,CAAA;gBACtD,MAAK;QACT,CAAC;IACH,CAAC;IAED,KAAK,CAAC,0BAA0B,CAAC,GAAG,EAAE,OAAgD;QACpF,IAAI,CAAC,IAAI,CAAC,cAAc,IAAI,CAAC,IAAI,CAAC,iBAAiB,EAAE,CAAC;YACpD,MAAM,CAAC,IAAI,CAAC,oEAAoE,CAAC,CAAA;YACjF,OAAM;QACR,CAAC;QACD,MAAM,UAAU,GAAG,OAAO,CAAC,UAAU,IAAI,CAAC,CAAA;QAC1C,IAAI,CAAC;YACH,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,cAAc,CAAC,QAAQ,CAAC,OAAO,CAAC,MAAM,CAAC,CAAA;YAC/D,IAAI,CAAC,IAAI,EAAE,CAAC;gBACV,OAAM;YACR,CAAC;YACD,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,iBAAiB,CAAC,YAAY,CAAC,OAAO,CAAC,MAAM,CAAC,CAAA;YAC1E,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBAC1B,OAAM;YACR,CAAC;YAED,MAAM,KAAK,GAAG,MAAM,QAAQ,CAAC,uBAAuB,CAAC,IAAI,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAA;YAC/E,MAAM,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,EAAE,EAAC,KAAK,EAAC,CAAC,CAAA;YAClD,4DAA4D;YAC5D,MAAM,WAAW,GAAG;gBAClB,GAAG,IAAI;gBACP,KAAK;aACN,CAAA;YACD,IAAI,IAAI,CAAC,MAAM;gBAAE,aAAa,CAAC,cAAc,CAAC,IAAI,CAAC,MAAM,EAAE,EAAC,GAAG,WAAW,EAAE,QAAQ,EAAC,CAAC,CAAC;QACzF,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,CAAC,KAAK,CAAC,wDAAwD,OAAO,CAAC,MAAM,iBAAiB,UAAU,EAAE,EAAE,KAAK,CAAC,CAAA;YAExH,IAAI,UAAU,GAAG,IAAI,CAAC,WAAW,EAAE,CAAC;gBAClC,MAAM,KAAK,GAAG,IAAI,CAAC,qBAAqB,CAAC,UAAU,CAAC,CAAA;gBACpD,MAAM,CAAC,IAAI,CAAC,2DAA2D,OAAO,CAAC,MAAM,UAAU,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,eAAe,UAAU,GAAG,CAAC,IAAI,IAAI,CAAC,WAAW,GAAG,CAAC,CAAA;gBAErK,MAAM,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAA;gBACvB,MAAM,IAAI,CAAC,4BAA4B,CAAC,OAAO,CAAC,MAAM,EAAE,UAAU,GAAG,CAAC,CAAC,CAAA;YACzE,CAAC;iBAAM,CAAC;gBACN,MAAM,CAAC,KAAK,CAAC,8BAA8B,IAAI,CAAC,WAAW,iDAAiD,OAAO,CAAC,MAAM,EAAE,CAAC,CAAA;YAC/H,CAAC;QACH,CAAC;IACH,CAAC;IAED,KAAK,CAAC,2BAA2B,CAAC,MAAc;QAC9C,IAAI,CAAC,IAAI,CAAC,cAAc,IAAI,CAAC,IAAI,CAAC,iBAAiB,EAAE,CAAC;YACpD,MAAM,CAAC,IAAI,CAAC,qEAAqE,CAAC,CAAA;YAClF,OAAM;QACR,CAAC;QACD,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,cAAc,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAA;QACvD,IAAI,CAAC,IAAI,EAAE,CAAC;YACV,OAAM;QACR,CAAC;QACD,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,iBAAiB,CAAC,YAAY,CAAC,MAAM,CAAC,CAAA;QAClE,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC1B,OAAM;QACR,CAAC;QAED,8CAA8C;IAEhD,CAAC;IAED,KAAK,CAAC,OAAO;QACX,MAAM,IAAI,CAAC,aAAa,CAAC,OAAO,EAAE,CAAA;QAClC,MAAM,IAAI,CAAC,aAAa,CAAC,SAAS,CAAC,MAAM,CAAC,KAAK,CAAC,wBAAwB,EAAE,IAAI,CAAC,0BAA0B,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAA;QACrH,MAAM,IAAI,CAAC,aAAa,CAAC,SAAS,CAAC,MAAM,CAAC,KAAK,CAAC,6BAA6B,EAAE,IAAI,CAAC,sBAAsB,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAA;QACtH,MAAM,IAAI,CAAC,aAAa,CAAC,MAAM,EAAE,CAAA;IACnC,CAAC;IAED,KAAK,CAAC,UAAU;QACd,MAAM,IAAI,CAAC,aAAa,CAAC,UAAU,EAAE,CAAA;IACvC,CAAC;CACF;AACD,MAAM,CAAC,MAAM,YAAY,GAAG,IAAI,YAAY,EAAE,CAAA"}
|