diff --git a/skyvern/__init__.py b/skyvern/__init__.py index f216bd7d..b81192bb 100644 --- a/skyvern/__init__.py +++ b/skyvern/__init__.py @@ -26,6 +26,7 @@ setup_logger() # noinspection PyUnresolvedReferences __all__ = [ "Skyvern", + "SkyvernSdk", "SkyvernPage", "RunContext", "action", @@ -54,6 +55,7 @@ __all__ = [ _lazy_imports = { "Skyvern": "skyvern.library", + "SkyvernSdk": "skyvern.library", "SkyvernPage": "skyvern.core.script_generations.skyvern_page", "RunContext": "skyvern.core.script_generations.skyvern_page", "setup": "skyvern.core.script_generations.run_initializer", diff --git a/skyvern/library/__init__.py b/skyvern/library/__init__.py index a8f539a0..67d54b0c 100644 --- a/skyvern/library/__init__.py +++ b/skyvern/library/__init__.py @@ -11,4 +11,9 @@ def __getattr__(name: str) -> Any: globals()["Skyvern"] = Skyvern return Skyvern + if name == "SkyvernSdk": + from skyvern.library.skyvern_sdk import SkyvernSdk # noqa: PLC0415 + + globals()["SkyvernSdk"] = SkyvernSdk + return SkyvernSdk raise AttributeError(f"module {__name__!r} has no attribute {name!r}")