|
|
@@ -721,6 +721,9 @@ async def handle_client(websocket: WebSocket,
|
|
|
# 正常输出
|
|
|
answer = data.get("answer", "")
|
|
|
result = {"message": answer, "type": "stream"}
|
|
|
+ elif data.get("event") == "error":
|
|
|
+ answer = data.get("message", "")
|
|
|
+ result = {"message": answer, "type": "system"}
|
|
|
elif data.get("event") == "workflow_finished":
|
|
|
answer = data.get("data", "")
|
|
|
if isinstance(answer, str):
|
|
|
@@ -873,6 +876,9 @@ async def handle_client(websocket: WebSocket,
|
|
|
# 正常输出
|
|
|
answer = data.get("answer", "")
|
|
|
result = {"message": answer, "type": "stream"}
|
|
|
+ elif data.get("event") == "error":
|
|
|
+ answer = data.get("message", "")
|
|
|
+ result = {"message": answer, "type": "system"}
|
|
|
elif data.get("event") == "workflow_finished":
|
|
|
answer = data.get("data", "")
|
|
|
if isinstance(answer, str):
|
|
|
@@ -888,8 +894,8 @@ async def handle_client(websocket: WebSocket,
|
|
|
else:
|
|
|
message = answer.get("error", "")
|
|
|
|
|
|
- # result = {"message": message, "type": "message",
|
|
|
- # "download_url": download_url}
|
|
|
+ result = {"message": message, "type": "system",
|
|
|
+ "download_url": download_url}
|
|
|
try:
|
|
|
SessionService(db).update_session(chat_id,
|
|
|
message={"role": "assistant",
|
|
|
@@ -902,7 +908,7 @@ async def handle_client(websocket: WebSocket,
|
|
|
logger.error("保存dify的会话异常！")
|
|
|
logger.error(e)
|
|
|
# await websocket.send_json(result)
|
|
|
- continue
|
|
|
+ # continue
|
|
|
elif data.get("event") == "message_end":
|
|
|
result = {"message": "", "type": "close"}
|
|
|
|
zhaoqingang