From 7c2b9e0d412cae6157bbdd4e4e6f21a33978dd09 Mon Sep 17 00:00:00 2001
From: Liuyang <2746366019@qq.com>
Date: Mon, 3 Mar 2025 10:54:58 +0800
Subject: [PATCH] =?UTF-8?q?refactor(yudao-module-llm):=20=E4=BC=98?=
 =?UTF-8?q?=E5=8C=96=E6=A8=A1=E5=9E=8B=E6=9C=8D=E5=8A=A1=E4=B8=AD=E7=9A=84?=
 =?UTF-8?q?=E6=95=B0=E6=8D=AE=E5=A4=84=E7=90=86=E9=80=BB=E8=BE=91?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

- 在处理数据行时，增加了对 "</think>"标记的检查
- 仅当数据字符串不包含 "[DONE]" 或 "</think>" 时，才解析为 JSON 对象
---
 .../cn/iocoder/yudao/module/llm/service/http/ModelService.java  | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/http/ModelService.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/http/ModelService.java
index e017bfc57..21bdbd881 100644
--- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/http/ModelService.java
+++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/http/ModelService.java
@@ -320,7 +320,7 @@ public class ModelService {
         if (StringUtils.isNotBlank(line)) {
             if (line.startsWith("data: ")) {
                 String dataString = extractJsonFromDataString(line);
-                if (!dataString.contains("[DONE]")) {
+                if (!dataString.contains("[DONE]")||!dataString.contains("</think>")) {
                     JSONObject jsonObject = JSON.parseObject(dataString);
                     // 获取 choices 数组
                     JSONArray choicesArray = jsonObject.getJSONArray("choices");