Skip to content

Commit ded7cd6

Browse files
committed
Refactor: addressed copoilet suggestions
Signed-off-by: Abhijay007 <Abhijay007j@gmail.com>
1 parent e285657 commit ded7cd6

File tree

1 file changed

+6
-4
lines changed

1 file changed

+6
-4
lines changed

ui/desktop/src/components/settings/models/bottom_bar/ModelsBottomBar.tsx

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -124,9 +124,9 @@ export default function ModelsBottomBar({
124124
if (effectiveProvider) {
125125
(async () => {
126126
if (recipeProvider) {
127-
const capitalizedProvider =
128-
recipeProvider.charAt(0).toUpperCase() + recipeProvider.slice(1);
129-
setDisplayProvider(capitalizedProvider);
127+
const metadata = await getProviderMetadata(recipeProvider, getProviders);
128+
const displayName = metadata?.display_name ?? metadata?.name ?? recipeProvider;
129+
setDisplayProvider(displayName);
130130
} else {
131131
const providerDisplayName = await getCurrentProviderDisplayName();
132132
if (providerDisplayName) {
@@ -144,6 +144,8 @@ export default function ModelsBottomBar({
144144
currentProvider,
145145
getCurrentProviderDisplayName,
146146
getCurrentModelAndProviderForDisplay,
147+
recipe,
148+
getProviders,
147149
]);
148150

149151
// Fetch provider default model when provider changes and no current model
@@ -173,7 +175,7 @@ export default function ModelsBottomBar({
173175
setDisplayModelName(displayName);
174176
}
175177
})();
176-
}, [effectiveModel, recipeModel, currentModel, getCurrentModelDisplayName]);
178+
}, [effectiveModel, recipeModel, currentModel, getCurrentModelDisplayName, recipe]);
177179

178180
return (
179181
<div className="relative flex items-center" ref={dropdownRef}>

0 commit comments

Comments
 (0)