diff --git a/webui/src/components/thread/ModelPresetBadge.tsx b/webui/src/components/thread/ModelPresetBadge.tsx
index 143fff734..224610ee3 100644
--- a/webui/src/components/thread/ModelPresetBadge.tsx
+++ b/webui/src/components/thread/ModelPresetBadge.tsx
@@ -548,14 +548,15 @@ function PresetPill({
zIndex: Math.round(scale * 100),
}}
>
-
+ {!needsSetup ? (
+
+ ) : null}
- {needsSetup ? (
-
- ) : logoUrl ? (
+ {logoUrl ? (
{
);
const badge = await screen.findByRole("button", { name: "Choose your AI" });
- const setupIcon = screen.getByTestId("composer-model-setup-icon");
- expect(setupIcon).toBeInTheDocument();
- expect(setupIcon.parentElement).not.toHaveClass("border-amber-500/35");
+ expect(screen.queryByTestId("composer-model-setup-icon")).not.toBeInTheDocument();
+ expect(badge).not.toHaveClass("border-amber-500/35");
expect(screen.queryByTestId("composer-model-logo-openai_codex")).not.toBeInTheDocument();
fireEvent.click(badge);
expect(await screen.findByRole("dialog", { name: "Choose your AI" })).toBeInTheDocument();