use acp_thread::{AgentConnection, Plan}; use agent_servers::AgentServer; use agent_settings::{AgentSettings, NotifyWhenAgentWaiting}; use audio::{Audio, Sound}; use std::cell::RefCell; use std::collections::BTreeMap; use std::path::Path; use std::process::ExitStatus; use std::rc::Rc; use std::sync::Arc; use std::time::Duration; use agent_client_protocol as acp; use assistant_tool::ActionLog; use buffer_diff::BufferDiff; use collections::{HashMap, HashSet}; use editor::{ AnchorRangeExt, ContextMenuOptions, ContextMenuPlacement, Editor, EditorElement, EditorMode, EditorStyle, MinimapVisibility, MultiBuffer, PathKey, }; use file_icons::FileIcons; use gpui::{ Action, Animation, AnimationExt, App, BorderStyle, EdgesRefinement, Empty, Entity, EntityId, FocusHandle, Focusable, Hsla, Length, ListOffset, ListState, PlatformDisplay, SharedString, StyleRefinement, Subscription, Task, TextStyle, TextStyleRefinement, Transformation, UnderlineStyle, WeakEntity, Window, WindowHandle, div, linear_color_stop, linear_gradient, list, percentage, point, prelude::*, pulsating_between, }; use language::language_settings::SoftWrap; use language::{Buffer, Language}; use markdown::{HeadingLevelStyles, Markdown, MarkdownElement, MarkdownStyle}; use parking_lot::Mutex; use project::Project; use settings::Settings as _; use text::{Anchor, BufferSnapshot}; use theme::ThemeSettings; use ui::{Disclosure, Divider, DividerColor, KeyBinding, Tooltip, prelude::*}; use util::ResultExt; use workspace::{CollaboratorId, Workspace}; use zed_actions::agent::{Chat, NextHistoryMessage, PreviousHistoryMessage}; use ::acp_thread::{ AcpThread, AcpThreadEvent, AgentThreadEntry, AssistantMessage, AssistantMessageChunk, Diff, LoadError, MentionPath, ThreadStatus, ToolCall, ToolCallContent, ToolCallStatus, }; use crate::acp::completion_provider::{ContextPickerCompletionProvider, MentionSet}; use crate::acp::message_history::MessageHistory; use crate::agent_diff::AgentDiff; use crate::message_editor::{MAX_EDITOR_LINES, MIN_EDITOR_LINES}; use crate::ui::{AgentNotification, AgentNotificationEvent}; use crate::{ AgentDiffPane, AgentPanel, ExpandMessageEditor, Follow, KeepAll, OpenAgentDiff, RejectAll, }; const RESPONSE_PADDING_X: Pixels = px(19.); pub struct AcpThreadView { agent: Rc, workspace: WeakEntity, project: Entity, thread_state: ThreadState, diff_editors: HashMap>, message_editor: Entity, message_set_from_history: Option, _message_editor_subscription: Subscription, mention_set: Arc>, notifications: Vec>, notification_subscriptions: HashMap, Vec>, last_error: Option>, list_state: ListState, auth_task: Option>, expanded_tool_calls: HashSet, expanded_thinking_blocks: HashSet<(usize, usize)>, edits_expanded: bool, plan_expanded: bool, editor_expanded: bool, message_history: Rc>>>, _cancel_task: Option>, } enum ThreadState { Loading { _task: Task<()>, }, Ready { thread: Entity, _subscription: [Subscription; 2], }, LoadError(LoadError), Unauthenticated { connection: Rc, }, ServerExited { status: ExitStatus, }, } impl AcpThreadView { pub fn new( agent: Rc, workspace: WeakEntity, project: Entity, message_history: Rc>>>, min_lines: usize, max_lines: Option, window: &mut Window, cx: &mut Context, ) -> Self { let language = Language::new( language::LanguageConfig { completion_query_characters: HashSet::from_iter(['.', '-', '_', '@']), ..Default::default() }, None, ); let mention_set = Arc::new(Mutex::new(MentionSet::default())); let message_editor = cx.new(|cx| { let buffer = cx.new(|cx| Buffer::local("", cx).with_language(Arc::new(language), cx)); let buffer = cx.new(|cx| MultiBuffer::singleton(buffer, cx)); let mut editor = Editor::new( editor::EditorMode::AutoHeight { min_lines, max_lines: max_lines, }, buffer, None, window, cx, ); editor.set_placeholder_text("Message the agent － @ to include files", cx); editor.set_show_indent_guides(false, cx); editor.set_soft_wrap(); editor.set_use_modal_editing(true); editor.set_completion_provider(Some(Rc::new(ContextPickerCompletionProvider::new( mention_set.clone(), workspace.clone(), cx.weak_entity(), )))); editor.set_context_menu_options(ContextMenuOptions { min_entries_visible: 12, max_entries_visible: 12, placement: Some(ContextMenuPlacement::Above), }); editor }); let message_editor_subscription = cx.subscribe(&message_editor, |this, editor, event, cx| { if let editor::EditorEvent::BufferEdited = &event { let buffer = editor .read(cx) .buffer() .read(cx) .as_singleton() .unwrap() .read(cx) .snapshot(); if let Some(message) = this.message_set_from_history.clone() && message.version() != buffer.version() { this.message_set_from_history = None; } if this.message_set_from_history.is_none() { this.message_history.borrow_mut().reset_position(); } } }); let mention_set = mention_set.clone(); let list_state = ListState::new(0, gpui::ListAlignment::Bottom, px(2048.0)); Self { agent: agent.clone(), workspace: workspace.clone(), project: project.clone(), thread_state: Self::initial_state(agent, workspace, project, window, cx), message_editor, message_set_from_history: None, _message_editor_subscription: message_editor_subscription, mention_set, notifications: Vec::new(), notification_subscriptions: HashMap::default(), diff_editors: Default::default(), list_state: list_state, last_error: None, auth_task: None, expanded_tool_calls: HashSet::default(), expanded_thinking_blocks: HashSet::default(), edits_expanded: false, plan_expanded: false, editor_expanded: false, message_history, _cancel_task: None, } } fn initial_state( agent: Rc, workspace: WeakEntity, project: Entity, window: &mut Window, cx: &mut Context, ) -> ThreadState { let root_dir = project .read(cx) .visible_worktrees(cx) .next() .map(|worktree| worktree.read(cx).abs_path()) .unwrap_or_else(|| paths::home_dir().as_path().into()); let connect_task = agent.connect(&root_dir, &project, cx); let load_task = cx.spawn_in(window, async move |this, cx| { let connection = match connect_task.await { Ok(connection) => connection, Err(err) => { this.update(cx, |this, cx| { this.handle_load_error(err, cx); cx.notify(); }) .log_err(); return; } }; // this.update_in(cx, |_this, _window, cx| { // let status = connection.exit_status(cx); // cx.spawn(async move |this, cx| { // let status = status.await.ok(); // this.update(cx, |this, cx| { // this.thread_state = ThreadState::ServerExited { status }; // cx.notify(); // }) // .ok(); // }) // .detach(); // }) // .ok(); let result = match connection .clone() .new_thread(project.clone(), &root_dir, cx) .await { Err(e) => { let mut cx = cx.clone(); if e.is::() { this.update(&mut cx, |this, cx| { this.thread_state = ThreadState::Unauthenticated { connection }; cx.notify(); }) .ok(); return; } else { Err(e) } } Ok(session_id) => Ok(session_id), }; this.update_in(cx, |this, window, cx| { match result { Ok(thread) => { let thread_subscription = cx.subscribe_in(&thread, window, Self::handle_thread_event); let action_log = thread.read(cx).action_log().clone(); let action_log_subscription = cx.observe(&action_log, |_, _, cx| cx.notify()); this.list_state .splice(0..0, thread.read(cx).entries().len()); AgentDiff::set_active_thread(&workspace, thread.clone(), window, cx); this.thread_state = ThreadState::Ready { thread, _subscription: [thread_subscription, action_log_subscription], }; cx.notify(); } Err(err) => { this.handle_load_error(err, cx); } }; }) .log_err(); }); ThreadState::Loading { _task: load_task } } fn handle_load_error(&mut self, err: anyhow::Error, cx: &mut Context) { if let Some(load_err) = err.downcast_ref::() { self.thread_state = ThreadState::LoadError(load_err.clone()); } else { self.thread_state = ThreadState::LoadError(LoadError::Other(err.to_string().into())) } cx.notify(); } pub fn thread(&self) -> Option<&Entity> { match &self.thread_state { ThreadState::Ready { thread, .. } => Some(thread), ThreadState::Unauthenticated { .. } | ThreadState::Loading { .. } | ThreadState::LoadError(..) | ThreadState::ServerExited { .. } => None, } } pub fn title(&self, cx: &App) -> SharedString { match &self.thread_state { ThreadState::Ready { thread, .. } => thread.read(cx).title(), ThreadState::Loading { .. } => "Loading…".into(), ThreadState::LoadError(_) => "Failed to load".into(), ThreadState::Unauthenticated { .. } => "Not authenticated".into(), ThreadState::ServerExited { .. } => "Server exited unexpectedly".into(), } } pub fn cancel(&mut self, cx: &mut Context) { self.last_error.take(); if let Some(thread) = self.thread() { self._cancel_task = Some(thread.update(cx, |thread, cx| thread.cancel(cx))); } } pub fn expand_message_editor( &mut self, _: &ExpandMessageEditor, _window: &mut Window, cx: &mut Context, ) { self.set_editor_is_expanded(!self.editor_expanded, cx); cx.notify(); } fn set_editor_is_expanded(&mut self, is_expanded: bool, cx: &mut Context) { self.editor_expanded = is_expanded; self.message_editor.update(cx, |editor, _| { if self.editor_expanded { editor.set_mode(EditorMode::Full { scale_ui_elements_with_buffer_font_size: false, show_active_line_background: false, sized_by_content: false, }) } else { editor.set_mode(EditorMode::AutoHeight { min_lines: MIN_EDITOR_LINES, max_lines: Some(MAX_EDITOR_LINES), }) } }); cx.notify(); } fn chat(&mut self, _: &Chat, window: &mut Window, cx: &mut Context) { self.last_error.take(); let mut ix = 0; let mut chunks: Vec = Vec::new(); let project = self.project.clone(); self.message_editor.update(cx, |editor, cx| { let text = editor.text(cx); editor.display_map.update(cx, |map, cx| { let snapshot = map.snapshot(cx); for (crease_id, crease) in snapshot.crease_snapshot.creases() { if let Some(project_path) = self.mention_set.lock().path_for_crease_id(crease_id) { let crease_range = crease.range().to_offset(&snapshot.buffer_snapshot); if crease_range.start > ix { chunks.push(text[ix..crease_range.start].into()); } if let Some(abs_path) = project.read(cx).absolute_path(&project_path, cx) { let path_str = abs_path.display().to_string(); chunks.push(acp::ContentBlock::ResourceLink(acp::ResourceLink { uri: path_str.clone(), name: path_str, annotations: None, description: None, mime_type: None, size: None, title: None, })); } ix = crease_range.end; } } if ix < text.len() { let last_chunk = text[ix..].trim(); if !last_chunk.is_empty() { chunks.push(last_chunk.into()); } } }) }); if chunks.is_empty() { return; } let Some(thread) = self.thread() else { return; }; let task = thread.update(cx, |thread, cx| thread.send(chunks.clone(), cx)); cx.spawn(async move |this, cx| { let result = task.await; this.update(cx, |this, cx| { if let Err(err) = result { this.last_error = Some(cx.new(|cx| Markdown::new(err.to_string().into(), None, None, cx))) } }) }) .detach(); let mention_set = self.mention_set.clone(); self.set_editor_is_expanded(false, cx); self.message_editor.update(cx, |editor, cx| { editor.clear(window, cx); editor.remove_creases(mention_set.lock().drain(), cx) }); self.scroll_to_bottom(cx); self.message_history.borrow_mut().push(chunks); } fn previous_history_message( &mut self, _: &PreviousHistoryMessage, window: &mut Window, cx: &mut Context, ) { if self.message_set_from_history.is_none() && !self.message_editor.read(cx).is_empty(cx) { self.message_editor.update(cx, |editor, cx| { editor.move_up(&Default::default(), window, cx); }); return; } self.message_set_from_history = Self::set_draft_message( self.message_editor.clone(), self.mention_set.clone(), self.project.clone(), self.message_history .borrow_mut() .prev() .map(|blocks| blocks.as_slice()), window, cx, ); } fn next_history_message( &mut self, _: &NextHistoryMessage, window: &mut Window, cx: &mut Context, ) { if self.message_set_from_history.is_none() { self.message_editor.update(cx, |editor, cx| { editor.move_down(&Default::default(), window, cx); }); return; } let mut message_history = self.message_history.borrow_mut(); let next_history = message_history.next(); let set_draft_message = Self::set_draft_message( self.message_editor.clone(), self.mention_set.clone(), self.project.clone(), Some( next_history .map(|blocks| blocks.as_slice()) .unwrap_or_else(|| &[]), ), window, cx, ); // If we reset the text to an empty string because we ran out of history, // we don't want to mark it as coming from the history self.message_set_from_history = if next_history.is_some() { set_draft_message } else { None }; } fn open_agent_diff(&mut self, _: &OpenAgentDiff, window: &mut Window, cx: &mut Context) { if let Some(thread) = self.thread() { AgentDiffPane::deploy(thread.clone(), self.workspace.clone(), window, cx).log_err(); } } fn open_edited_buffer( &mut self, buffer: &Entity, window: &mut Window, cx: &mut Context, ) { let Some(thread) = self.thread() else { return; }; let Some(diff) = AgentDiffPane::deploy(thread.clone(), self.workspace.clone(), window, cx).log_err() else { return; }; diff.update(cx, |diff, cx| { diff.move_to_path(PathKey::for_buffer(&buffer, cx), window, cx) }) } fn set_draft_message( message_editor: Entity, mention_set: Arc>, project: Entity, message: Option<&[acp::ContentBlock]>, window: &mut Window, cx: &mut Context, ) -> Option { cx.notify(); let message = message?; let mut text = String::new(); let mut mentions = Vec::new(); for chunk in message { match chunk { acp::ContentBlock::Text(text_content) => { text.push_str(&text_content.text); } acp::ContentBlock::ResourceLink(resource_link) => { let path = Path::new(&resource_link.uri); let start = text.len(); let content = MentionPath::new(&path).to_string(); text.push_str(&content); let end = text.len(); if let Some(project_path) = project.read(cx).project_path_for_absolute_path(&path, cx) { let filename: SharedString = path .file_name() .unwrap_or_default() .to_string_lossy() .to_string() .into(); mentions.push((start..end, project_path, filename)); } } acp::ContentBlock::Image(_) | acp::ContentBlock::Audio(_) | acp::ContentBlock::Resource(_) => {} } } let snapshot = message_editor.update(cx, |editor, cx| { editor.set_text(text, window, cx); editor.buffer().read(cx).snapshot(cx) }); for (range, project_path, filename) in mentions { let crease_icon_path = if project_path.path.is_dir() { FileIcons::get_folder_icon(false, cx) .unwrap_or_else(|| IconName::Folder.path().into()) } else { FileIcons::get_icon(Path::new(project_path.path.as_ref()), cx) .unwrap_or_else(|| IconName::File.path().into()) }; let anchor = snapshot.anchor_before(range.start); let crease_id = crate::context_picker::insert_crease_for_mention( anchor.excerpt_id, anchor.text_anchor, range.end - range.start, filename, crease_icon_path, message_editor.clone(), window, cx, ); if let Some(crease_id) = crease_id { mention_set.lock().insert(crease_id, project_path); } } let snapshot = snapshot.as_singleton().unwrap().2.clone(); Some(snapshot.text) } fn handle_thread_event( &mut self, thread: &Entity, event: &AcpThreadEvent, window: &mut Window, cx: &mut Context, ) { let count = self.list_state.item_count(); match event { AcpThreadEvent::NewEntry => { let index = thread.read(cx).entries().len() - 1; self.sync_thread_entry_view(index, window, cx); self.list_state.splice(count..count, 1); } AcpThreadEvent::EntryUpdated(index) => { let index = *index; self.sync_thread_entry_view(index, window, cx); self.list_state.splice(index..index + 1, 1); } AcpThreadEvent::ToolAuthorizationRequired => { self.notify_with_sound("Waiting for tool confirmation", IconName::Info, window, cx); } AcpThreadEvent::Stopped => { let used_tools = thread.read(cx).used_tools_since_last_user_message(); self.notify_with_sound( if used_tools { "Finished running tools" } else { "New message" }, IconName::ZedAssistant, window, cx, ); } AcpThreadEvent::Error => { self.notify_with_sound( "Agent stopped due to an error", IconName::Warning, window, cx, ); } AcpThreadEvent::ServerExited(status) => { self.thread_state = ThreadState::ServerExited { status: *status }; } } cx.notify(); } fn sync_thread_entry_view( &mut self, entry_ix: usize, window: &mut Window, cx: &mut Context, ) { let Some(multibuffers) = self.entry_diff_multibuffers(entry_ix, cx) else { return; }; let multibuffers = multibuffers.collect::>(); for multibuffer in multibuffers { if self.diff_editors.contains_key(&multibuffer.entity_id()) { return; } let editor = cx.new(|cx| { let mut editor = Editor::new( EditorMode::Full { scale_ui_elements_with_buffer_font_size: false, show_active_line_background: false, sized_by_content: true, }, multibuffer.clone(), None, window, cx, ); editor.set_show_gutter(false, cx); editor.disable_inline_diagnostics(); editor.disable_expand_excerpt_buttons(cx); editor.set_show_vertical_scrollbar(false, cx); editor.set_minimap_visibility(MinimapVisibility::Disabled, window, cx); editor.set_soft_wrap_mode(SoftWrap::None, cx); editor.scroll_manager.set_forbid_vertical_scroll(true); editor.set_show_indent_guides(false, cx); editor.set_read_only(true); editor.set_show_breakpoints(false, cx); editor.set_show_code_actions(false, cx); editor.set_show_git_diff_gutter(false, cx); editor.set_expand_all_diff_hunks(cx); editor.set_text_style_refinement(TextStyleRefinement { font_size: Some( TextSize::Small .rems(cx) .to_pixels(ThemeSettings::get_global(cx).agent_font_size(cx)) .into(), ), ..Default::default() }); editor }); let entity_id = multibuffer.entity_id(); cx.observe_release(&multibuffer, move |this, _, _| { this.diff_editors.remove(&entity_id); }) .detach(); self.diff_editors.insert(entity_id, editor); } } fn entry_diff_multibuffers( &self, entry_ix: usize, cx: &App, ) -> Option>> { let entry = self.thread()?.read(cx).entries().get(entry_ix)?; Some(entry.diffs().map(|diff| diff.multibuffer.clone())) } fn authenticate( &mut self, method: acp::AuthMethodId, window: &mut Window, cx: &mut Context, ) { let ThreadState::Unauthenticated { ref connection } = self.thread_state else { return; }; self.last_error.take(); let authenticate = connection.authenticate(method, cx); self.auth_task = Some(cx.spawn_in(window, { let project = self.project.clone(); let agent = self.agent.clone(); async move |this, cx| { let result = authenticate.await; this.update_in(cx, |this, window, cx| { if let Err(err) = result { this.last_error = Some(cx.new(|cx| { Markdown::new(format!("Error: {err}").into(), None, None, cx) })) } else { this.thread_state = Self::initial_state( agent, this.workspace.clone(), project.clone(), window, cx, ) } this.auth_task.take() }) .ok(); } })); } fn authorize_tool_call( &mut self, tool_call_id: acp::ToolCallId, option_id: acp::PermissionOptionId, option_kind: acp::PermissionOptionKind, cx: &mut Context, ) { let Some(thread) = self.thread() else { return; }; thread.update(cx, |thread, cx| { thread.authorize_tool_call(tool_call_id, option_id, option_kind, cx); }); cx.notify(); } fn render_entry( &self, index: usize, total_entries: usize, entry: &AgentThreadEntry, window: &mut Window, cx: &Context, ) -> AnyElement { let primary = match &entry { AgentThreadEntry::UserMessage(message) => div() .py_4() .px_2() .child( v_flex() .p_3() .gap_1p5() .rounded_lg() .shadow_md() .bg(cx.theme().colors().editor_background) .border_1() .border_color(cx.theme().colors().border) .text_xs() .children(message.content.markdown().map(|md| { self.render_markdown( md.clone(), user_message_markdown_style(window, cx), ) })), ) .into_any(), AgentThreadEntry::AssistantMessage(AssistantMessage { chunks }) => { let style = default_markdown_style(false, window, cx); let message_body = v_flex() .w_full() .gap_2p5() .children(chunks.iter().enumerate().filter_map( |(chunk_ix, chunk)| match chunk { AssistantMessageChunk::Message { block } => { block.markdown().map(|md| { self.render_markdown(md.clone(), style.clone()) .into_any_element() }) } AssistantMessageChunk::Thought { block } => { block.markdown().map(|md| { self.render_thinking_block( index, chunk_ix, md.clone(), window, cx, ) .into_any_element() }) } }, )) .into_any(); v_flex() .px_5() .py_1() .when(index + 1 == total_entries, |this| this.pb_4()) .w_full() .text_ui(cx) .child(message_body) .into_any() } AgentThreadEntry::ToolCall(tool_call) => div() .py_1p5() .px_5() .child(self.render_tool_call(index, tool_call, window, cx)) .into_any(), }; let Some(thread) = self.thread() else { return primary; }; let is_generating = matches!(thread.read(cx).status(), ThreadStatus::Generating); if index == total_entries - 1 && !is_generating { v_flex() .child(primary) .child(self.render_thread_controls(cx)) .into_any_element() } else { primary } } fn tool_card_header_bg(&self, cx: &Context) -> Hsla { cx.theme() .colors() .element_background .blend(cx.theme().colors().editor_foreground.opacity(0.025)) } fn tool_card_border_color(&self, cx: &Context) -> Hsla { cx.theme().colors().border.opacity(0.6) } fn tool_name_font_size(&self) -> Rems { rems_from_px(13.) } fn render_thinking_block( &self, entry_ix: usize, chunk_ix: usize, chunk: Entity, window: &Window, cx: &Context, ) -> AnyElement { let header_id = SharedString::from(format!("thinking-block-header-{}", entry_ix)); let key = (entry_ix, chunk_ix); let is_open = self.expanded_thinking_blocks.contains(&key); v_flex() .child( h_flex() .id(header_id) .group("disclosure-header") .w_full() .justify_between() .opacity(0.8) .hover(|style| style.opacity(1.)) .child( h_flex() .gap_1p5() .child( Icon::new(IconName::ToolBulb) .size(IconSize::Small) .color(Color::Muted), ) .child( div() .text_size(self.tool_name_font_size()) .child("Thinking"), ), ) .child( div().visible_on_hover("disclosure-header").child( Disclosure::new("thinking-disclosure", is_open) .opened_icon(IconName::ChevronUp) .closed_icon(IconName::ChevronDown) .on_click(cx.listener({ move |this, _event, _window, cx| { if is_open { this.expanded_thinking_blocks.remove(&key); } else { this.expanded_thinking_blocks.insert(key); } cx.notify(); } })), ), ) .on_click(cx.listener({ move |this, _event, _window, cx| { if is_open { this.expanded_thinking_blocks.remove(&key); } else { this.expanded_thinking_blocks.insert(key); } cx.notify(); } })), ) .when(is_open, |this| { this.child( div() .relative() .mt_1p5() .ml(px(7.)) .pl_4() .border_l_1() .border_color(self.tool_card_border_color(cx)) .text_ui_sm(cx) .child( self.render_markdown(chunk, default_markdown_style(false, window, cx)), ), ) }) .into_any_element() } fn render_tool_call( &self, entry_ix: usize, tool_call: &ToolCall, window: &Window, cx: &Context, ) -> Div { let header_id = SharedString::from(format!("tool-call-header-{}", entry_ix)); let status_icon = match &tool_call.status { ToolCallStatus::Allowed { status: acp::ToolCallStatus::Pending, } | ToolCallStatus::WaitingForConfirmation { .. } => None, ToolCallStatus::Allowed { status: acp::ToolCallStatus::InProgress, .. } => Some( Icon::new(IconName::ArrowCircle) .color(Color::Accent) .size(IconSize::Small) .with_animation( "running", Animation::new(Duration::from_secs(2)).repeat(), |icon, delta| icon.transform(Transformation::rotate(percentage(delta))), ) .into_any(), ), ToolCallStatus::Allowed { status: acp::ToolCallStatus::Completed, .. } => None, ToolCallStatus::Rejected | ToolCallStatus::Canceled | ToolCallStatus::Allowed { status: acp::ToolCallStatus::Failed, .. } => Some( Icon::new(IconName::X) .color(Color::Error) .size(IconSize::Small) .into_any_element(), ), }; let needs_confirmation = match &tool_call.status { ToolCallStatus::WaitingForConfirmation { .. } => true, _ => tool_call .content .iter() .any(|content| matches!(content, ToolCallContent::Diff { .. })), }; let is_collapsible = !tool_call.content.is_empty() && !needs_confirmation; let is_open = !is_collapsible || self.expanded_tool_calls.contains(&tool_call.id); v_flex() .when(needs_confirmation, |this| { this.rounded_lg() .border_1() .border_color(self.tool_card_border_color(cx)) .bg(cx.theme().colors().editor_background) .overflow_hidden() }) .child( h_flex() .id(header_id) .w_full() .gap_1() .justify_between() .map(|this| { if needs_confirmation { this.px_2() .py_1() .rounded_t_md() .bg(self.tool_card_header_bg(cx)) .border_b_1() .border_color(self.tool_card_border_color(cx)) } else { this.opacity(0.8).hover(|style| style.opacity(1.)) } }) .child( h_flex() .id("tool-call-header") .overflow_x_scroll() .map(|this| { if needs_confirmation { this.text_xs() } else { this.text_size(self.tool_name_font_size()) } }) .gap_1p5() .child( Icon::new(match tool_call.kind { acp::ToolKind::Read => IconName::ToolRead, acp::ToolKind::Edit => IconName::ToolPencil, acp::ToolKind::Delete => IconName::ToolDeleteFile, acp::ToolKind::Move => IconName::ArrowRightLeft, acp::ToolKind::Search => IconName::ToolSearch, acp::ToolKind::Execute => IconName::ToolTerminal, acp::ToolKind::Think => IconName::ToolBulb, acp::ToolKind::Fetch => IconName::ToolWeb, acp::ToolKind::Other => IconName::ToolHammer, }) .size(IconSize::Small) .color(Color::Muted), ) .child(if tool_call.locations.len() == 1 { let name = tool_call.locations[0] .path .file_name() .unwrap_or_default() .display() .to_string(); h_flex() .id(("open-tool-call-location", entry_ix)) .child(name) .w_full() .max_w_full() .pr_1() .gap_0p5() .cursor_pointer() .rounded_sm() .opacity(0.8) .hover(|label| { label.opacity(1.).bg(cx .theme() .colors() .element_hover .opacity(0.5)) }) .tooltip(Tooltip::text("Jump to File")) .on_click(cx.listener(move |this, _, window, cx| { this.open_tool_call_location(entry_ix, 0, window, cx); })) .into_any_element() } else { self.render_markdown( tool_call.label.clone(), default_markdown_style(needs_confirmation, window, cx), ) .into_any() }), ) .child( h_flex() .gap_0p5() .when(is_collapsible, |this| { this.child( Disclosure::new(("expand", entry_ix), is_open) .opened_icon(IconName::ChevronUp) .closed_icon(IconName::ChevronDown) .on_click(cx.listener({ let id = tool_call.id.clone(); move |this: &mut Self, _, _, cx: &mut Context| { if is_open { this.expanded_tool_calls.remove(&id); } else { this.expanded_tool_calls.insert(id.clone()); } cx.notify(); } })), ) }) .children(status_icon), ) .on_click(cx.listener({ let id = tool_call.id.clone(); move |this: &mut Self, _, _, cx: &mut Context| { if is_open { this.expanded_tool_calls.remove(&id); } else { this.expanded_tool_calls.insert(id.clone()); } cx.notify(); } })), ) .when(is_open, |this| { this.child( v_flex() .text_xs() .when(is_collapsible, |this| { this.mt_1() .border_1() .border_color(self.tool_card_border_color(cx)) .bg(cx.theme().colors().editor_background) .rounded_lg() }) .map(|this| { if is_open { match &tool_call.status { ToolCallStatus::WaitingForConfirmation { options, .. } => this .children(tool_call.content.iter().map(|content| { div() .py_1p5() .child( self.render_tool_call_content( content, window, cx, ), ) .into_any_element() })) .child(self.render_permission_buttons( options, entry_ix, tool_call.id.clone(), tool_call.content.is_empty(), cx, )), ToolCallStatus::Allowed { .. } | ToolCallStatus::Canceled => { this.children(tool_call.content.iter().map(|content| { div() .py_1p5() .child( self.render_tool_call_content( content, window, cx, ), ) .into_any_element() })) } ToolCallStatus::Rejected => this, } } else { this } }), ) }) } fn render_tool_call_content( &self, content: &ToolCallContent, window: &Window, cx: &Context, ) -> AnyElement { match content { ToolCallContent::ContentBlock { content } => { if let Some(md) = content.markdown() { div() .p_2() .child( self.render_markdown( md.clone(), default_markdown_style(false, window, cx), ), ) .into_any_element() } else { Empty.into_any_element() } } ToolCallContent::Diff { diff: Diff { multibuffer, .. }, .. } => self.render_diff_editor(multibuffer), } } fn render_permission_buttons( &self, options: &[acp::PermissionOption], entry_ix: usize, tool_call_id: acp::ToolCallId, empty_content: bool, cx: &Context, ) -> Div { h_flex() .py_1p5() .px_1p5() .gap_1() .justify_end() .when(!empty_content, |this| { this.border_t_1() .border_color(self.tool_card_border_color(cx)) }) .children(options.iter().map(|option| { let option_id = SharedString::from(option.id.0.clone()); Button::new((option_id, entry_ix), option.name.clone()) .map(|this| match option.kind { acp::PermissionOptionKind::AllowOnce => { this.icon(IconName::Check).icon_color(Color::Success) } acp::PermissionOptionKind::AllowAlways => { this.icon(IconName::CheckDouble).icon_color(Color::Success) } acp::PermissionOptionKind::RejectOnce => { this.icon(IconName::X).icon_color(Color::Error) } acp::PermissionOptionKind::RejectAlways => { this.icon(IconName::X).icon_color(Color::Error) } }) .icon_position(IconPosition::Start) .icon_size(IconSize::XSmall) .on_click(cx.listener({ let tool_call_id = tool_call_id.clone(); let option_id = option.id.clone(); let option_kind = option.kind; move |this, _, _, cx| { this.authorize_tool_call( tool_call_id.clone(), option_id.clone(), option_kind, cx, ); } })) })) } fn render_diff_editor(&self, multibuffer: &Entity) -> AnyElement { v_flex() .h_full() .child( if let Some(editor) = self.diff_editors.get(&multibuffer.entity_id()) { editor.clone().into_any_element() } else { Empty.into_any() }, ) .into_any() } fn render_agent_logo(&self) -> AnyElement { Icon::new(self.agent.logo()) .color(Color::Muted) .size(IconSize::XLarge) .into_any_element() } fn render_error_agent_logo(&self) -> AnyElement { let logo = Icon::new(self.agent.logo()) .color(Color::Muted) .size(IconSize::XLarge) .into_any_element(); h_flex() .relative() .justify_center() .child(div().opacity(0.3).child(logo)) .child( h_flex().absolute().right_1().bottom_0().child( Icon::new(IconName::XCircle) .color(Color::Error) .size(IconSize::Small), ), ) .into_any_element() } fn render_empty_state(&self, cx: &App) -> AnyElement { let loading = matches!(&self.thread_state, ThreadState::Loading { .. }); v_flex() .size_full() .items_center() .justify_center() .child(if loading { h_flex() .justify_center() .child(self.render_agent_logo()) .with_animation( "pulsating_icon", Animation::new(Duration::from_secs(2)) .repeat() .with_easing(pulsating_between(0.4, 1.0)), |icon, delta| icon.opacity(delta), ) .into_any() } else { self.render_agent_logo().into_any_element() }) .child(h_flex().mt_4().mb_1().justify_center().child(if loading { div() .child(LoadingLabel::new("").size(LabelSize::Large)) .into_any_element() } else { Headline::new(self.agent.empty_state_headline()) .size(HeadlineSize::Medium) .into_any_element() })) .child( div() .max_w_1_2() .text_sm() .text_center() .map(|this| { if loading { this.invisible() } else { this.text_color(cx.theme().colors().text_muted) } }) .child(self.agent.empty_state_message()), ) .into_any() } fn render_pending_auth_state(&self) -> AnyElement { v_flex() .items_center() .justify_center() .child(self.render_error_agent_logo()) .child( h_flex() .mt_4() .mb_1() .justify_center() .child(Headline::new("Not Authenticated").size(HeadlineSize::Medium)), ) .into_any() } fn render_server_exited(&self, status: ExitStatus, _cx: &Context) -> AnyElement { v_flex() .items_center() .justify_center() .child(self.render_error_agent_logo()) .child( v_flex() .mt_4() .mb_2() .gap_0p5() .text_center() .items_center() .child(Headline::new("Server exited unexpectedly").size(HeadlineSize::Medium)) .child( Label::new(format!("Exit status: {}", status.code().unwrap_or(-127))) .size(LabelSize::Small) .color(Color::Muted), ), ) .into_any_element() } fn render_load_error(&self, e: &LoadError, cx: &Context) -> AnyElement { let mut container = v_flex() .items_center() .justify_center() .child(self.render_error_agent_logo()) .child( v_flex() .mt_4() .mb_2() .gap_0p5() .text_center() .items_center() .child(Headline::new("Failed to launch").size(HeadlineSize::Medium)) .child( Label::new(e.to_string()) .size(LabelSize::Small) .color(Color::Muted), ), ); if let LoadError::Unsupported { upgrade_message, upgrade_command, .. } = &e { let upgrade_message = upgrade_message.clone(); let upgrade_command = upgrade_command.clone(); container = container.child(Button::new("upgrade", upgrade_message).on_click( cx.listener(move |this, _, window, cx| { this.workspace .update(cx, |workspace, cx| { let project = workspace.project().read(cx); let cwd = project.first_project_directory(cx); let shell = project.terminal_settings(&cwd, cx).shell.clone(); let spawn_in_terminal = task::SpawnInTerminal { id: task::TaskId("install".to_string()), full_label: upgrade_command.clone(), label: upgrade_command.clone(), command: Some(upgrade_command.clone()), args: Vec::new(), command_label: upgrade_command.clone(), cwd, env: Default::default(), use_new_terminal: true, allow_concurrent_runs: true, reveal: Default::default(), reveal_target: Default::default(), hide: Default::default(), shell, show_summary: true, show_command: true, show_rerun: false, }; workspace .spawn_in_terminal(spawn_in_terminal, window, cx) .detach(); }) .ok(); }), )); } container.into_any() } fn render_activity_bar( &self, thread_entity: &Entity, window: &mut Window, cx: &Context, ) -> Option { let thread = thread_entity.read(cx); let action_log = thread.action_log(); let changed_buffers = action_log.read(cx).changed_buffers(cx); let plan = thread.plan(); if changed_buffers.is_empty() && plan.is_empty() { return None; } let editor_bg_color = cx.theme().colors().editor_background; let active_color = cx.theme().colors().element_selected; let bg_edit_files_disclosure = editor_bg_color.blend(active_color.opacity(0.3)); let pending_edits = thread.has_pending_edit_tool_calls(); v_flex() .mt_1() .mx_2() .bg(bg_edit_files_disclosure) .border_1() .border_b_0() .border_color(cx.theme().colors().border) .rounded_t_md() .shadow(vec![gpui::BoxShadow { color: gpui::black().opacity(0.15), offset: point(px(1.), px(-1.)), blur_radius: px(3.), spread_radius: px(0.), }]) .when(!plan.is_empty(), |this| { this.child(self.render_plan_summary(plan, window, cx)) .when(self.plan_expanded, |parent| { parent.child(self.render_plan_entries(plan, window, cx)) }) }) .when(!changed_buffers.is_empty(), |this| { this.child(Divider::horizontal()) .child(self.render_edits_summary( action_log, &changed_buffers, self.edits_expanded, pending_edits, window, cx, )) .when(self.edits_expanded, |parent| { parent.child(self.render_edited_files( action_log, &changed_buffers, pending_edits, cx, )) }) }) .into_any() .into() } fn render_plan_summary(&self, plan: &Plan, window: &mut Window, cx: &Context) -> Div { let stats = plan.stats(); let title = if let Some(entry) = stats.in_progress_entry && !self.plan_expanded { h_flex() .w_full() .gap_1() .text_xs() .text_color(cx.theme().colors().text_muted) .justify_between() .child( h_flex() .gap_1() .child( Label::new("Current:") .size(LabelSize::Small) .color(Color::Muted), ) .child(MarkdownElement::new( entry.content.clone(), plan_label_markdown_style(&entry.status, window, cx), )), ) .when(stats.pending > 0, |this| { this.child( Label::new(format!("{} left", stats.pending)) .size(LabelSize::Small) .color(Color::Muted) .mr_1(), ) }) } else { let status_label = if stats.pending == 0 { "All Done".to_string() } else if stats.completed == 0 { format!("{}", plan.entries.len()) } else { format!("{}/{}", stats.completed, plan.entries.len()) }; h_flex() .w_full() .gap_1() .justify_between() .child( Label::new("Plan") .size(LabelSize::Small) .color(Color::Muted), ) .child( Label::new(status_label) .size(LabelSize::Small) .color(Color::Muted) .mr_1(), ) }; h_flex() .p_1() .justify_between() .when(self.plan_expanded, |this| { this.border_b_1().border_color(cx.theme().colors().border) }) .child( h_flex() .id("plan_summary") .w_full() .gap_1() .child(Disclosure::new("plan_disclosure", self.plan_expanded)) .child(title) .on_click(cx.listener(|this, _, _, cx| { this.plan_expanded = !this.plan_expanded; cx.notify(); })), ) } fn render_plan_entries(&self, plan: &Plan, window: &mut Window, cx: &Context) -> Div { v_flex().children(plan.entries.iter().enumerate().flat_map(|(index, entry)| { let element = h_flex() .py_1() .px_2() .gap_2() .justify_between() .bg(cx.theme().colors().editor_background) .when(index < plan.entries.len() - 1, |parent| { parent.border_color(cx.theme().colors().border).border_b_1() }) .child( h_flex() .id(("plan_entry", index)) .gap_1p5() .max_w_full() .overflow_x_scroll() .text_xs() .text_color(cx.theme().colors().text_muted) .child(match entry.status { acp::PlanEntryStatus::Pending => Icon::new(IconName::TodoPending) .size(IconSize::Small) .color(Color::Muted) .into_any_element(), acp::PlanEntryStatus::InProgress => Icon::new(IconName::TodoProgress) .size(IconSize::Small) .color(Color::Accent) .with_animation( "running", Animation::new(Duration::from_secs(2)).repeat(), |icon, delta| { icon.transform(Transformation::rotate(percentage(delta))) }, ) .into_any_element(), acp::PlanEntryStatus::Completed => Icon::new(IconName::TodoComplete) .size(IconSize::Small) .color(Color::Success) .into_any_element(), }) .child(MarkdownElement::new( entry.content.clone(), plan_label_markdown_style(&entry.status, window, cx), )), ); Some(element) })) } fn render_edits_summary( &self, action_log: &Entity, changed_buffers: &BTreeMap, Entity>, expanded: bool, pending_edits: bool, window: &mut Window, cx: &Context, ) -> Div { const EDIT_NOT_READY_TOOLTIP_LABEL: &str = "Wait until file edits are complete."; let focus_handle = self.focus_handle(cx); h_flex() .p_1() .justify_between() .when(expanded, |this| { this.border_b_1().border_color(cx.theme().colors().border) }) .child( h_flex() .id("edits-container") .cursor_pointer() .w_full() .gap_1() .child(Disclosure::new("edits-disclosure", expanded)) .map(|this| { if pending_edits { this.child( Label::new(format!( "Editing {} {}…", changed_buffers.len(), if changed_buffers.len() == 1 { "file" } else { "files" } )) .color(Color::Muted) .size(LabelSize::Small) .with_animation( "edit-label", Animation::new(Duration::from_secs(2)) .repeat() .with_easing(pulsating_between(0.3, 0.7)), |label, delta| label.alpha(delta), ), ) } else { this.child( Label::new("Edits") .size(LabelSize::Small) .color(Color::Muted), ) .child(Label::new("•").size(LabelSize::XSmall).color(Color::Muted)) .child( Label::new(format!( "{} {}", changed_buffers.len(), if changed_buffers.len() == 1 { "file" } else { "files" } )) .size(LabelSize::Small) .color(Color::Muted), ) } }) .on_click(cx.listener(|this, _, _, cx| { this.edits_expanded = !this.edits_expanded; cx.notify(); })), ) .child( h_flex() .gap_1() .child( IconButton::new("review-changes", IconName::ListTodo) .icon_size(IconSize::Small) .tooltip({ let focus_handle = focus_handle.clone(); move |window, cx| { Tooltip::for_action_in( "Review Changes", &OpenAgentDiff, &focus_handle, window, cx, ) } }) .on_click(cx.listener(|_, _, window, cx| { window.dispatch_action(OpenAgentDiff.boxed_clone(), cx); })), ) .child(Divider::vertical().color(DividerColor::Border)) .child( Button::new("reject-all-changes", "Reject All") .label_size(LabelSize::Small) .disabled(pending_edits) .when(pending_edits, |this| { this.tooltip(Tooltip::text(EDIT_NOT_READY_TOOLTIP_LABEL)) }) .key_binding( KeyBinding::for_action_in( &RejectAll, &focus_handle.clone(), window, cx, ) .map(|kb| kb.size(rems_from_px(10.))), ) .on_click({ let action_log = action_log.clone(); cx.listener(move |_, _, _, cx| { action_log.update(cx, |action_log, cx| { action_log.reject_all_edits(cx).detach(); }) }) }), ) .child( Button::new("keep-all-changes", "Keep All") .label_size(LabelSize::Small) .disabled(pending_edits) .when(pending_edits, |this| { this.tooltip(Tooltip::text(EDIT_NOT_READY_TOOLTIP_LABEL)) }) .key_binding( KeyBinding::for_action_in(&KeepAll, &focus_handle, window, cx) .map(|kb| kb.size(rems_from_px(10.))), ) .on_click({ let action_log = action_log.clone(); cx.listener(move |_, _, _, cx| { action_log.update(cx, |action_log, cx| { action_log.keep_all_edits(cx); }) }) }), ), ) } fn render_edited_files( &self, action_log: &Entity, changed_buffers: &BTreeMap, Entity>, pending_edits: bool, cx: &Context, ) -> Div { let editor_bg_color = cx.theme().colors().editor_background; v_flex().children(changed_buffers.into_iter().enumerate().flat_map( |(index, (buffer, _diff))| { let file = buffer.read(cx).file()?; let path = file.path(); let file_path = path.parent().and_then(|parent| { let parent_str = parent.to_string_lossy(); if parent_str.is_empty() { None } else { Some( Label::new(format!("/{}{}", parent_str, std::path::MAIN_SEPARATOR_STR)) .color(Color::Muted) .size(LabelSize::XSmall) .buffer_font(cx), ) } }); let file_name = path.file_name().map(|name| { Label::new(name.to_string_lossy().to_string()) .size(LabelSize::XSmall) .buffer_font(cx) }); let file_icon = FileIcons::get_icon(&path, cx) .map(Icon::from_path) .map(|icon| icon.color(Color::Muted).size(IconSize::Small)) .unwrap_or_else(|| { Icon::new(IconName::File) .color(Color::Muted) .size(IconSize::Small) }); let overlay_gradient = linear_gradient( 90., linear_color_stop(editor_bg_color, 1.), linear_color_stop(editor_bg_color.opacity(0.2), 0.), ); let element = h_flex() .group("edited-code") .id(("file-container", index)) .relative() .py_1() .pl_2() .pr_1() .gap_2() .justify_between() .bg(editor_bg_color) .when(index < changed_buffers.len() - 1, |parent| { parent.border_color(cx.theme().colors().border).border_b_1() }) .child( h_flex() .id(("file-name", index)) .pr_8() .gap_1p5() .max_w_full() .overflow_x_scroll() .child(file_icon) .child(h_flex().gap_0p5().children(file_name).children(file_path)) .on_click({ let buffer = buffer.clone(); cx.listener(move |this, _, window, cx| { this.open_edited_buffer(&buffer, window, cx); }) }), ) .child( h_flex() .gap_1() .visible_on_hover("edited-code") .child( Button::new("review", "Review") .label_size(LabelSize::Small) .on_click({ let buffer = buffer.clone(); cx.listener(move |this, _, window, cx| { this.open_edited_buffer(&buffer, window, cx); }) }), ) .child(Divider::vertical().color(DividerColor::BorderVariant)) .child( Button::new("reject-file", "Reject") .label_size(LabelSize::Small) .disabled(pending_edits) .on_click({ let buffer = buffer.clone(); let action_log = action_log.clone(); move |_, _, cx| { action_log.update(cx, |action_log, cx| { action_log .reject_edits_in_ranges( buffer.clone(), vec![Anchor::MIN..Anchor::MAX], cx, ) .detach_and_log_err(cx); }) } }), ) .child( Button::new("keep-file", "Keep") .label_size(LabelSize::Small) .disabled(pending_edits) .on_click({ let buffer = buffer.clone(); let action_log = action_log.clone(); move |_, _, cx| { action_log.update(cx, |action_log, cx| { action_log.keep_edits_in_range( buffer.clone(), Anchor::MIN..Anchor::MAX, cx, ); }) } }), ), ) .child( div() .id("gradient-overlay") .absolute() .h_full() .w_12() .top_0() .bottom_0() .right(px(152.)) .bg(overlay_gradient), ); Some(element) }, )) } fn render_message_editor(&mut self, window: &mut Window, cx: &mut Context) -> AnyElement { let focus_handle = self.message_editor.focus_handle(cx); let editor_bg_color = cx.theme().colors().editor_background; let (expand_icon, expand_tooltip) = if self.editor_expanded { (IconName::Minimize, "Minimize Message Editor") } else { (IconName::Maximize, "Expand Message Editor") }; v_flex() .on_action(cx.listener(Self::expand_message_editor)) .p_2() .gap_2() .border_t_1() .border_color(cx.theme().colors().border) .bg(editor_bg_color) .when(self.editor_expanded, |this| { this.h(vh(0.8, window)).size_full().justify_between() }) .child( v_flex() .relative() .size_full() .pt_1() .pr_2p5() .child(div().flex_1().child({ let settings = ThemeSettings::get_global(cx); let font_size = TextSize::Small .rems(cx) .to_pixels(settings.agent_font_size(cx)); let line_height = settings.buffer_line_height.value() * font_size; let text_style = TextStyle { color: cx.theme().colors().text, font_family: settings.buffer_font.family.clone(), font_fallbacks: settings.buffer_font.fallbacks.clone(), font_features: settings.buffer_font.features.clone(), font_size: font_size.into(), line_height: line_height.into(), ..Default::default() }; EditorElement::new( &self.message_editor, EditorStyle { background: editor_bg_color, local_player: cx.theme().players().local(), text: text_style, syntax: cx.theme().syntax().clone(), ..Default::default() }, ) })) .child( h_flex() .absolute() .top_0() .right_0() .opacity(0.5) .hover(|this| this.opacity(1.0)) .child( IconButton::new("toggle-height", expand_icon) .icon_size(IconSize::XSmall) .icon_color(Color::Muted) .tooltip({ let focus_handle = focus_handle.clone(); move |window, cx| { Tooltip::for_action_in( expand_tooltip, &ExpandMessageEditor, &focus_handle, window, cx, ) } }) .on_click(cx.listener(|_, _, window, cx| { window.dispatch_action(Box::new(ExpandMessageEditor), cx); })), ), ), ) .child( h_flex() .flex_none() .justify_between() .child(self.render_follow_toggle(cx)) .child(self.render_send_button(cx)), ) .into_any() } fn render_send_button(&self, cx: &mut Context) -> AnyElement { if self.thread().map_or(true, |thread| { thread.read(cx).status() == ThreadStatus::Idle }) { let is_editor_empty = self.message_editor.read(cx).is_empty(cx); IconButton::new("send-message", IconName::Send) .icon_color(Color::Accent) .style(ButtonStyle::Filled) .disabled(self.thread().is_none() || is_editor_empty) .when(!is_editor_empty, |button| { button.tooltip(move |window, cx| Tooltip::for_action("Send", &Chat, window, cx)) }) .when(is_editor_empty, |button| { button.tooltip(Tooltip::text("Type a message to submit")) }) .on_click(cx.listener(|this, _, window, cx| { this.chat(&Chat, window, cx); })) .into_any_element() } else { IconButton::new("stop-generation", IconName::StopFilled) .icon_color(Color::Error) .style(ButtonStyle::Tinted(ui::TintColor::Error)) .tooltip(move |window, cx| { Tooltip::for_action("Stop Generation", &editor::actions::Cancel, window, cx) }) .on_click(cx.listener(|this, _event, _, cx| this.cancel(cx))) .into_any_element() } } fn render_follow_toggle(&self, cx: &mut Context) -> impl IntoElement { let following = self .workspace .read_with(cx, |workspace, _| { workspace.is_being_followed(CollaboratorId::Agent) }) .unwrap_or(false); IconButton::new("follow-agent", IconName::Crosshair) .icon_size(IconSize::Small) .icon_color(Color::Muted) .toggle_state(following) .selected_icon_color(Some(Color::Custom(cx.theme().players().agent().cursor))) .tooltip(move |window, cx| { if following { Tooltip::for_action("Stop Following Agent", &Follow, window, cx) } else { Tooltip::with_meta( "Follow Agent", Some(&Follow), "Track the agent's location as it reads and edits files.", window, cx, ) } }) .on_click(cx.listener(move |this, _, window, cx| { this.workspace .update(cx, |workspace, cx| { if following { workspace.unfollow(CollaboratorId::Agent, window, cx); } else { workspace.follow(CollaboratorId::Agent, window, cx); } }) .ok(); })) } fn render_markdown(&self, markdown: Entity, style: MarkdownStyle) -> MarkdownElement { let workspace = self.workspace.clone(); MarkdownElement::new(markdown, style).on_url_click(move |text, window, cx| { Self::open_link(text, &workspace, window, cx); }) } fn open_link( url: SharedString, workspace: &WeakEntity, window: &mut Window, cx: &mut App, ) { let Some(workspace) = workspace.upgrade() else { cx.open_url(&url); return; }; if let Some(mention_path) = MentionPath::try_parse(&url) { workspace.update(cx, |workspace, cx| { let project = workspace.project(); let Some((path, entry)) = project.update(cx, |project, cx| { let path = project.find_project_path(mention_path.path(), cx)?; let entry = project.entry_for_path(&path, cx)?; Some((path, entry)) }) else { return; }; if entry.is_dir() { project.update(cx, |_, cx| { cx.emit(project::Event::RevealInProjectPanel(entry.id)); }); } else { workspace .open_path(path, None, true, window, cx) .detach_and_log_err(cx); } }) } else { cx.open_url(&url); } } fn open_tool_call_location( &self, entry_ix: usize, location_ix: usize, window: &mut Window, cx: &mut Context, ) -> Option<()> { let location = self .thread()? .read(cx) .entries() .get(entry_ix)? .locations()? .get(location_ix)?; let project_path = self .project .read(cx) .find_project_path(&location.path, cx)?; let open_task = self .workspace .update(cx, |worskpace, cx| { worskpace.open_path(project_path, None, true, window, cx) }) .log_err()?; window .spawn(cx, async move |cx| { let item = open_task.await?; let Some(active_editor) = item.downcast::() else { return anyhow::Ok(()); }; active_editor.update_in(cx, |editor, window, cx| { let snapshot = editor.buffer().read(cx).snapshot(cx); let first_hunk = editor .diff_hunks_in_ranges( &[editor::Anchor::min()..editor::Anchor::max()], &snapshot, ) .next(); if let Some(first_hunk) = first_hunk { let first_hunk_start = first_hunk.multi_buffer_range().start; editor.change_selections(Default::default(), window, cx, |selections| { selections.select_anchor_ranges([first_hunk_start..first_hunk_start]); }) } })?; anyhow::Ok(()) }) .detach_and_log_err(cx); None } pub fn open_thread_as_markdown( &self, workspace: Entity, window: &mut Window, cx: &mut App, ) -> Task> { let markdown_language_task = workspace .read(cx) .app_state() .languages .language_for_name("Markdown"); let (thread_summary, markdown) = if let Some(thread) = self.thread() { let thread = thread.read(cx); (thread.title().to_string(), thread.to_markdown(cx)) } else { return Task::ready(Ok(())); }; window.spawn(cx, async move |cx| { let markdown_language = markdown_language_task.await?; workspace.update_in(cx, |workspace, window, cx| { let project = workspace.project().clone(); if !project.read(cx).is_local() { anyhow::bail!("failed to open active thread as markdown in remote project"); } let buffer = project.update(cx, |project, cx| { project.create_local_buffer(&markdown, Some(markdown_language), cx) }); let buffer = cx.new(|cx| { MultiBuffer::singleton(buffer, cx).with_title(thread_summary.clone()) }); workspace.add_item_to_active_pane( Box::new(cx.new(|cx| { let mut editor = Editor::for_multibuffer(buffer, Some(project.clone()), window, cx); editor.set_breadcrumb_header(thread_summary); editor })), None, true, window, cx, ); anyhow::Ok(()) })??; anyhow::Ok(()) }) } fn scroll_to_top(&mut self, cx: &mut Context) { self.list_state.scroll_to(ListOffset::default()); cx.notify(); } pub fn scroll_to_bottom(&mut self, cx: &mut Context) { if let Some(thread) = self.thread() { let entry_count = thread.read(cx).entries().len(); self.list_state.reset(entry_count); cx.notify(); } } fn notify_with_sound( &mut self, caption: impl Into, icon: IconName, window: &mut Window, cx: &mut Context, ) { self.play_notification_sound(window, cx); self.show_notification(caption, icon, window, cx); } fn play_notification_sound(&self, window: &Window, cx: &mut App) { let settings = AgentSettings::get_global(cx); if settings.play_sound_when_agent_done && !window.is_window_active() { Audio::play_sound(Sound::AgentDone, cx); } } fn show_notification( &mut self, caption: impl Into, icon: IconName, window: &mut Window, cx: &mut Context, ) { if window.is_window_active() || !self.notifications.is_empty() { return; } let title = self.title(cx); match AgentSettings::get_global(cx).notify_when_agent_waiting { NotifyWhenAgentWaiting::PrimaryScreen => { if let Some(primary) = cx.primary_display() { self.pop_up(icon, caption.into(), title, window, primary, cx); } } NotifyWhenAgentWaiting::AllScreens => { let caption = caption.into(); for screen in cx.displays() { self.pop_up(icon, caption.clone(), title.clone(), window, screen, cx); } } NotifyWhenAgentWaiting::Never => { // Don't show anything } } } fn pop_up( &mut self, icon: IconName, caption: SharedString, title: SharedString, window: &mut Window, screen: Rc, cx: &mut Context, ) { let options = AgentNotification::window_options(screen, cx); let project_name = self.workspace.upgrade().and_then(|workspace| { workspace .read(cx) .project() .read(cx) .visible_worktrees(cx) .next() .map(|worktree| worktree.read(cx).root_name().to_string()) }); if let Some(screen_window) = cx .open_window(options, |_, cx| { cx.new(|_| { AgentNotification::new(title.clone(), caption.clone(), icon, project_name) }) }) .log_err() { if let Some(pop_up) = screen_window.entity(cx).log_err() { self.notification_subscriptions .entry(screen_window) .or_insert_with(Vec::new) .push(cx.subscribe_in(&pop_up, window, { |this, _, event, window, cx| match event { AgentNotificationEvent::Accepted => { let handle = window.window_handle(); cx.activate(true); let workspace_handle = this.workspace.clone(); // If there are multiple Zed windows, activate the correct one. cx.defer(move |cx| { handle .update(cx, |_view, window, _cx| { window.activate_window(); if let Some(workspace) = workspace_handle.upgrade() { workspace.update(_cx, |workspace, cx| { workspace.focus_panel::(window, cx); }); } }) .log_err(); }); this.dismiss_notifications(cx); } AgentNotificationEvent::Dismissed => { this.dismiss_notifications(cx); } } })); self.notifications.push(screen_window); // If the user manually refocuses the original window, dismiss the popup. self.notification_subscriptions .entry(screen_window) .or_insert_with(Vec::new) .push({ let pop_up_weak = pop_up.downgrade(); cx.observe_window_activation(window, move |_, window, cx| { if window.is_window_active() { if let Some(pop_up) = pop_up_weak.upgrade() { pop_up.update(cx, |_, cx| { cx.emit(AgentNotificationEvent::Dismissed); }); } } }) }); } } } fn dismiss_notifications(&mut self, cx: &mut Context) { for window in self.notifications.drain(..) { window .update(cx, |_, window, _| { window.remove_window(); }) .ok(); self.notification_subscriptions.remove(&window); } } fn render_thread_controls(&self, cx: &Context) -> impl IntoElement { let open_as_markdown = IconButton::new("open-as-markdown", IconName::FileText) .icon_size(IconSize::XSmall) .icon_color(Color::Ignored) .tooltip(Tooltip::text("Open Thread as Markdown")) .on_click(cx.listener(move |this, _, window, cx| { if let Some(workspace) = this.workspace.upgrade() { this.open_thread_as_markdown(workspace, window, cx) .detach_and_log_err(cx); } })); let scroll_to_top = IconButton::new("scroll_to_top", IconName::ArrowUpAlt) .icon_size(IconSize::XSmall) .icon_color(Color::Ignored) .tooltip(Tooltip::text("Scroll To Top")) .on_click(cx.listener(move |this, _, _, cx| { this.scroll_to_top(cx); })); h_flex() .mr_1() .pb_2() .px(RESPONSE_PADDING_X) .opacity(0.4) .hover(|style| style.opacity(1.)) .flex_wrap() .justify_end() .child(open_as_markdown) .child(scroll_to_top) } } impl Focusable for AcpThreadView { fn focus_handle(&self, cx: &App) -> FocusHandle { self.message_editor.focus_handle(cx) } } impl Render for AcpThreadView { fn render(&mut self, window: &mut Window, cx: &mut Context) -> impl IntoElement { v_flex() .size_full() .key_context("AcpThread") .on_action(cx.listener(Self::chat)) .on_action(cx.listener(Self::previous_history_message)) .on_action(cx.listener(Self::next_history_message)) .on_action(cx.listener(Self::open_agent_diff)) .bg(cx.theme().colors().panel_background) .child(match &self.thread_state { ThreadState::Unauthenticated { connection } => v_flex() .p_2() .flex_1() .items_center() .justify_center() .child(self.render_pending_auth_state()) .child(h_flex().mt_1p5().justify_center().children( connection.auth_methods().into_iter().map(|method| { Button::new( SharedString::from(method.id.0.clone()), method.name.clone(), ) .on_click({ let method_id = method.id.clone(); cx.listener(move |this, _, window, cx| { this.authenticate(method_id.clone(), window, cx) }) }) }), )), ThreadState::Loading { .. } => v_flex().flex_1().child(self.render_empty_state(cx)), ThreadState::LoadError(e) => v_flex() .p_2() .flex_1() .items_center() .justify_center() .child(self.render_load_error(e, cx)), ThreadState::ServerExited { status } => v_flex() .p_2() .flex_1() .items_center() .justify_center() .child(self.render_server_exited(*status, cx)), ThreadState::Ready { thread, .. } => { let thread_clone = thread.clone(); v_flex().flex_1().map(|this| { if self.list_state.item_count() > 0 { this.child( list( self.list_state.clone(), cx.processor(|this, index: usize, window, cx| { let Some((entry, len)) = this.thread().and_then(|thread| { let entries = &thread.read(cx).entries(); Some((entries.get(index)?, entries.len())) }) else { return Empty.into_any(); }; this.render_entry(index, len, entry, window, cx) }), ) .with_sizing_behavior(gpui::ListSizingBehavior::Auto) .flex_grow() .into_any(), ) .children(match thread_clone.read(cx).status() { ThreadStatus::Idle | ThreadStatus::WaitingForToolConfirmation => { None } ThreadStatus::Generating => div() .px_5() .py_2() .child(LoadingLabel::new("").size(LabelSize::Small)) .into(), }) .children(self.render_activity_bar(&thread_clone, window, cx)) } else { this.child(self.render_empty_state(cx)) } }) } }) .when_some(self.last_error.clone(), |el, error| { el.child( div() .p_2() .text_xs() .border_t_1() .border_color(cx.theme().colors().border) .bg(cx.theme().status().error_background) .child( self.render_markdown(error, default_markdown_style(false, window, cx)), ), ) }) .child(self.render_message_editor(window, cx)) } } fn user_message_markdown_style(window: &Window, cx: &App) -> MarkdownStyle { let mut style = default_markdown_style(false, window, cx); let mut text_style = window.text_style(); let theme_settings = ThemeSettings::get_global(cx); let buffer_font = theme_settings.buffer_font.family.clone(); let buffer_font_size = TextSize::Small.rems(cx); text_style.refine(&TextStyleRefinement { font_family: Some(buffer_font), font_size: Some(buffer_font_size.into()), ..Default::default() }); style.base_text_style = text_style; style.link_callback = Some(Rc::new(move |url, cx| { if MentionPath::try_parse(url).is_some() { let colors = cx.theme().colors(); Some(TextStyleRefinement { background_color: Some(colors.element_background), ..Default::default() }) } else { None } })); style } fn default_markdown_style(buffer_font: bool, window: &Window, cx: &App) -> MarkdownStyle { let theme_settings = ThemeSettings::get_global(cx); let colors = cx.theme().colors(); let buffer_font_size = TextSize::Small.rems(cx); let mut text_style = window.text_style(); let line_height = buffer_font_size * 1.75; let font_family = if buffer_font { theme_settings.buffer_font.family.clone() } else { theme_settings.ui_font.family.clone() }; let font_size = if buffer_font { TextSize::Small.rems(cx) } else { TextSize::Default.rems(cx) }; text_style.refine(&TextStyleRefinement { font_family: Some(font_family), font_fallbacks: theme_settings.ui_font.fallbacks.clone(), font_features: Some(theme_settings.ui_font.features.clone()), font_size: Some(font_size.into()), line_height: Some(line_height.into()), color: Some(cx.theme().colors().text), ..Default::default() }); MarkdownStyle { base_text_style: text_style.clone(), syntax: cx.theme().syntax().clone(), selection_background_color: cx.theme().colors().element_selection_background, code_block_overflow_x_scroll: true, table_overflow_x_scroll: true, heading_level_styles: Some(HeadingLevelStyles { h1: Some(TextStyleRefinement { font_size: Some(rems(1.15).into()), ..Default::default() }), h2: Some(TextStyleRefinement { font_size: Some(rems(1.1).into()), ..Default::default() }), h3: Some(TextStyleRefinement { font_size: Some(rems(1.05).into()), ..Default::default() }), h4: Some(TextStyleRefinement { font_size: Some(rems(1.).into()), ..Default::default() }), h5: Some(TextStyleRefinement { font_size: Some(rems(0.95).into()), ..Default::default() }), h6: Some(TextStyleRefinement { font_size: Some(rems(0.875).into()), ..Default::default() }), }), code_block: StyleRefinement { padding: EdgesRefinement { top: Some(DefiniteLength::Absolute(AbsoluteLength::Pixels(Pixels(8.)))), left: Some(DefiniteLength::Absolute(AbsoluteLength::Pixels(Pixels(8.)))), right: Some(DefiniteLength::Absolute(AbsoluteLength::Pixels(Pixels(8.)))), bottom: Some(DefiniteLength::Absolute(AbsoluteLength::Pixels(Pixels(8.)))), }, margin: EdgesRefinement { top: Some(Length::Definite(Pixels(8.).into())), left: Some(Length::Definite(Pixels(0.).into())), right: Some(Length::Definite(Pixels(0.).into())), bottom: Some(Length::Definite(Pixels(12.).into())), }, border_style: Some(BorderStyle::Solid), border_widths: EdgesRefinement { top: Some(AbsoluteLength::Pixels(Pixels(1.))), left: Some(AbsoluteLength::Pixels(Pixels(1.))), right: Some(AbsoluteLength::Pixels(Pixels(1.))), bottom: Some(AbsoluteLength::Pixels(Pixels(1.))), }, border_color: Some(colors.border_variant), background: Some(colors.editor_background.into()), text: Some(TextStyleRefinement { font_family: Some(theme_settings.buffer_font.family.clone()), font_fallbacks: theme_settings.buffer_font.fallbacks.clone(), font_features: Some(theme_settings.buffer_font.features.clone()), font_size: Some(buffer_font_size.into()), ..Default::default() }), ..Default::default() }, inline_code: TextStyleRefinement { font_family: Some(theme_settings.buffer_font.family.clone()), font_fallbacks: theme_settings.buffer_font.fallbacks.clone(), font_features: Some(theme_settings.buffer_font.features.clone()), font_size: Some(buffer_font_size.into()), background_color: Some(colors.editor_foreground.opacity(0.08)), ..Default::default() }, link: TextStyleRefinement { background_color: Some(colors.editor_foreground.opacity(0.025)), underline: Some(UnderlineStyle { color: Some(colors.text_accent.opacity(0.5)), thickness: px(1.), ..Default::default() }), ..Default::default() }, ..Default::default() } } fn plan_label_markdown_style( status: &acp::PlanEntryStatus, window: &Window, cx: &App, ) -> MarkdownStyle { let default_md_style = default_markdown_style(false, window, cx); MarkdownStyle { base_text_style: TextStyle { color: cx.theme().colors().text_muted, strikethrough: if matches!(status, acp::PlanEntryStatus::Completed) { Some(gpui::StrikethroughStyle { thickness: px(1.), color: Some(cx.theme().colors().text_muted.opacity(0.8)), }) } else { None }, ..default_md_style.base_text_style }, ..default_md_style } } #[cfg(test)] mod tests { use agent_client_protocol::SessionId; use editor::EditorSettings; use fs::FakeFs; use futures::future::try_join_all; use gpui::{SemanticVersion, TestAppContext, VisualTestContext}; use rand::Rng; use settings::SettingsStore; use super::*; #[gpui::test] async fn test_drop(cx: &mut TestAppContext) { init_test(cx); let (thread_view, _cx) = setup_thread_view(StubAgentServer::default(), cx).await; let weak_view = thread_view.downgrade(); drop(thread_view); assert!(!weak_view.is_upgradable()); } #[gpui::test] async fn test_notification_for_stop_event(cx: &mut TestAppContext) { init_test(cx); let (thread_view, cx) = setup_thread_view(StubAgentServer::default(), cx).await; let message_editor = cx.read(|cx| thread_view.read(cx).message_editor.clone()); message_editor.update_in(cx, |editor, window, cx| { editor.set_text("Hello", window, cx); }); cx.deactivate_window(); thread_view.update_in(cx, |thread_view, window, cx| { thread_view.chat(&Chat, window, cx); }); cx.run_until_parked(); assert!( cx.windows() .iter() .any(|window| window.downcast::().is_some()) ); } #[gpui::test] async fn test_notification_for_error(cx: &mut TestAppContext) { init_test(cx); let (thread_view, cx) = setup_thread_view(StubAgentServer::new(SaboteurAgentConnection), cx).await; let message_editor = cx.read(|cx| thread_view.read(cx).message_editor.clone()); message_editor.update_in(cx, |editor, window, cx| { editor.set_text("Hello", window, cx); }); cx.deactivate_window(); thread_view.update_in(cx, |thread_view, window, cx| { thread_view.chat(&Chat, window, cx); }); cx.run_until_parked(); assert!( cx.windows() .iter() .any(|window| window.downcast::().is_some()) ); } #[gpui::test] async fn test_notification_for_tool_authorization(cx: &mut TestAppContext) { init_test(cx); let tool_call_id = acp::ToolCallId("1".into()); let tool_call = acp::ToolCall { id: tool_call_id.clone(), title: "Label".into(), kind: acp::ToolKind::Edit, status: acp::ToolCallStatus::Pending, content: vec!["hi".into()], locations: vec![], raw_input: None, }; let connection = StubAgentConnection::new(vec![acp::SessionUpdate::ToolCall(tool_call)]) .with_permission_requests(HashMap::from_iter([( tool_call_id, vec![acp::PermissionOption { id: acp::PermissionOptionId("1".into()), name: "Allow".into(), kind: acp::PermissionOptionKind::AllowOnce, }], )])); let (thread_view, cx) = setup_thread_view(StubAgentServer::new(connection), cx).await; let message_editor = cx.read(|cx| thread_view.read(cx).message_editor.clone()); message_editor.update_in(cx, |editor, window, cx| { editor.set_text("Hello", window, cx); }); cx.deactivate_window(); thread_view.update_in(cx, |thread_view, window, cx| { thread_view.chat(&Chat, window, cx); }); cx.run_until_parked(); assert!( cx.windows() .iter() .any(|window| window.downcast::().is_some()) ); } async fn setup_thread_view( agent: impl AgentServer + 'static, cx: &mut TestAppContext, ) -> (Entity, &mut VisualTestContext) { let fs = FakeFs::new(cx.executor()); let project = Project::test(fs, [], cx).await; let (workspace, cx) = cx.add_window_view(|window, cx| Workspace::test_new(project.clone(), window, cx)); let thread_view = cx.update(|window, cx| { cx.new(|cx| { AcpThreadView::new( Rc::new(agent), workspace.downgrade(), project, Rc::new(RefCell::new(MessageHistory::default())), 1, None, window, cx, ) }) }); cx.run_until_parked(); (thread_view, cx) } struct StubAgentServer { connection: C, } impl StubAgentServer { fn new(connection: C) -> Self { Self { connection } } } impl StubAgentServer { fn default() -> Self { Self::new(StubAgentConnection::default()) } } impl AgentServer for StubAgentServer where C: 'static + AgentConnection + Send + Clone, { fn logo(&self) -> ui::IconName { unimplemented!() } fn name(&self) -> &'static str { unimplemented!() } fn empty_state_headline(&self) -> &'static str { unimplemented!() } fn empty_state_message(&self) -> &'static str { unimplemented!() } fn connect( &self, _root_dir: &Path, _project: &Entity, _cx: &mut App, ) -> Task>> { Task::ready(Ok(Rc::new(self.connection.clone()))) } } #[derive(Clone, Default)] struct StubAgentConnection { sessions: Arc>>>, permission_requests: HashMap>, updates: Vec, } impl StubAgentConnection { fn new(updates: Vec) -> Self { Self { updates, permission_requests: HashMap::default(), sessions: Arc::default(), } } fn with_permission_requests( mut self, permission_requests: HashMap>, ) -> Self { self.permission_requests = permission_requests; self } } impl AgentConnection for StubAgentConnection { fn auth_methods(&self) -> &[acp::AuthMethod] { &[] } fn new_thread( self: Rc, project: Entity, _cwd: &Path, cx: &mut gpui::AsyncApp, ) -> Task>> { let session_id = SessionId( rand::thread_rng() .sample_iter(&rand::distributions::Alphanumeric) .take(7) .map(char::from) .collect::() .into(), ); let thread = cx .new(|cx| AcpThread::new("Test", self.clone(), project, session_id.clone(), cx)) .unwrap(); self.sessions.lock().insert(session_id, thread.downgrade()); Task::ready(Ok(thread)) } fn authenticate( &self, _method_id: acp::AuthMethodId, _cx: &mut App, ) -> Task> { unimplemented!() } fn prompt( &self, params: acp::PromptRequest, cx: &mut App, ) -> Task> { let sessions = self.sessions.lock(); let thread = sessions.get(¶ms.session_id).unwrap(); let mut tasks = vec![]; for update in &self.updates { let thread = thread.clone(); let update = update.clone(); let permission_request = if let acp::SessionUpdate::ToolCall(tool_call) = &update && let Some(options) = self.permission_requests.get(&tool_call.id) { Some((tool_call.clone(), options.clone())) } else { None }; let task = cx.spawn(async move |cx| { if let Some((tool_call, options)) = permission_request { let permission = thread.update(cx, |thread, cx| { thread.request_tool_call_permission( tool_call.clone(), options.clone(), cx, ) })?; permission.await?; } thread.update(cx, |thread, cx| { thread.handle_session_update(update.clone(), cx).unwrap(); })?; anyhow::Ok(()) }); tasks.push(task); } cx.spawn(async move |_| { try_join_all(tasks).await?; Ok(acp::PromptResponse { stop_reason: acp::StopReason::EndTurn, }) }) } fn cancel(&self, _session_id: &acp::SessionId, _cx: &mut App) { unimplemented!() } } #[derive(Clone)] struct SaboteurAgentConnection; impl AgentConnection for SaboteurAgentConnection { fn new_thread( self: Rc, project: Entity, _cwd: &Path, cx: &mut gpui::AsyncApp, ) -> Task>> { Task::ready(Ok(cx .new(|cx| { AcpThread::new( "SaboteurAgentConnection", self, project, SessionId("test".into()), cx, ) }) .unwrap())) } fn auth_methods(&self) -> &[acp::AuthMethod] { &[] } fn authenticate( &self, _method_id: acp::AuthMethodId, _cx: &mut App, ) -> Task> { unimplemented!() } fn prompt( &self, _params: acp::PromptRequest, _cx: &mut App, ) -> Task> { Task::ready(Err(anyhow::anyhow!("Error prompting"))) } fn cancel(&self, _session_id: &acp::SessionId, _cx: &mut App) { unimplemented!() } } fn init_test(cx: &mut TestAppContext) { cx.update(|cx| { let settings_store = SettingsStore::test(cx); cx.set_global(settings_store); language::init(cx); Project::init_settings(cx); AgentSettings::register(cx); workspace::init_settings(cx); ThemeSettings::register(cx); release_channel::init(SemanticVersion::default(), cx); EditorSettings::register(cx); }); } }