From b663fe4535cdb13a28936ea21d908e9191a64c16 Mon Sep 17 00:00:00 2001
From: zhaoqingang <zhaoqg0118@163.com>
Date: 星期一, 31 三月 2025 11:00:57 +0800
Subject: [PATCH] 问题优化

---
 app/models/v2/chat.py |    4 ++--
 1 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/app/models/v2/chat.py b/app/models/v2/chat.py
index 1397edc..1ee30d2 100644
--- a/app/models/v2/chat.py
+++ b/app/models/v2/chat.py
@@ -6,7 +6,7 @@
 from sqlalchemy import Column, Integer, String, BigInteger, ForeignKey, DateTime, Text, TEXT
 from sqlalchemy.orm import Session
 
-from app.config.const import Dialog_STATSU_DELETE, Dialog_STATSU_ON, complex_knowledge_chat
+from app.config.const import Dialog_STATSU_DELETE, Dialog_STATSU_ON, complex_knowledge_chat, complex_knowledge_chat_deep
 from app.models.base_model import Base
 from app.utils.common import current_time
 
@@ -198,7 +198,7 @@
                 "isDeep": query.get("isDeep", 1),
                 "mindmap": True if self.mindmap else False,
             }
-            if self.chat_mode == complex_knowledge_chat:
+            if self.chat_mode == complex_knowledge_chat or self.chat_mode == complex_knowledge_chat_deep:
                 res['reference'] = json.loads(self.node_data) if self.node_data else {}
             else:
                 res['node_list'] = json.loads(self.node_data) if self.node_data else []

--
Gitblit v1.8.0