1mod active_thread;
  2mod agent_diff;
  3mod assistant_configuration;
  4mod assistant_model_selector;
  5mod assistant_panel;
  6mod buffer_codegen;
  7mod context;
  8mod context_picker;
  9mod context_server_configuration;
 10mod context_store;
 11mod context_strip;
 12mod history_store;
 13mod inline_assistant;
 14mod inline_prompt_editor;
 15mod message_editor;
 16mod profile_selector;
 17mod terminal_codegen;
 18mod terminal_inline_assistant;
 19mod thread;
 20mod thread_history;
 21mod thread_store;
 22mod tool_compatibility;
 23mod tool_use;
 24mod ui;
 25
 26use std::sync::Arc;
 27
 28use assistant_settings::{AgentProfileId, AssistantSettings};
 29use client::Client;
 30use command_palette_hooks::CommandPaletteFilter;
 31use feature_flags::{Assistant2FeatureFlag, FeatureFlagAppExt};
 32use fs::Fs;
 33use gpui::{App, actions, impl_actions};
 34use language::LanguageRegistry;
 35use prompt_store::PromptBuilder;
 36use schemars::JsonSchema;
 37use serde::Deserialize;
 38use settings::Settings as _;
 39use thread::ThreadId;
 40
 41pub use crate::active_thread::ActiveThread;
 42use crate::assistant_configuration::{AddContextServerModal, ManageProfilesModal};
 43pub use crate::assistant_panel::{AssistantPanel, ConcreteAssistantPanelDelegate};
 44pub use crate::context::{ContextLoadResult, LoadedContext};
 45pub use crate::inline_assistant::InlineAssistant;
 46pub use crate::thread::{Message, MessageSegment, Thread, ThreadEvent};
 47pub use crate::thread_store::ThreadStore;
 48pub use agent_diff::{AgentDiff, AgentDiffToolbar};
 49
 50actions!(
 51    agent,
 52    [
 53        NewTextThread,
 54        ToggleContextPicker,
 55        ToggleNavigationMenu,
 56        ToggleOptionsMenu,
 57        DeleteRecentlyOpenThread,
 58        ToggleProfileSelector,
 59        RemoveAllContext,
 60        ExpandMessageEditor,
 61        OpenHistory,
 62        AddContextServer,
 63        RemoveSelectedThread,
 64        Chat,
 65        ChatMode,
 66        CycleNextInlineAssist,
 67        CyclePreviousInlineAssist,
 68        FocusUp,
 69        FocusDown,
 70        FocusLeft,
 71        FocusRight,
 72        RemoveFocusedContext,
 73        AcceptSuggestedContext,
 74        OpenActiveThreadAsMarkdown,
 75        OpenAgentDiff,
 76        Keep,
 77        Reject,
 78        RejectAll,
 79        KeepAll
 80    ]
 81);
 82
 83#[derive(Default, Clone, PartialEq, Deserialize, JsonSchema)]
 84pub struct NewThread {
 85    #[serde(default)]
 86    from_thread_id: Option<ThreadId>,
 87}
 88
 89#[derive(PartialEq, Clone, Default, Debug, Deserialize, JsonSchema)]
 90pub struct ManageProfiles {
 91    #[serde(default)]
 92    pub customize_tools: Option<AgentProfileId>,
 93}
 94
 95impl ManageProfiles {
 96    pub fn customize_tools(profile_id: AgentProfileId) -> Self {
 97        Self {
 98            customize_tools: Some(profile_id),
 99        }
100    }
101}
102
103impl_actions!(agent, [NewThread, ManageProfiles]);
104
105const NAMESPACE: &str = "agent";
106
107/// Initializes the `agent` crate.
108pub fn init(
109    fs: Arc<dyn Fs>,
110    client: Arc<Client>,
111    prompt_builder: Arc<PromptBuilder>,
112    language_registry: Arc<LanguageRegistry>,
113    cx: &mut App,
114) {
115    AssistantSettings::register(cx);
116    thread_store::init(cx);
117    assistant_panel::init(cx);
118    context_server_configuration::init(language_registry, cx);
119
120    inline_assistant::init(
121        fs.clone(),
122        prompt_builder.clone(),
123        client.telemetry().clone(),
124        cx,
125    );
126    terminal_inline_assistant::init(
127        fs.clone(),
128        prompt_builder.clone(),
129        client.telemetry().clone(),
130        cx,
131    );
132    cx.observe_new(AddContextServerModal::register).detach();
133    cx.observe_new(ManageProfilesModal::register).detach();
134
135    feature_gate_agent_actions(cx);
136}
137
138fn feature_gate_agent_actions(cx: &mut App) {
139    CommandPaletteFilter::update_global(cx, |filter, _cx| {
140        filter.hide_namespace(NAMESPACE);
141    });
142
143    cx.observe_flag::<Assistant2FeatureFlag, _>(move |is_enabled, cx| {
144        if is_enabled {
145            CommandPaletteFilter::update_global(cx, |filter, _cx| {
146                filter.show_namespace(NAMESPACE);
147            });
148        } else {
149            CommandPaletteFilter::update_global(cx, |filter, _cx| {
150                filter.hide_namespace(NAMESPACE);
151            });
152        }
153    })
154    .detach();
155}