feat: add LLM model dropdown for trae-cli (no rebuild needed)

Author: Shuming-Yang

.github/workflows/test-e2e-advanced.yml

@@ -36,6 +36,27 @@ on:
           Maximum wait time in seconds for the agent to respond.
         required: false
         default: 120
+      model:
+        description: |
+          LLM model to use (only for trae-cli). Select "default" to use image config.
+        required: false
+        type: choice
+        default: default
+        options:
+          - default
+          - deepseek/deepseek-v4-flash
+          - deepseek/deepseek-v4-pro
+          - deepseek/deepseek-chat
+          - deepseek/deepseek-r1
+          - openai/gpt-4o-mini
+          - openai/gpt-4o
+          - openai/o3-mini
+          - google/gemini-2.0-flash-exp
+          - google/gemini-2.5-flash
+          - google/gemini-2.5-pro
+          - anthropic/claude-3.5-sonnet
+          - anthropic/claude-opus-4
+          - meta-llama/llama-3.3-70b-instruct
       opencode_api_key:
         description: |
           OPENCODE_API_KEY (for opencode agent)
@@ -109,4 +130,5 @@ jobs:
             --agent ${{ inputs.agent }} \
             --prompt "${{ inputs.prompt }}" \
             --timeout ${{ inputs.timeout }} \
+            --model ${{ inputs.model }} \
             -v --tb=short -s

tests/e2e/agents/py-devkit-image-base.yaml

@@ -23,5 +23,6 @@ agents:
 
   - name: trae-cli
     binary: trae-cli
+    supports_model: true
     env_vars: [OPENAI_API_KEY, ANTHROPIC_API_KEY, OPENROUTER_API_KEY, DOUBAO_API_KEY]
     run_cmd: 'trae-cli run {prompt}'

tests/e2e/conftest.py

@@ -12,6 +12,7 @@ def pytest_addoption(parser):
     parser.addoption("--agent", action="store", default=None)
     parser.addoption("--prompt", action="store", default=None)
     parser.addoption("--timeout", action="store", type=int, default=120)
+    parser.addoption("--model", action="store", default="default")
 
 
 def load_agent_config(image_name):
@@ -69,6 +70,11 @@ def cmd_timeout(request):
     return request.config.getoption("--timeout")
 
 
+@pytest.fixture(scope="module")
+def llm_model(request):
+    return request.config.getoption("--model")
+
+
 @pytest.fixture(scope="module")
 def check_env_vars(selected_agent):
     missing = [v for v in selected_agent.get("env_vars", []) if v not in os.environ]

tests/e2e/test_advanced.py

@@ -4,12 +4,14 @@
 import pytest
 
 
-def test_agent_prompt_response(container_id, selected_agent, custom_prompt, cmd_timeout):
+def test_agent_prompt_response(container_id, selected_agent, custom_prompt, cmd_timeout, llm_model):
     run_template = selected_agent.get("run_cmd")
     if not run_template:
         pytest.skip(f"No run_cmd defined for {selected_agent['name']}")
 
     cmd = run_template.replace("{prompt}", shlex.quote(custom_prompt))
+    if selected_agent.get("supports_model") and llm_model and llm_model != "default":
+        cmd += f" --model {shlex.quote(llm_model)}"
 
     proc = subprocess.Popen(
         ["docker", "exec", "-i", container_id, "bash", "-c", cmd],