diff --git a/studio/frontend/src/components/assistant-ui/model-selector.tsx b/studio/frontend/src/components/assistant-ui/model-selector.tsx index 403d674d24..e6cba20b9c 100644 --- a/studio/frontend/src/components/assistant-ui/model-selector.tsx +++ b/studio/frontend/src/components/assistant-ui/model-selector.tsx @@ -129,6 +129,17 @@ function ModelSelectorContent({ const chatOnly = usePlatformStore((s) => s.isChatOnly()); const hasExternal = externalModels.length > 0; + const chatOnlyTabsDefault = useMemo( + () => (value && externalModels.some((m) => m.id === value) ? "external" : "hub"), + [value, externalModels], + ); + + const studioTabsDefault = useMemo((): "hub" | "lora" | "external" => { + if (value && externalModels.some((m) => m.id === value)) return "external"; + if (value && loraModels.some((l) => l.id === value)) return "lora"; + return "hub"; + }, [value, externalModels, loraModels]); + return ( {chatOnly ? ( - + hasExternal ? ( + + + Hub models + External + + + + + + + + + ) : ( + + ) ) : ( - + Hub models Fine-tuned