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