enable ui tars cua (#2715)

2025-06-13 19:55:41 -07:00
parent 346b36fa4d
commit 57f4619ef4
1 changed files with 5 additions and 1 deletions
--- a/skyvern/forge/agent.py
+++ b/skyvern/forge/agent.py
@@ -871,7 +871,11 @@ class ForgeAgent:
                    scraped_page=scraped_page,
                    llm_caller=llm_caller,
                )
-            elif engine == RunEngine.ui_tars:
+            elif engine == RunEngine.ui_tars and not app.EXPERIMENTATION_PROVIDER.is_feature_enabled_cached(
+                "DISABLE_UI_TARS_CUA",
+                task.workflow_run_id or task.task_id,
+                properties={"organization_id": task.organization_id},
+            ):
                assert llm_caller is not None
                actions = await self._generate_ui_tars_actions(
                    task=task,