From af86455055918d26a0f6eebc270074c4863db0be Mon Sep 17 00:00:00 2001
From: zhaoqingang <zhaoqg0118@163.com>
Date: 星期五, 07 三月 2025 09:57:19 +0800
Subject: [PATCH] 增加深度思考逻辑
---
app/service/v2/mindmap.py | 9 +++++++--
1 files changed, 7 insertions(+), 2 deletions(-)
diff --git a/app/service/v2/mindmap.py b/app/service/v2/mindmap.py
index dd5b994..79620f5 100644
--- a/app/service/v2/mindmap.py
+++ b/app/service/v2/mindmap.py
@@ -73,8 +73,9 @@
token = await get_chat_token(db, session.chat_id)
chat = ChatAgent()
url = settings.dify_base_url + DF_CHAT_AGENT
+ chat_request = json.loads(session.query)
if session.mindmap:
- chat_request = json.loads(session.query)
+
inputs = {"is_deep": chat_request.get("isDeep", 1)}
if session.chat_mode == complex_knowledge_chat:
inputs["query_json"] = json.dumps(
@@ -96,9 +97,12 @@
return res
else:
mindmap_query = session.content
-
+ # print("-----------------", mindmap_query)
try:
+ if chat_request.get("isDeep", 1) == 2:
+ mindmap_query = mindmap_query.split("</think>")[-1]
mindmap_str = ""
+ # print("mindmap_query", mindmap_query)
token = await get_chat_token(db, mindmap_chat_id)
async for ans in chat.chat_completions(url,
await chat.complex_request_data(mindmap_query, "",
@@ -114,6 +118,7 @@
except Exception as e:
logger.error(e)
return res
+ # print(mindmap_str)
if "```json" in mindmap_str:
mindmap_list = mindmap_str.split("```")
mindmap_str = mindmap_list[1].lstrip("json")
--
Gitblit v1.8.0