vodorod/Dorod-Sky
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

ui-tars integration (#2656)

Browse Source
This commit is contained in:
Wyatt Marshall
2025-06-13 01:23:39 -04:00
committed by GitHub
parent 47cf755d9c
commit 15d46aab82
18 changed files with 986 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
skyvern/cli/llm_setup.py
View File

@@ -160,6 +160,31 @@ def setup_llm_providers() -> None:
else:
update_or_add_env_var("ENABLE_NOVITA", "false")
console.print("\n[bold blue]--- UI-TARS Configuration ---[/bold blue]")
console.print("To enable UI-TARS (Seed1.5-VL), you must have a ByteDance Doubao API key.")
console.print("UI-TARS now uses direct VolcEngine API calls for improved compatibility.")
enable_ui_tars = Confirm.ask("Do you want to enable UI-TARS?")
if enable_ui_tars:
ui_tars_api_key = Prompt.ask("Enter your ByteDance Doubao API key", password=True)
if not ui_tars_api_key:
console.print("[red]Error: UI-TARS API key is required. UI-TARS will not be enabled.[/red]")
else:
update_or_add_env_var("UI_TARS_API_KEY", ui_tars_api_key)
update_or_add_env_var("ENABLE_UI_TARS", "true")
# Optional: Allow customizing model endpoint
custom_model = Confirm.ask(
"Do you want to use a custom model endpoint? (default: doubao-1-5-thinking-vision-pro-250428)"
)
if custom_model:
ui_tars_model = Prompt.ask("Enter your UI-TARS model endpoint ID")
if ui_tars_model:
update_or_add_env_var("UI_TARS_MODEL", ui_tars_model)
model_options.append("UI_TARS_SEED1_5_VL")
else:
update_or_add_env_var("ENABLE_UI_TARS", "false")
console.print("\n[bold blue]--- OpenAI-Compatible Provider Configuration ---[/bold blue]")
console.print("To enable an OpenAI-compatible provider, you must have a model name, API key, and API base URL.")
enable_openai_compatible = Confirm.ask("Do you want to enable an OpenAI-compatible provider?")