|
@@ -81,13 +81,12 @@ class BishengService:
|
|
|
data = self._check_response(response)
|
|
data = self._check_response(response)
|
|
|
return data
|
|
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"
|
|
url = f"{self.base_url}/api/v1/knowledge/upload"
|
|
|
headers = {'cookie': f"access_token_cookie={token};"}
|
|
headers = {'cookie': f"access_token_cookie={token};"}
|
|
|
|
|
|
|
|
# 创建表单数据,包含文件
|
|
# 创建表单数据,包含文件
|
|
|
- files = {"file": ("file", file)} # 使用默认文件名 "file"
|
|
|
|
|
-
|
|
|
|
|
|
|
+ files = {"file": (filename, file)}
|
|
|
async with httpx.AsyncClient() as client:
|
|
async with httpx.AsyncClient() as client:
|
|
|
response = await client.post(url, headers=headers, files=files)
|
|
response = await client.post(url, headers=headers, files=files)
|
|
|
data = self._check_response(response)
|
|
data = self._check_response(response)
|