From 7a9f9cb3c34ebc0836720d0f5738bbe70070bbb3 Mon Sep 17 00:00:00 2001
From: zhaoqingang <zhaoqg0118@163.com>
Date: 星期四, 05 十二月 2024 14:22:24 +0800
Subject: [PATCH] 增加菜单和路由控制

---
 app/service/bisheng.py |   22 +++++++---------------
 1 files changed, 7 insertions(+), 15 deletions(-)

diff --git a/app/service/bisheng.py b/app/service/bisheng.py
index 9be6692..d4ff068 100644
--- a/app/service/bisheng.py
+++ b/app/service/bisheng.py
@@ -90,18 +90,11 @@
                         name = message_json['query']
                     elif 'report_name' in message_json:
                         name = message_json['report_name']
-                except json.JSONDecodeError:
+                except Exception as e:
                     pass
                 if not name:
-                    name =  item.get("flow_name")
-                return name
-
-            def process_name_report(item):
-                # logger.error("-----------------------process_name-------------------------------------")
-                # logger.error(item)
-                name = item.get("flow_name", "鎶ュ憡鐢熸垚")
-
-                return name
+                    name = item.get("flow_name")
+                return name[:50]
 
             result = [
                 {
@@ -126,16 +119,15 @@
         async with httpx.AsyncClient() as client:
             response = await client.get(url, headers=headers)
             response.raise_for_status()
-            # print(response.text)
             data = self._check_response(response)
             session_log = [
                 {
-                    "message": message.get("intermediate_steps", ""),
-                    "intermediate_steps": message.get("message", ""),
-                    "role": message.get("category"),
+                    "message":message.get("message", "") if message.get("message", "") else message.get("intermediate_steps", ""),
+                    "files": message.get("files", ""),
+                    "role": "question" if message.get("category") == "question" and message.get("message", "") else "answer",
                     "ts": message.get("create_time")
                 }
-                for message in data
+                for message in data if message.get("category") != "system"
             ]
 
             # 鎶妔ession_log 鎸塼s 鍗囧簭鎺掑簭

--
Gitblit v1.8.0