From 6bac1630e5af5890a6922bdc624e591eb19a12eb Mon Sep 17 00:00:00 2001
From: zhaoqingang <zhaoqg0118@163.com>
Date: 星期四, 13 三月 2025 18:36:07 +0800
Subject: [PATCH] 知识库对接rg
---
app/service/v2/mindmap.py | 8 ++++++--
1 files changed, 6 insertions(+), 2 deletions(-)
diff --git a/app/service/v2/mindmap.py b/app/service/v2/mindmap.py
index dd5b994..f6e576c 100644
--- a/app/service/v2/mindmap.py
+++ b/app/service/v2/mindmap.py
@@ -73,8 +73,8 @@
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 +96,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 +117,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