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