diff --git a/src/renderer/components/providers/ProviderIcon.test.tsx b/src/renderer/components/providers/ProviderIcon.test.tsx index 7c4d2722..4ca3479a 100644 --- a/src/renderer/components/providers/ProviderIcon.test.tsx +++ b/src/renderer/components/providers/ProviderIcon.test.tsx @@ -10,4 +10,11 @@ describe("ProviderIcon", () => { expect(screen.getByText("H")).toBeInTheDocument(); expect(screen.queryByText("C")).not.toBeInTheDocument(); }); + + it("uses the Claude profile id for the badge when no display label is provided", () => { + render(); + + expect(screen.getByText("P")).toBeInTheDocument(); + expect(screen.queryByText("C")).not.toBeInTheDocument(); + }); }); diff --git a/src/renderer/components/providers/ProviderIcon.tsx b/src/renderer/components/providers/ProviderIcon.tsx index 0ceeeb06..d6b2b2c4 100644 --- a/src/renderer/components/providers/ProviderIcon.tsx +++ b/src/renderer/components/providers/ProviderIcon.tsx @@ -79,6 +79,7 @@ function claudeProfileBadgeLabel(kind: string, fallbackLabel: string | undefined const profileId = kind.slice("claude:".length); const label = fallbackLabel?.trim(); if (!label) return profileId; + if (label === kind || label.toLowerCase().startsWith("claude:")) return profileId; const profileLabel = label.replace(/^claude\s+/i, "").trim(); return profileLabel || profileId; }