mirror of
https://github.com/OpenHands/OpenHands.git
synced 2025-12-26 05:48:36 +08:00
18 lines
739 B
Python
18 lines
739 B
Python
from openhands.storage.files import FileStore
|
|
from openhands.storage.google_cloud import GoogleCloudFileStore
|
|
from openhands.storage.local import LocalFileStore
|
|
from openhands.storage.memory import InMemoryFileStore
|
|
from openhands.storage.s3 import S3FileStore
|
|
|
|
|
|
def get_file_store(file_store: str, file_store_path: str | None = None) -> FileStore:
|
|
if file_store == 'local':
|
|
if file_store_path is None:
|
|
raise ValueError('file_store_path is required for local file store')
|
|
return LocalFileStore(file_store_path)
|
|
elif file_store == 's3':
|
|
return S3FileStore(file_store_path)
|
|
elif file_store == 'google_cloud':
|
|
return GoogleCloudFileStore(file_store_path)
|
|
return InMemoryFileStore()
|