From 9ea07e00fc8b92e9b75849859c1ecce77c7096d1 Mon Sep 17 00:00:00 2001
From: zhaoqingang <zhaoqg0118@163.com>
Date: 星期四, 10 四月 2025 13:31:23 +0800
Subject: [PATCH] 优化
---
app/models/v2/chat.py | 12 +++++++-----
1 files changed, 7 insertions(+), 5 deletions(-)
diff --git a/app/models/v2/chat.py b/app/models/v2/chat.py
index e7fa060..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
@@ -177,16 +177,17 @@
}
def log_to_json(self):
+ query = {}
+ if self.query:
+ query = json.loads(self.query)
if self.message_type == 1:
return {
'id': self.id,
'role': "user",
'content': self.content,
+ 'files': query.get("files", []),
}
else:
- query = {}
- if self.query:
- query = json.loads(self.query)
res = {
'id': self.id,
@@ -195,8 +196,9 @@
'chat_mode': self.chat_mode,
"parentId": query.get("parentId"),
"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