From da346d781499f01a2dfad5fcc4187a81c4954dfe Mon Sep 17 00:00:00 2001
From: "nikos.tsiougkos" <nikos.tsiougkos@keragon.com>
Date: Wed, 24 Jan 2024 18:05:32 +0200
Subject: [PATCH] include chat message id in result

---
 packages/server/src/index.ts | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/packages/server/src/index.ts b/packages/server/src/index.ts
index b11653ee..6d4c1887 100644
--- a/packages/server/src/index.ts
+++ b/packages/server/src/index.ts
@@ -1814,7 +1814,8 @@ export class App {
             if (result?.sourceDocuments) apiMessage.sourceDocuments = JSON.stringify(result.sourceDocuments)
             if (result?.usedTools) apiMessage.usedTools = JSON.stringify(result.usedTools)
             if (result?.fileAnnotations) apiMessage.fileAnnotations = JSON.stringify(result.fileAnnotations)
-            await this.addChatMessage(apiMessage)
+            const chatMessage = await this.addChatMessage(apiMessage)
+            result.chatMessageId = chatMessage.id
 
             logger.debug(`[server]: Finished running ${nodeToExecuteData.label} (${nodeToExecuteData.id})`)
             await this.telemetry.sendTelemetry('prediction_sent', {