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

package/dist/esm/services/KafkaService.js

@@ -1,120 +0,0 @@
-import { KafkaConsumer, KafkaProducer } from '../libs/kafka/index.js';
-import { config } from '../config.js';
-import { AIHelper } from '../helpers/AIHelper.js';
-import { logger } from "../libs/logger/index.js";
-import { socketService } from "./SocketService.js";
-var BackgroundChatProcessingEvent;
-(function (BackgroundChatProcessingEvent) {
-    BackgroundChatProcessingEvent["Start"] = "start";
-})(BackgroundChatProcessingEvent || (BackgroundChatProcessingEvent = {}));
-class KafkaService {
-    constructor(chatRepository, messageRepository) {
-        this.MAX_RETRIES = 5;
-        this.BASE_DELAY_MS = 1000; // 1 second base delay
-        this.MAX_DELAY_MS = 60000; // 60 seconds max delay
-        this.kafkaConsumer = new KafkaConsumer(config.kafka.groupId);
-        this.kafkaProducer = new KafkaProducer();
-        this.chatRepository = chatRepository;
-        this.messageRepository = messageRepository;
-    }
-    setRepositories(chatRepository, messageRepository) {
-        this.chatRepository = chatRepository;
-        this.messageRepository = messageRepository;
-    }
-    async start() {
-        await Promise.all([this.connect(), this.kafkaProducer.connect()]);
-    }
-    async stop() {
-        await Promise.all([this.kafkaConsumer.disconnect(), this.kafkaProducer.disconnect()]);
-    }
-    async sendStartBackgroundAgentEvent(chatId) {
-        await this.kafkaProducer.send(config.kafka.backgroundChatProcessingTopic, { event: BackgroundChatProcessingEvent.Start, chatId });
-    }
-    async sendChatTitleGenerationEvent(chatId, retryCount = 0) {
-        await this.kafkaProducer.send(config.kafka.chatTitleGenerationTopic, { chatId, retryCount });
-    }
-    async sleep(ms) {
-        return new Promise(resolve => setTimeout(resolve, ms));
-    }
-    calculateBackoffDelay(retryCount) {
-        // Exponential backoff: baseDelay * 2^retryCount with jitter
-        const exponentialDelay = this.BASE_DELAY_MS * Math.pow(2, retryCount);
-        // Add jitter (random 0-25% of delay) to prevent thundering herd
-        const jitter = Math.random() * 0.25 * exponentialDelay;
-        const delay = exponentialDelay + jitter;
-        // Cap at max delay
-        return Math.min(delay, this.MAX_DELAY_MS);
-    }
-    async onBackgroundAgentEvent(key, message) {
-        switch (message.event) {
-            case BackgroundChatProcessingEvent.Start:
-                await this.onStartBackgroundAgentEvent(message.chatId);
-                break;
-        }
-    }
-    async onChatTitleGenerationEvent(key, message) {
-        var _a;
-        if (!this.chatRepository || !this.messageRepository) {
-            logger.warn('KafkaService: Repositories not set, skipping chat title generation');
-            return;
-        }
-        const retryCount = (_a = message.retryCount) !== null && _a !== void 0 ? _a : 0;
-        try {
-            const chat = await this.chatRepository.findById(message.chatId);
-            if (!chat) {
-                return;
-            }
-            const messages = await this.messageRepository.findByChatId(message.chatId);
-            if (messages.length === 0) {
-                return;
-            }
-            const title = await AIHelper.generateTitleForMessage(chat.contextKey, messages);
-            await this.chatRepository.update(chat.id, { title });
-            // Update the chat object with the new title before emitting
-            const updatedChat = {
-                ...chat,
-                title
-            };
-            if (chat.userId)
-                socketService.emitChatUpdate(chat.userId, { ...updatedChat, messages });
-        }
-        catch (error) {
-            logger.error(`KafkaService: Error generating chat title for chatId ${message.chatId}, retryCount: ${retryCount}`, error);
-            if (retryCount < this.MAX_RETRIES) {
-                const delay = this.calculateBackoffDelay(retryCount);
-                logger.info(`KafkaService: Retrying chat title generation for chatId ${message.chatId} after ${Math.round(delay)}ms (attempt ${retryCount + 1}/${this.MAX_RETRIES})`);
-                await this.sleep(delay);
-                await this.sendChatTitleGenerationEvent(message.chatId, retryCount + 1);
-            }
-            else {
-                logger.error(`KafkaService: Max retries (${this.MAX_RETRIES}) exceeded for chat title generation, chatId: ${message.chatId}`);
-            }
-        }
-    }
-    async onStartBackgroundAgentEvent(chatId) {
-        if (!this.chatRepository || !this.messageRepository) {
-            logger.warn('KafkaService: Repositories not set, skipping background agent event');
-            return;
-        }
-        const chat = await this.chatRepository.findById(chatId);
-        if (!chat) {
-            return;
-        }
-        const messages = await this.messageRepository.findByChatId(chatId);
-        if (messages.length === 0) {
-            return;
-        }
-        // TODO: Implement background agent processing
-    }
-    async connect() {
-        await this.kafkaConsumer.connect();
-        await this.kafkaConsumer.subscribe(config.kafka.chatTitleGenerationTopic, this.onChatTitleGenerationEvent.bind(this));
-        await this.kafkaConsumer.subscribe(config.kafka.backgroundChatProcessingTopic, this.onBackgroundAgentEvent.bind(this));
-        await this.kafkaConsumer.listen();
-    }
-    async disconnect() {
-        await this.kafkaConsumer.disconnect();
-    }
-}
-export const kafkaService = new KafkaService();
-//# sourceMappingURL=KafkaService.js.map

package/dist/esm/services/KafkaService.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"KafkaService.js","sourceRoot":"","sources":["../../../src/services/KafkaService.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,aAAa,EAAE,aAAa,EAAE,MAAM,wBAAwB,CAAA;AACrE,OAAO,EAAE,MAAM,EAAE,MAAM,cAAc,CAAA;AAErC,OAAO,EAAE,QAAQ,EAAE,MAAM,wBAAwB,CAAA;AACjD,OAAO,EAAC,MAAM,EAAC,MAAM,yBAAyB,CAAC;AAC/C,OAAO,EAAC,aAAa,EAAC,MAAM,oBAAoB,CAAC;AAGjD,IAAK,6BAEJ;AAFD,WAAK,6BAA6B;IAChC,gDAAe,CAAA;AACjB,CAAC,EAFI,6BAA6B,KAA7B,6BAA6B,QAEjC;AAED,MAAM,YAAY;IAUhB,YAAY,cAAoC,EAAE,iBAA0C;QAJ3E,gBAAW,GAAG,CAAC,CAAA;QACf,kBAAa,GAAG,IAAI,CAAA,CAAC,sBAAsB;QAC3C,iBAAY,GAAG,KAAK,CAAA,CAAC,uBAAuB;QAG3D,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,MAAA,OAAO,CAAC,UAAU,mCAAI,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"}