From 39eba302f3887d83ce0584b7e79440a92f9037c2 Mon Sep 17 00:00:00 2001
From: zhangqian <zhangqian@123.com>
Date: 星期五, 18 十月 2024 11:23:19 +0800
Subject: [PATCH] ragflow返回的消息改个类型

---
 app/api/chat.py |    6 +++---
 1 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/app/api/chat.py b/app/api/chat.py
index 1e9218c..07a05da 100644
--- a/app/api/chat.py
+++ b/app/api/chat.py
@@ -79,12 +79,12 @@
                                     result = {"message": "", "type": "close"}
                                 elif data is None:  # 鍙戠敓閿欒
                                     answer = json_data.get("retmsg", json_data.get("retcode"))
-                                    result = {"message": "鍐呴儴閿欒锛�" + answer, "type": "stream"}
+                                    result = {"message": "鍐呴儴閿欒锛�" + answer, "type": "message"}
                                 else:  # 姝e父杈撳嚭
                                     answer = data.get("answer", "")
-                                    result = {"message": answer, "type": "stream"}
+                                    result = {"message": answer, "type": "message"}
                             except json.JSONDecodeError:
-                                result = {"message": text, "type": "stream"}
+                                result = {"message": text, "type": "message"}
                             await websocket.send_json(result)
                             print(f"Forwarded to client {chat_id}: {result}")
                         except Exception as e:

--
Gitblit v1.8.0