mirrors/zed
1
0
Fork 0
mirror of https://github.com/zed-industries/zed.git synced 2024-12-27 02:48:34 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 8

Never insert an empty prefix when splitting a message with a non-empty range

Browse source 
Co-Authored-By: Antonio Scandurra <antonio@zed.dev>
Co-Authored-By: Piotr Osiewicz <piotr@zed.dev>
This commit is contained in:
Nathan Sobo 2023-06-15 09:02:15 -06:00
parent 21e8e8763e
commit 8c6ba13fef
1 changed files with 23 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

29
crates/ai/src/assistant.rs
View file

@ -727,15 +727,15 @@ impl Assistant {
let start_message = self.message_for_offset(range.start, cx);
let end_message = self.message_for_offset(range.end, cx);
if let Some((start_message, end_message)) = start_message.zip(end_message) {
let (start_message_ix, _, start_message_metadata) = start_message;
let (end_message_ix, _, _) = end_message;
let (start_message_ix, _, metadata, message_range) = start_message;
let (end_message_ix, _, _, _) = end_message;
// Prevent splitting when range spans multiple messages.
if start_message_ix != end_message_ix {
return (None, None);
}
let role = start_message_metadata.role;
let role = metadata.role;
self.buffer.update(cx, |buffer, cx| {
buffer.edit([(range.end..range.end, "\n")], None, cx)
});
@ -753,7 +753,7 @@ impl Assistant {
},
);
if range.start == range.end {
if range.start == range.end || range.start == message_range.start {
(None, Some(suffix))
} else {
self.buffer.update(cx, |buffer, cx| {
@ -835,11 +835,11 @@ impl Assistant {
&'a self,
offset: usize,
cx: &'a AppContext,
) -> Option<(usize, &Message, &MessageMetadata)> {
) -> Option<(usize, &Message, &MessageMetadata, Range<usize>)> {
let mut messages = self.messages(cx).peekable();
while let Some((ix, message, metadata, range)) = messages.next() {
if range.contains(&offset) || messages.peek().is_none() {
return Some((ix, message, metadata));
return Some((ix, message, metadata, range));
}
}
None
@ -1600,6 +1600,23 @@ mod tests {
(message_3.id, Role::User, 7..8)
]
);
// Don't include an empty prefix when splitting with a non-empty range
let (no_message, message_8) =
assistant.update(cx, |assistant, cx| assistant.split_message(3..4, cx));
assert!(no_message.is_none());
let message_8 = message_8.unwrap();
assert_eq!(
messages(&assistant, cx),
vec![
(message_1.id, Role::User, 0..3),
(message_6.id, Role::User, 3..5),
(message_8.id, Role::User, 5..6),
(message_7.id, Role::User, 6..7),
(message_5.id, Role::System, 7..8),
(message_3.id, Role::User, 8..9)
]
);
}
fn messages(