fix(trae-cn): reject ambiguous model selections

Author: jackwener

clis/trae-cn/trae-cn.test.js

@@ -1,6 +1,6 @@
 import { describe, expect, it, vi } from 'vitest';
 import { JSDOM } from 'jsdom';
-import { CommandExecutionError, TimeoutError } from '@jackwener/opencli/errors';
+import { ArgumentError, CommandExecutionError, TimeoutError } from '@jackwener/opencli/errors';
 import { activityCommand } from './activity.js';
 import { approveCommand } from './approve.js';
 import { askCommand } from './ask.js';
@@ -335,6 +335,24 @@ describe('trae-cn commands', () => {
       .rejects.toBeInstanceOf(CommandExecutionError);
   });
 
+  it('select-model rejects ambiguous partial model names before changing selection', async () => {
+    const page = {
+      evaluate: vi.fn()
+        .mockResolvedValueOnce([])
+        .mockResolvedValueOnce([
+          { Index: 0, Model: 'GPT 5.4' },
+          { Index: 1, Model: 'GPT 5.5' },
+        ]),
+      click: vi.fn().mockResolvedValue(undefined),
+      wait: vi.fn().mockResolvedValue(undefined),
+    };
+
+    await expect(selectModelCommand.func(page, { name: 'GPT 5' }))
+      .rejects.toBeInstanceOf(ArgumentError);
+    expect(page.click).toHaveBeenCalledTimes(1);
+    expect(page.click).toHaveBeenCalledWith(expect.stringContaining('icd-model-select-trigger'));
+  });
+
   it('new fails closed when clicking new task does not prove a fresh empty composer', async () => {
     const page = {
       evaluate: vi.fn()

clis/trae-cn/utils.js

@@ -716,8 +716,16 @@ export async function selectTraeModel(page, name) {
     await page.wait(0.8);
     models = await page.evaluate(listOpenModelItemsScript());
   }
-  const match = models.find((item) => normalizeModelLabel(item.Model) === wantedKey)
-    || models.find((item) => normalizeModelLabel(item.Model).includes(wantedKey));
+  const exactMatch = models.find((item) => normalizeModelLabel(item.Model) === wantedKey);
+  const containsMatches = exactMatch
+    ? []
+    : models.filter((item) => normalizeModelLabel(item.Model).includes(wantedKey));
+  if (!exactMatch && containsMatches.length > 1) {
+    throw new ArgumentError(
+      `Model "${wanted}" is ambiguous in Trae CN model menu: ${containsMatches.map(item => item.Model).join(', ')}`,
+    );
+  }
+  const match = exactMatch || containsMatches[0];
   if (!match) {
     throw new ArgumentError(`Model "${wanted}" not found in Trae CN model menu`);
   }