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/app_driver/chat_dialog.py | 13 ++++++++++---
1 files changed, 10 insertions(+), 3 deletions(-)
diff --git a/app/service/v2/app_driver/chat_dialog.py b/app/service/v2/app_driver/chat_dialog.py
index bfbe645..9aa750e 100644
--- a/app/service/v2/app_driver/chat_dialog.py
+++ b/app/service/v2/app_driver/chat_dialog.py
@@ -22,6 +22,8 @@
yield json_data
except json.JSONDecodeError as e:
+ # print(e)
+ # print(complete_response)
logger.info("Invalid JSON data------------------")
# print(e)
@@ -43,9 +45,14 @@
"session_id": session_id
}
-
-
-
+ @staticmethod
+ async def complex_request_data(question, dataset_ids, session_id=""):
+ return {
+ "question": question,
+ "stream": True,
+ "session_id": session_id,
+ "kb_ids": dataset_ids
+ }
if __name__ == "__main__":
--
Gitblit v1.8.0