use anyhow::{anyhow, Result};use collections::BTreeMap;use editor::{Editor, EditorElement, EditorStyle};use futures::{future::BoxFuture, FutureExt, StreamExt};use google_ai::stream_generate_content;use gpui::{ AnyView, AppContext, AsyncAppContext, FontStyle, ModelContext, Subscription, Task, TextStyle, View, WhiteSpace,};use http_client::HttpClient;use language_model::LanguageModelCompletionEvent;use language_model::{ LanguageModel, LanguageModelId, LanguageModelName, LanguageModelProvider, LanguageModelProviderId, LanguageModelProviderName, LanguageModelProviderState, LanguageModelRequest, RateLimiter,};use schemars::JsonSchema;use serde::{Deserialize, Serialize};use settings::{Settings, SettingsStore};use std::{future, sync::Arc};use strum::IntoEnumIterator;use theme::ThemeSettings;use ui::{prelude::*, Icon, IconName, Tooltip};use util::ResultExt; use crate::AllLanguageModelSettings; const PROVIDER_ID: &str = "google";const PROVIDER_NAME: &str = "Google AI"; #[derive(Default, Clone, Debug, PartialEq)]pub struct GoogleSettings { pub api_url: String, pub available_models: Vec<AvailableModel>,} #[derive(Clone, Debug, PartialEq, Serialize, Deserialize, JsonSchema)]pub struct AvailableModel { name: String, display_name: Option<String>, max_tokens: usize,} pub struct GoogleLanguageModelProvider { http_client: Arc<dyn HttpClient>, state: gpui::Model<State>,} pub struct State { api_key: Option<String>, api_key_from_env: bool, _subscription: Subscription,} const GOOGLE_AI_API_KEY_VAR: &str = "GOOGLE_AI_API_KEY"; impl State { fn is_authenticated(&self) -> bool { self.api_key.is_some() } fn reset_api_key(&self, cx: &mut ModelContext<Self>) -> Task<Result<()>> { let delete_credentials = cx.delete_credentials(&AllLanguageModelSettings::get_global(cx).google.api_url); cx.spawn(|this, mut cx| async move { delete_credentials.await.ok(); this.update(&mut cx, |this, cx| { this.api_key = None; this.api_key_from_env = false; cx.notify(); }) }) } fn set_api_key(&mut self, api_key: String, cx: &mut ModelContext<Self>) -> Task<Result<()>> { let settings = &AllLanguageModelSettings::get_global(cx).google; let write_credentials = cx.write_credentials(&settings.api_url, "Bearer", api_key.as_bytes()); cx.spawn(|this, mut cx| async move { write_credentials.await?; this.update(&mut cx, |this, cx| { this.api_key = Some(api_key); cx.notify(); }) }) } fn authenticate(&self, cx: &mut ModelContext<Self>) -> Task<Result<()>> { if self.is_authenticated() { Task::ready(Ok(())) } else { let api_url = AllLanguageModelSettings::get_global(cx) .google .api_url .clone(); cx.spawn(|this, mut cx| async move { let (api_key, from_env) = if let Ok(api_key) = std::env::var(GOOGLE_AI_API_KEY_VAR) { (api_key, true) } else { let (_, api_key) = cx .update(|cx| cx.read_credentials(&api_url))? .await? .ok_or_else(|| anyhow!("credentials not found"))?; (String::from_utf8(api_key)?, false) }; this.update(&mut cx, |this, cx| { this.api_key = Some(api_key); this.api_key_from_env = from_env; cx.notify(); }) }) } }} impl GoogleLanguageModelProvider { pub fn new(http_client: Arc<dyn HttpClient>, cx: &mut AppContext) -> Self { let state = cx.new_model(|cx| State { api_key: None, api_key_from_env: false, _subscription: cx.observe_global::<SettingsStore>(|_, cx| { cx.notify(); }), }); Self { http_client, state } }} impl LanguageModelProviderState for GoogleLanguageModelProvider { type ObservableEntity = State; fn observable_entity(&self) -> Option<gpui::Model<Self::ObservableEntity>> { Some(self.state.clone()) }} impl LanguageModelProvider for GoogleLanguageModelProvider { fn id(&self) -> LanguageModelProviderId { LanguageModelProviderId(PROVIDER_ID.into()) } fn name(&self) -> LanguageModelProviderName { LanguageModelProviderName(PROVIDER_NAME.into()) } fn icon(&self) -> IconName { IconName::AiGoogle } fn provided_models(&self, cx: &AppContext) -> Vec<Arc<dyn LanguageModel>> { let mut models = BTreeMap::default(); // Add base models from google_ai::Model::iter() for model in google_ai::Model::iter() { if !matches!(model, google_ai::Model::Custom { .. }) { models.insert(model.id().to_string(), model); } } // Override with available models from settings for model in &AllLanguageModelSettings::get_global(cx) .google .available_models { models.insert( model.name.clone(), google_ai::Model::Custom { name: model.name.clone(), display_name: model.display_name.clone(), max_tokens: model.max_tokens, }, ); } models .into_values() .map(|model| { Arc::new(GoogleLanguageModel { id: LanguageModelId::from(model.id().to_string()), model, state: self.state.clone(), http_client: self.http_client.clone(), rate_limiter: RateLimiter::new(4), }) as Arc<dyn LanguageModel> }) .collect() } fn is_authenticated(&self, cx: &AppContext) -> bool { self.state.read(cx).is_authenticated() } fn authenticate(&self, cx: &mut AppContext) -> Task<Result<()>> { self.state.update(cx, |state, cx| state.authenticate(cx)) } fn configuration_view(&self, cx: &mut WindowContext) -> AnyView { cx.new_view(|cx| ConfigurationView::new(self.state.clone(), cx)) .into() } fn reset_credentials(&self, cx: &mut AppContext) -> Task<Result<()>> { let state = self.state.clone(); let delete_credentials = cx.delete_credentials(&AllLanguageModelSettings::get_global(cx).google.api_url); cx.spawn(|mut cx| async move { delete_credentials.await.log_err(); state.update(&mut cx, |this, cx| { this.api_key = None; cx.notify(); }) }) }} pub struct GoogleLanguageModel { id: LanguageModelId, model: google_ai::Model, state: gpui::Model<State>, http_client: Arc<dyn HttpClient>, rate_limiter: RateLimiter,} impl LanguageModel for GoogleLanguageModel { fn id(&self) -> LanguageModelId { self.id.clone() } fn name(&self) -> LanguageModelName { LanguageModelName::from(self.model.display_name().to_string()) } fn provider_id(&self) -> LanguageModelProviderId { LanguageModelProviderId(PROVIDER_ID.into()) } fn provider_name(&self) -> LanguageModelProviderName { LanguageModelProviderName(PROVIDER_NAME.into()) } fn telemetry_id(&self) -> String { format!("google/{}", self.model.id()) } fn max_token_count(&self) -> usize { self.model.max_token_count() } fn count_tokens( &self, request: LanguageModelRequest, cx: &AppContext, ) -> BoxFuture<'static, Result<usize>> { let request = request.into_google(self.model.id().to_string()); let http_client = self.http_client.clone(); let api_key = self.state.read(cx).api_key.clone(); let settings = &AllLanguageModelSettings::get_global(cx).google; let api_url = settings.api_url.clone(); async move { let api_key = api_key.ok_or_else(|| anyhow!("Missing Google API key"))?; let response = google_ai::count_tokens( http_client.as_ref(), &api_url, &api_key, google_ai::CountTokensRequest { contents: request.contents, }, ) .await?; Ok(response.total_tokens) } .boxed() } fn stream_completion( &self, request: LanguageModelRequest, cx: &AsyncAppContext, ) -> BoxFuture< 'static, Result<futures::stream::BoxStream<'static, Result<LanguageModelCompletionEvent>>>, > { let request = request.into_google(self.model.id().to_string()); let http_client = self.http_client.clone(); let Ok((api_key, api_url)) = cx.read_model(&self.state, |state, cx| { let settings = &AllLanguageModelSettings::get_global(cx).google; (state.api_key.clone(), settings.api_url.clone()) }) else { return futures::future::ready(Err(anyhow!("App state dropped"))).boxed(); }; let future = self.rate_limiter.stream(async move { let api_key = api_key.ok_or_else(|| anyhow!("Missing Google API Key"))?; let response = stream_generate_content(http_client.as_ref(), &api_url, &api_key, request); let events = response.await?; Ok(google_ai::extract_text_from_events(events).boxed()) }); async move { Ok(future .await? .map(|result| result.map(LanguageModelCompletionEvent::Text)) .boxed()) } .boxed() } fn use_any_tool( &self, _request: LanguageModelRequest, _name: String, _description: String, _schema: serde_json::Value, _cx: &AsyncAppContext, ) -> BoxFuture<'static, Result<futures::stream::BoxStream<'static, Result<String>>>> { future::ready(Err(anyhow!("not implemented"))).boxed() }} struct ConfigurationView { api_key_editor: View<Editor>, state: gpui::Model<State>, load_credentials_task: Option<Task<()>>,} impl ConfigurationView { fn new(state: gpui::Model<State>, cx: &mut ViewContext<Self>) -> Self { cx.observe(&state, |_, _, cx| { cx.notify(); }) .detach(); let load_credentials_task = Some(cx.spawn({ let state = state.clone(); |this, mut cx| async move { if let Some(task) = state .update(&mut cx, |state, cx| state.authenticate(cx)) .log_err() { // We don't log an error, because "not signed in" is also an error. let _ = task.await; } this.update(&mut cx, |this, cx| { this.load_credentials_task = None; cx.notify(); }) .log_err(); } })); Self { api_key_editor: cx.new_view(|cx| { let mut editor = Editor::single_line(cx); editor.set_placeholder_text("AIzaSy...", cx); editor }), state, load_credentials_task, } } fn save_api_key(&mut self, _: &menu::Confirm, cx: &mut ViewContext<Self>) { let api_key = self.api_key_editor.read(cx).text(cx); if api_key.is_empty() { return; } let state = self.state.clone(); cx.spawn(|_, mut cx| async move { state .update(&mut cx, |state, cx| state.set_api_key(api_key, cx))? .await }) .detach_and_log_err(cx); cx.notify(); } fn reset_api_key(&mut self, cx: &mut ViewContext<Self>) { self.api_key_editor .update(cx, |editor, cx| editor.set_text("", cx)); let state = self.state.clone(); cx.spawn(|_, mut cx| async move { state .update(&mut cx, |state, cx| state.reset_api_key(cx))? .await }) .detach_and_log_err(cx); cx.notify(); } fn render_api_key_editor(&self, cx: &mut ViewContext<Self>) -> impl IntoElement { let settings = ThemeSettings::get_global(cx); let text_style = TextStyle { color: cx.theme().colors().text, font_family: settings.ui_font.family.clone(), font_features: settings.ui_font.features.clone(), font_fallbacks: settings.ui_font.fallbacks.clone(), font_size: rems(0.875).into(), font_weight: settings.ui_font.weight, font_style: FontStyle::Normal, line_height: relative(1.3), background_color: None, underline: None, strikethrough: None, white_space: WhiteSpace::Normal, truncate: None, }; EditorElement::new( &self.api_key_editor, EditorStyle { background: cx.theme().colors().editor_background, local_player: cx.theme().players().local(), text: text_style, ..Default::default() }, ) } fn should_render_editor(&self, cx: &mut ViewContext<Self>) -> bool { !self.state.read(cx).is_authenticated() }} impl Render for ConfigurationView { fn render(&mut self, cx: &mut ViewContext<Self>) -> impl IntoElement { const GOOGLE_CONSOLE_URL: &str = "https://aistudio.google.com/app/apikey"; const INSTRUCTIONS: [&str; 3] = [ "To use Zed's assistant with Google AI, you need to add an API key. Follow these steps:", "- Create one by visiting:", "- Paste your API key below and hit enter to use the assistant", ]; let env_var_set = self.state.read(cx).api_key_from_env; if self.load_credentials_task.is_some() { div().child(Label::new("Loading credentials...")).into_any() } else if self.should_render_editor(cx) { v_flex() .size_full() .on_action(cx.listener(Self::save_api_key)) .child(Label::new(INSTRUCTIONS[0])) .child(h_flex().child(Label::new(INSTRUCTIONS[1])).child( Button::new("google_console", GOOGLE_CONSOLE_URL) .style(ButtonStyle::Subtle) .icon(IconName::ExternalLink) .icon_size(IconSize::XSmall) .icon_color(Color::Muted) .on_click(move |_, cx| cx.open_url(GOOGLE_CONSOLE_URL)) ) ) .child(Label::new(INSTRUCTIONS[2])) .child( h_flex() .w_full() .my_2() .px_2() .py_1() .bg(cx.theme().colors().editor_background) .rounded_md() .child(self.render_api_key_editor(cx)), ) .child( Label::new( format!("You can also assign the {GOOGLE_AI_API_KEY_VAR} environment variable and restart Zed."), ) .size(LabelSize::Small), ) .into_any() } else { h_flex() .size_full() .justify_between() .child( h_flex() .gap_1() .child(Icon::new(IconName::Check).color(Color::Success)) .child(Label::new(if env_var_set { format!("API key set in {GOOGLE_AI_API_KEY_VAR} environment variable.") } else { "API key configured.".to_string() })), ) .child( Button::new("reset-key", "Reset key") .icon(Some(IconName::Trash)) .icon_size(IconSize::Small) .icon_position(IconPosition::Start) .disabled(env_var_set) .when(env_var_set, |this| {Lines 1–500 of 508. Load more
use anyhow::{anyhow, Result}; use collections::BTreeMap; use editor::{Editor, EditorElement, EditorStyle}; use futures::{future::BoxFuture, FutureExt, StreamExt}; use google_ai::stream_generate_content; use gpui::{ AnyView, AppContext, AsyncAppContext, FontStyle, ModelContext, Subscription, Task, TextStyle, View, WhiteSpace, }; use http_client::HttpClient; use language_model::LanguageModelCompletionEvent; use language_model::{ LanguageModel, LanguageModelId, LanguageModelName, LanguageModelProvider, LanguageModelProviderId, LanguageModelProviderName, LanguageModelProviderState, LanguageModelRequest, RateLimiter, }; use schemars::JsonSchema; use serde::{Deserialize, Serialize}; use settings::{Settings, SettingsStore}; use std::{future, sync::Arc}; use strum::IntoEnumIterator; use theme::ThemeSettings; use ui::{prelude::*, Icon, IconName, Tooltip}; use util::ResultExt; use crate::AllLanguageModelSettings; const PROVIDER_ID: &str = "google"; const PROVIDER_NAME: &str = "Google AI"; #[derive(Default, Clone, Debug, PartialEq)] pub struct GoogleSettings { pub api_url: String, pub available_models: Vec<AvailableModel>, } #[derive(Clone, Debug, PartialEq, Serialize, Deserialize, JsonSchema)] pub struct AvailableModel { name: String, display_name: Option<String>, max_tokens: usize, } pub struct GoogleLanguageModelProvider { http_client: Arc<dyn HttpClient>, state: gpui::Model<State>, } pub struct State { api_key: Option<String>, api_key_from_env: bool, _subscription: Subscription, } const GOOGLE_AI_API_KEY_VAR: &str = "GOOGLE_AI_API_KEY"; impl State { fn is_authenticated(&self) -> bool { self.api_key.is_some() } fn reset_api_key(&self, cx: &mut ModelContext<Self>) -> Task<Result<()>> { let delete_credentials = cx.delete_credentials(&AllLanguageModelSettings::get_global(cx).google.api_url); cx.spawn(|this, mut cx| async move { delete_credentials.await.ok(); this.update(&mut cx, |this, cx| { this.api_key = None; this.api_key_from_env = false; cx.notify(); }) }) } fn set_api_key(&mut self, api_key: String, cx: &mut ModelContext<Self>) -> Task<Result<()>> { let settings = &AllLanguageModelSettings::get_global(cx).google; let write_credentials = cx.write_credentials(&settings.api_url, "Bearer", api_key.as_bytes()); cx.spawn(|this, mut cx| async move { write_credentials.await?; this.update(&mut cx, |this, cx| { this.api_key = Some(api_key); cx.notify(); }) }) } fn authenticate(&self, cx: &mut ModelContext<Self>) -> Task<Result<()>> { if self.is_authenticated() { Task::ready(Ok(())) } else { let api_url = AllLanguageModelSettings::get_global(cx) .google .api_url .clone(); cx.spawn(|this, mut cx| async move { let (api_key, from_env) = if let Ok(api_key) = std::env::var(GOOGLE_AI_API_KEY_VAR) { (api_key, true) } else { let (_, api_key) = cx .update(|cx| cx.read_credentials(&api_url))? .await? .ok_or_else(|| anyhow!("credentials not found"))?; (String::from_utf8(api_key)?, false) }; this.update(&mut cx, |this, cx| { this.api_key = Some(api_key); this.api_key_from_env = from_env; cx.notify(); }) }) } } } impl GoogleLanguageModelProvider { pub fn new(http_client: Arc<dyn HttpClient>, cx: &mut AppContext) -> Self { let state = cx.new_model(|cx| State { api_key: None, api_key_from_env: false, _subscription: cx.observe_global::<SettingsStore>(|_, cx| { cx.notify(); }), }); Self { http_client, state } } } impl LanguageModelProviderState for GoogleLanguageModelProvider { type ObservableEntity = State; fn observable_entity(&self) -> Option<gpui::Model<Self::ObservableEntity>> { Some(self.state.clone()) } } impl LanguageModelProvider for GoogleLanguageModelProvider { fn id(&self) -> LanguageModelProviderId { LanguageModelProviderId(PROVIDER_ID.into()) } fn name(&self) -> LanguageModelProviderName { LanguageModelProviderName(PROVIDER_NAME.into()) } fn icon(&self) -> IconName { IconName::AiGoogle } fn provided_models(&self, cx: &AppContext) -> Vec<Arc<dyn LanguageModel>> { let mut models = BTreeMap::default(); // Add base models from google_ai::Model::iter() for model in google_ai::Model::iter() { if !matches!(model, google_ai::Model::Custom { .. }) { models.insert(model.id().to_string(), model); } } // Override with available models from settings for model in &AllLanguageModelSettings::get_global(cx) .google .available_models { models.insert( model.name.clone(), google_ai::Model::Custom { name: model.name.clone(), display_name: model.display_name.clone(), max_tokens: model.max_tokens, }, ); } models .into_values() .map(|model| { Arc::new(GoogleLanguageModel { id: LanguageModelId::from(model.id().to_string()), model, state: self.state.clone(), http_client: self.http_client.clone(), rate_limiter: RateLimiter::new(4), }) as Arc<dyn LanguageModel> }) .collect() } fn is_authenticated(&self, cx: &AppContext) -> bool { self.state.read(cx).is_authenticated() } fn authenticate(&self, cx: &mut AppContext) -> Task<Result<()>> { self.state.update(cx, |state, cx| state.authenticate(cx)) } fn configuration_view(&self, cx: &mut WindowContext) -> AnyView { cx.new_view(|cx| ConfigurationView::new(self.state.clone(), cx)) .into() } fn reset_credentials(&self, cx: &mut AppContext) -> Task<Result<()>> { let state = self.state.clone(); let delete_credentials = cx.delete_credentials(&AllLanguageModelSettings::get_global(cx).google.api_url); cx.spawn(|mut cx| async move { delete_credentials.await.log_err(); state.update(&mut cx, |this, cx| { this.api_key = None; cx.notify(); }) }) } } pub struct GoogleLanguageModel { id: LanguageModelId, model: google_ai::Model, state: gpui::Model<State>, http_client: Arc<dyn HttpClient>, rate_limiter: RateLimiter, } impl LanguageModel for GoogleLanguageModel { fn id(&self) -> LanguageModelId { self.id.clone() } fn name(&self) -> LanguageModelName { LanguageModelName::from(self.model.display_name().to_string()) } fn provider_id(&self) -> LanguageModelProviderId { LanguageModelProviderId(PROVIDER_ID.into()) } fn provider_name(&self) -> LanguageModelProviderName { LanguageModelProviderName(PROVIDER_NAME.into()) } fn telemetry_id(&self) -> String { format!("google/{}", self.model.id()) } fn max_token_count(&self) -> usize { self.model.max_token_count() } fn count_tokens( &self, request: LanguageModelRequest, cx: &AppContext, ) -> BoxFuture<'static, Result<usize>> { let request = request.into_google(self.model.id().to_string()); let http_client = self.http_client.clone(); let api_key = self.state.read(cx).api_key.clone(); let settings = &AllLanguageModelSettings::get_global(cx).google; let api_url = settings.api_url.clone(); async move { let api_key = api_key.ok_or_else(|| anyhow!("Missing Google API key"))?; let response = google_ai::count_tokens( http_client.as_ref(), &api_url, &api_key, google_ai::CountTokensRequest { contents: request.contents, }, ) .await?; Ok(response.total_tokens) } .boxed() } fn stream_completion( &self, request: LanguageModelRequest, cx: &AsyncAppContext, ) -> BoxFuture< 'static, Result<futures::stream::BoxStream<'static, Result<LanguageModelCompletionEvent>>>, > { let request = request.into_google(self.model.id().to_string()); let http_client = self.http_client.clone(); let Ok((api_key, api_url)) = cx.read_model(&self.state, |state, cx| { let settings = &AllLanguageModelSettings::get_global(cx).google; (state.api_key.clone(), settings.api_url.clone()) }) else { return futures::future::ready(Err(anyhow!("App state dropped"))).boxed(); }; let future = self.rate_limiter.stream(async move { let api_key = api_key.ok_or_else(|| anyhow!("Missing Google API Key"))?; let response = stream_generate_content(http_client.as_ref(), &api_url, &api_key, request); let events = response.await?; Ok(google_ai::extract_text_from_events(events).boxed()) }); async move { Ok(future .await? .map(|result| result.map(LanguageModelCompletionEvent::Text)) .boxed()) } .boxed() } fn use_any_tool( &self, _request: LanguageModelRequest, _name: String, _description: String, _schema: serde_json::Value, _cx: &AsyncAppContext, ) -> BoxFuture<'static, Result<futures::stream::BoxStream<'static, Result<String>>>> { future::ready(Err(anyhow!("not implemented"))).boxed() } } struct ConfigurationView { api_key_editor: View<Editor>, state: gpui::Model<State>, load_credentials_task: Option<Task<()>>, } impl ConfigurationView { fn new(state: gpui::Model<State>, cx: &mut ViewContext<Self>) -> Self { cx.observe(&state, |_, _, cx| { cx.notify(); }) .detach(); let load_credentials_task = Some(cx.spawn({ let state = state.clone(); |this, mut cx| async move { if let Some(task) = state .update(&mut cx, |state, cx| state.authenticate(cx)) .log_err() { // We don't log an error, because "not signed in" is also an error. let _ = task.await; } this.update(&mut cx, |this, cx| { this.load_credentials_task = None; cx.notify(); }) .log_err(); } })); Self { api_key_editor: cx.new_view(|cx| { let mut editor = Editor::single_line(cx); editor.set_placeholder_text("AIzaSy...", cx); editor }), state, load_credentials_task, } } fn save_api_key(&mut self, _: &menu::Confirm, cx: &mut ViewContext<Self>) { let api_key = self.api_key_editor.read(cx).text(cx); if api_key.is_empty() { return; } let state = self.state.clone(); cx.spawn(|_, mut cx| async move { state .update(&mut cx, |state, cx| state.set_api_key(api_key, cx))? .await }) .detach_and_log_err(cx); cx.notify(); } fn reset_api_key(&mut self, cx: &mut ViewContext<Self>) { self.api_key_editor .update(cx, |editor, cx| editor.set_text("", cx)); let state = self.state.clone(); cx.spawn(|_, mut cx| async move { state .update(&mut cx, |state, cx| state.reset_api_key(cx))? .await }) .detach_and_log_err(cx); cx.notify(); } fn render_api_key_editor(&self, cx: &mut ViewContext<Self>) -> impl IntoElement { let settings = ThemeSettings::get_global(cx); let text_style = TextStyle { color: cx.theme().colors().text, font_family: settings.ui_font.family.clone(), font_features: settings.ui_font.features.clone(), font_fallbacks: settings.ui_font.fallbacks.clone(), font_size: rems(0.875).into(), font_weight: settings.ui_font.weight, font_style: FontStyle::Normal, line_height: relative(1.3), background_color: None, underline: None, strikethrough: None, white_space: WhiteSpace::Normal, truncate: None, }; EditorElement::new( &self.api_key_editor, EditorStyle { background: cx.theme().colors().editor_background, local_player: cx.theme().players().local(), text: text_style, ..Default::default() }, ) } fn should_render_editor(&self, cx: &mut ViewContext<Self>) -> bool { !self.state.read(cx).is_authenticated() } } impl Render for ConfigurationView { fn render(&mut self, cx: &mut ViewContext<Self>) -> impl IntoElement { const GOOGLE_CONSOLE_URL: &str = "https://aistudio.google.com/app/apikey"; const INSTRUCTIONS: [&str; 3] = [ "To use Zed's assistant with Google AI, you need to add an API key. Follow these steps:", "- Create one by visiting:", "- Paste your API key below and hit enter to use the assistant", ]; let env_var_set = self.state.read(cx).api_key_from_env; if self.load_credentials_task.is_some() { div().child(Label::new("Loading credentials...")).into_any() } else if self.should_render_editor(cx) { v_flex() .size_full() .on_action(cx.listener(Self::save_api_key)) .child(Label::new(INSTRUCTIONS[0])) .child(h_flex().child(Label::new(INSTRUCTIONS[1])).child( Button::new("google_console", GOOGLE_CONSOLE_URL) .style(ButtonStyle::Subtle) .icon(IconName::ExternalLink) .icon_size(IconSize::XSmall) .icon_color(Color::Muted) .on_click(move |_, cx| cx.open_url(GOOGLE_CONSOLE_URL)) ) ) .child(Label::new(INSTRUCTIONS[2])) .child( h_flex() .w_full() .my_2() .px_2() .py_1() .bg(cx.theme().colors().editor_background) .rounded_md() .child(self.render_api_key_editor(cx)), ) .child( Label::new( format!("You can also assign the {GOOGLE_AI_API_KEY_VAR} environment variable and restart Zed."), ) .size(LabelSize::Small), ) .into_any() } else { h_flex() .size_full() .justify_between() .child( h_flex() .gap_1() .child(Icon::new(IconName::Check).color(Color::Success)) .child(Label::new(if env_var_set { format!("API key set in {GOOGLE_AI_API_KEY_VAR} environment variable.") } else { "API key configured.".to_string() })), ) .child( Button::new("reset-key", "Reset key") .icon(Some(IconName::Trash)) .icon_size(IconSize::Small) .icon_position(IconPosition::Start) .disabled(env_var_set) .when(env_var_set, |this| {