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');
 };