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

Clean up language model selector (#26134)

Browse source 
This PR does some cleanup for the language model selector after
https://github.com/zed-industries/zed/pull/26090.

Release Notes:

- N/A
This commit is contained in:
Marshall Bowers 2025-03-05 11:18:01 -05:00 • committed by GitHub
parent 3975d8ea93
commit 728c161e8d
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
5 changed files with 48 additions and 42 deletions
Download patch file Download diff file Expand all files Collapse all files

23
crates/language_model_selector/src/language_model_selector.rs
View file

@ -484,17 +484,17 @@ impl PickerDelegate for LanguageModelPickerDelegate {
}
pub fn inline_language_model_selector(
f: impl Fn(Arc<dyn LanguageModel>, &App) + 'static,
) -> AnyElement {
let f = Rc::new(f);
on_model_changed: impl Fn(Arc<dyn LanguageModel>, &App) + 'static,
) -> PopoverMenu<LanguageModelSelector> {
let on_model_changed = Rc::new(on_model_changed);
PopoverMenu::new("popover-button")
.menu(move |window, cx| {
Some(cx.new(|cx| {
LanguageModelSelector::new(
{
let f = f.clone();
let on_model_changed = on_model_changed.clone();
move |model, cx| {
f(model, cx);
on_model_changed(model, cx);
}
},
window,
@ -524,36 +524,34 @@ pub fn inline_language_model_selector(
},
)
.anchor(gpui::Corner::TopRight)
// .when_some(menu_handle, |el, handle| el.with_handle(handle))
.offset(gpui::Point {
x: px(0.0),
y: px(-2.0),
})
.into_any_element()
}
pub fn assistant_language_model_selector(
keybinding_target: FocusHandle,
menu_handle: Option<PopoverMenuHandle<LanguageModelSelector>>,
cx: &App,
f: impl Fn(Arc<dyn LanguageModel>, &App) + 'static,
) -> AnyElement {
on_model_changed: impl Fn(Arc<dyn LanguageModel>, &App) + 'static,
) -> PopoverMenu<LanguageModelSelector> {
let active_model = LanguageModelRegistry::read_global(cx).active_model();
let model_name = match active_model {
Some(model) => model.name().0,
_ => SharedString::from("No model selected"),
};
let f = Rc::new(f);
let on_model_changed = Rc::new(on_model_changed);
PopoverMenu::new("popover-button")
.menu(move |window, cx| {
Some(cx.new(|cx| {
LanguageModelSelector::new(
{
let f = f.clone();
let on_model_changed = on_model_changed.clone();
move |model, cx| {
f(model, cx);
on_model_changed(model, cx);
}
},
window,
@ -594,5 +592,4 @@ pub fn assistant_language_model_selector(
x: px(0.0),
y: px(-2.0),
})
.into_any_element()
}