diff --git a/assets/icons/ai_anthropic_hosted.svg b/assets/icons/ai_anthropic_hosted.svg index 12d731fb0b4438fcf6c263bd6c071bc8873823de..b0885204908a3e21e87d921b6c6d4e1ddb8abe67 100644 --- a/assets/icons/ai_anthropic_hosted.svg +++ b/assets/icons/ai_anthropic_hosted.svg @@ -1,11 +1,12 @@ - - - + + + + - - + + diff --git a/crates/assistant/src/model_selector.rs b/crates/assistant/src/model_selector.rs index 514bb3ee870fe2814f03364b990e81455c70a314..9a53c07dba025da53932d8413d03252ce5a6193a 100644 --- a/crates/assistant/src/model_selector.rs +++ b/crates/assistant/src/model_selector.rs @@ -1,6 +1,7 @@ use feature_flags::ZedPro; use gpui::Action; use gpui::DismissEvent; + use language_model::{LanguageModel, LanguageModelAvailability, LanguageModelRegistry}; use proto::Plan; use workspace::ShowConfiguration; @@ -149,6 +150,8 @@ impl PickerDelegate for ModelPickerDelegate { use feature_flags::FeatureFlagAppExt; let model_info = self.filtered_models.get(ix)?; let show_badges = cx.has_flag::(); + let provider_name: String = model_info.model.provider_name().0.into(); + Some( ListItem::new(ix) .inset(true) @@ -166,11 +169,16 @@ impl PickerDelegate for ModelPickerDelegate { .w_full() .justify_between() .font_buffer(cx) - .min_w(px(200.)) + .min_w(px(240.)) .child( h_flex() .gap_2() .child(Label::new(model_info.model.name().0.clone())) + .child( + Label::new(provider_name) + .size(LabelSize::XSmall) + .color(Color::Muted), + ) .children(match model_info.availability { LanguageModelAvailability::Public => None, LanguageModelAvailability::RequiresPlan(Plan::Free) => None,