From 37022fb11efacb2348a92c17e46e42cedb20cbcc Mon Sep 17 00:00:00 2001
From: ItzCrazyKns <95534749+ItzCrazyKns@users.noreply.github.com>
Date: Tue, 22 Jul 2025 21:20:16 +0530
Subject: [PATCH] feat(format-history): update role determination

---
 src/lib/utils/formatHistory.ts | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/src/lib/utils/formatHistory.ts b/src/lib/utils/formatHistory.ts
index 6d0d309..6fdede6 100644
--- a/src/lib/utils/formatHistory.ts
+++ b/src/lib/utils/formatHistory.ts
@@ -1,8 +1,11 @@
-import { BaseMessage } from '@langchain/core/messages';
+import { BaseMessage, isAIMessage } from '@langchain/core/messages';
 
 const formatChatHistoryAsString = (history: BaseMessage[]) => {
   return history
-    .map((message) => `${message._getType()}: ${message.content}`)
+    .map(
+      (message) =>
+        `${isAIMessage(message) ? 'AI' : 'User:'}: ${message.content}`,
+    )
     .join('\n');
 };