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) } ] })