From d4ab4357a4eaf61dcab6bba427e620b9c5333e42 Mon Sep 17 00:00:00 2001
From: zhaoqingang <zhaoqg0118@163.com>
Date: 星期四, 13 二月 2025 17:45:06 +0800
Subject: [PATCH] tmp

---
 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 2d2845d..8845322 100644
--- a/app/api/chat.py
+++ b/app/api/chat.py
@@ -323,7 +323,7 @@
         # token = get_dify_token(db, current_user.id)
         try:
             async def forward_to_dify():
-                if chat_type == "imageTalk":
+                if agent.type == "imageTalk":
                     token = DfTokenDao(db).get_token_by_id(IMAGE_TO_TEXT)
                     if not token:
                         await websocket.send_json({"message": "Invalid token", "type": "error"})
@@ -429,7 +429,7 @@
                                 result = {"message": f"鍐呴儴閿欒锛� {e2}", "type": "close"}
                                 await websocket.send_json(result)
                                 print(f"Error process message of ragflow: {e2}")
-                elif chat_type == "reportWorkflow":
+                elif agent.type == "reportWorkflow":
 
                     while True:
                         receive_message = await websocket.receive_json()

--
Gitblit v1.8.0