diff --git a/skyvern/config.py b/skyvern/config.py index 4cc859c4..3e2a4410 100644 --- a/skyvern/config.py +++ b/skyvern/config.py @@ -374,6 +374,7 @@ class Settings(BaseSettings): BITWARDEN_SERVER_PORT: int = 8002 SVG_MAX_LENGTH: int = 100000 + SVG_MAX_PARSING_ELEMENT_CNT: int = 3000 ENABLE_LOG_ARTIFACTS: bool = False ENABLE_CODE_BLOCK: bool = True diff --git a/skyvern/forge/agent_functions.py b/skyvern/forge/agent_functions.py index 26804fcd..b4234065 100644 --- a/skyvern/forge/agent_functions.py +++ b/skyvern/forge/agent_functions.py @@ -504,7 +504,7 @@ class AgentFunction: task: Task | None = None, step: Step | None = None, ) -> CleanupElementTreeFunc: - MAX_ELEMENT_CNT = 3000 + MAX_ELEMENT_CNT = settings.SVG_MAX_PARSING_ELEMENT_CNT @TraceManager.traced_async(ignore_input=True) async def cleanup_element_tree_func(frame: Page | Frame, url: str, element_tree: list[dict]) -> list[dict]: