From 345f2822f5072c34d9f969ae077830af968043b9 Mon Sep 17 00:00:00 2001 From: zhaoqingang <zhaoqg0118@163.com> Date: 星期四, 27 二月 2025 11:44:31 +0800 Subject: [PATCH] 优化输出 --- app/api/chat.py | 4 ++-- 1 files changed, 2 insertions(+), 2 deletions(-) diff --git a/app/api/chat.py b/app/api/chat.py index 241815a..16003e2 100644 --- a/app/api/chat.py +++ b/app/api/chat.py @@ -325,7 +325,7 @@ # token = get_dify_token(db, current_user.id) try: async def forward_to_dify(): - if agent.type == "imageTalk": + if chat_type == "imageTalk": token = DfTokenDao(db).get_token_by_id(IMAGE_TO_TEXT) if not token: await websocket.send_json({"message": "Invalid token", "type": "error"}) @@ -431,7 +431,7 @@ result = {"message": f"鍐呴儴閿欒锛� {e2}", "type": "close"} await websocket.send_json(result) print(f"Error process message of ragflow: {e2}") - elif agent.type == "reportWorkflow": + elif chat_type == "reportWorkflow": while True: receive_message = await websocket.receive_json() -- Gitblit v1.8.0