Bradley Axen
92 lines
3.4 KiB
Rust
92 lines
3.4 KiB
Rust
use anyhow::Result;
|
|
use futures::future::BoxFuture;
|
|
use std::collections::HashMap;
|
|
use std::path::PathBuf;
|
|
|
|
use crate::acp::{
|
|
extension_configs_to_mcp_servers, AcpProvider, AcpProviderConfig, ACP_CURRENT_MODEL,
|
|
};
|
|
use crate::config::search_path::SearchPaths;
|
|
use crate::config::{Config, GooseMode};
|
|
use crate::model::ModelConfig;
|
|
use crate::providers::acp_tooling::{acp_adapter_installed, acp_inventory_identity};
|
|
use crate::providers::base::{ProviderDef, ProviderMetadata};
|
|
use crate::providers::inventory::InventoryIdentityInput;
|
|
|
|
const AMP_ACP_PROVIDER_NAME: &str = "amp-acp";
|
|
const AMP_ACP_DOC_URL: &str = "https://ampcode.com";
|
|
const AMP_ACP_BINARY: &str = "amp-acp";
|
|
|
|
pub struct AmpAcpProvider;
|
|
|
|
impl ProviderDef for AmpAcpProvider {
|
|
type Provider = AcpProvider;
|
|
|
|
fn metadata() -> ProviderMetadata {
|
|
ProviderMetadata::new(
|
|
AMP_ACP_PROVIDER_NAME,
|
|
"Amp",
|
|
"Use goose with your Amp subscription via the amp-acp adapter.",
|
|
ACP_CURRENT_MODEL,
|
|
vec![],
|
|
AMP_ACP_DOC_URL,
|
|
vec![],
|
|
)
|
|
.with_setup_steps(vec![
|
|
"Install the Amp CLI: `curl -fsSL https://ampcode.com/install.sh | bash`",
|
|
"Install the ACP adapter: `npm install -g amp-acp`",
|
|
"Ensure your Amp CLI is authenticated (run `amp` to verify)",
|
|
"Set in your goose config file (`~/.config/goose/config.yaml` on macOS/Linux):\n GOOSE_PROVIDER: amp-acp\n GOOSE_MODEL: current",
|
|
"Restart goose for changes to take effect",
|
|
])
|
|
.with_model_selection_hint("Use the Amp CLI to configure models")
|
|
}
|
|
|
|
fn from_env(
|
|
model: ModelConfig,
|
|
extensions: Vec<crate::config::ExtensionConfig>,
|
|
) -> BoxFuture<'static, Result<AcpProvider>> {
|
|
Box::pin(async move {
|
|
let config = Config::global();
|
|
let resolved_command = SearchPaths::builder().with_npm().resolve(AMP_ACP_BINARY)?;
|
|
let goose_mode = config.get_goose_mode().unwrap_or(GooseMode::Auto);
|
|
|
|
let mode_mapping = HashMap::from([
|
|
// "bypass" skips confirmations, closest to autonomous mode.
|
|
(GooseMode::Auto, "bypass".to_string()),
|
|
// "default" prompts before risky actions.
|
|
(GooseMode::Approve, "default".to_string()),
|
|
(GooseMode::SmartApprove, "default".to_string()),
|
|
(GooseMode::Chat, "default".to_string()),
|
|
]);
|
|
|
|
let provider_config = AcpProviderConfig {
|
|
command: resolved_command,
|
|
args: vec![],
|
|
env: vec![],
|
|
env_remove: vec![],
|
|
work_dir: std::env::current_dir().unwrap_or_else(|_| PathBuf::from(".")),
|
|
mcp_servers: extension_configs_to_mcp_servers(&extensions),
|
|
session_mode_id: Some(mode_mapping[&goose_mode].clone()),
|
|
mode_mapping,
|
|
notification_callback: None,
|
|
};
|
|
|
|
let metadata = Self::metadata();
|
|
AcpProvider::connect(metadata.name, model, goose_mode, provider_config).await
|
|
})
|
|
}
|
|
|
|
fn supports_inventory_refresh() -> bool {
|
|
false
|
|
}
|
|
|
|
fn inventory_identity() -> Result<InventoryIdentityInput> {
|
|
acp_inventory_identity(AMP_ACP_PROVIDER_NAME, AMP_ACP_BINARY)
|
|
}
|
|
|
|
fn inventory_configured() -> bool {
|
|
acp_adapter_installed(AMP_ACP_BINARY)
|
|
}
|
|
}
|