Yehowshua/ZIm
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Wrap earlier

Browse source
This commit is contained in:
Agus Zubiaga 2025-08-12 10:20:30 -03:00
parent cf8e056ec4
commit 38128bab3e
2 changed files with 10 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

15
crates/agent_ui/src/acp/completion_provider.rs
View file

@ -1,5 +1,5 @@
use std::ops::Range; use std::ops::Range;
use std::path::{Path, PathBuf}; use std::path::Path;
use std::sync::Arc; use std::sync::Arc;
use std::sync::atomic::AtomicBool; use std::sync::atomic::AtomicBool;
@ -25,17 +25,16 @@ use crate::context_picker::file_context_picker::{extract_file_name_and_directory
#[derive(Default)] #[derive(Default)]
pub struct MentionSet { pub struct MentionSet {
paths_by_crease_id: HashMap<CreaseId, MentionUri>, uri_by_crease_id: HashMap<CreaseId, MentionUri>,
} }
impl MentionSet { impl MentionSet {
pub fn insert(&mut self, crease_id: CreaseId, path: PathBuf) { pub fn insert(&mut self, crease_id: CreaseId, uri: MentionUri) {
self.paths_by_crease_id self.uri_by_crease_id.insert(crease_id, uri);
.insert(crease_id, MentionUri::File(path));
} }
pub fn drain(&mut self) -> impl Iterator<Item = CreaseId> { pub fn drain(&mut self) -> impl Iterator<Item = CreaseId> {
self.paths_by_crease_id.drain().map(|(id, _)| id) self.uri_by_crease_id.drain().map(|(id, _)| id)
} }
pub fn contents( pub fn contents(
@ -44,7 +43,7 @@ impl MentionSet {
cx: &mut App, cx: &mut App,
) -> Task<Result<HashMap<CreaseId, Mention>>> { ) -> Task<Result<HashMap<CreaseId, Mention>>> {
let contents = self let contents = self
.paths_by_crease_id .uri_by_crease_id
.iter() .iter()
.map(|(crease_id, uri)| match uri { .map(|(crease_id, uri)| match uri {
MentionUri::File(path) => { MentionUri::File(path) => {
@ -328,7 +327,7 @@ fn confirm_completion_callback(
}; };
if let Some(crease_id) = crease_id { if let Some(crease_id) = crease_id {
mention_set.lock().insert(crease_id, path); mention_set.lock().insert(crease_id, MentionUri::File(path));
} }
}); });
false false

4
crates/agent_ui/src/acp/thread_view.rs
View file

@ -640,7 +640,9 @@ impl AcpThreadView {
); );
if let Some(crease_id) = crease_id { if let Some(crease_id) = crease_id {
mention_set.lock().insert(crease_id, project_path); mention_set
.lock()
.insert(crease_id, MentionUri::File(project_path));
} }
} }
} }