apunkt/MetaGPT
1
0
Fork 0
mirror of https://github.com/FoundationAgents/MetaGPT.git synced 2026-04-29 19:06:23 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat: test s3

Browse source
This commit is contained in:
莘权 马 2023-09-02 20:47:14 +08:00
parent 86e3ca0ba9
commit 7881937e8f
1 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

6
metagpt/utils/s3.py
View file

@ -129,13 +129,13 @@ class S3:
logger.error(f"Failed to download the file from S3: {e}")
raise e
async def cache(self, data: str, format: str = "") -> str:
async def cache(self, data: str, file_ext: str, format: str = "") -> str:
"""Save data to remote S3 and return url"""
object_name = str(uuid.uuid4()).replace("-", "")
object_name = str(uuid.uuid4()).replace("-", "") + file_ext
path = Path(__file__).parent
pathname = path / object_name
try:
async with aiofiles.open(str(pathname), mode="w") as file:
async with aiofiles.open(str(pathname), mode="wb") as file:
if format == BASE64_FORMAT:
data = base64.b64decode(data)
await file.write(data)