From 5580958d49e5aab48908000614e47ecb75ff4797 Mon Sep 17 00:00:00 2001
From: zhaoqingang <zhaoqg0118@163.com>
Date: 星期四, 28 十一月 2024 19:14:26 +0800
Subject: [PATCH] 智能数据问题优化
---
app/api/agent.py | 12 ++++++++++++
1 files changed, 12 insertions(+), 0 deletions(-)
diff --git a/app/api/agent.py b/app/api/agent.py
index 2f27f23..698ca0d 100644
--- a/app/api/agent.py
+++ b/app/api/agent.py
@@ -154,12 +154,24 @@
if i.get("role") == "user":
tmp_data["question"]=i.get("content")
elif i.get("role") == "assistant":
+
if isinstance(i.get("content"), dict):
tmp_data["answer"] = i.get("content", {}).get("message")
if "file_name" in i.get("content", {}):
tmp_data["files"] = [{"file_name":i.get("content", {}).get("file_name"), "file_url":i.get("content", {}).get("file_url")}]
else:
tmp_data["answer"] = i.get("content")
+
+ if "excel_url" in i:
+ tmp_data["excel_url"] = i.get("excel_url")
+ if "image_url" in i:
+ tmp_data["image_url"] = i.get("image_url")
+ if "sql" in i:
+ tmp_data["sql"] = i.get("sql")
+ if "code" in i:
+ tmp_data["code"] = i.get("code")
+ if "e" in i:
+ tmp_data["e"] = i.get("e")
data.append(tmp_data)
tmp_data = {}
--
Gitblit v1.8.0