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