SDK: improve IDE highlighting/type support (#3967)
This commit is contained in:
committed by
GitHub
parent
6794bbf70f
commit
c8baa9f3ec
@@ -1,4 +1,5 @@
|
|||||||
import re
|
import re
|
||||||
|
import typing
|
||||||
from typing import Any
|
from typing import Any
|
||||||
|
|
||||||
from ddtrace import tracer
|
from ddtrace import tracer
|
||||||
@@ -7,6 +8,10 @@ from ddtrace.trace import TraceFilter, Span
|
|||||||
|
|
||||||
from skyvern.forge.sdk.forge_log import setup_logger
|
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):
|
class FilterHeartbeat(TraceFilter):
|
||||||
_HB_URL = re.compile(r"http://.*/heartbeat$")
|
_HB_URL = re.compile(r"http://.*/heartbeat$")
|
||||||
|
|||||||
@@ -1,7 +1,12 @@
|
|||||||
|
import typing
|
||||||
from typing import Any
|
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
|
# noinspection PyUnresolvedReferences
|
||||||
__all__ = ["Skyvern"]
|
__all__ = ["Skyvern", "SkyvernSdk"]
|
||||||
|
|
||||||
|
|
||||||
def __getattr__(name: str) -> Any:
|
def __getattr__(name: str) -> Any:
|
||||||
|
|||||||
Reference in New Issue
Block a user