SDK: improve IDE highlighting/type support (#3967)

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$")

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: