assistant: Add Bedrock support (#21092)

Closes #16544 Release Notes: - Added support for AWS Bedrock to the Assistant. --------- Co-authored-by: Marshall Bowers <[email protected]> Co-authored-by: Anthony <[email protected]> Co-authored-by: Anthony Eid <[email protected]>
zed-industries · Feb 24, 2025 · eda2333 · eda2333
1 parent 17a483c
commit eda2333
Show file tree

Hide file tree

Showing 10 changed files with 1,045 additions and 0 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/assets/icons/ai_bedrock.svg b/assets/icons/ai_bedrock.svg
diff --git a/crates/assistant_settings/src/assistant_settings.rs b/crates/assistant_settings/src/assistant_settings.rs
@@ -359,6 +359,7 @@ fn providers_schema(_: &mut schemars::gen::SchemaGenerator) -> schemars::schema:
     schemars::schema::SchemaObject {
         enum_values: Some(vec![
             "anthropic".into(),
+            "bedrock".into(),
             "google".into(),
             "lmstudio".into(),
             "ollama".into(),

diff --git a/crates/language_models/Cargo.toml b/crates/language_models/Cargo.toml
@@ -14,6 +14,10 @@ path = "src/language_models.rs"
 [dependencies]
 anthropic = { workspace = true, features = ["schemars"] }
 anyhow.workspace = true
+aws-config = { workspace = true, features = ["behavior-version-latest"] }
+aws-credential-types = { workspace = true, features = ["hardcoded-credentials"] }
+aws_http_client.workspace = true
+bedrock.workspace = true
 client.workspace = true
 collections.workspace = true
 credentials_provider.workspace = true
@@ -25,6 +29,7 @@ fs.workspace = true
 futures.workspace = true
 google_ai = { workspace = true, features = ["schemars"] }
 gpui.workspace = true
+gpui_tokio.workspace = true
 http_client.workspace = true
 language_model.workspace = true
 lmstudio = { workspace = true, features = ["schemars"] }
@@ -42,6 +47,7 @@ smol.workspace = true
 strum.workspace = true
 theme.workspace = true
 tiktoken-rs.workspace = true
+tokio = { workspace = true, features = ["rt", "rt-multi-thread"] }
 ui.workspace = true
 util.workspace = true
 

diff --git a/crates/language_models/src/language_models.rs b/crates/language_models/src/language_models.rs
@@ -10,6 +10,7 @@ pub mod provider;
 mod settings;
 
 use crate::provider::anthropic::AnthropicLanguageModelProvider;
+use crate::provider::bedrock::BedrockLanguageModelProvider;
 use crate::provider::cloud::CloudLanguageModelProvider;
 use crate::provider::copilot_chat::CopilotChatLanguageModelProvider;
 use crate::provider::google::GoogleLanguageModelProvider;
@@ -63,6 +64,10 @@ fn register_language_model_providers(
         MistralLanguageModelProvider::new(client.http_client(), cx),
         cx,
     );
+    registry.register_provider(
+        BedrockLanguageModelProvider::new(client.http_client(), cx),
+        cx,
+    );
     registry.register_provider(CopilotChatLanguageModelProvider::new(cx), cx);
 
     cx.observe_flag::<feature_flags::LanguageModels, _>(move |enabled, cx| {

diff --git a/crates/language_models/src/provider.rs b/crates/language_models/src/provider.rs
@@ -1,4 +1,5 @@
 pub mod anthropic;
+pub mod bedrock;
 pub mod cloud;
 pub mod copilot_chat;
 pub mod deepseek;