From bd0c9bc825d92048bd0875112ff6e4dd3b780ba5 Mon Sep 17 00:00:00 2001
From: n4ze3m <n4ze3m@gmail.com>
Date: Sun, 9 Feb 2025 12:32:18 +0530
Subject: [PATCH] feat: remove reasoning from message content in history
 generation

---
 src/libs/reasoning.ts         | 2 +-
 src/utils/generate-history.ts | 5 +++--
 2 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/src/libs/reasoning.ts b/src/libs/reasoning.ts
index d6c1b23..883995e 100644
--- a/src/libs/reasoning.ts
+++ b/src/libs/reasoning.ts
@@ -96,5 +96,5 @@ export function mergeReasoningContent(
 
   originalText = originalText.replace(reasoningTag, "")
 
-  return `${reasoningTag}${originalText + reasoning}`.trim()
+  return `${reasoningTag}${originalText + reasoning}`
 }
diff --git a/src/utils/generate-history.ts b/src/utils/generate-history.ts
index cb4b466..312e075 100644
--- a/src/utils/generate-history.ts
+++ b/src/utils/generate-history.ts
@@ -1,4 +1,5 @@
 import { isCustomModel } from "@/db/models"
+import { removeReasoning } from "@/libs/reasoning"
 import {
   HumanMessage,
   AIMessage,
@@ -51,11 +52,11 @@ export const generateHistory = (
       history.push(
         new AIMessage({
           content: isCustom
-            ? message.content
+            ? removeReasoning(message.content)
             : [
                 {
                   type: "text",
-                  text: message.content
+                  text: removeReasoning(message.content)
                 }
               ]
         })