From 3cee1dd5768ae10d3834afd3294850af7d626633 Mon Sep 17 00:00:00 2001 From: zhangqian <zhangqian@123.com> Date: 星期五, 18 十月 2024 10:17:57 +0800 Subject: [PATCH] fix --- app/api/chat.py | 8 ++++++-- 1 files changed, 6 insertions(+), 2 deletions(-) diff --git a/app/api/chat.py b/app/api/chat.py index a847431..1e9218c 100644 --- a/app/api/chat.py +++ b/app/api/chat.py @@ -58,7 +58,11 @@ await websocket.close() return else: - chat_history.append(message) + chat_history.append({ + "content": message["message"], + "doc_ids": message.get("doc_ids", []), + "role": "user" + }) async for rag_response in ragflow_service.chat(token, chat_id, chat_history): try: print(f"Received from ragflow: {rag_response}") @@ -87,6 +91,7 @@ result = {"message": f"鍐呴儴閿欒锛� {e}", "type": "close"} await websocket.send_json(result) print(f"Error process message of ragflow: {e}") + # 鍚姩浠诲姟澶勭悊瀹㈡埛绔秷鎭� tasks = [ asyncio.create_task(forward_to_ragflow()) @@ -155,4 +160,3 @@ else: ret = {"message": "Agent not found", "type": "close"} await websocket.send_json(ret) - -- Gitblit v1.8.0