Start on an assistant panel based on multi-buffers
Each message is represented as a multibuffer excerpt to allow for fluid editing of the conversation transcript. Co-Authored-By: Antonio Scandurra <antonio@zed.dev>
This commit is contained in:
Nathan Sobo
parent
128c19875d
commit
80080a43e4
7 changed files with 381 additions and 34 deletions
316
crates/ai/src/assistant.rs
Normal file
316
crates/ai/src/assistant.rs
Normal file
|
|
@ -0,0 +1,316 @@
|
|||
use crate::{stream_completion, OpenAIRequest, RequestMessage, Role};
|
||||
use editor::{Editor, ExcerptRange, MultiBuffer};
|
||||
use futures::StreamExt;
|
||||
use gpui::{
|
||||
actions, elements::*, Action, AppContext, Entity, ModelHandle, Subscription, View, ViewContext,
|
||||
ViewHandle, WeakViewHandle, WindowContext,
|
||||
};
|
||||
use language::{language_settings::SoftWrap, Anchor, Buffer};
|
||||
use std::sync::Arc;
|
||||
use util::ResultExt;
|
||||
use workspace::{
|
||||
dock::{DockPosition, Panel},
|
||||
item::Item,
|
||||
pane, Pane, Workspace,
|
||||
};
|
||||
|
||||
actions!(assistant, [NewContext, Assist]);
|
||||
|
||||
pub fn init(cx: &mut AppContext) {
|
||||
cx.add_action(ContextEditor::assist);
|
||||
}
|
||||
|
||||
pub enum AssistantPanelEvent {
|
||||
ZoomIn,
|
||||
ZoomOut,
|
||||
Focus,
|
||||
Close,
|
||||
}
|
||||
|
||||
pub struct AssistantPanel {
|
||||
width: Option<f32>,
|
||||
pane: ViewHandle<Pane>,
|
||||
workspace: WeakViewHandle<Workspace>,
|
||||
_subscriptions: Vec<Subscription>,
|
||||
}
|
||||
|
||||
impl AssistantPanel {
|
||||
pub fn new(workspace: &Workspace, cx: &mut ViewContext<Self>) -> Self {
|
||||
let weak_self = cx.weak_handle();
|
||||
let pane = cx.add_view(|cx| {
|
||||
let window_id = cx.window_id();
|
||||
let mut pane = Pane::new(
|
||||
workspace.weak_handle(),
|
||||
workspace.app_state().background_actions,
|
||||
Default::default(),
|
||||
cx,
|
||||
);
|
||||
pane.set_can_split(false, cx);
|
||||
pane.set_can_navigate(false, cx);
|
||||
pane.on_can_drop(move |_, cx| false);
|
||||
pane.set_render_tab_bar_buttons(cx, move |pane, cx| {
|
||||
let this = weak_self.clone();
|
||||
Flex::row()
|
||||
.with_child(Pane::render_tab_bar_button(
|
||||
0,
|
||||
"icons/plus_12.svg",
|
||||
Some(("New Context".into(), Some(Box::new(NewContext)))),
|
||||
cx,
|
||||
move |_, cx| {},
|
||||
None,
|
||||
))
|
||||
.with_child(Pane::render_tab_bar_button(
|
||||
1,
|
||||
if pane.is_zoomed() {
|
||||
"icons/minimize_8.svg"
|
||||
} else {
|
||||
"icons/maximize_8.svg"
|
||||
},
|
||||
Some(("Toggle Zoom".into(), Some(Box::new(workspace::ToggleZoom)))),
|
||||
cx,
|
||||
move |pane, cx| pane.toggle_zoom(&Default::default(), cx),
|
||||
None,
|
||||
))
|
||||
.into_any()
|
||||
});
|
||||
let buffer_search_bar = cx.add_view(search::BufferSearchBar::new);
|
||||
pane.toolbar()
|
||||
.update(cx, |toolbar, cx| toolbar.add_item(buffer_search_bar, cx));
|
||||
pane
|
||||
});
|
||||
let subscriptions = vec![
|
||||
cx.observe(&pane, |_, _, cx| cx.notify()),
|
||||
cx.subscribe(&pane, Self::handle_pane_event),
|
||||
];
|
||||
|
||||
Self {
|
||||
pane,
|
||||
workspace: workspace.weak_handle(),
|
||||
width: None,
|
||||
_subscriptions: subscriptions,
|
||||
}
|
||||
}
|
||||
|
||||
fn handle_pane_event(
|
||||
&mut self,
|
||||
_pane: ViewHandle<Pane>,
|
||||
event: &pane::Event,
|
||||
cx: &mut ViewContext<Self>,
|
||||
) {
|
||||
match event {
|
||||
pane::Event::ZoomIn => cx.emit(AssistantPanelEvent::ZoomIn),
|
||||
pane::Event::ZoomOut => cx.emit(AssistantPanelEvent::ZoomOut),
|
||||
pane::Event::Focus => cx.emit(AssistantPanelEvent::Focus),
|
||||
pane::Event::Remove => cx.emit(AssistantPanelEvent::Close),
|
||||
_ => {}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
impl Entity for AssistantPanel {
|
||||
type Event = AssistantPanelEvent;
|
||||
}
|
||||
|
||||
impl View for AssistantPanel {
|
||||
fn ui_name() -> &'static str {
|
||||
"AssistantPanel"
|
||||
}
|
||||
|
||||
fn render(&mut self, cx: &mut ViewContext<Self>) -> AnyElement<Self> {
|
||||
ChildView::new(&self.pane, cx).into_any()
|
||||
}
|
||||
}
|
||||
|
||||
impl Panel for AssistantPanel {
|
||||
fn position(&self, cx: &WindowContext) -> DockPosition {
|
||||
DockPosition::Right
|
||||
}
|
||||
|
||||
fn position_is_valid(&self, position: DockPosition) -> bool {
|
||||
matches!(position, DockPosition::Right)
|
||||
}
|
||||
|
||||
fn set_position(&mut self, position: DockPosition, cx: &mut ViewContext<Self>) {}
|
||||
|
||||
fn size(&self, cx: &WindowContext) -> f32 {
|
||||
self.width.unwrap_or(480.)
|
||||
}
|
||||
|
||||
fn set_size(&mut self, size: f32, cx: &mut ViewContext<Self>) {
|
||||
self.width = Some(size);
|
||||
cx.notify();
|
||||
}
|
||||
|
||||
fn should_zoom_in_on_event(event: &AssistantPanelEvent) -> bool {
|
||||
matches!(event, AssistantPanelEvent::ZoomIn)
|
||||
}
|
||||
|
||||
fn should_zoom_out_on_event(event: &AssistantPanelEvent) -> bool {
|
||||
matches!(event, AssistantPanelEvent::ZoomOut)
|
||||
}
|
||||
|
||||
fn is_zoomed(&self, cx: &WindowContext) -> bool {
|
||||
self.pane.read(cx).is_zoomed()
|
||||
}
|
||||
|
||||
fn set_zoomed(&mut self, zoomed: bool, cx: &mut ViewContext<Self>) {
|
||||
self.pane.update(cx, |pane, cx| pane.set_zoomed(zoomed, cx));
|
||||
}
|
||||
|
||||
fn set_active(&mut self, active: bool, cx: &mut ViewContext<Self>) {
|
||||
if active && self.pane.read(cx).items_len() == 0 {
|
||||
cx.defer(|this, cx| {
|
||||
if let Some(workspace) = this.workspace.upgrade(cx) {
|
||||
workspace.update(cx, |workspace, cx| {
|
||||
let focus = this.pane.read(cx).has_focus();
|
||||
let editor = Box::new(cx.add_view(|cx| ContextEditor::new(cx)));
|
||||
Pane::add_item(workspace, &this.pane, editor, true, focus, None, cx);
|
||||
})
|
||||
}
|
||||
});
|
||||
}
|
||||
}
|
||||
|
||||
fn icon_path(&self) -> &'static str {
|
||||
"icons/speech_bubble_12.svg"
|
||||
}
|
||||
|
||||
fn icon_tooltip(&self) -> (String, Option<Box<dyn Action>>) {
|
||||
("Assistant Panel".into(), None)
|
||||
}
|
||||
|
||||
fn should_change_position_on_event(event: &Self::Event) -> bool {
|
||||
false
|
||||
}
|
||||
|
||||
fn should_activate_on_event(_: &Self::Event) -> bool {
|
||||
false
|
||||
}
|
||||
|
||||
fn should_close_on_event(event: &AssistantPanelEvent) -> bool {
|
||||
matches!(event, AssistantPanelEvent::Close)
|
||||
}
|
||||
|
||||
fn has_focus(&self, cx: &WindowContext) -> bool {
|
||||
self.pane.read(cx).has_focus()
|
||||
}
|
||||
|
||||
fn is_focus_event(event: &Self::Event) -> bool {
|
||||
matches!(event, AssistantPanelEvent::Focus)
|
||||
}
|
||||
}
|
||||
|
||||
struct ContextEditor {
|
||||
messages: Vec<Message>,
|
||||
editor: ViewHandle<Editor>,
|
||||
}
|
||||
|
||||
impl ContextEditor {
|
||||
fn new(cx: &mut ViewContext<Self>) -> Self {
|
||||
let messages = vec![Message {
|
||||
role: Role::User,
|
||||
content: cx.add_model(|cx| Buffer::new(0, "", cx)),
|
||||
}];
|
||||
|
||||
let multibuffer = cx.add_model(|cx| {
|
||||
let mut multibuffer = MultiBuffer::new(0);
|
||||
for message in &messages {
|
||||
multibuffer.push_excerpts_with_context_lines(
|
||||
message.content.clone(),
|
||||
vec![Anchor::MIN..Anchor::MAX],
|
||||
0,
|
||||
cx,
|
||||
);
|
||||
}
|
||||
multibuffer
|
||||
});
|
||||
let editor = cx.add_view(|cx| {
|
||||
let mut editor = Editor::for_multibuffer(multibuffer, None, cx);
|
||||
editor.set_soft_wrap_mode(SoftWrap::EditorWidth, cx);
|
||||
editor
|
||||
});
|
||||
|
||||
Self { messages, editor }
|
||||
}
|
||||
|
||||
fn assist(&mut self, _: &Assist, cx: &mut ViewContext<Self>) {
|
||||
let messages = self
|
||||
.messages
|
||||
.iter()
|
||||
.map(|message| RequestMessage {
|
||||
role: message.role,
|
||||
content: message.content.read(cx).text(),
|
||||
})
|
||||
.collect();
|
||||
let request = OpenAIRequest {
|
||||
model: "gpt-3.5-turbo".into(),
|
||||
messages,
|
||||
stream: true,
|
||||
};
|
||||
|
||||
if let Some(api_key) = std::env::var("OPENAI_API_KEY").log_err() {
|
||||
let stream = stream_completion(api_key, cx.background_executor().clone(), request);
|
||||
let content = cx.add_model(|cx| Buffer::new(0, "", cx));
|
||||
self.messages.push(Message {
|
||||
role: Role::Assistant,
|
||||
content: content.clone(),
|
||||
});
|
||||
self.editor.update(cx, |editor, cx| {
|
||||
editor.buffer().update(cx, |multibuffer, cx| {
|
||||
multibuffer.push_excerpts_with_context_lines(
|
||||
content.clone(),
|
||||
vec![Anchor::MIN..Anchor::MAX],
|
||||
0,
|
||||
cx,
|
||||
);
|
||||
});
|
||||
});
|
||||
cx.spawn(|_, mut cx| async move {
|
||||
let mut messages = stream.await?;
|
||||
|
||||
while let Some(message) = messages.next().await {
|
||||
let mut message = message?;
|
||||
if let Some(choice) = message.choices.pop() {
|
||||
content.update(&mut cx, |content, cx| {
|
||||
let text: Arc<str> = choice.delta.content?.into();
|
||||
content.edit([(content.len()..content.len(), text)], None, cx);
|
||||
Some(())
|
||||
});
|
||||
}
|
||||
}
|
||||
anyhow::Ok(())
|
||||
})
|
||||
.detach_and_log_err(cx);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
impl Entity for ContextEditor {
|
||||
type Event = ();
|
||||
}
|
||||
|
||||
impl View for ContextEditor {
|
||||
fn ui_name() -> &'static str {
|
||||
"ContextEditor"
|
||||
}
|
||||
|
||||
fn render(&mut self, cx: &mut ViewContext<Self>) -> AnyElement<Self> {
|
||||
ChildView::new(&self.editor, cx).into_any()
|
||||
}
|
||||
}
|
||||
|
||||
impl Item for ContextEditor {
|
||||
fn tab_content<V: View>(
|
||||
&self,
|
||||
_: Option<usize>,
|
||||
style: &theme::Tab,
|
||||
_: &gpui::AppContext,
|
||||
) -> AnyElement<V> {
|
||||
Label::new("New Context", style.label.clone()).into_any()
|
||||
}
|
||||
}
|
||||
|
||||
struct Message {
|
||||
role: Role,
|
||||
content: ModelHandle<Buffer>,
|
||||
}
|
||||
Loading…
Add table
Add a link
Reference in a new issue