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