From c8baa9f3ecdac9814c9279d4aab79067a7d02c80 Mon Sep 17 00:00:00 2001
From: Stanislav Novosad <stas@skyvern.com>
Date: Tue, 11 Nov 2025 15:34:04 -0700
Subject: [PATCH] SDK: improve IDE highlighting/type support (#3967)

---
 skyvern/__init__.py         | 5 +++++
 skyvern/library/__init__.py | 7 ++++++-
 2 files changed, 11 insertions(+), 1 deletion(-)

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: