diff --git a/skyvern/__init__.py b/skyvern/__init__.py index f2332f5d..502cde74 100644 --- a/skyvern/__init__.py +++ b/skyvern/__init__.py @@ -1,3 +1,13 @@ +from ddtrace import tracer +from ddtrace.filters import FilterRequestsOnUrl + from skyvern.forge.sdk.forge_log import setup_logger +tracer.configure( + settings={ + "FILTERS": [ + FilterRequestsOnUrl(r"http://.*/heartbeat$"), + ], + }, +) setup_logger() diff --git a/skyvern/forge/app.py b/skyvern/forge/app.py index 0f6bd6ae..18a6d1ee 100644 --- a/skyvern/forge/app.py +++ b/skyvern/forge/app.py @@ -1,7 +1,5 @@ from typing import Awaitable, Callable -from ddtrace import tracer -from ddtrace.filters import FilterRequestsOnUrl from fastapi import FastAPI from playwright.async_api import Frame, Page @@ -18,14 +16,6 @@ from skyvern.forge.sdk.workflow.context_manager import WorkflowContextManager from skyvern.forge.sdk.workflow.service import WorkflowService from skyvern.webeye.browser_manager import BrowserManager -tracer.configure( - settings={ - "FILTERS": [ - FilterRequestsOnUrl(r"http://.*/heartbeat$"), - ], - }, -) - SETTINGS_MANAGER = SettingsManager.get_settings() DATABASE = AgentDB( SettingsManager.get_settings().DATABASE_STRING,