From 30ff0afd5d76a3a5aa48058210ae411253574ada Mon Sep 17 00:00:00 2001
From: zhaoqingang <zhaoqg0118@163.com>
Date: 星期四, 13 三月 2025 14:55:30 +0800
Subject: [PATCH] 增加文件多轮问答
---
app/models/v2/chat.py | 7 +++++++
1 files changed, 7 insertions(+), 0 deletions(-)
diff --git a/app/models/v2/chat.py b/app/models/v2/chat.py
index e41edcf..7aed562 100644
--- a/app/models/v2/chat.py
+++ b/app/models/v2/chat.py
@@ -23,6 +23,7 @@
class ChatDataRequest(BaseModel):
sessionId: str
+ parentId: Optional[str] = ""
query: str
chatMode: Optional[int] = 1 # 1= 鏅�氬璇濓紝2=鑱旂綉锛�3=鐭ヨ瘑搴�,4=娣卞害
isDeep: Optional[int] = 1 # 1= 鏅��, 2=娣卞害
@@ -40,6 +41,7 @@
"files": self.files,
"isDeep": self.isDeep,
"optimizeType": self.optimizeType,
+ "parentId": self.parentId,
}
@@ -182,11 +184,16 @@
'content': self.content,
}
else:
+ query = {}
+ if self.query:
+ query = json.loads(self.query)
return {
'id': self.id,
'role': "assistant",
'answer': self.content,
+ 'chat_mode': self.chat_mode,
'node_list': json.loads(self.node_data) if self.node_data else [],
+ "parentId": query.get("parentId")
}
--
Gitblit v1.8.0