zed-industries/zed
1
1
Fork 0
mirror of https://github.com/zed-industries/zed.git synced 2024-11-10 05:37:29 +03:00
Code Issues Packages Projects Releases Wiki Activity

Insert empty user message when assisting with the current last message

Browse Source 
Co-Authored-By: Nathan Sobo <nathan@zed.dev>
This commit is contained in:
Antonio Scandurra 2023-06-26 15:57:36 +02:00
parent 92d7b6aa3b
commit 9d4dd5c42b
1 changed files with 21 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
crates/ai/src/assistant.rs
View File

@ -1023,6 +1023,14 @@ impl Conversation {
) -> Vec<MessageAnchor> {
let mut user_messages = Vec::new();
let mut tasks = Vec::new();
let last_message_id = self.message_anchors.iter().rev().find_map(|message| {
message
.start
.is_valid(self.buffer.read(cx))
.then_some(message.id)
});
for selected_message_id in selected_messages {
let selected_message_role =
if let Some(metadata) = self.messages_metadata.get(&selected_message_id) {
@ -1085,6 +1093,19 @@ impl Conversation {
)
.unwrap();
// Queue up the user's next reply
if Some(selected_message_id) == last_message_id {
let user_message = self
.insert_message_after(
assistant_message.id,
Role::User,
MessageStatus::Done,
cx,
)
.unwrap();
user_messages.push(user_message);
}
tasks.push(cx.spawn_weak({
|this, mut cx| async move {
let assistant_message_id = assistant_message.id;