소스 검색

fix reference为对象

xuyonghao 1 년 전
부모
커밋
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}")