From 5ef590b70cc8e2de16083af2ee2d977daae5587c Mon Sep 17 00:00:00 2001 From: zhaoqingang <zhaoqg0118@163.com> Date: 星期二, 19 十一月 2024 16:34:29 +0800 Subject: [PATCH] 会话缓存本地数据库 --- app/api/chat.py | 10 ++++++++-- 1 files changed, 8 insertions(+), 2 deletions(-) diff --git a/app/api/chat.py b/app/api/chat.py index fe86fb5..ea1be48 100644 --- a/app/api/chat.py +++ b/app/api/chat.py @@ -10,6 +10,7 @@ from app.models.agent_model import AgentModel, AgentType from app.models.base_model import get_db from app.models.user_model import UserModel +from app.service.dialog import update_session_history from app.service.ragflow import RagflowService from app.service.service_token import get_bisheng_token, get_ragflow_token @@ -44,6 +45,7 @@ try: async def forward_to_ragflow(): while True: + is_new = False message = await websocket.receive_json() print(f"Received from client {chat_id}: {message}") chat_history = message.get('chatHistory', []) @@ -51,9 +53,10 @@ if len(chat_history) == 0: chat_history = await ragflow_service.get_session_history(token, chat_id) if len(chat_history) == 0: + is_new = True chat_history = await ragflow_service.set_session(token, agent_id, message, chat_id, True) - print("chat_history------------------------", chat_history) + # print("chat_history------------------------", chat_history) if len(chat_history) == 0: result = {"message": "鍐呴儴閿欒锛氬垱寤轰細璇濆け璐�", "type": "close"} await websocket.send_json(result) @@ -91,11 +94,14 @@ complete_response = "" except json.JSONDecodeError as e: print(f"Error decoding JSON: {e}") - print(f"Response text: {text}") + # print(f"Response text: {text}") except Exception as e2: result = {"message": f"鍐呴儴閿欒锛� {e2}", "type": "close"} await websocket.send_json(result) print(f"Error process message of ragflow: {e2}") + dialog_chat_history = await ragflow_service.get_session_history(token, chat_id, 1) + await update_session_history(db, dialog_chat_history, current_user.id, is_new) + # 鍚姩浠诲姟澶勭悊瀹㈡埛绔秷鎭� tasks = [ asyncio.create_task(forward_to_ragflow()) -- Gitblit v1.8.0