Bladeren bron

ragflow返回的消息改个类型

zhangqian 1 jaar geleden
bovenliggende
commit
39eba302f3
1 gewijzigde bestanden met toevoegingen van 3 en 3 verwijderingen
  1. 3 3
      app/api/chat.py

+ 3 - 3
app/api/chat.py

@@ -79,12 +79,12 @@ async def handle_client(websocket: WebSocket,
                                     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:  # 正常输出
                                     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: