feat: add basic support for aws bedrock to desktop app (#1271)

crates/goose-server/src/routes/providers_and_keys.json

@@ -52,5 +52,11 @@
         "description": "Connect to Azure OpenAI Service",
         "models": ["gpt-4o", "gpt-4o-mini"],
         "required_keys": ["AZURE_OPENAI_API_KEY", "AZURE_OPENAI_ENDPOINT", "AZURE_OPENAI_DEPLOYMENT_NAME"]
+    },
+    "aws_bedrock": {
+        "name": "AWS Bedrock",
+        "description": "Connect to LLMs via AWS Bedrock",
+        "models": ["us.anthropic.claude-3-7-sonnet-20250219-v1:0"],
+        "required_keys": ["AWS_PROFILE"]
     }
 }

crates/goose/Cargo.toml

@@ -63,9 +63,9 @@ rand = "0.8.5"
 utoipa = "4.1"
 
 # For Bedrock provider
-aws-config = { version = "1.1.7", features = ["behavior-version-latest"] }
-aws-smithy-types = "1.2.12"
-aws-sdk-bedrockruntime = "1.72.0"
+aws-config = { version = "1.5.16", features = ["behavior-version-latest"] }
+aws-smithy-types = "1.2.13"
+aws-sdk-bedrockruntime = "1.74.0"
 
 # For GCP Vertex AI provider auth
 jsonwebtoken = "9.3.1"

crates/goose/src/config/base.rs

@@ -191,7 +191,7 @@ impl Config {
     }
 
     // Load current secrets from the keyring
-    fn load_secrets(&self) -> Result<HashMap<String, Value>, ConfigError> {
+    pub fn load_secrets(&self) -> Result<HashMap<String, Value>, ConfigError> {
         let entry = Entry::new(&self.keyring_service, KEYRING_USERNAME)?;
 
         match entry.get_password() {

crates/goose/src/providers/bedrock.rs

@@ -4,7 +4,7 @@ use aws_sdk_bedrockruntime::operation::converse::ConverseError;
 use aws_sdk_bedrockruntime::{types as bedrock, Client};
 use mcp_core::Tool;
 
-use super::base::{Provider, ProviderMetadata, ProviderUsage};
+use super::base::{ConfigKey, Provider, ProviderMetadata, ProviderUsage};
 use super::errors::ProviderError;
 use crate::message::Message;
 use crate::model::ModelConfig;
@@ -33,6 +33,12 @@ pub struct BedrockProvider {
 
 impl BedrockProvider {
     pub fn from_env(model: ModelConfig) -> Result<Self> {
+        let config = crate::config::Config::global();
+        for (key, value) in config.load_secrets()?.iter() {
+            if key.starts_with("AWS_") && value.is_string() {
+                std::env::set_var(key, value.as_str().unwrap());
+            }
+        }
         let sdk_config = futures::executor::block_on(aws_config::load_from_env());
         let client = Client::new(&sdk_config);
 
@@ -51,13 +57,13 @@ impl Default for BedrockProvider {
 impl Provider for BedrockProvider {
     fn metadata() -> ProviderMetadata {
         ProviderMetadata::new(
-            "bedrock",
+            "aws_bedrock",
             "Amazon Bedrock",
-            "Run models through Amazon Bedrock. You may have to set AWS_ACCESS_KEY_ID, AWS_ACCESS_KEY, and AWS_REGION as env vars before configuring.",
+            "Run models through Amazon Bedrock. You may have to set 'AWS_' environment variables to configure authentication.",
             BEDROCK_DEFAULT_MODEL,
             BEDROCK_KNOWN_MODELS.iter().map(|s| s.to_string()).collect(),
             BEDROCK_DOC_LINK,
-            vec![],
+            vec![ConfigKey::new("AWS_PROFILE", true, false, Some("us-west-2"))],
         )
     }
 

crates/goose/src/providers/factory.rs

@@ -34,7 +34,7 @@ pub fn create(name: &str, model: ModelConfig) -> Result<Box<dyn Provider + Send
         "openai" => Ok(Box::new(OpenAiProvider::from_env(model)?)),
         "anthropic" => Ok(Box::new(AnthropicProvider::from_env(model)?)),
         "azure_openai" => Ok(Box::new(AzureProvider::from_env(model)?)),
-        "bedrock" => Ok(Box::new(BedrockProvider::from_env(model)?)),
+        "aws_bedrock" => Ok(Box::new(BedrockProvider::from_env(model)?)),
         "databricks" => Ok(Box::new(DatabricksProvider::from_env(model)?)),
         "groq" => Ok(Box::new(GroqProvider::from_env(model)?)),
         "ollama" => Ok(Box::new(OllamaProvider::from_env(model)?)),

crates/goose/tests/truncate_agent.rs

@@ -37,7 +37,7 @@ impl ProviderType {
             ],
             ProviderType::OpenAi => &["OPENAI_API_KEY"],
             ProviderType::Anthropic => &["ANTHROPIC_API_KEY"],
-            ProviderType::Bedrock => &["AWS_PROFILE", "AWS_REGION"],
+            ProviderType::Bedrock => &["AWS_PROFILE"],
             ProviderType::Databricks => &["DATABRICKS_HOST"],
             ProviderType::Google => &["GOOGLE_API_KEY"],
             ProviderType::Groq => &["GROQ_API_KEY"],