From 42cf97314526722d3151f6a66cc2eab83584ec99 Mon Sep 17 00:00:00 2001
From: Liuyang <2746366019@qq.com>
Date: Tue, 11 Mar 2025 17:55:19 +0800
Subject: [PATCH] =?UTF-8?q?refactor(module-llm):=E4=BC=98=E5=8C=96?=
 =?UTF-8?q?=E7=9F=A5=E8=AF=86=E5=BA=93=E5=AD=97=E7=AC=A6=E4=B8=B2=E5=A4=84?=
 =?UTF-8?q?=E7=90=86=E9=80=BB=E8=BE=91?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

- 移除了不必要的空字符串赋值操作
- 添加了对 knowledgeBase
---
 .../conversation/ConversationServiceImpl.java      | 14 +++++++-------
 1 file changed, 7 insertions(+), 7 deletions(-)

diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/conversation/ConversationServiceImpl.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/conversation/ConversationServiceImpl.java
index c9123e5ed..f5e343ade 100644
--- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/conversation/ConversationServiceImpl.java
+++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/conversation/ConversationServiceImpl.java
@@ -434,19 +434,19 @@ public class ConversationServiceImpl implements ConversationService {
         String knowledgeBaseString="";
         if (chatReqVO.getKnowledge() != null) {
             StringBuilder knowledgeBase = getKnowledgeBase(chatReqVO);
-             knowledgeBaseString = knowledgeBase.toString();
+            knowledgeBaseString = knowledgeBase.toString();
 
-             if (org.apache.commons.lang3.StringUtils.isBlank(knowledgeBaseString)){
-                 knowledgeBaseString="";
-             }
+            // 处理 knowledgeBaseString
+            if (StringUtils.isNotBlank(knowledgeBaseString)) {
+                knowledgeBaseString = "<context>" + knowledgeBaseString + "</context>";
+            }
 
             // 处理 systemPrompt
-             systemPrompt = StringUtils.isBlank(chatReqVO.getSystemPrompt())
+            systemPrompt = StringUtils.isBlank(chatReqVO.getSystemPrompt())
                     ? PROMPT
                     : chatReqVO.getSystemPrompt() + "\n" + PROMPT;
-
         }
-        String mess = systemPrompt +  "<content>"+knowledgeBaseString+"</content>";
+        String mess = systemPrompt + knowledgeBaseString;
 
         // 查询历史记录消息，并将查询出来的知识信息放入到 role = system 的消息中
         List<String> messageHistoryList = stringRedisTemplate.opsForList().range(CHAT_HIStORY_REDIS_KEY + ":" + chatReqVO.getUuid(), 0, -1);