zhangqian
2024-10-18 39eba302f3887d83ce0584b7e79440a92f9037c2
ragflow返回的消息改个类型
1个文件已修改
6 ■■■■ 已修改文件
app/api/chat.py 6 ●●●● 补丁 | 查看 | 原始文档 | blame | 历史
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:  # 正常输出
                                    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: