1- from typing import BinaryIO , Optional , Union
1+ from typing import BinaryIO , Dict , Optional , Tuple , Union
22
33from huntflow_api_client .entities .base import BaseEntity
44from huntflow_api_client .models .request .file import UploadFileHeaders
@@ -12,6 +12,7 @@ async def upload(
1212 headers : UploadFileHeaders ,
1313 file : Union [bytes , BinaryIO ],
1414 preset : Optional [str ] = None ,
15+ filename : Optional [str ] = None ,
1516 ) -> UploadResponse :
1617 """
1718 API method reference https://api.huntflow.ai/v2/docs#post-/accounts/-account_id-/upload
@@ -20,16 +21,24 @@ async def upload(
2021 :param file: File
2122 :param preset: Preset
2223 :param headers: Headers
24+ :param filename: Filename
2325 :return: Additional data
2426 """
2527
2628 data = {}
2729 if preset :
2830 data ["preset" ] = preset
31+
32+ files : Dict [str , Union [Union [bytes , BinaryIO ], Tuple [str , Union [bytes , BinaryIO ]]]] = {}
33+ if filename :
34+ files ["file" ] = (filename , file )
35+ else :
36+ files ["file" ] = file
37+
2938 response = await self ._api .request (
3039 "POST" ,
3140 f"/accounts/{ account_id } /upload" ,
32- files = { "file" : file } ,
41+ files = files ,
3342 data = data ,
3443 headers = headers .jsonable_dict (exclude_none = True , by_alias = True ),
3544 )
0 commit comments