From f249dda1376dd5a525e74617900351c5bbc61705 Mon Sep 17 00:00:00 2001
From: zhaoqingang <zhaoqg0118@163.com>
Date: 星期一, 02 十二月 2024 09:26:18 +0800
Subject: [PATCH] 报告生成
---
app/api/chat.py | 18 ++++++++++++------
1 files changed, 12 insertions(+), 6 deletions(-)
diff --git a/app/api/chat.py b/app/api/chat.py
index f21f07c..51c8409 100644
--- a/app/api/chat.py
+++ b/app/api/chat.py
@@ -255,26 +255,32 @@
logger.error("---------------------excel_talk-----------------------------")
excel_url = ""
image_url = ""
+ image_name = ""
+ excel_name = ""
async for data in service.excel_talk(question, chat_id):
# logger.error(data)
output = data.get("output", "")
- excel_name = data.get("excel_name", "")
- image_name = data.get("image_name", "")
+ e_name = data.get("excel_name", "")
+ i_name = data.get("image_name", "")
def build_file_url(name, file_type):
if not name:
return None
return (f"/api/files/download/?agent_id={agent_id}&file_id={name}"
f"&file_type={file_type}")
- if excel_name:
- excel_url = build_file_url(excel_name, 'excel')
- if image_name:
- image_url = build_file_url(image_name, 'image')
+ if e_name:
+ excel_url = build_file_url(e_name, 'excel')
+ excel_name = e_name
+ if i_name:
+ image_url = build_file_url(i_name, 'image')
+ image_name = i_name
if data["type"] == "message":
message_data = {
"content": output,
"excel_url": excel_url,
"image_url": image_url,
+ "image_name": image_name,
+ "excel_name": excel_name,
"sql": data.get("sql", ""),
"code": data.get("code", ""),
"e": data.get("e", ""),
--
Gitblit v1.8.0