From a9caccfe34e6126270f41533c4b6dc7c79c66273 Mon Sep 17 00:00:00 2001
From: zhangqian <zhangqian@123.com>
Date: 星期四, 17 十月 2024 13:58:35 +0800
Subject: [PATCH] 上传文件时传入文件名
---
app/service/bisheng.py | 5 ++---
app/api/files.py | 2 +-
2 files changed, 3 insertions(+), 4 deletions(-)
diff --git a/app/api/files.py b/app/api/files.py
index 401428a..5e2640b 100644
--- a/app/api/files.py
+++ b/app/api/files.py
@@ -35,7 +35,7 @@
bisheng_service = BishengService(base_url=settings.bisheng_base_url)
try:
token = get_bisheng_token(db, current_user.id)
- result = await bisheng_service.upload(token, file_content)
+ result = await bisheng_service.upload(token, file.filename, file_content)
except Exception as e:
raise HTTPException(status_code=500, detail=str(e))
return Response(code=200, msg="", data=result)
diff --git a/app/service/bisheng.py b/app/service/bisheng.py
index d56a00d..e7c92fa 100644
--- a/app/service/bisheng.py
+++ b/app/service/bisheng.py
@@ -81,13 +81,12 @@
data = self._check_response(response)
return data
- async def upload(self, token: str, file: bytes) -> dict:
+ async def upload(self, token: str, filename: str, file: bytes) -> dict:
url = f"{self.base_url}/api/v1/knowledge/upload"
headers = {'cookie': f"access_token_cookie={token};"}
# 鍒涘缓琛ㄥ崟鏁版嵁锛屽寘鍚枃浠�
- files = {"file": ("file", file)} # 浣跨敤榛樿鏂囦欢鍚� "file"
-
+ files = {"file": (filename, file)}
async with httpx.AsyncClient() as client:
response = await client.post(url, headers=headers, files=files)
data = self._check_response(response)
--
Gitblit v1.8.0