UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-01-19 09:16:44 +00:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

add function to retrieve the storage provider so it can be easily tested

Browse Source
This commit is contained in:
Rodrigo Agundez 2025-01-16 08:09:56 +08:00
parent 0129255478
commit dd16c3d5c1
1 changed files with 11 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
backend/open_webui/storage/provider.py
View File

@ -138,9 +138,14 @@ class S3StorageProvider(StorageProvider):
LocalStorageProvider.delete_all_files() LocalStorageProvider.delete_all_files()
if STORAGE_PROVIDER == "local": def get_storage_provider(storage_provider: str):
if storage_provider == "local":
Storage = LocalStorageProvider() Storage = LocalStorageProvider()
elif STORAGE_PROVIDER == "s3": elif storage_provider == "s3":
Storage = S3StorageProvider() Storage = S3StorageProvider()
else: else:
raise RuntimeError(f"Unsupported storage provider: {STORAGE_PROVIDER}") raise RuntimeError(f"Unsupported storage provider: {storage_provider}")
return Storage
Storage = get_storage_provider(STORAGE_PROVIDER)