mirror of
https://github.com/Skyvern-AI/skyvern.git
synced 2025-09-01 18:20:06 +00:00
46 lines
1.8 KiB
Python
46 lines
1.8 KiB
Python
from typing import Awaitable, Callable
|
|
|
|
from ddtrace import tracer
|
|
from ddtrace.filters import FilterRequestsOnUrl
|
|
from fastapi import FastAPI
|
|
|
|
from skyvern.forge.agent import ForgeAgent
|
|
from skyvern.forge.agent_functions import AgentFunction
|
|
from skyvern.forge.sdk.api.llm.api_handler_factory import LLMAPIHandlerFactory
|
|
from skyvern.forge.sdk.artifact.manager import ArtifactManager
|
|
from skyvern.forge.sdk.artifact.storage.factory import StorageFactory
|
|
from skyvern.forge.sdk.db.client import AgentDB
|
|
from skyvern.forge.sdk.experimentation.providers import BaseExperimentationProvider, NoOpExperimentationProvider
|
|
from skyvern.forge.sdk.forge_log import setup_logger
|
|
from skyvern.forge.sdk.models import Organization
|
|
from skyvern.forge.sdk.settings_manager import SettingsManager
|
|
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$"),
|
|
],
|
|
},
|
|
)
|
|
|
|
setup_logger()
|
|
SETTINGS_MANAGER = SettingsManager.get_settings()
|
|
DATABASE = AgentDB(
|
|
SettingsManager.get_settings().DATABASE_STRING,
|
|
debug_enabled=SettingsManager.get_settings().DEBUG_MODE,
|
|
)
|
|
STORAGE = StorageFactory.get_storage()
|
|
ARTIFACT_MANAGER = ArtifactManager()
|
|
BROWSER_MANAGER = BrowserManager()
|
|
EXPERIMENTATION_PROVIDER: BaseExperimentationProvider = NoOpExperimentationProvider()
|
|
LLM_API_HANDLER = LLMAPIHandlerFactory.get_llm_api_handler(SettingsManager.get_settings().LLM_KEY)
|
|
WORKFLOW_CONTEXT_MANAGER = WorkflowContextManager()
|
|
WORKFLOW_SERVICE = WorkflowService()
|
|
AGENT_FUNCTION = AgentFunction()
|
|
authentication_function: Callable[[str], Awaitable[Organization]] | None = None
|
|
setup_api_app: Callable[[FastAPI], None] | None = None
|
|
|
|
agent = ForgeAgent()
|