From 66f5df3ec8004e91ec2f440d69755caa52ac33bd Mon Sep 17 00:00:00 2001
From: zhangqian <zhangqian@123.com>
Date: 星期五, 22 十一月 2024 01:22:19 +0800
Subject: [PATCH] 接收excel_talk返回的消息和缓冲区的数据拼接，然后解析JSON。如果解析失败，存入缓存区继续累积，如何解析成功，给前端返回并清空缓冲区。 && 修复 保存会话消息历史不成功的bug
---
app/api/files.py | 15 +++++++++++++++
1 files changed, 15 insertions(+), 0 deletions(-)
diff --git a/app/api/files.py b/app/api/files.py
index 968b6b3..863f174 100644
--- a/app/api/files.py
+++ b/app/api/files.py
@@ -68,6 +68,12 @@
return Response(code=200, msg="", data=result)
+ elif agent_id == "basic_paper_agent":
+ service = BasicService(base_url=settings.basic_paper_url)
+ result = await service.paper_file_upload(chat_id, file.filename, file_content)
+
+ return Response(code=200, msg="", data=result)
+
else:
return Response(code=200, msg="Unsupported agent type")
@@ -101,6 +107,8 @@
if agent_id == "basic_excel_talk":
return await download_basic_file(file_id, file_type)
+ elif agent_id == "basic_question_talk":
+ return await download_basic_file(file_id, file_type)
else:
return Response(code=400, msg="Unsupported agent type")
@@ -137,5 +145,12 @@
media_type=mimetype,
headers={"Content-Disposition": f"attachment; filename={filename}"}
)
+ elif file_type == "word":
+ content, filename, mimetype = await service.questions_talk_word_download(file_id)
+ return StreamingResponse(
+ io.BytesIO(content),
+ media_type=mimetype,
+ headers={"Content-Disposition": f"attachment; filename={filename}"}
+ )
else:
return Response(code=400, msg="Unsupported file type")
\ No newline at end of file
--
Gitblit v1.8.0