Fix draft mode chip for non-thinking models

Keep the desktop controls row alive when desktop extras exist, so draft mode chips still render even when a model has no thinking options.\n\nRestore the replacement draft e2e assertion for the mode chip now that it renders again.
2026-05-26 20:15:11 +07:00
parent f79101a1f0
commit 88914ccba6
2 changed files with 6 additions and 5 deletions
@@ -126,10 +126,7 @@ async function expectReplacementDraftMatchesPreviousSetup(page: Page): Promise<v
  await expect(
    page.getByRole("button", { name: "Select model (Ten second stream)" }),
  ).toBeVisible();
-  // TODO(boudra): the replacement draft's mode picker stopped rendering after
-  // the composer refactor — the model carries over but modes aren't surfaced
-  // in the draft's provider snapshot. Restore this assertion once the draft
-  // mode picker is fixed.
+  await expect(page.getByRole("button", { name: "Select agent mode (Load test)" })).toBeVisible();
 }

 async function createAgentFromReplacementDraft(page: Page): Promise<void> {