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

Fix assistant panel tests

Browse source
This commit is contained in:
Antonio Scandurra 2023-06-07 15:24:08 +02:00
parent d26cc2c897
commit d3e0d38bef
1 changed files with 7 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -1365,16 +1365,18 @@ mod tests {
cx.add_model(|cx| {
let mut assistant = Assistant::new(Default::default(), registry, cx);
let message_1 = assistant.insert_message_after(ExcerptId::max(), Role::Assistant, cx);
let message_2 = assistant.insert_message_after(message_1.excerpt_id, Role::User, cx);
let message_3 = assistant.insert_message_after(message_1.excerpt_id, Role::User, cx);
let message_1 = assistant.messages[0].clone();
let message_2 = assistant.insert_message_after(ExcerptId::max(), Role::Assistant, cx);
let message_3 = assistant.insert_message_after(message_2.excerpt_id, Role::User, cx);
let message_4 = assistant.insert_message_after(message_2.excerpt_id, Role::User, cx);
assistant.remove_empty_messages(
HashSet::from_iter([message_2.excerpt_id, message_3.excerpt_id]),
HashSet::from_iter([message_3.excerpt_id, message_4.excerpt_id]),
Default::default(),
cx,
);
assert_eq!(assistant.messages.len(), 1);
assert_eq!(assistant.messages.len(), 2);
assert_eq!(assistant.messages[0].excerpt_id, message_1.excerpt_id);
assert_eq!(assistant.messages[1].excerpt_id, message_2.excerpt_id);
assistant
});
}