From e078028f1a8da34f3cc2cb5095b8e103a996f553 Mon Sep 17 00:00:00 2001
From: zhaoqingang <zhaoqg0118@163.com>
Date: 星期五, 11 四月 2025 10:16:42 +0800
Subject: [PATCH] 首页智能体中心过滤掉自研类型
---
app/service/v2/app_driver/chat_dialog.py | 38 +++++++++++++++++++++++++-------------
1 files changed, 25 insertions(+), 13 deletions(-)
diff --git a/app/service/v2/app_driver/chat_dialog.py b/app/service/v2/app_driver/chat_dialog.py
index 27807fd..9aa750e 100644
--- a/app/service/v2/app_driver/chat_dialog.py
+++ b/app/service/v2/app_driver/chat_dialog.py
@@ -6,22 +6,10 @@
class ChatDialog(ChatBase):
-
- def __init__(self, token):
- self.token = token
-
-
- async def get_headers(self):
- return {
- 'Content-Type': 'application/json',
- 'Authorization': f'Bearer {self.token}'
- }
-
-
async def chat_completions(self, url, data, headers):
complete_response = ""
async for line in self.http_stream(url, data, headers):
- # logger.error(line)
+ # print(line)
if line.startswith("data:"):
complete_response = line.strip("data:").strip()
else:
@@ -34,13 +22,37 @@
yield json_data
except json.JSONDecodeError as e:
+ # print(e)
+ # print(complete_response)
logger.info("Invalid JSON data------------------")
# print(e)
+ async def chat_sessions(self, url, data, headers):
+
+ res = await self.http_post(url, data, headers)
+ if res.status_code == 200:
+ return res.json()
+ else:
+ return {}
+ @staticmethod
+ async def request_data(question, session_id=""):
+ return {
+ "question": question,
+ "stream": True,
+ "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