248 lines
8 KiB
Rust
248 lines
8 KiB
Rust
#![cfg_attr(target_os = "windows", allow(unused, dead_code))]
|
|
|
|
mod assistant_configuration;
|
|
pub mod assistant_panel;
|
|
mod inline_assistant;
|
|
pub mod slash_command_settings;
|
|
mod terminal_inline_assistant;
|
|
|
|
use std::sync::Arc;
|
|
|
|
use assistant_settings::AssistantSettings;
|
|
use assistant_slash_command::SlashCommandRegistry;
|
|
use client::Client;
|
|
use command_palette_hooks::CommandPaletteFilter;
|
|
use feature_flags::FeatureFlagAppExt;
|
|
use fs::Fs;
|
|
use gpui::{App, Global, ReadGlobal, UpdateGlobal, actions};
|
|
use language_model::{
|
|
LanguageModelId, LanguageModelProviderId, LanguageModelRegistry, LanguageModelResponseMessage,
|
|
};
|
|
use prompt_store::PromptBuilder;
|
|
use serde::Deserialize;
|
|
use settings::{Settings, SettingsStore};
|
|
|
|
pub use crate::assistant_panel::{AssistantPanel, AssistantPanelEvent};
|
|
pub(crate) use crate::inline_assistant::*;
|
|
use crate::slash_command_settings::SlashCommandSettings;
|
|
|
|
actions!(
|
|
assistant,
|
|
[
|
|
InsertActivePrompt,
|
|
DeployHistory,
|
|
NewChat,
|
|
CycleNextInlineAssist,
|
|
CyclePreviousInlineAssist
|
|
]
|
|
);
|
|
|
|
const DEFAULT_CONTEXT_LINES: usize = 50;
|
|
|
|
#[derive(Deserialize, Debug)]
|
|
pub struct LanguageModelUsage {
|
|
pub prompt_tokens: u32,
|
|
pub completion_tokens: u32,
|
|
pub total_tokens: u32,
|
|
}
|
|
|
|
#[derive(Deserialize, Debug)]
|
|
pub struct LanguageModelChoiceDelta {
|
|
pub index: u32,
|
|
pub delta: LanguageModelResponseMessage,
|
|
pub finish_reason: Option<String>,
|
|
}
|
|
|
|
/// The state pertaining to the Assistant.
|
|
#[derive(Default)]
|
|
struct Assistant {
|
|
/// Whether the Assistant is enabled.
|
|
enabled: bool,
|
|
}
|
|
|
|
impl Global for Assistant {}
|
|
|
|
impl Assistant {
|
|
const NAMESPACE: &'static str = "assistant";
|
|
|
|
fn set_enabled(&mut self, enabled: bool, cx: &mut App) {
|
|
if self.enabled == enabled {
|
|
return;
|
|
}
|
|
|
|
self.enabled = enabled;
|
|
|
|
if !enabled {
|
|
CommandPaletteFilter::update_global(cx, |filter, _cx| {
|
|
filter.hide_namespace(Self::NAMESPACE);
|
|
});
|
|
|
|
return;
|
|
}
|
|
|
|
CommandPaletteFilter::update_global(cx, |filter, _cx| {
|
|
filter.show_namespace(Self::NAMESPACE);
|
|
});
|
|
}
|
|
|
|
pub fn enabled(cx: &App) -> bool {
|
|
Self::global(cx).enabled
|
|
}
|
|
}
|
|
|
|
pub fn init(
|
|
fs: Arc<dyn Fs>,
|
|
client: Arc<Client>,
|
|
prompt_builder: Arc<PromptBuilder>,
|
|
cx: &mut App,
|
|
) {
|
|
cx.set_global(Assistant::default());
|
|
AssistantSettings::register(cx);
|
|
SlashCommandSettings::register(cx);
|
|
|
|
assistant_context_editor::init(client.clone(), cx);
|
|
prompt_library::init(cx);
|
|
init_language_model_settings(cx);
|
|
assistant_slash_command::init(cx);
|
|
assistant_tool::init(cx);
|
|
assistant_panel::init(cx);
|
|
context_server::init(cx);
|
|
|
|
register_slash_commands(cx);
|
|
inline_assistant::init(
|
|
fs.clone(),
|
|
prompt_builder.clone(),
|
|
client.telemetry().clone(),
|
|
cx,
|
|
);
|
|
terminal_inline_assistant::init(
|
|
fs.clone(),
|
|
prompt_builder.clone(),
|
|
client.telemetry().clone(),
|
|
cx,
|
|
);
|
|
indexed_docs::init(cx);
|
|
|
|
CommandPaletteFilter::update_global(cx, |filter, _cx| {
|
|
filter.hide_namespace(Assistant::NAMESPACE);
|
|
});
|
|
Assistant::update_global(cx, |assistant, cx| {
|
|
let settings = AssistantSettings::get_global(cx);
|
|
|
|
assistant.set_enabled(settings.enabled, cx);
|
|
});
|
|
cx.observe_global::<SettingsStore>(|cx| {
|
|
Assistant::update_global(cx, |assistant, cx| {
|
|
let settings = AssistantSettings::get_global(cx);
|
|
assistant.set_enabled(settings.enabled, cx);
|
|
});
|
|
})
|
|
.detach();
|
|
}
|
|
|
|
fn init_language_model_settings(cx: &mut App) {
|
|
update_active_language_model_from_settings(cx);
|
|
|
|
cx.observe_global::<SettingsStore>(update_active_language_model_from_settings)
|
|
.detach();
|
|
cx.subscribe(
|
|
&LanguageModelRegistry::global(cx),
|
|
|_, event: &language_model::Event, cx| match event {
|
|
language_model::Event::ProviderStateChanged
|
|
| language_model::Event::AddedProvider(_)
|
|
| language_model::Event::RemovedProvider(_) => {
|
|
update_active_language_model_from_settings(cx);
|
|
}
|
|
_ => {}
|
|
},
|
|
)
|
|
.detach();
|
|
}
|
|
|
|
fn update_active_language_model_from_settings(cx: &mut App) {
|
|
let settings = AssistantSettings::get_global(cx);
|
|
let active_model_provider_name =
|
|
LanguageModelProviderId::from(settings.default_model.provider.clone());
|
|
let active_model_id = LanguageModelId::from(settings.default_model.model.clone());
|
|
let editor_provider_name =
|
|
LanguageModelProviderId::from(settings.editor_model.provider.clone());
|
|
let editor_model_id = LanguageModelId::from(settings.editor_model.model.clone());
|
|
let inline_alternatives = settings
|
|
.inline_alternatives
|
|
.iter()
|
|
.map(|alternative| {
|
|
(
|
|
LanguageModelProviderId::from(alternative.provider.clone()),
|
|
LanguageModelId::from(alternative.model.clone()),
|
|
)
|
|
})
|
|
.collect::<Vec<_>>();
|
|
LanguageModelRegistry::global(cx).update(cx, |registry, cx| {
|
|
registry.select_active_model(&active_model_provider_name, &active_model_id, cx);
|
|
registry.select_editor_model(&editor_provider_name, &editor_model_id, cx);
|
|
registry.select_inline_alternative_models(inline_alternatives, cx);
|
|
});
|
|
}
|
|
|
|
fn register_slash_commands(cx: &mut App) {
|
|
let slash_command_registry = SlashCommandRegistry::global(cx);
|
|
|
|
slash_command_registry.register_command(assistant_slash_commands::FileSlashCommand, true);
|
|
slash_command_registry.register_command(assistant_slash_commands::DeltaSlashCommand, true);
|
|
slash_command_registry.register_command(assistant_slash_commands::OutlineSlashCommand, true);
|
|
slash_command_registry.register_command(assistant_slash_commands::TabSlashCommand, true);
|
|
slash_command_registry
|
|
.register_command(assistant_slash_commands::CargoWorkspaceSlashCommand, true);
|
|
slash_command_registry.register_command(assistant_slash_commands::PromptSlashCommand, true);
|
|
slash_command_registry.register_command(assistant_slash_commands::SelectionCommand, true);
|
|
slash_command_registry.register_command(assistant_slash_commands::DefaultSlashCommand, false);
|
|
slash_command_registry.register_command(assistant_slash_commands::TerminalSlashCommand, true);
|
|
slash_command_registry.register_command(assistant_slash_commands::NowSlashCommand, false);
|
|
slash_command_registry
|
|
.register_command(assistant_slash_commands::DiagnosticsSlashCommand, true);
|
|
slash_command_registry.register_command(assistant_slash_commands::FetchSlashCommand, true);
|
|
|
|
cx.observe_flag::<assistant_slash_commands::StreamingExampleSlashCommandFeatureFlag, _>({
|
|
let slash_command_registry = slash_command_registry.clone();
|
|
move |is_enabled, _cx| {
|
|
if is_enabled {
|
|
slash_command_registry.register_command(
|
|
assistant_slash_commands::StreamingExampleSlashCommand,
|
|
false,
|
|
);
|
|
}
|
|
}
|
|
})
|
|
.detach();
|
|
|
|
update_slash_commands_from_settings(cx);
|
|
cx.observe_global::<SettingsStore>(update_slash_commands_from_settings)
|
|
.detach();
|
|
}
|
|
|
|
fn update_slash_commands_from_settings(cx: &mut App) {
|
|
let slash_command_registry = SlashCommandRegistry::global(cx);
|
|
let settings = SlashCommandSettings::get_global(cx);
|
|
|
|
if settings.docs.enabled {
|
|
slash_command_registry.register_command(assistant_slash_commands::DocsSlashCommand, true);
|
|
} else {
|
|
slash_command_registry.unregister_command(assistant_slash_commands::DocsSlashCommand);
|
|
}
|
|
|
|
if settings.cargo_workspace.enabled {
|
|
slash_command_registry
|
|
.register_command(assistant_slash_commands::CargoWorkspaceSlashCommand, true);
|
|
} else {
|
|
slash_command_registry
|
|
.unregister_command(assistant_slash_commands::CargoWorkspaceSlashCommand);
|
|
}
|
|
}
|
|
|
|
#[cfg(test)]
|
|
#[ctor::ctor]
|
|
fn init_logger() {
|
|
if std::env::var("RUST_LOG").is_ok() {
|
|
env_logger::init();
|
|
}
|
|
}
|