From 7be58d7783c8ef4b0b10c384dcd76d6b011f0817 Mon Sep 17 00:00:00 2001
From: zhangqian <zhangqian@123.com>
Date: 星期三, 30 十月 2024 10:04:00 +0800
Subject: [PATCH] fix报告生成
---
app/service/bisheng.py | 13 +++++++++----
1 files changed, 9 insertions(+), 4 deletions(-)
diff --git a/app/service/bisheng.py b/app/service/bisheng.py
index e263c65..e7c92fa 100644
--- a/app/service/bisheng.py
+++ b/app/service/bisheng.py
@@ -16,7 +16,13 @@
status_code = response_data.get("status_code", 0)
if status_code != 200:
raise Exception(f"Failed to fetch data from Bisheng API: {response.text}")
- return response_data.get("data", {})
+ # 妫�鏌ヨ繑鍥炵殑鏁版嵁绫诲瀷
+ if isinstance(response_data.get("data"), dict):
+ return response_data.get("data", {})
+ elif isinstance(response_data.get("data"), list):
+ return response_data.get("data", [])
+ else:
+ return {}
async def register(self, username: str, password: str):
public_key = await self.get_public_key_api()
@@ -75,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