|
@@ -65,7 +65,6 @@ async def handle_client(websocket: WebSocket,
|
|
|
})
|
|
})
|
|
|
async for rag_response in ragflow_service.chat(token, chat_id, chat_history):
|
|
async for rag_response in ragflow_service.chat(token, chat_id, chat_history):
|
|
|
try:
|
|
try:
|
|
|
- print(f"Received from ragflow: {rag_response}")
|
|
|
|
|
if rag_response[:5] == "data:":
|
|
if rag_response[:5] == "data:":
|
|
|
# 如果是,则截取掉前5个字符,并去除首尾空白符
|
|
# 如果是,则截取掉前5个字符,并去除首尾空白符
|
|
|
text = rag_response[5:].strip()
|
|
text = rag_response[5:].strip()
|
|
@@ -83,10 +82,11 @@ async def handle_client(websocket: WebSocket,
|
|
|
else: # 正常输出
|
|
else: # 正常输出
|
|
|
answer = data.get("answer", "")
|
|
answer = data.get("answer", "")
|
|
|
result = {"message": answer, "type": "message"}
|
|
result = {"message": answer, "type": "message"}
|
|
|
|
|
+ await websocket.send_json(result)
|
|
|
except json.JSONDecodeError:
|
|
except json.JSONDecodeError:
|
|
|
- result = {"message": text, "type": "message"}
|
|
|
|
|
- await websocket.send_json(result)
|
|
|
|
|
- print(f"Forwarded to client {chat_id}: {result}")
|
|
|
|
|
|
|
+ print(f"Error decode ragflow response: {text}")
|
|
|
|
|
+ pass
|
|
|
|
|
+
|
|
|
except Exception as e:
|
|
except Exception as e:
|
|
|
result = {"message": f"内部错误: {e}", "type": "close"}
|
|
result = {"message": f"内部错误: {e}", "type": "close"}
|
|
|
await websocket.send_json(result)
|
|
await websocket.send_json(result)
|