diff --git a/packages/session-ui/src/tools/tool-renderer.tsx b/packages/session-ui/src/tools/tool-renderer.tsx index c2ae0a2ba6b0..ab985cfe4834 100644 --- a/packages/session-ui/src/tools/tool-renderer.tsx +++ b/packages/session-ui/src/tools/tool-renderer.tsx @@ -259,15 +259,13 @@ function agentColor(value: string | undefined, themeColors: Record) { const name = - provider === "parallel" - ? "Parallel" - : provider === "exa" - ? "Exa" - : provider === "firecrawl" - ? "Firecrawl" - : provider === "tavily" - ? "Tavily" - : undefined + typeof provider !== "string" || !provider + ? undefined + : provider === "tinyfish" + ? "TinyFish" + : provider === "opencode" + ? "OpenCode" + : `${provider[0].toUpperCase()}${provider.slice(1)}` if (name) return i18n.t("ui.tool.websearch.provider", { provider: name }) return i18n.t("ui.tool.websearch") }