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

assistant2: Add rudimentary chat functionality (#21178)

Browse source 
This PR adds in rudimentary functionality for sending messages to the
LLM in `assistant2`.

<img width="1079" alt="Screenshot 2024-11-25 at 1 49 11 PM"
src="https://github.com/user-attachments/assets/5accb749-c034-4fb2-bf55-3ae5bc9529ad">

Release Notes:

- N/A
This commit is contained in:
Marshall Bowers 2024-11-25 14:08:40 -05:00 • committed by GitHub
parent bd02b35ba9
commit a02684b2f7
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
7 changed files with 211 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

28
crates/assistant2/src/assistant_panel.rs
View file

@ -1,7 +1,7 @@
use anyhow::Result;
use gpui::{
prelude::*, px, Action, AppContext, AsyncWindowContext, EventEmitter, FocusHandle,
FocusableView, Pixels, Task, View, ViewContext, WeakView, WindowContext,
FocusableView, Model, Pixels, Task, View, ViewContext, WeakView, WindowContext,
};
use language_model::LanguageModelRegistry;
use language_model_selector::LanguageModelSelector;
@ -10,6 +10,7 @@ use workspace::dock::{DockPosition, Panel, PanelEvent};
use workspace::{Pane, Workspace};
use crate::message_editor::MessageEditor;
use crate::thread::Thread;
use crate::{NewThread, ToggleFocus, ToggleModelSelector};
pub fn init(cx: &mut AppContext) {
@ -25,6 +26,7 @@ pub fn init(cx: &mut AppContext) {
pub struct AssistantPanel {
pane: View<Pane>,
thread: Model<Thread>,
message_editor: View<MessageEditor>,
}
@ -56,9 +58,12 @@ impl AssistantPanel {
pane
});
let thread = cx.new_model(Thread::new);
Self {
pane,
message_editor: cx.new_view(MessageEditor::new),
thread: thread.clone(),
message_editor: cx.new_view(|cx| MessageEditor::new(thread, cx)),
}
}
}
@ -247,7 +252,24 @@ impl Render for AssistantPanel {
println!("Action: New Thread");
}))
.child(self.render_toolbar(cx))
.child(v_flex().bg(cx.theme().colors().panel_background))
.child(
v_flex()
.id("message-list")
.gap_2()
.size_full()
.p_2()
.overflow_y_scroll()
.bg(cx.theme().colors().panel_background)
.children(self.thread.read(cx).messages.iter().map(|message| {
v_flex()
.p_2()
.border_1()
.border_color(cx.theme().colors().border_variant)
.rounded_md()
.child(Label::new(message.role.to_string()))
.child(Label::new(message.text.clone()))
})),
)
.child(
h_flex()
.border_t_1()