From 48e2e613bbb322650f78c26dc497d26eacd83bdc Mon Sep 17 00:00:00 2001
From: lanceyq <1982376970@qq.com>
Date: Mon, 23 Mar 2026 17:34:54 +0800
Subject: [PATCH] =?UTF-8?q?=E3=80=90change=E3=80=91Restore=20chat=20mode?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../core/memory/llm_tools/openai_client.py    | 19 +++++--------------
 1 file changed, 5 insertions(+), 14 deletions(-)

diff --git a/api/app/core/memory/llm_tools/openai_client.py b/api/app/core/memory/llm_tools/openai_client.py
index 4536f62d..0b75de3a 100644
--- a/api/app/core/memory/llm_tools/openai_client.py
+++ b/api/app/core/memory/llm_tools/openai_client.py
@@ -82,26 +82,17 @@ class OpenAIClient(LLMClient):
             LLMClientException: LLM 调用失败
         """
         try:
-            from langchain_core.messages import HumanMessage, SystemMessage, AIMessage
-
-            # 将 dict 消息列表转换为 LangChain 消息对象
-            lc_messages = []
-            for m in messages:
-                role = m.get("role", "user")
-                content = m.get("content", "")
-                if role == "system":
-                    lc_messages.append(SystemMessage(content=content))
-                elif role == "assistant":
-                    lc_messages.append(AIMessage(content=content))
-                else:
-                    lc_messages.append(HumanMessage(content=content))
+            # 使用 Langfuse 回调（如果可用）
+            template = """{messages}"""
+            prompt = ChatPromptTemplate.from_template(template)
+            chain = prompt | self.client
 
             # 添加 Langfuse 回调（如果可用）
             config = {}
             if self.langfuse_handler:
                 config["callbacks"] = [self.langfuse_handler]
 
-            response = await self.client.ainvoke(lc_messages, config=config)
+            response = await chain.ainvoke({"messages": messages}, config=config)
             return response
 
         except Exception as e: