Yehowshua/ZIm
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

chore: Remove assistant labels for users without feature flag enabled (#15899)

Browse source 
Release Notes:

- N/A
This commit is contained in:
Piotr Osiewicz 2024-08-07 12:46:49 +02:00 • committed by GitHub
parent 990774247e
commit bf403ca788
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 10 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

16
crates/assistant/src/model_selector.rs
View file

@ -1,4 +1,5 @@
use feature_flags::LanguageModels; use feature_flags::LanguageModels;
use feature_flags::ZedPro;
use language_model::{LanguageModel, LanguageModelAvailability, LanguageModelRegistry}; use language_model::{LanguageModel, LanguageModelAvailability, LanguageModelRegistry};
use proto::Plan; use proto::Plan;
@ -143,8 +144,9 @@ impl PickerDelegate for ModelPickerDelegate {
selected: bool, selected: bool,
cx: &mut ViewContext<Picker<Self>>, cx: &mut ViewContext<Picker<Self>>,
) -> Option<Self::ListItem> { ) -> Option<Self::ListItem> {
use feature_flags::FeatureFlagAppExt;
let model_info = self.filtered_models.get(ix)?; let model_info = self.filtered_models.get(ix)?;
let show_badges = cx.has_flag::<ZedPro>();
Some( Some(
ListItem::new(ix) ListItem::new(ix)
.inset(true) .inset(true)
@ -170,11 +172,13 @@ impl PickerDelegate for ModelPickerDelegate {
.children(match model_info.availability { .children(match model_info.availability {
LanguageModelAvailability::Public => None, LanguageModelAvailability::Public => None,
LanguageModelAvailability::RequiresPlan(Plan::Free) => None, LanguageModelAvailability::RequiresPlan(Plan::Free) => None,
LanguageModelAvailability::RequiresPlan(Plan::ZedPro) => Some( LanguageModelAvailability::RequiresPlan(Plan::ZedPro) => {
Label::new("Pro") show_badges.then(|| {
.size(LabelSize::XSmall) Label::new("Pro")
.color(Color::Muted), .size(LabelSize::XSmall)
), .color(Color::Muted)
})
}
}), }),
) )
.child(div().when(model_info.is_selected, |this| { .child(div().when(model_info.is_selected, |this| {