1mod active_thread;
  2mod assistant_model_selector;
  3mod assistant_panel;
  4mod buffer_codegen;
  5mod context;
  6mod context_picker;
  7mod context_store;
  8mod context_strip;
  9mod inline_assistant;
 10mod inline_prompt_editor;
 11mod message_editor;
 12mod terminal_codegen;
 13mod terminal_inline_assistant;
 14mod thread;
 15mod thread_history;
 16mod thread_store;
 17mod ui;
 18
 19use std::sync::Arc;
 20
 21use assistant_settings::AssistantSettings;
 22use client::Client;
 23use command_palette_hooks::CommandPaletteFilter;
 24use feature_flags::{Assistant2FeatureFlag, FeatureFlagAppExt};
 25use fs::Fs;
 26use gpui::{actions, AppContext};
 27use prompt_library::{PromptBuilder, PromptLoadingParams};
 28use settings::Settings as _;
 29use util::ResultExt;
 30
 31pub use crate::assistant_panel::AssistantPanel;
 32pub use crate::inline_assistant::InlineAssistant;
 33
 34actions!(
 35    assistant2,
 36    [
 37        ToggleFocus,
 38        NewThread,
 39        ToggleContextPicker,
 40        ToggleModelSelector,
 41        RemoveAllContext,
 42        OpenHistory,
 43        Chat,
 44        ChatMode,
 45        CycleNextInlineAssist,
 46        CyclePreviousInlineAssist,
 47        FocusUp,
 48        FocusDown,
 49        FocusLeft,
 50        FocusRight,
 51        RemoveFocusedContext,
 52        AcceptSuggestedContext
 53    ]
 54);
 55
 56const NAMESPACE: &str = "assistant2";
 57
 58/// Initializes the `assistant2` crate.
 59pub fn init(fs: Arc<dyn Fs>, client: Arc<Client>, stdout_is_a_pty: bool, cx: &mut AppContext) {
 60    AssistantSettings::register(cx);
 61    assistant_panel::init(cx);
 62
 63    let prompt_builder = PromptBuilder::new(Some(PromptLoadingParams {
 64        fs: fs.clone(),
 65        repo_path: stdout_is_a_pty
 66            .then(|| std::env::current_dir().log_err())
 67            .flatten(),
 68        cx,
 69    }))
 70    .log_err()
 71    .map(Arc::new)
 72    .unwrap_or_else(|| Arc::new(PromptBuilder::new(None).unwrap()));
 73    inline_assistant::init(
 74        fs.clone(),
 75        prompt_builder.clone(),
 76        client.telemetry().clone(),
 77        cx,
 78    );
 79    terminal_inline_assistant::init(
 80        fs.clone(),
 81        prompt_builder.clone(),
 82        client.telemetry().clone(),
 83        cx,
 84    );
 85
 86    feature_gate_assistant2_actions(cx);
 87}
 88
 89fn feature_gate_assistant2_actions(cx: &mut AppContext) {
 90    CommandPaletteFilter::update_global(cx, |filter, _cx| {
 91        filter.hide_namespace(NAMESPACE);
 92    });
 93
 94    cx.observe_flag::<Assistant2FeatureFlag, _>(move |is_enabled, cx| {
 95        if is_enabled {
 96            CommandPaletteFilter::update_global(cx, |filter, _cx| {
 97                filter.show_namespace(NAMESPACE);
 98            });
 99        } else {
100            CommandPaletteFilter::update_global(cx, |filter, _cx| {
101                filter.hide_namespace(NAMESPACE);
102            });
103        }
104    })
105    .detach();
106}