mirror of
https://github.com/Skyvern-AI/skyvern.git
synced 2026-04-28 03:30:10 +00:00
14 lines
408 B
Python
14 lines
408 B
Python
from skyvern.forge.sdk.artifact.storage.base import BaseStorage
|
|
from skyvern.forge.sdk.artifact.storage.local import LocalStorage
|
|
|
|
|
|
class StorageFactory:
|
|
__storage: BaseStorage = LocalStorage()
|
|
|
|
@staticmethod
|
|
def set_storage(storage: BaseStorage) -> None:
|
|
StorageFactory.__storage = storage
|
|
|
|
@staticmethod
|
|
def get_storage() -> BaseStorage:
|
|
return StorageFactory.__storage
|