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

assistant2: Render messages in the thread using a list (#21491)

Browse source 
This PR updates the rendering of the messages in the current thread to
use a `gpui::list`.

Release Notes:

- N/A
This commit is contained in:
Marshall Bowers 2024-12-03 16:25:09 -05:00 • committed by GitHub
parent db34f29300
commit aca23da971
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 57 additions and 40 deletions
Download patch file Download diff file Expand all files Collapse all files

15
crates/assistant2/src/thread.rs
View file

@ -63,8 +63,8 @@ impl Thread {
}
}
pub fn messages(&self) -> impl Iterator<Item = &Message> {
self.messages.iter()
pub fn message(&self, id: MessageId) -> Option<&Message> {
self.messages.iter().find(|message| message.id == id)
}
pub fn tools(&self) -> &Arc<ToolWorkingSet> {
@ -75,12 +75,14 @@ impl Thread {
self.pending_tool_uses_by_id.values().collect()
}
pub fn insert_user_message(&mut self, text: impl Into<String>) {
pub fn insert_user_message(&mut self, text: impl Into<String>, cx: &mut ModelContext<Self>) {
let id = self.next_message_id.post_inc();
self.messages.push(Message {
id: self.next_message_id.post_inc(),
id,
role: Role::User,
text: text.into(),
});
cx.emit(ThreadEvent::MessageAdded(id));
}
pub fn to_completion_request(
@ -150,11 +152,13 @@ impl Thread {
thread.update(&mut cx, |thread, cx| {
match event {
LanguageModelCompletionEvent::StartMessage { .. } => {
let id = thread.next_message_id.post_inc();
thread.messages.push(Message {
id: thread.next_message_id.post_inc(),
id,
role: Role::Assistant,
text: String::new(),
});
cx.emit(ThreadEvent::MessageAdded(id));
}
LanguageModelCompletionEvent::Stop(reason) => {
stop_reason = reason;
@ -316,6 +320,7 @@ pub enum ThreadError {
pub enum ThreadEvent {
ShowError(ThreadError),
StreamedCompletion,
MessageAdded(MessageId),
UsePendingTools,
ToolFinished {
#[allow(unused)]