hz/ebiz-dify-ai
1
0
Fork 0
mirror of http://112.124.100.131/huang.ze/ebiz-dify-ai.git synced 2025-12-07 18:06:52 +08:00
Code Issues Packages Projects Releases Wiki Activity

add keyword table s3 storage support (#3065)

Browse Source 
Co-authored-by: jyong <jyong@dify.ai>
This commit is contained in:
Jyong
2024-04-01 20:19:30 +08:00
committed by GitHub
parent 5e591fc1b7
commit a94d86da6d
6 changed files with 95 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
api/extensions/ext_storage.py
View File

@@ -172,6 +172,20 @@ class Storage:
return os.path.exists(filename)
def delete(self, filename):
if self.storage_type == 's3':
self.client.delete_object(Bucket=self.bucket_name, Key=filename)
elif self.storage_type == 'azure-blob':
blob_container = self.client.get_container_client(container=self.bucket_name)
blob_container.delete_blob(filename)
else:
if not self.folder or self.folder.endswith('/'):
filename = self.folder + filename
else:
filename = self.folder + '/' + filename
if os.path.exists(filename):
os.remove(filename)
storage = Storage()