From e65b52a457c2b92b9ae595287c14f438e3fb0b5f Mon Sep 17 00:00:00 2001 From: cyri113 Date: Fri, 4 Apr 2025 16:09:18 +0200 Subject: [PATCH] add chat id --- src/controllers/hivemind.controller.ts | 4 ++-- src/services/temporal/hivemind.service.ts | 4 +++- 2 files changed, 5 insertions(+), 3 deletions(-) diff --git a/src/controllers/hivemind.controller.ts b/src/controllers/hivemind.controller.ts index 3d20614..5752ff1 100644 --- a/src/controllers/hivemind.controller.ts +++ b/src/controllers/hivemind.controller.ts @@ -6,10 +6,10 @@ import HivemindTemporalService from '../services/temporal/hivemind.service'; import { catchAsync } from '../utils'; const askQuestion = catchAsync(async function (req: IAuthRequest, res: Response) { - console.debug('Body', req.body.communityId, req.body.question); + console.debug('Body', req.body.communityId, req.body.question, req.body.chatId); req.setTimeout(6 * 60 * 1000); res.setTimeout(6 * 60 * 1000); - const answer = await HivemindTemporalService.triggerWorkflow(req.body.communityId, req.body.question, false); + const answer = await HivemindTemporalService.triggerWorkflow(req.body.communityId, req.body.question, false, req.body.chatId); res.status(httpStatus.OK).send({ answer }); }); diff --git a/src/services/temporal/hivemind.service.ts b/src/services/temporal/hivemind.service.ts index f3fee86..db924bb 100644 --- a/src/services/temporal/hivemind.service.ts +++ b/src/services/temporal/hivemind.service.ts @@ -9,6 +9,7 @@ interface HivemindPayload { community_id: string; query: string; enable_answer_skipping: boolean; + chat_id: string; } class HivemindTemporalService extends TemporalCoreService { @@ -24,7 +25,7 @@ class HivemindTemporalService extends TemporalCoreService { * @param enableAnswerSkipping - Flag indicating if answer skipping is enabled. * @returns The workflow ID of the started workflow. */ - public async triggerWorkflow(communityId: string, query: string, enableAnswerSkipping: boolean) { + public async triggerWorkflow(communityId: string, query: string, enableAnswerSkipping: boolean, chatId: string = '') { const client: Client = await this.getClient(); // // Construct the payload as specified @@ -32,6 +33,7 @@ class HivemindTemporalService extends TemporalCoreService { community_id: communityId, query: query, enable_answer_skipping: enableAnswerSkipping, + chat_id: chatId, }; try { const hivemindTaskQueue = 'HIVEMIND_AGENT_QUEUE';