From c5e16330a21231abbf2f326889e941ce3a890995 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E8=8E=98=E6=9D=83=20=E9=A9=AC?= <mashenquan@fuzhi.ai>
Date: Sat, 2 Sep 2023 18:51:46 +0800
Subject: [PATCH] feat: +path

---
 metagpt/utils/s3.py | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/metagpt/utils/s3.py b/metagpt/utils/s3.py
index 85837fedb..d13030292 100644
--- a/metagpt/utils/s3.py
+++ b/metagpt/utils/s3.py
@@ -1,4 +1,5 @@
 import base64
+import os.path
 import traceback
 import uuid
 from typing import Optional
@@ -138,8 +139,11 @@ class S3:
                 await file.write(data)
 
             bucket = CONFIG.S3.get("bucket")
-            await self.upload_file(bucket=bucket, local_path=pathname, object_name=object_name)
-            return await self.get_object_url(bucket=bucket, object_name=object_name)
+            object_pathname = CONFIG.S3.get("path") or "system"
+            object_pathname += f"/{object_name}"
+            object_pathname = os.path.normpath(object_pathname)
+            await self.upload_file(bucket=bucket, local_path=pathname, object_name=object_pathname)
+            return await self.get_object_url(bucket=bucket, object_name=object_pathname)
         except Exception as e:
             logger.exception(f"{e}, stack:{traceback.format_exc()}")
             return None