1use assistant_settings::AssistantSettings;
 2use fs::Fs;
 3use gpui::{Entity, FocusHandle};
 4use language_model_selector::{AssistantLanguageModelSelector, LanguageModelSelector};
 5use settings::update_settings_file;
 6use std::sync::Arc;
 7use ui::prelude::*;
 8
 9pub struct AssistantModelSelector {
10    pub selector: Entity<LanguageModelSelector>,
11    focus_handle: FocusHandle,
12}
13
14impl AssistantModelSelector {
15    pub(crate) fn new(
16        fs: Arc<dyn Fs>,
17        focus_handle: FocusHandle,
18        window: &mut Window,
19        cx: &mut App,
20    ) -> Self {
21        Self {
22            selector: cx.new(|cx| {
23                let fs = fs.clone();
24                LanguageModelSelector::new(
25                    move |model, cx| {
26                        update_settings_file::<AssistantSettings>(
27                            fs.clone(),
28                            cx,
29                            move |settings, _cx| settings.set_model(model.clone()),
30                        );
31                    },
32                    window,
33                    cx,
34                )
35            }),
36            focus_handle,
37        }
38    }
39}
40
41impl Render for AssistantModelSelector {
42    fn render(&mut self, window: &mut Window, cx: &mut Context<Self>) -> impl IntoElement {
43        AssistantLanguageModelSelector::new(self.focus_handle.clone(), self.selector.clone())
44            .render(window, cx)
45    }
46}