support volcengine + migrate ui tars to volcengine (#2705)

2025-06-13 21:17:32 +08:00
parent 40e608f9cd
commit 296d2f903b
9 changed files with 95 additions and 87 deletions
--- a/skyvern/forge/agent.py
+++ b/skyvern/forge/agent.py
@@ -404,7 +404,7 @@ class ForgeAgent:
                llm_caller = LLMCallerManager.get_llm_caller(task.task_id)
                if not llm_caller:
                    # create a new UI-TARS llm_caller
-                    llm_key = task.llm_key or settings.UI_TARS_LLM_KEY
+                    llm_key = task.llm_key or settings.VOLCENGINE_CUA_LLM_KEY
                    llm_caller = UITarsLLMCaller(llm_key=llm_key, screenshot_scaling_enabled=True)
                    llm_caller.initialize_conversation(task)