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::{AgentDiffPane, AgentDiffToolbar};
 49pub use context_store::ContextStore;
 50pub use ui::{all_agent_previews, get_agent_preview};
 51
 52actions!(
 53    agent,
 54    [
 55        NewTextThread,
 56        ToggleContextPicker,
 57        ToggleNavigationMenu,
 58        ToggleOptionsMenu,
 59        DeleteRecentlyOpenThread,
 60        ToggleProfileSelector,
 61        RemoveAllContext,
 62        ExpandMessageEditor,
 63        OpenHistory,
 64        AddContextServer,
 65        RemoveSelectedThread,
 66        Chat,
 67        CycleNextInlineAssist,
 68        CyclePreviousInlineAssist,
 69        FocusUp,
 70        FocusDown,
 71        FocusLeft,
 72        FocusRight,
 73        RemoveFocusedContext,
 74        AcceptSuggestedContext,
 75        OpenActiveThreadAsMarkdown,
 76        OpenAgentDiff,
 77        Keep,
 78        Reject,
 79        RejectAll,
 80        KeepAll,
 81        Follow
 82    ]
 83);
 84
 85#[derive(Default, Clone, PartialEq, Deserialize, JsonSchema)]
 86pub struct NewThread {
 87    #[serde(default)]
 88    from_thread_id: Option<ThreadId>,
 89}
 90
 91#[derive(PartialEq, Clone, Default, Debug, Deserialize, JsonSchema)]
 92pub struct ManageProfiles {
 93    #[serde(default)]
 94    pub customize_tools: Option<AgentProfileId>,
 95}
 96
 97impl ManageProfiles {
 98    pub fn customize_tools(profile_id: AgentProfileId) -> Self {
 99        Self {
100            customize_tools: Some(profile_id),
101        }
102    }
103}
104
105impl_actions!(agent, [NewThread, ManageProfiles]);
106
107const NAMESPACE: &str = "agent";
108
109/// Initializes the `agent` crate.
110pub fn init(
111    fs: Arc<dyn Fs>,
112    client: Arc<Client>,
113    prompt_builder: Arc<PromptBuilder>,
114    language_registry: Arc<LanguageRegistry>,
115    cx: &mut App,
116) {
117    AssistantSettings::register(cx);
118    thread_store::init(cx);
119    assistant_panel::init(cx);
120    context_server_configuration::init(language_registry, cx);
121
122    inline_assistant::init(
123        fs.clone(),
124        prompt_builder.clone(),
125        client.telemetry().clone(),
126        cx,
127    );
128    terminal_inline_assistant::init(
129        fs.clone(),
130        prompt_builder.clone(),
131        client.telemetry().clone(),
132        cx,
133    );
134    cx.observe_new(AddContextServerModal::register).detach();
135    cx.observe_new(ManageProfilesModal::register).detach();
136
137    feature_gate_agent_actions(cx);
138}
139
140fn feature_gate_agent_actions(cx: &mut App) {
141    CommandPaletteFilter::update_global(cx, |filter, _cx| {
142        filter.hide_namespace(NAMESPACE);
143    });
144
145    cx.observe_flag::<Assistant2FeatureFlag, _>(move |is_enabled, cx| {
146        if is_enabled {
147            CommandPaletteFilter::update_global(cx, |filter, _cx| {
148                filter.show_namespace(NAMESPACE);
149            });
150        } else {
151            CommandPaletteFilter::update_global(cx, |filter, _cx| {
152                filter.hide_namespace(NAMESPACE);
153            });
154        }
155    })
156    .detach();
157}