From 600d2205367bf914dceaf97e6f082383aa54348c Mon Sep 17 00:00:00 2001
From: zhaoqingang <zhaoqg0118@163.com>
Date: 星期五, 22 十一月 2024 09:55:16 +0800
Subject: [PATCH] merge
---
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