xuyonghao 1 год назад
Родитель
Сommit
53471a68e4
1 измененных файлов с 2 добавлено и 2 удалено
  1. 2 2
      app/api/chat.py

+ 2 - 2
app/api/chat.py

@@ -82,8 +82,8 @@ async def handle_client(websocket: WebSocket,
                                     result = {"message": "内部错误:" + answer, "type": "message"}
                                 else:  # 正常输出
                                     answer = data.get("answer", "")
-                                    reference = data.get("reference", "")
-                                    result = {"message": answer, "type": "message", "reference": reference }
+                                    reference = data.get("reference", {})
+                                    result = {"message": answer, "type": "message", "reference": reference}
                                 await websocket.send_json(result)
                             except json.JSONDecodeError:
                                 print(f"Error decode ragflow response: {text}")