Dorod-Sky/skyvern/__init__.py

from ddtrace import tracer
from ddtrace.trace import TraceFilter, Span
from ddtrace.ext import http
import re
from skyvern.forge.sdk.forge_log import setup_logger


class FilterHeartbeat(TraceFilter):
    _HB_URL = re.compile(r"http://.*/heartbeat$")

    def process_trace(self, trace: list[Span]) -> list[Span] | None:
        for span in trace:
            url = span.get_tag(http.URL)
            if span.parent_id is None and url is not None and self._HB_URL.match(url):
                # drop the full trace chunk
                return None
        return trace


tracer.configure(trace_processors=[FilterHeartbeat()])
setup_logger()


from skyvern.forge import app  # noqa: E402, F401
from skyvern.library import Skyvern  # noqa: E402
from skyvern.core.script_generations.skyvern_page import RunContext, SkyvernPage  # noqa: E402
from skyvern.core.script_generations.run_initializer import setup  # noqa: E402
from skyvern.core.script_generations.workflow_wrappers import (  # noqa: E402
    cached,  # noqa: E402
    workflow,  # noqa: E402
)  # noqa: E402
from skyvern.services.script_service import (  # noqa: E402
    action,  # noqa: E402
    download,  # noqa: E402
    extract,  # noqa: E402
    http_request,  # noqa: E402
    goto,  # noqa: E402
    login,  # noqa: E402
    loop,  # noqa: E402
    parse_file,  # noqa: E402
    parse_pdf,  # noqa: E402
    prompt,  # noqa: E402
    render_list,  # noqa: E402
    render_template,  # noqa: E402
    run_code,  # noqa: E402
    run_script,  # noqa: E402
    run_task,  # noqa: E402
    send_email,  # noqa: E402
    upload_file,  # noqa: E402
    validate,  # noqa: E402
    wait,  # noqa: E402
)  # noqa: E402


__all__ = [
    "Skyvern",
    "SkyvernPage",
    "RunContext",
    "action",
    "cached",
    "download",
    "extract",
    "http_request",
    "goto",
    "login",
    "loop",
    "parse_file",
    "parse_pdf",
    "prompt",
    "render_list",
    "render_template",
    "run_code",
    "run_script",
    "run_task",
    "send_email",
    "setup",
    "upload_file",
    "validate",
    "wait",
    "workflow",
]