mirror of
https://github.com/open-webui/open-webui
synced 2025-02-23 21:53:37 +00:00
17 lines
665 B
Python
17 lines
665 B
Python
from open_webui.constants import ERROR_MESSAGES
|
|
from open_webui.config import STORAGE_PROVIDER
|
|
|
|
from openwebui.backend.open_webui.storage.base_storage_provider import StorageProvider
|
|
from openwebui.backend.open_webui.storage.local_storage_provider import LocalStorageProvider
|
|
from openwebui.backend.open_webui.storage.s3_storage_provider import S3StorageProvider
|
|
|
|
def get_storage_provider() -> StorageProvider:
|
|
if STORAGE_PROVIDER == "s3":
|
|
return S3StorageProvider()
|
|
elif STORAGE_PROVIDER == "local":
|
|
return LocalStorageProvider()
|
|
else:
|
|
raise ValueError(ERROR_MESSAGES.INVALID_STORAGE_PROVIDER)
|
|
|
|
Storage = get_storage_provider()
|