diff --git a/skyvern/__init__.py b/skyvern/__init__.py
index b81192bb..90a1ce66 100644
--- a/skyvern/__init__.py
+++ b/skyvern/__init__.py
@@ -1,4 +1,5 @@
 import re
+import typing
 from typing import Any
 
 from ddtrace import tracer
@@ -7,6 +8,10 @@ from ddtrace.trace import TraceFilter, Span
 
 from skyvern.forge.sdk.forge_log import setup_logger
 
+if typing.TYPE_CHECKING:
+    from skyvern.library import Skyvern  # noqa: E402
+    from skyvern.library.skyvern_sdk import SkyvernSdk  # noqa: E402
+
 
 class FilterHeartbeat(TraceFilter):
     _HB_URL = re.compile(r"http://.*/heartbeat$")
diff --git a/skyvern/library/__init__.py b/skyvern/library/__init__.py
index 67d54b0c..dc87fe64 100644
--- a/skyvern/library/__init__.py
+++ b/skyvern/library/__init__.py
@@ -1,7 +1,12 @@
+import typing
 from typing import Any
 
+if typing.TYPE_CHECKING:
+    from skyvern.library.skyvern import Skyvern  # noqa: E402
+    from skyvern.library.skyvern_sdk import SkyvernSdk  # noqa: E402
+
 # noinspection PyUnresolvedReferences
-__all__ = ["Skyvern"]
+__all__ = ["Skyvern", "SkyvernSdk"]
 
 
 def __getattr__(name: str) -> Any: