assistant-panel: Update model selector to a combo-box (#15693)
This updates the model selector to be a combobox (filterable list) This PR causes the following regression: There is no longer a message in the inline assistant explaining context is included from the assistant panel. Will follow up with some sort of solution soon. Before:  After:   Release Notes: - N/A --------- Co-authored-by: Marshall Bowers <1486634+maxdeviant@users.noreply.github.com>
This commit is contained in:
Nate Butler
GitHub
parent
f11f3f2599
commit
03cc18dd20
3 changed files with 472 additions and 189 deletions
|
|
@ -2801,21 +2801,19 @@ pub struct ContextEditorToolbarItem {
|
||||||
fs: Arc<dyn Fs>,
|
fs: Arc<dyn Fs>,
|
||||||
workspace: WeakView<Workspace>,
|
workspace: WeakView<Workspace>,
|
||||||
active_context_editor: Option<WeakView<ContextEditor>>,
|
active_context_editor: Option<WeakView<ContextEditor>>,
|
||||||
model_selector_menu_handle: PopoverMenuHandle<ContextMenu>,
|
|
||||||
model_summary_editor: View<Editor>,
|
model_summary_editor: View<Editor>,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl ContextEditorToolbarItem {
|
impl ContextEditorToolbarItem {
|
||||||
pub fn new(
|
pub fn new(
|
||||||
workspace: &Workspace,
|
workspace: &Workspace,
|
||||||
model_selector_menu_handle: PopoverMenuHandle<ContextMenu>,
|
_model_selector_menu_handle: PopoverMenuHandle<ContextMenu>,
|
||||||
model_summary_editor: View<Editor>,
|
model_summary_editor: View<Editor>,
|
||||||
) -> Self {
|
) -> Self {
|
||||||
Self {
|
Self {
|
||||||
fs: workspace.app_state().fs.clone(),
|
fs: workspace.app_state().fs.clone(),
|
||||||
workspace: workspace.weak_handle(),
|
workspace: workspace.weak_handle(),
|
||||||
active_context_editor: None,
|
active_context_editor: None,
|
||||||
model_selector_menu_handle,
|
|
||||||
model_summary_editor,
|
model_summary_editor,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
@ -2946,49 +2944,46 @@ impl Render for ContextEditorToolbarItem {
|
||||||
});
|
});
|
||||||
let right_side = h_flex()
|
let right_side = h_flex()
|
||||||
.gap_2()
|
.gap_2()
|
||||||
.child(
|
.child(ModelSelector::new(
|
||||||
ModelSelector::new(
|
self.fs.clone(),
|
||||||
self.fs.clone(),
|
ButtonLike::new("active-model")
|
||||||
ButtonLike::new("active-model")
|
.style(ButtonStyle::Subtle)
|
||||||
.style(ButtonStyle::Subtle)
|
.child(
|
||||||
.child(
|
h_flex()
|
||||||
h_flex()
|
.w_full()
|
||||||
.w_full()
|
.gap_0p5()
|
||||||
.gap_0p5()
|
.child(
|
||||||
.child(
|
div()
|
||||||
div()
|
.overflow_x_hidden()
|
||||||
.overflow_x_hidden()
|
.flex_grow()
|
||||||
.flex_grow()
|
.whitespace_nowrap()
|
||||||
.whitespace_nowrap()
|
.child(
|
||||||
.child(
|
Label::new(
|
||||||
Label::new(
|
LanguageModelRegistry::read_global(cx)
|
||||||
LanguageModelRegistry::read_global(cx)
|
.active_model()
|
||||||
.active_model()
|
.map(|model| {
|
||||||
.map(|model| {
|
format!(
|
||||||
format!(
|
"{}: {}",
|
||||||
"{}: {}",
|
model.provider_name().0,
|
||||||
model.provider_name().0,
|
model.name().0
|
||||||
model.name().0
|
)
|
||||||
)
|
})
|
||||||
})
|
.unwrap_or_else(|| "No model selected".into()),
|
||||||
.unwrap_or_else(|| "No model selected".into()),
|
)
|
||||||
)
|
.size(LabelSize::Small)
|
||||||
.size(LabelSize::Small)
|
.color(Color::Muted),
|
||||||
.color(Color::Muted),
|
),
|
||||||
),
|
)
|
||||||
)
|
.child(
|
||||||
.child(
|
Icon::new(IconName::ChevronDown)
|
||||||
Icon::new(IconName::ChevronDown)
|
.color(Color::Muted)
|
||||||
.color(Color::Muted)
|
.size(IconSize::XSmall),
|
||||||
.size(IconSize::XSmall),
|
),
|
||||||
),
|
)
|
||||||
)
|
.tooltip(move |cx| {
|
||||||
.tooltip(move |cx| {
|
Tooltip::for_action("Change Model", &ToggleModelSelector, cx)
|
||||||
Tooltip::for_action("Change Model", &ToggleModelSelector, cx)
|
}),
|
||||||
}),
|
))
|
||||||
)
|
|
||||||
.with_handle(self.model_selector_menu_handle.clone()),
|
|
||||||
)
|
|
||||||
.children(self.render_remaining_tokens(cx))
|
.children(self.render_remaining_tokens(cx))
|
||||||
.child(self.render_inject_context_menu(cx));
|
.child(self.render_inject_context_menu(cx));
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -1,21 +1,45 @@
|
||||||
use std::sync::Arc;
|
use language_model::{LanguageModel, LanguageModelAvailability, LanguageModelRegistry};
|
||||||
|
|
||||||
use crate::{assistant_settings::AssistantSettings, ShowConfiguration};
|
|
||||||
use fs::Fs;
|
|
||||||
use gpui::{Action, SharedString};
|
|
||||||
use language_model::{LanguageModelAvailability, LanguageModelRegistry};
|
|
||||||
use proto::Plan;
|
use proto::Plan;
|
||||||
|
|
||||||
|
use std::sync::Arc;
|
||||||
|
use ui::ListItemSpacing;
|
||||||
|
|
||||||
|
use crate::assistant_settings::AssistantSettings;
|
||||||
|
use crate::ShowConfiguration;
|
||||||
|
use fs::Fs;
|
||||||
|
use gpui::Action;
|
||||||
|
use gpui::SharedString;
|
||||||
|
use gpui::Task;
|
||||||
|
use picker::{Picker, PickerDelegate};
|
||||||
use settings::update_settings_file;
|
use settings::update_settings_file;
|
||||||
use ui::{prelude::*, ContextMenu, PopoverMenu, PopoverMenuHandle, PopoverTrigger};
|
use ui::{prelude::*, ListItem, PopoverMenu, PopoverMenuHandle, PopoverTrigger};
|
||||||
|
|
||||||
|
const TRY_ZED_PRO_URL: &str = "https://zed.dev/pro";
|
||||||
|
|
||||||
#[derive(IntoElement)]
|
#[derive(IntoElement)]
|
||||||
pub struct ModelSelector<T: PopoverTrigger> {
|
pub struct ModelSelector<T: PopoverTrigger> {
|
||||||
handle: Option<PopoverMenuHandle<ContextMenu>>,
|
handle: Option<PopoverMenuHandle<Picker<ModelPickerDelegate>>>,
|
||||||
fs: Arc<dyn Fs>,
|
fs: Arc<dyn Fs>,
|
||||||
trigger: T,
|
trigger: T,
|
||||||
info_text: Option<SharedString>,
|
info_text: Option<SharedString>,
|
||||||
}
|
}
|
||||||
|
|
||||||
|
pub struct ModelPickerDelegate {
|
||||||
|
fs: Arc<dyn Fs>,
|
||||||
|
all_models: Vec<ModelInfo>,
|
||||||
|
filtered_models: Vec<ModelInfo>,
|
||||||
|
selected_index: usize,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Clone)]
|
||||||
|
struct ModelInfo {
|
||||||
|
model: Arc<dyn LanguageModel>,
|
||||||
|
_provider_name: SharedString,
|
||||||
|
provider_icon: IconName,
|
||||||
|
availability: LanguageModelAvailability,
|
||||||
|
is_selected: bool,
|
||||||
|
}
|
||||||
|
|
||||||
impl<T: PopoverTrigger> ModelSelector<T> {
|
impl<T: PopoverTrigger> ModelSelector<T> {
|
||||||
pub fn new(fs: Arc<dyn Fs>, trigger: T) -> Self {
|
pub fn new(fs: Arc<dyn Fs>, trigger: T) -> Self {
|
||||||
ModelSelector {
|
ModelSelector {
|
||||||
|
|
@ -26,7 +50,7 @@ impl<T: PopoverTrigger> ModelSelector<T> {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn with_handle(mut self, handle: PopoverMenuHandle<ContextMenu>) -> Self {
|
pub fn with_handle(mut self, handle: PopoverMenuHandle<Picker<ModelPickerDelegate>>) -> Self {
|
||||||
self.handle = Some(handle);
|
self.handle = Some(handle);
|
||||||
self
|
self
|
||||||
}
|
}
|
||||||
|
|
@ -37,148 +61,228 @@ impl<T: PopoverTrigger> ModelSelector<T> {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
impl<T: PopoverTrigger> RenderOnce for ModelSelector<T> {
|
impl PickerDelegate for ModelPickerDelegate {
|
||||||
fn render(self, _cx: &mut WindowContext) -> impl IntoElement {
|
type ListItem = ListItem;
|
||||||
let mut menu = PopoverMenu::new("model-switcher");
|
|
||||||
if let Some(handle) = self.handle {
|
|
||||||
menu = menu.with_handle(handle);
|
|
||||||
}
|
|
||||||
|
|
||||||
let info_text = self.info_text.clone();
|
fn match_count(&self) -> usize {
|
||||||
|
self.filtered_models.len()
|
||||||
|
}
|
||||||
|
|
||||||
menu.menu(move |cx| {
|
fn selected_index(&self) -> usize {
|
||||||
ContextMenu::build(cx, |mut menu, cx| {
|
self.selected_index
|
||||||
if let Some(info_text) = info_text.clone() {
|
}
|
||||||
menu = menu
|
|
||||||
.custom_row(move |_cx| {
|
|
||||||
Label::new(info_text.clone())
|
|
||||||
.color(Color::Muted)
|
|
||||||
.into_any_element()
|
|
||||||
})
|
|
||||||
.separator();
|
|
||||||
}
|
|
||||||
|
|
||||||
for (index, provider) in LanguageModelRegistry::global(cx)
|
fn set_selected_index(&mut self, ix: usize, cx: &mut ViewContext<Picker<Self>>) {
|
||||||
.read(cx)
|
self.selected_index = ix.min(self.filtered_models.len().saturating_sub(1));
|
||||||
.providers()
|
cx.notify();
|
||||||
.into_iter()
|
}
|
||||||
.enumerate()
|
|
||||||
{
|
|
||||||
let provider_icon = provider.icon();
|
|
||||||
let provider_name = provider.name().0.clone();
|
|
||||||
|
|
||||||
if index > 0 {
|
fn placeholder_text(&self, _cx: &mut WindowContext) -> Arc<str> {
|
||||||
menu = menu.separator();
|
"Select a model...".into()
|
||||||
|
}
|
||||||
|
|
||||||
|
fn update_matches(&mut self, query: String, cx: &mut ViewContext<Picker<Self>>) -> Task<()> {
|
||||||
|
let all_models = self.all_models.clone();
|
||||||
|
cx.spawn(|this, mut cx| async move {
|
||||||
|
let filtered_models = cx
|
||||||
|
.background_executor()
|
||||||
|
.spawn(async move {
|
||||||
|
if query.is_empty() {
|
||||||
|
all_models
|
||||||
|
} else {
|
||||||
|
all_models
|
||||||
|
.into_iter()
|
||||||
|
.filter(|model_info| {
|
||||||
|
model_info
|
||||||
|
.model
|
||||||
|
.name()
|
||||||
|
.0
|
||||||
|
.to_lowercase()
|
||||||
|
.contains(&query.to_lowercase())
|
||||||
|
})
|
||||||
|
.collect()
|
||||||
}
|
}
|
||||||
menu = menu.custom_row(move |_| {
|
})
|
||||||
h_flex()
|
.await;
|
||||||
.pb_1()
|
|
||||||
.gap_1p5()
|
this.update(&mut cx, |this, cx| {
|
||||||
.w_full()
|
this.delegate.filtered_models = filtered_models;
|
||||||
.child(
|
this.delegate.set_selected_index(0, cx);
|
||||||
Icon::new(provider_icon)
|
cx.notify();
|
||||||
.color(Color::Muted)
|
})
|
||||||
|
.ok();
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
fn confirm(&mut self, _secondary: bool, cx: &mut ViewContext<Picker<Self>>) {
|
||||||
|
if let Some(model_info) = self.filtered_models.get(self.selected_index) {
|
||||||
|
let model = model_info.model.clone();
|
||||||
|
update_settings_file::<AssistantSettings>(self.fs.clone(), cx, move |settings, _| {
|
||||||
|
settings.set_model(model.clone())
|
||||||
|
});
|
||||||
|
|
||||||
|
// Update the selection status
|
||||||
|
let selected_model_id = model_info.model.id();
|
||||||
|
for model in &mut self.all_models {
|
||||||
|
model.is_selected = model.model.id() == selected_model_id;
|
||||||
|
}
|
||||||
|
for model in &mut self.filtered_models {
|
||||||
|
model.is_selected = model.model.id() == selected_model_id;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn dismissed(&mut self, _cx: &mut ViewContext<Picker<Self>>) {}
|
||||||
|
|
||||||
|
fn render_match(
|
||||||
|
&self,
|
||||||
|
ix: usize,
|
||||||
|
selected: bool,
|
||||||
|
cx: &mut ViewContext<Picker<Self>>,
|
||||||
|
) -> Option<Self::ListItem> {
|
||||||
|
let model_info = self.filtered_models.get(ix)?;
|
||||||
|
|
||||||
|
Some(
|
||||||
|
ListItem::new(ix)
|
||||||
|
.inset(true)
|
||||||
|
.spacing(ListItemSpacing::Sparse)
|
||||||
|
.selected(selected)
|
||||||
|
.start_slot(
|
||||||
|
div().pr_1().child(
|
||||||
|
Icon::new(model_info.provider_icon)
|
||||||
|
.color(Color::Muted)
|
||||||
|
.size(IconSize::XSmall),
|
||||||
|
),
|
||||||
|
)
|
||||||
|
.child(
|
||||||
|
h_flex()
|
||||||
|
.w_full()
|
||||||
|
.justify_between()
|
||||||
|
.font_buffer(cx)
|
||||||
|
.min_w(px(200.))
|
||||||
|
.child(
|
||||||
|
h_flex()
|
||||||
|
.gap_2()
|
||||||
|
.child(Label::new(model_info.model.name().0.clone()))
|
||||||
|
.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),
|
||||||
|
),
|
||||||
|
}),
|
||||||
|
)
|
||||||
|
.child(div().when(model_info.is_selected, |this| {
|
||||||
|
this.child(
|
||||||
|
Icon::new(IconName::Check)
|
||||||
|
.color(Color::Accent)
|
||||||
.size(IconSize::Small),
|
.size(IconSize::Small),
|
||||||
)
|
)
|
||||||
.child(Label::new(provider_name.clone()))
|
})),
|
||||||
.into_any_element()
|
),
|
||||||
});
|
)
|
||||||
|
}
|
||||||
|
|
||||||
let available_models = provider.provided_models(cx);
|
fn render_footer(&self, cx: &mut ViewContext<Picker<Self>>) -> Option<gpui::AnyElement> {
|
||||||
if available_models.is_empty() {
|
let plan = proto::Plan::ZedPro;
|
||||||
menu = menu.custom_entry(
|
let is_trial = false;
|
||||||
{
|
|
||||||
move |_| {
|
|
||||||
h_flex()
|
|
||||||
.w_full()
|
|
||||||
.gap_1()
|
|
||||||
.child(Icon::new(IconName::Settings))
|
|
||||||
.child(Label::new("Configure"))
|
|
||||||
.into_any()
|
|
||||||
}
|
|
||||||
},
|
|
||||||
{
|
|
||||||
|cx| {
|
|
||||||
cx.dispatch_action(ShowConfiguration.boxed_clone());
|
|
||||||
}
|
|
||||||
},
|
|
||||||
);
|
|
||||||
}
|
|
||||||
|
|
||||||
let selected_provider = LanguageModelRegistry::read_global(cx)
|
Some(
|
||||||
.active_provider()
|
h_flex()
|
||||||
.map(|m| m.id());
|
.w_full()
|
||||||
let selected_model = LanguageModelRegistry::read_global(cx)
|
.border_t_1()
|
||||||
.active_model()
|
.border_color(cx.theme().colors().border)
|
||||||
.map(|m| m.id());
|
.p_1()
|
||||||
|
.gap_4()
|
||||||
for available_model in available_models {
|
.justify_between()
|
||||||
menu = menu.custom_entry(
|
.child(match plan {
|
||||||
{
|
// Already a zed pro subscriber
|
||||||
let id = available_model.id();
|
Plan::ZedPro => Button::new("zed-pro", "Zed Pro")
|
||||||
let provider_id = available_model.provider_id();
|
.icon(IconName::ZedAssistant)
|
||||||
let model_name = available_model.name().0.clone();
|
.icon_size(IconSize::Small)
|
||||||
let availability = available_model.availability();
|
.icon_color(Color::Muted)
|
||||||
let selected_model = selected_model.clone();
|
.icon_position(IconPosition::Start)
|
||||||
let selected_provider = selected_provider.clone();
|
.on_click(|_, cx| {
|
||||||
move |cx| {
|
cx.dispatch_action(Box::new(zed_actions::OpenAccountSettings))
|
||||||
h_flex()
|
}),
|
||||||
.w_full()
|
// Free user
|
||||||
.justify_between()
|
Plan::Free => Button::new(
|
||||||
.font_buffer(cx)
|
"try-pro",
|
||||||
.min_w(px(260.))
|
if is_trial {
|
||||||
.child(
|
"Upgrade to Pro"
|
||||||
h_flex()
|
} else {
|
||||||
.gap_2()
|
"Try Pro"
|
||||||
.child(Label::new(model_name.clone()))
|
},
|
||||||
.children(match availability {
|
)
|
||||||
LanguageModelAvailability::Public => None,
|
.on_click(|_, cx| cx.open_url(TRY_ZED_PRO_URL)),
|
||||||
LanguageModelAvailability::RequiresPlan(
|
})
|
||||||
Plan::Free,
|
.child(
|
||||||
) => None,
|
Button::new("configure", "Configure")
|
||||||
LanguageModelAvailability::RequiresPlan(
|
.icon(IconName::Settings)
|
||||||
Plan::ZedPro,
|
.icon_size(IconSize::Small)
|
||||||
) => Some(
|
.icon_color(Color::Muted)
|
||||||
Label::new("Pro")
|
.icon_position(IconPosition::Start)
|
||||||
.size(LabelSize::XSmall)
|
.on_click(|_, cx| {
|
||||||
.color(Color::Muted),
|
cx.dispatch_action(ShowConfiguration.boxed_clone());
|
||||||
),
|
}),
|
||||||
}),
|
)
|
||||||
)
|
.into_any(),
|
||||||
.child(div().when(
|
)
|
||||||
selected_model.as_ref() == Some(&id)
|
}
|
||||||
&& selected_provider.as_ref() == Some(&provider_id),
|
}
|
||||||
|this| {
|
|
||||||
this.child(
|
impl<T: PopoverTrigger> RenderOnce for ModelSelector<T> {
|
||||||
Icon::new(IconName::Check)
|
fn render(self, cx: &mut WindowContext) -> impl IntoElement {
|
||||||
.color(Color::Accent)
|
let selected_provider = LanguageModelRegistry::read_global(cx)
|
||||||
.size(IconSize::Small),
|
.active_provider()
|
||||||
)
|
.map(|m| m.id());
|
||||||
},
|
let selected_model = LanguageModelRegistry::read_global(cx)
|
||||||
))
|
.active_model()
|
||||||
.into_any()
|
.map(|m| m.id());
|
||||||
}
|
|
||||||
},
|
let all_models = LanguageModelRegistry::global(cx)
|
||||||
{
|
.read(cx)
|
||||||
let fs = self.fs.clone();
|
.providers()
|
||||||
let model = available_model.clone();
|
.iter()
|
||||||
move |cx| {
|
.flat_map(|provider| {
|
||||||
let model = model.clone();
|
let provider_name = provider.name().0.clone();
|
||||||
update_settings_file::<AssistantSettings>(
|
let provider_icon = provider.icon();
|
||||||
fs.clone(),
|
let provider_id = provider.id();
|
||||||
cx,
|
let selected_model = selected_model.clone();
|
||||||
move |settings, _| settings.set_model(model),
|
let selected_provider = selected_provider.clone();
|
||||||
);
|
|
||||||
}
|
provider.provided_models(cx).into_iter().map(move |model| {
|
||||||
},
|
let model = model.clone();
|
||||||
);
|
|
||||||
}
|
ModelInfo {
|
||||||
}
|
model: model.clone(),
|
||||||
menu
|
_provider_name: provider_name.clone(),
|
||||||
})
|
provider_icon,
|
||||||
.into()
|
availability: model.availability(),
|
||||||
})
|
is_selected: selected_model.as_ref() == Some(&model.id())
|
||||||
.trigger(self.trigger)
|
&& selected_provider.as_ref() == Some(&provider_id),
|
||||||
.attach(gpui::AnchorCorner::BottomLeft)
|
}
|
||||||
|
})
|
||||||
|
})
|
||||||
|
.collect::<Vec<_>>();
|
||||||
|
|
||||||
|
let delegate = ModelPickerDelegate {
|
||||||
|
fs: self.fs.clone(),
|
||||||
|
all_models: all_models.clone(),
|
||||||
|
filtered_models: all_models,
|
||||||
|
selected_index: 0,
|
||||||
|
};
|
||||||
|
|
||||||
|
let picker_view = cx.new_view(|cx| {
|
||||||
|
let picker = Picker::uniform_list(delegate, cx).max_height(Some(rems(20.).into()));
|
||||||
|
picker
|
||||||
|
});
|
||||||
|
|
||||||
|
PopoverMenu::new("model-switcher")
|
||||||
|
.menu(move |_cx| Some(picker_view.clone()))
|
||||||
|
.trigger(self.trigger)
|
||||||
|
.attach(gpui::AnchorCorner::BottomLeft)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
|
||||||
184
crates/assistant/src/model_selector_old.rs
Normal file
184
crates/assistant/src/model_selector_old.rs
Normal file
|
|
@ -0,0 +1,184 @@
|
||||||
|
use std::sync::Arc;
|
||||||
|
|
||||||
|
use crate::{assistant_settings::AssistantSettings, ShowConfiguration};
|
||||||
|
use fs::Fs;
|
||||||
|
use gpui::{Action, SharedString};
|
||||||
|
use language_model::{LanguageModelAvailability, LanguageModelRegistry};
|
||||||
|
use proto::Plan;
|
||||||
|
use settings::update_settings_file;
|
||||||
|
use ui::{prelude::*, ContextMenu, PopoverMenu, PopoverMenuHandle, PopoverTrigger};
|
||||||
|
|
||||||
|
#[derive(IntoElement)]
|
||||||
|
pub struct ModelSelector<T: PopoverTrigger> {
|
||||||
|
handle: Option<PopoverMenuHandle<ContextMenu>>,
|
||||||
|
fs: Arc<dyn Fs>,
|
||||||
|
trigger: T,
|
||||||
|
info_text: Option<SharedString>,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl<T: PopoverTrigger> ModelSelector<T> {
|
||||||
|
pub fn new(fs: Arc<dyn Fs>, trigger: T) -> Self {
|
||||||
|
ModelSelector {
|
||||||
|
handle: None,
|
||||||
|
fs,
|
||||||
|
trigger,
|
||||||
|
info_text: None,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn with_handle(mut self, handle: PopoverMenuHandle<ContextMenu>) -> Self {
|
||||||
|
self.handle = Some(handle);
|
||||||
|
self
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn with_info_text(mut self, text: impl Into<SharedString>) -> Self {
|
||||||
|
self.info_text = Some(text.into());
|
||||||
|
self
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl<T: PopoverTrigger> RenderOnce for ModelSelector<T> {
|
||||||
|
fn render(self, _cx: &mut WindowContext) -> impl IntoElement {
|
||||||
|
let mut menu = PopoverMenu::new("model-switcher");
|
||||||
|
if let Some(handle) = self.handle {
|
||||||
|
menu = menu.with_handle(handle);
|
||||||
|
}
|
||||||
|
|
||||||
|
let info_text = self.info_text.clone();
|
||||||
|
|
||||||
|
menu.menu(move |cx| {
|
||||||
|
ContextMenu::build(cx, |mut menu, cx| {
|
||||||
|
if let Some(info_text) = info_text.clone() {
|
||||||
|
menu = menu
|
||||||
|
.custom_row(move |_cx| {
|
||||||
|
Label::new(info_text.clone())
|
||||||
|
.color(Color::Muted)
|
||||||
|
.into_any_element()
|
||||||
|
})
|
||||||
|
.separator();
|
||||||
|
}
|
||||||
|
|
||||||
|
for (index, provider) in LanguageModelRegistry::global(cx)
|
||||||
|
.read(cx)
|
||||||
|
.providers()
|
||||||
|
.into_iter()
|
||||||
|
.enumerate()
|
||||||
|
{
|
||||||
|
let provider_icon = provider.icon();
|
||||||
|
let provider_name = provider.name().0.clone();
|
||||||
|
|
||||||
|
if index > 0 {
|
||||||
|
menu = menu.separator();
|
||||||
|
}
|
||||||
|
menu = menu.custom_row(move |_| {
|
||||||
|
h_flex()
|
||||||
|
.pb_1()
|
||||||
|
.gap_1p5()
|
||||||
|
.w_full()
|
||||||
|
.child(
|
||||||
|
Icon::new(provider_icon)
|
||||||
|
.color(Color::Muted)
|
||||||
|
.size(IconSize::Small),
|
||||||
|
)
|
||||||
|
.child(Label::new(provider_name.clone()))
|
||||||
|
.into_any_element()
|
||||||
|
});
|
||||||
|
|
||||||
|
let available_models = provider.provided_models(cx);
|
||||||
|
if available_models.is_empty() {
|
||||||
|
menu = menu.custom_entry(
|
||||||
|
{
|
||||||
|
move |_| {
|
||||||
|
h_flex()
|
||||||
|
.w_full()
|
||||||
|
.gap_1()
|
||||||
|
.child(Icon::new(IconName::Settings))
|
||||||
|
.child(Label::new("Configure"))
|
||||||
|
.into_any()
|
||||||
|
}
|
||||||
|
},
|
||||||
|
{
|
||||||
|
|cx| {
|
||||||
|
cx.dispatch_action(ShowConfiguration.boxed_clone());
|
||||||
|
}
|
||||||
|
},
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
let selected_provider = LanguageModelRegistry::read_global(cx)
|
||||||
|
.active_provider()
|
||||||
|
.map(|m| m.id());
|
||||||
|
let selected_model = LanguageModelRegistry::read_global(cx)
|
||||||
|
.active_model()
|
||||||
|
.map(|m| m.id());
|
||||||
|
|
||||||
|
for available_model in available_models {
|
||||||
|
menu = menu.custom_entry(
|
||||||
|
{
|
||||||
|
let id = available_model.id();
|
||||||
|
let provider_id = available_model.provider_id();
|
||||||
|
let model_name = available_model.name().0.clone();
|
||||||
|
let availability = available_model.availability();
|
||||||
|
let selected_model = selected_model.clone();
|
||||||
|
let selected_provider = selected_provider.clone();
|
||||||
|
move |cx| {
|
||||||
|
h_flex()
|
||||||
|
.w_full()
|
||||||
|
.justify_between()
|
||||||
|
.font_buffer(cx)
|
||||||
|
.min_w(px(260.))
|
||||||
|
.child(
|
||||||
|
h_flex()
|
||||||
|
.gap_2()
|
||||||
|
.child(Label::new(model_name.clone()))
|
||||||
|
.children(match availability {
|
||||||
|
LanguageModelAvailability::Public => None,
|
||||||
|
LanguageModelAvailability::RequiresPlan(
|
||||||
|
Plan::Free,
|
||||||
|
) => None,
|
||||||
|
LanguageModelAvailability::RequiresPlan(
|
||||||
|
Plan::ZedPro,
|
||||||
|
) => Some(
|
||||||
|
Label::new("Pro")
|
||||||
|
.size(LabelSize::XSmall)
|
||||||
|
.color(Color::Muted),
|
||||||
|
),
|
||||||
|
}),
|
||||||
|
)
|
||||||
|
.child(div().when(
|
||||||
|
selected_model.as_ref() == Some(&id)
|
||||||
|
&& selected_provider.as_ref() == Some(&provider_id),
|
||||||
|
|this| {
|
||||||
|
this.child(
|
||||||
|
Icon::new(IconName::Check)
|
||||||
|
.color(Color::Accent)
|
||||||
|
.size(IconSize::Small),
|
||||||
|
)
|
||||||
|
},
|
||||||
|
))
|
||||||
|
.into_any()
|
||||||
|
}
|
||||||
|
},
|
||||||
|
{
|
||||||
|
let fs = self.fs.clone();
|
||||||
|
let model = available_model.clone();
|
||||||
|
move |cx| {
|
||||||
|
let model = model.clone();
|
||||||
|
update_settings_file::<AssistantSettings>(
|
||||||
|
fs.clone(),
|
||||||
|
cx,
|
||||||
|
move |settings, _| settings.set_model(model),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
},
|
||||||
|
);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
menu
|
||||||
|
})
|
||||||
|
.into()
|
||||||
|
})
|
||||||
|
.trigger(self.trigger)
|
||||||
|
.attach(gpui::AnchorCorner::BottomLeft)
|
||||||
|
}
|
||||||
|
}
|
||||||
Loading…
Add table
Add a link
Reference in a new issue