1use crate::{
2 slash_command::SlashCommandLine, CompletionProvider, LanguageModelRequest,
3 LanguageModelRequestMessage, MessageId, MessageStatus, Role,
4};
5use anyhow::{anyhow, Context as _, Result};
6use assistant_slash_command::{
7 SlashCommandOutput, SlashCommandOutputSection, SlashCommandRegistry,
8};
9use client::{proto, telemetry::Telemetry};
10use clock::ReplicaId;
11use collections::{HashMap, HashSet};
12use fs::Fs;
13use futures::{future::Shared, FutureExt, StreamExt};
14use gpui::{AppContext, Context as _, EventEmitter, Model, ModelContext, Subscription, Task};
15use language::{AnchorRangeExt, Bias, Buffer, LanguageRegistry, OffsetRangeExt, Point, ToOffset};
16use open_ai::Model as OpenAiModel;
17use paths::contexts_dir;
18use serde::{Deserialize, Serialize};
19use std::{
20 cmp::Ordering,
21 iter, mem,
22 ops::Range,
23 path::{Path, PathBuf},
24 sync::Arc,
25 time::{Duration, Instant},
26};
27use telemetry_events::AssistantKind;
28use ui::SharedString;
29use util::{post_inc, TryFutureExt};
30use uuid::Uuid;
31
32#[derive(Clone, Eq, PartialEq, Hash, PartialOrd, Ord, Serialize, Deserialize)]
33pub struct ContextId(String);
34
35impl ContextId {
36 pub fn new() -> Self {
37 Self(Uuid::new_v4().to_string())
38 }
39
40 pub fn from_proto(id: String) -> Self {
41 Self(id)
42 }
43
44 pub fn to_proto(&self) -> String {
45 self.0.clone()
46 }
47}
48
49#[derive(Clone, Debug)]
50pub enum ContextOperation {
51 InsertMessage {
52 anchor: MessageAnchor,
53 metadata: MessageMetadata,
54 version: clock::Global,
55 },
56 UpdateMessage {
57 message_id: MessageId,
58 metadata: MessageMetadata,
59 version: clock::Global,
60 },
61 UpdateSummary {
62 summary: ContextSummary,
63 version: clock::Global,
64 },
65 SlashCommandFinished {
66 id: SlashCommandId,
67 output_range: Range<language::Anchor>,
68 sections: Vec<SlashCommandOutputSection<language::Anchor>>,
69 version: clock::Global,
70 },
71 BufferOperation(language::Operation),
72}
73
74impl ContextOperation {
75 pub fn from_proto(op: proto::ContextOperation) -> Result<Self> {
76 match op.variant.context("invalid variant")? {
77 proto::context_operation::Variant::InsertMessage(insert) => {
78 let message = insert.message.context("invalid message")?;
79 let id = MessageId(language::proto::deserialize_timestamp(
80 message.id.context("invalid id")?,
81 ));
82 Ok(Self::InsertMessage {
83 anchor: MessageAnchor {
84 id,
85 start: language::proto::deserialize_anchor(
86 message.start.context("invalid anchor")?,
87 )
88 .context("invalid anchor")?,
89 },
90 metadata: MessageMetadata {
91 role: Role::from_proto(message.role),
92 status: MessageStatus::from_proto(
93 message.status.context("invalid status")?,
94 ),
95 timestamp: id.0,
96 },
97 version: language::proto::deserialize_version(&insert.version),
98 })
99 }
100 proto::context_operation::Variant::UpdateMessage(update) => Ok(Self::UpdateMessage {
101 message_id: MessageId(language::proto::deserialize_timestamp(
102 update.message_id.context("invalid message id")?,
103 )),
104 metadata: MessageMetadata {
105 role: Role::from_proto(update.role),
106 status: MessageStatus::from_proto(update.status.context("invalid status")?),
107 timestamp: language::proto::deserialize_timestamp(
108 update.timestamp.context("invalid timestamp")?,
109 ),
110 },
111 version: language::proto::deserialize_version(&update.version),
112 }),
113 proto::context_operation::Variant::UpdateSummary(update) => Ok(Self::UpdateSummary {
114 summary: ContextSummary {
115 text: update.summary,
116 done: update.done,
117 timestamp: language::proto::deserialize_timestamp(
118 update.timestamp.context("invalid timestamp")?,
119 ),
120 },
121 version: language::proto::deserialize_version(&update.version),
122 }),
123 proto::context_operation::Variant::SlashCommandFinished(finished) => {
124 Ok(Self::SlashCommandFinished {
125 id: SlashCommandId(language::proto::deserialize_timestamp(
126 finished.id.context("invalid id")?,
127 )),
128 output_range: language::proto::deserialize_anchor_range(
129 finished.output_range.context("invalid range")?,
130 )?,
131 sections: finished
132 .sections
133 .into_iter()
134 .map(|section| {
135 Ok(SlashCommandOutputSection {
136 range: language::proto::deserialize_anchor_range(
137 section.range.context("invalid range")?,
138 )?,
139 icon: section.icon_name.parse()?,
140 label: section.label.into(),
141 })
142 })
143 .collect::<Result<Vec<_>>>()?,
144 version: language::proto::deserialize_version(&finished.version),
145 })
146 }
147 proto::context_operation::Variant::BufferOperation(op) => Ok(Self::BufferOperation(
148 language::proto::deserialize_operation(
149 op.operation.context("invalid buffer operation")?,
150 )?,
151 )),
152 }
153 }
154
155 pub fn to_proto(&self) -> proto::ContextOperation {
156 match self {
157 Self::InsertMessage {
158 anchor,
159 metadata,
160 version,
161 } => proto::ContextOperation {
162 variant: Some(proto::context_operation::Variant::InsertMessage(
163 proto::context_operation::InsertMessage {
164 message: Some(proto::ContextMessage {
165 id: Some(language::proto::serialize_timestamp(anchor.id.0)),
166 start: Some(language::proto::serialize_anchor(&anchor.start)),
167 role: metadata.role.to_proto() as i32,
168 status: Some(metadata.status.to_proto()),
169 }),
170 version: language::proto::serialize_version(version),
171 },
172 )),
173 },
174 Self::UpdateMessage {
175 message_id,
176 metadata,
177 version,
178 } => proto::ContextOperation {
179 variant: Some(proto::context_operation::Variant::UpdateMessage(
180 proto::context_operation::UpdateMessage {
181 message_id: Some(language::proto::serialize_timestamp(message_id.0)),
182 role: metadata.role.to_proto() as i32,
183 status: Some(metadata.status.to_proto()),
184 timestamp: Some(language::proto::serialize_timestamp(metadata.timestamp)),
185 version: language::proto::serialize_version(version),
186 },
187 )),
188 },
189 Self::UpdateSummary { summary, version } => proto::ContextOperation {
190 variant: Some(proto::context_operation::Variant::UpdateSummary(
191 proto::context_operation::UpdateSummary {
192 summary: summary.text.clone(),
193 done: summary.done,
194 timestamp: Some(language::proto::serialize_timestamp(summary.timestamp)),
195 version: language::proto::serialize_version(version),
196 },
197 )),
198 },
199 Self::SlashCommandFinished {
200 id,
201 output_range,
202 sections,
203 version,
204 } => proto::ContextOperation {
205 variant: Some(proto::context_operation::Variant::SlashCommandFinished(
206 proto::context_operation::SlashCommandFinished {
207 id: Some(language::proto::serialize_timestamp(id.0)),
208 output_range: Some(language::proto::serialize_anchor_range(
209 output_range.clone(),
210 )),
211 sections: sections
212 .iter()
213 .map(|section| {
214 let icon_name: &'static str = section.icon.into();
215 proto::SlashCommandOutputSection {
216 range: Some(language::proto::serialize_anchor_range(
217 section.range.clone(),
218 )),
219 icon_name: icon_name.to_string(),
220 label: section.label.to_string(),
221 }
222 })
223 .collect(),
224 version: language::proto::serialize_version(version),
225 },
226 )),
227 },
228 Self::BufferOperation(operation) => proto::ContextOperation {
229 variant: Some(proto::context_operation::Variant::BufferOperation(
230 proto::context_operation::BufferOperation {
231 operation: Some(language::proto::serialize_operation(operation)),
232 },
233 )),
234 },
235 }
236 }
237
238 fn timestamp(&self) -> clock::Lamport {
239 match self {
240 Self::InsertMessage { anchor, .. } => anchor.id.0,
241 Self::UpdateMessage { metadata, .. } => metadata.timestamp,
242 Self::UpdateSummary { summary, .. } => summary.timestamp,
243 Self::SlashCommandFinished { id, .. } => id.0,
244 Self::BufferOperation(_) => {
245 panic!("reading the timestamp of a buffer operation is not supported")
246 }
247 }
248 }
249
250 /// Returns the current version of the context operation.
251 pub fn version(&self) -> &clock::Global {
252 match self {
253 Self::InsertMessage { version, .. }
254 | Self::UpdateMessage { version, .. }
255 | Self::UpdateSummary { version, .. }
256 | Self::SlashCommandFinished { version, .. } => version,
257 Self::BufferOperation(_) => {
258 panic!("reading the version of a buffer operation is not supported")
259 }
260 }
261 }
262}
263
264#[derive(Clone)]
265pub enum ContextEvent {
266 MessagesEdited,
267 SummaryChanged,
268 EditSuggestionsChanged,
269 StreamedCompletion,
270 PendingSlashCommandsUpdated {
271 removed: Vec<Range<language::Anchor>>,
272 updated: Vec<PendingSlashCommand>,
273 },
274 SlashCommandFinished {
275 output_range: Range<language::Anchor>,
276 sections: Vec<SlashCommandOutputSection<language::Anchor>>,
277 run_commands_in_output: bool,
278 },
279 Operation(ContextOperation),
280}
281
282#[derive(Clone, Default, Debug)]
283pub struct ContextSummary {
284 pub text: String,
285 done: bool,
286 timestamp: clock::Lamport,
287}
288
289#[derive(Clone, Debug, Eq, PartialEq)]
290pub struct MessageAnchor {
291 pub id: MessageId,
292 pub start: language::Anchor,
293}
294
295#[derive(Clone, Debug, Eq, PartialEq, Serialize, Deserialize)]
296pub struct MessageMetadata {
297 pub role: Role,
298 status: MessageStatus,
299 timestamp: clock::Lamport,
300}
301
302#[derive(Clone, Debug, PartialEq, Eq)]
303pub struct Message {
304 pub offset_range: Range<usize>,
305 pub index_range: Range<usize>,
306 pub id: MessageId,
307 pub anchor: language::Anchor,
308 pub role: Role,
309 pub status: MessageStatus,
310}
311
312impl Message {
313 fn to_request_message(&self, buffer: &Buffer) -> LanguageModelRequestMessage {
314 LanguageModelRequestMessage {
315 role: self.role,
316 content: buffer.text_for_range(self.offset_range.clone()).collect(),
317 }
318 }
319}
320
321struct PendingCompletion {
322 id: usize,
323 _task: Task<()>,
324}
325
326#[derive(Copy, Clone, Debug, Hash, Eq, PartialEq)]
327pub struct SlashCommandId(clock::Lamport);
328
329pub struct Context {
330 id: ContextId,
331 timestamp: clock::Lamport,
332 version: clock::Global,
333 pending_ops: Vec<ContextOperation>,
334 operations: Vec<ContextOperation>,
335 buffer: Model<Buffer>,
336 edit_suggestions: Vec<EditSuggestion>,
337 pending_slash_commands: Vec<PendingSlashCommand>,
338 edits_since_last_slash_command_parse: language::Subscription,
339 finished_slash_commands: HashSet<SlashCommandId>,
340 slash_command_output_sections: Vec<SlashCommandOutputSection<language::Anchor>>,
341 message_anchors: Vec<MessageAnchor>,
342 messages_metadata: HashMap<MessageId, MessageMetadata>,
343 summary: Option<ContextSummary>,
344 pending_summary: Task<Option<()>>,
345 completion_count: usize,
346 pending_completions: Vec<PendingCompletion>,
347 token_count: Option<usize>,
348 pending_token_count: Task<Option<()>>,
349 pending_edit_suggestion_parse: Option<Task<()>>,
350 pending_save: Task<Result<()>>,
351 path: Option<PathBuf>,
352 _subscriptions: Vec<Subscription>,
353 telemetry: Option<Arc<Telemetry>>,
354 language_registry: Arc<LanguageRegistry>,
355}
356
357impl EventEmitter<ContextEvent> for Context {}
358
359impl Context {
360 pub fn local(
361 language_registry: Arc<LanguageRegistry>,
362 telemetry: Option<Arc<Telemetry>>,
363 cx: &mut ModelContext<Self>,
364 ) -> Self {
365 Self::new(
366 ContextId::new(),
367 ReplicaId::default(),
368 language::Capability::ReadWrite,
369 language_registry,
370 telemetry,
371 cx,
372 )
373 }
374
375 pub fn new(
376 id: ContextId,
377 replica_id: ReplicaId,
378 capability: language::Capability,
379 language_registry: Arc<LanguageRegistry>,
380 telemetry: Option<Arc<Telemetry>>,
381 cx: &mut ModelContext<Self>,
382 ) -> Self {
383 let buffer = cx.new_model(|_cx| {
384 let mut buffer = Buffer::remote(
385 language::BufferId::new(1).unwrap(),
386 replica_id,
387 capability,
388 "",
389 );
390 buffer.set_language_registry(language_registry.clone());
391 buffer
392 });
393 let edits_since_last_slash_command_parse =
394 buffer.update(cx, |buffer, _| buffer.subscribe());
395 let mut this = Self {
396 id,
397 timestamp: clock::Lamport::new(replica_id),
398 version: clock::Global::new(),
399 pending_ops: Vec::new(),
400 operations: Vec::new(),
401 message_anchors: Default::default(),
402 messages_metadata: Default::default(),
403 edit_suggestions: Vec::new(),
404 pending_slash_commands: Vec::new(),
405 finished_slash_commands: HashSet::default(),
406 slash_command_output_sections: Vec::new(),
407 edits_since_last_slash_command_parse,
408 summary: None,
409 pending_summary: Task::ready(None),
410 completion_count: Default::default(),
411 pending_completions: Default::default(),
412 token_count: None,
413 pending_token_count: Task::ready(None),
414 pending_edit_suggestion_parse: None,
415 _subscriptions: vec![cx.subscribe(&buffer, Self::handle_buffer_event)],
416 pending_save: Task::ready(Ok(())),
417 path: None,
418 buffer,
419 telemetry,
420 language_registry,
421 };
422
423 let first_message_id = MessageId(clock::Lamport {
424 replica_id: 0,
425 value: 0,
426 });
427 let message = MessageAnchor {
428 id: first_message_id,
429 start: language::Anchor::MIN,
430 };
431 this.messages_metadata.insert(
432 first_message_id,
433 MessageMetadata {
434 role: Role::User,
435 status: MessageStatus::Done,
436 timestamp: first_message_id.0,
437 },
438 );
439 this.message_anchors.push(message);
440
441 this.set_language(cx);
442 this.count_remaining_tokens(cx);
443 this
444 }
445
446 fn serialize(&self, cx: &AppContext) -> SavedContext {
447 let buffer = self.buffer.read(cx);
448 SavedContext {
449 id: Some(self.id.clone()),
450 zed: "context".into(),
451 version: SavedContext::VERSION.into(),
452 text: buffer.text(),
453 messages: self
454 .messages(cx)
455 .map(|message| SavedMessage {
456 id: message.id,
457 start: message.offset_range.start,
458 metadata: self.messages_metadata[&message.id].clone(),
459 })
460 .collect(),
461 summary: self
462 .summary
463 .as_ref()
464 .map(|summary| summary.text.clone())
465 .unwrap_or_default(),
466 slash_command_output_sections: self
467 .slash_command_output_sections
468 .iter()
469 .filter_map(|section| {
470 let range = section.range.to_offset(buffer);
471 if section.range.start.is_valid(buffer) && !range.is_empty() {
472 Some(assistant_slash_command::SlashCommandOutputSection {
473 range,
474 icon: section.icon,
475 label: section.label.clone(),
476 })
477 } else {
478 None
479 }
480 })
481 .collect(),
482 }
483 }
484
485 #[allow(clippy::too_many_arguments)]
486 pub fn deserialize(
487 saved_context: SavedContext,
488 path: PathBuf,
489 language_registry: Arc<LanguageRegistry>,
490 telemetry: Option<Arc<Telemetry>>,
491 cx: &mut ModelContext<Self>,
492 ) -> Self {
493 let id = saved_context.id.clone().unwrap_or_else(|| ContextId::new());
494 let mut this = Self::new(
495 id,
496 ReplicaId::default(),
497 language::Capability::ReadWrite,
498 language_registry,
499 telemetry,
500 cx,
501 );
502 this.path = Some(path);
503 this.buffer.update(cx, |buffer, cx| {
504 buffer.set_text(saved_context.text.as_str(), cx)
505 });
506 let operations = saved_context.into_ops(&this.buffer, cx);
507 this.apply_ops(operations, cx).unwrap();
508 this
509 }
510
511 pub fn id(&self) -> &ContextId {
512 &self.id
513 }
514
515 pub fn replica_id(&self) -> ReplicaId {
516 self.timestamp.replica_id
517 }
518
519 pub fn version(&self, cx: &AppContext) -> ContextVersion {
520 ContextVersion {
521 context: self.version.clone(),
522 buffer: self.buffer.read(cx).version(),
523 }
524 }
525
526 pub fn set_capability(
527 &mut self,
528 capability: language::Capability,
529 cx: &mut ModelContext<Self>,
530 ) {
531 self.buffer
532 .update(cx, |buffer, cx| buffer.set_capability(capability, cx));
533 }
534
535 fn next_timestamp(&mut self) -> clock::Lamport {
536 let timestamp = self.timestamp.tick();
537 self.version.observe(timestamp);
538 timestamp
539 }
540
541 pub fn serialize_ops(
542 &self,
543 since: &ContextVersion,
544 cx: &AppContext,
545 ) -> Task<Vec<proto::ContextOperation>> {
546 let buffer_ops = self
547 .buffer
548 .read(cx)
549 .serialize_ops(Some(since.buffer.clone()), cx);
550
551 let mut context_ops = self
552 .operations
553 .iter()
554 .filter(|op| !since.context.observed(op.timestamp()))
555 .cloned()
556 .collect::<Vec<_>>();
557 context_ops.extend(self.pending_ops.iter().cloned());
558
559 cx.background_executor().spawn(async move {
560 let buffer_ops = buffer_ops.await;
561 context_ops.sort_unstable_by_key(|op| op.timestamp());
562 buffer_ops
563 .into_iter()
564 .map(|op| proto::ContextOperation {
565 variant: Some(proto::context_operation::Variant::BufferOperation(
566 proto::context_operation::BufferOperation {
567 operation: Some(op),
568 },
569 )),
570 })
571 .chain(context_ops.into_iter().map(|op| op.to_proto()))
572 .collect()
573 })
574 }
575
576 pub fn apply_ops(
577 &mut self,
578 ops: impl IntoIterator<Item = ContextOperation>,
579 cx: &mut ModelContext<Self>,
580 ) -> Result<()> {
581 let mut buffer_ops = Vec::new();
582 for op in ops {
583 match op {
584 ContextOperation::BufferOperation(buffer_op) => buffer_ops.push(buffer_op),
585 op @ _ => self.pending_ops.push(op),
586 }
587 }
588 self.buffer
589 .update(cx, |buffer, cx| buffer.apply_ops(buffer_ops, cx))?;
590 self.flush_ops(cx);
591
592 Ok(())
593 }
594
595 fn flush_ops(&mut self, cx: &mut ModelContext<Context>) {
596 let mut messages_changed = false;
597 let mut summary_changed = false;
598
599 self.pending_ops.sort_unstable_by_key(|op| op.timestamp());
600 for op in mem::take(&mut self.pending_ops) {
601 if !self.can_apply_op(&op, cx) {
602 self.pending_ops.push(op);
603 continue;
604 }
605
606 let timestamp = op.timestamp();
607 match op.clone() {
608 ContextOperation::InsertMessage {
609 anchor, metadata, ..
610 } => {
611 if self.messages_metadata.contains_key(&anchor.id) {
612 // We already applied this operation.
613 } else {
614 self.insert_message(anchor, metadata, cx);
615 messages_changed = true;
616 }
617 }
618 ContextOperation::UpdateMessage {
619 message_id,
620 metadata: new_metadata,
621 ..
622 } => {
623 let metadata = self.messages_metadata.get_mut(&message_id).unwrap();
624 if new_metadata.timestamp > metadata.timestamp {
625 *metadata = new_metadata;
626 messages_changed = true;
627 }
628 }
629 ContextOperation::UpdateSummary {
630 summary: new_summary,
631 ..
632 } => {
633 if self
634 .summary
635 .as_ref()
636 .map_or(true, |summary| new_summary.timestamp > summary.timestamp)
637 {
638 self.summary = Some(new_summary);
639 summary_changed = true;
640 }
641 }
642 ContextOperation::SlashCommandFinished {
643 id,
644 output_range,
645 sections,
646 ..
647 } => {
648 if self.finished_slash_commands.insert(id) {
649 let buffer = self.buffer.read(cx);
650 self.slash_command_output_sections
651 .extend(sections.iter().cloned());
652 self.slash_command_output_sections
653 .sort_by(|a, b| a.range.cmp(&b.range, buffer));
654 cx.emit(ContextEvent::SlashCommandFinished {
655 output_range,
656 sections,
657 run_commands_in_output: false,
658 });
659 }
660 }
661 ContextOperation::BufferOperation(_) => unreachable!(),
662 }
663
664 self.version.observe(timestamp);
665 self.timestamp.observe(timestamp);
666 self.operations.push(op);
667 }
668
669 if messages_changed {
670 cx.emit(ContextEvent::MessagesEdited);
671 cx.notify();
672 }
673
674 if summary_changed {
675 cx.emit(ContextEvent::SummaryChanged);
676 cx.notify();
677 }
678 }
679
680 fn can_apply_op(&self, op: &ContextOperation, cx: &AppContext) -> bool {
681 if !self.version.observed_all(op.version()) {
682 return false;
683 }
684
685 match op {
686 ContextOperation::InsertMessage { anchor, .. } => self
687 .buffer
688 .read(cx)
689 .version
690 .observed(anchor.start.timestamp),
691 ContextOperation::UpdateMessage { message_id, .. } => {
692 self.messages_metadata.contains_key(message_id)
693 }
694 ContextOperation::UpdateSummary { .. } => true,
695 ContextOperation::SlashCommandFinished {
696 output_range,
697 sections,
698 ..
699 } => {
700 let version = &self.buffer.read(cx).version;
701 sections
702 .iter()
703 .map(|section| §ion.range)
704 .chain([output_range])
705 .all(|range| {
706 let observed_start = range.start == language::Anchor::MIN
707 || range.start == language::Anchor::MAX
708 || version.observed(range.start.timestamp);
709 let observed_end = range.end == language::Anchor::MIN
710 || range.end == language::Anchor::MAX
711 || version.observed(range.end.timestamp);
712 observed_start && observed_end
713 })
714 }
715 ContextOperation::BufferOperation(_) => {
716 panic!("buffer operations should always be applied")
717 }
718 }
719 }
720
721 fn push_op(&mut self, op: ContextOperation, cx: &mut ModelContext<Self>) {
722 self.operations.push(op.clone());
723 cx.emit(ContextEvent::Operation(op));
724 }
725
726 pub fn buffer(&self) -> &Model<Buffer> {
727 &self.buffer
728 }
729
730 pub fn path(&self) -> Option<&Path> {
731 self.path.as_deref()
732 }
733
734 pub fn summary(&self) -> Option<&ContextSummary> {
735 self.summary.as_ref()
736 }
737
738 pub fn edit_suggestions(&self) -> &[EditSuggestion] {
739 &self.edit_suggestions
740 }
741
742 pub fn pending_slash_commands(&self) -> &[PendingSlashCommand] {
743 &self.pending_slash_commands
744 }
745
746 pub fn slash_command_output_sections(&self) -> &[SlashCommandOutputSection<language::Anchor>] {
747 &self.slash_command_output_sections
748 }
749
750 fn set_language(&mut self, cx: &mut ModelContext<Self>) {
751 let markdown = self.language_registry.language_for_name("Markdown");
752 cx.spawn(|this, mut cx| async move {
753 let markdown = markdown.await?;
754 this.update(&mut cx, |this, cx| {
755 this.buffer
756 .update(cx, |buffer, cx| buffer.set_language(Some(markdown), cx));
757 })
758 })
759 .detach_and_log_err(cx);
760 }
761
762 fn handle_buffer_event(
763 &mut self,
764 _: Model<Buffer>,
765 event: &language::Event,
766 cx: &mut ModelContext<Self>,
767 ) {
768 match event {
769 language::Event::Operation(operation) => cx.emit(ContextEvent::Operation(
770 ContextOperation::BufferOperation(operation.clone()),
771 )),
772 language::Event::Edited => {
773 self.count_remaining_tokens(cx);
774 self.reparse_edit_suggestions(cx);
775 self.reparse_slash_commands(cx);
776 cx.emit(ContextEvent::MessagesEdited);
777 }
778 _ => {}
779 }
780 }
781
782 pub(crate) fn token_count(&self) -> Option<usize> {
783 self.token_count
784 }
785
786 pub(crate) fn count_remaining_tokens(&mut self, cx: &mut ModelContext<Self>) {
787 let request = self.to_completion_request(cx);
788 self.pending_token_count = cx.spawn(|this, mut cx| {
789 async move {
790 cx.background_executor()
791 .timer(Duration::from_millis(200))
792 .await;
793
794 let token_count = cx
795 .update(|cx| CompletionProvider::global(cx).count_tokens(request, cx))?
796 .await?;
797
798 this.update(&mut cx, |this, cx| {
799 this.token_count = Some(token_count);
800 cx.notify()
801 })?;
802 anyhow::Ok(())
803 }
804 .log_err()
805 });
806 }
807
808 pub fn reparse_slash_commands(&mut self, cx: &mut ModelContext<Self>) {
809 let buffer = self.buffer.read(cx);
810 let mut row_ranges = self
811 .edits_since_last_slash_command_parse
812 .consume()
813 .into_iter()
814 .map(|edit| {
815 let start_row = buffer.offset_to_point(edit.new.start).row;
816 let end_row = buffer.offset_to_point(edit.new.end).row + 1;
817 start_row..end_row
818 })
819 .peekable();
820
821 let mut removed = Vec::new();
822 let mut updated = Vec::new();
823 while let Some(mut row_range) = row_ranges.next() {
824 while let Some(next_row_range) = row_ranges.peek() {
825 if row_range.end >= next_row_range.start {
826 row_range.end = next_row_range.end;
827 row_ranges.next();
828 } else {
829 break;
830 }
831 }
832
833 let start = buffer.anchor_before(Point::new(row_range.start, 0));
834 let end = buffer.anchor_after(Point::new(
835 row_range.end - 1,
836 buffer.line_len(row_range.end - 1),
837 ));
838
839 let old_range = self.pending_command_indices_for_range(start..end, cx);
840
841 let mut new_commands = Vec::new();
842 let mut lines = buffer.text_for_range(start..end).lines();
843 let mut offset = lines.offset();
844 while let Some(line) = lines.next() {
845 if let Some(command_line) = SlashCommandLine::parse(line) {
846 let name = &line[command_line.name.clone()];
847 let argument = command_line.argument.as_ref().and_then(|argument| {
848 (!argument.is_empty()).then_some(&line[argument.clone()])
849 });
850 if let Some(command) = SlashCommandRegistry::global(cx).command(name) {
851 if !command.requires_argument() || argument.is_some() {
852 let start_ix = offset + command_line.name.start - 1;
853 let end_ix = offset
854 + command_line
855 .argument
856 .map_or(command_line.name.end, |argument| argument.end);
857 let source_range =
858 buffer.anchor_after(start_ix)..buffer.anchor_after(end_ix);
859 let pending_command = PendingSlashCommand {
860 name: name.to_string(),
861 argument: argument.map(ToString::to_string),
862 source_range,
863 status: PendingSlashCommandStatus::Idle,
864 };
865 updated.push(pending_command.clone());
866 new_commands.push(pending_command);
867 }
868 }
869 }
870
871 offset = lines.offset();
872 }
873
874 let removed_commands = self.pending_slash_commands.splice(old_range, new_commands);
875 removed.extend(removed_commands.map(|command| command.source_range));
876 }
877
878 if !updated.is_empty() || !removed.is_empty() {
879 cx.emit(ContextEvent::PendingSlashCommandsUpdated { removed, updated });
880 }
881 }
882
883 fn reparse_edit_suggestions(&mut self, cx: &mut ModelContext<Self>) {
884 self.pending_edit_suggestion_parse = Some(cx.spawn(|this, mut cx| async move {
885 cx.background_executor()
886 .timer(Duration::from_millis(200))
887 .await;
888
889 this.update(&mut cx, |this, cx| {
890 this.reparse_edit_suggestions_in_range(0..this.buffer.read(cx).len(), cx);
891 })
892 .ok();
893 }));
894 }
895
896 fn reparse_edit_suggestions_in_range(
897 &mut self,
898 range: Range<usize>,
899 cx: &mut ModelContext<Self>,
900 ) {
901 self.buffer.update(cx, |buffer, _| {
902 let range_start = buffer.anchor_before(range.start);
903 let range_end = buffer.anchor_after(range.end);
904 let start_ix = self
905 .edit_suggestions
906 .binary_search_by(|probe| {
907 probe
908 .source_range
909 .end
910 .cmp(&range_start, buffer)
911 .then(Ordering::Greater)
912 })
913 .unwrap_err();
914 let end_ix = self
915 .edit_suggestions
916 .binary_search_by(|probe| {
917 probe
918 .source_range
919 .start
920 .cmp(&range_end, buffer)
921 .then(Ordering::Less)
922 })
923 .unwrap_err();
924
925 let mut new_edit_suggestions = Vec::new();
926 let mut message_lines = buffer.as_rope().chunks_in_range(range).lines();
927 while let Some(suggestion) = parse_next_edit_suggestion(&mut message_lines) {
928 let start_anchor = buffer.anchor_after(suggestion.outer_range.start);
929 let end_anchor = buffer.anchor_before(suggestion.outer_range.end);
930 new_edit_suggestions.push(EditSuggestion {
931 source_range: start_anchor..end_anchor,
932 full_path: suggestion.path,
933 });
934 }
935 self.edit_suggestions
936 .splice(start_ix..end_ix, new_edit_suggestions);
937 });
938 cx.emit(ContextEvent::EditSuggestionsChanged);
939 cx.notify();
940 }
941
942 pub fn pending_command_for_position(
943 &mut self,
944 position: language::Anchor,
945 cx: &mut ModelContext<Self>,
946 ) -> Option<&mut PendingSlashCommand> {
947 let buffer = self.buffer.read(cx);
948 match self
949 .pending_slash_commands
950 .binary_search_by(|probe| probe.source_range.end.cmp(&position, buffer))
951 {
952 Ok(ix) => Some(&mut self.pending_slash_commands[ix]),
953 Err(ix) => {
954 let cmd = self.pending_slash_commands.get_mut(ix)?;
955 if position.cmp(&cmd.source_range.start, buffer).is_ge()
956 && position.cmp(&cmd.source_range.end, buffer).is_le()
957 {
958 Some(cmd)
959 } else {
960 None
961 }
962 }
963 }
964 }
965
966 pub fn pending_commands_for_range(
967 &self,
968 range: Range<language::Anchor>,
969 cx: &AppContext,
970 ) -> &[PendingSlashCommand] {
971 let range = self.pending_command_indices_for_range(range, cx);
972 &self.pending_slash_commands[range]
973 }
974
975 fn pending_command_indices_for_range(
976 &self,
977 range: Range<language::Anchor>,
978 cx: &AppContext,
979 ) -> Range<usize> {
980 let buffer = self.buffer.read(cx);
981 let start_ix = match self
982 .pending_slash_commands
983 .binary_search_by(|probe| probe.source_range.end.cmp(&range.start, &buffer))
984 {
985 Ok(ix) | Err(ix) => ix,
986 };
987 let end_ix = match self
988 .pending_slash_commands
989 .binary_search_by(|probe| probe.source_range.start.cmp(&range.end, &buffer))
990 {
991 Ok(ix) => ix + 1,
992 Err(ix) => ix,
993 };
994 start_ix..end_ix
995 }
996
997 pub fn insert_command_output(
998 &mut self,
999 command_range: Range<language::Anchor>,
1000 output: Task<Result<SlashCommandOutput>>,
1001 insert_trailing_newline: bool,
1002 cx: &mut ModelContext<Self>,
1003 ) {
1004 self.reparse_slash_commands(cx);
1005
1006 let insert_output_task = cx.spawn(|this, mut cx| {
1007 let command_range = command_range.clone();
1008 async move {
1009 let output = output.await;
1010 this.update(&mut cx, |this, cx| match output {
1011 Ok(mut output) => {
1012 if insert_trailing_newline {
1013 output.text.push('\n');
1014 }
1015
1016 let version = this.version.clone();
1017 let command_id = SlashCommandId(this.next_timestamp());
1018 let (operation, event) = this.buffer.update(cx, |buffer, cx| {
1019 let start = command_range.start.to_offset(buffer);
1020 let old_end = command_range.end.to_offset(buffer);
1021 let new_end = start + output.text.len();
1022 buffer.edit([(start..old_end, output.text)], None, cx);
1023
1024 let mut sections = output
1025 .sections
1026 .into_iter()
1027 .map(|section| SlashCommandOutputSection {
1028 range: buffer.anchor_after(start + section.range.start)
1029 ..buffer.anchor_before(start + section.range.end),
1030 icon: section.icon,
1031 label: section.label,
1032 })
1033 .collect::<Vec<_>>();
1034 sections.sort_by(|a, b| a.range.cmp(&b.range, buffer));
1035
1036 this.slash_command_output_sections
1037 .extend(sections.iter().cloned());
1038 this.slash_command_output_sections
1039 .sort_by(|a, b| a.range.cmp(&b.range, buffer));
1040
1041 let output_range =
1042 buffer.anchor_after(start)..buffer.anchor_before(new_end);
1043 this.finished_slash_commands.insert(command_id);
1044
1045 (
1046 ContextOperation::SlashCommandFinished {
1047 id: command_id,
1048 output_range: output_range.clone(),
1049 sections: sections.clone(),
1050 version,
1051 },
1052 ContextEvent::SlashCommandFinished {
1053 output_range,
1054 sections,
1055 run_commands_in_output: output.run_commands_in_text,
1056 },
1057 )
1058 });
1059
1060 this.push_op(operation, cx);
1061 cx.emit(event);
1062 }
1063 Err(error) => {
1064 if let Some(pending_command) =
1065 this.pending_command_for_position(command_range.start, cx)
1066 {
1067 pending_command.status =
1068 PendingSlashCommandStatus::Error(error.to_string());
1069 cx.emit(ContextEvent::PendingSlashCommandsUpdated {
1070 removed: vec![pending_command.source_range.clone()],
1071 updated: vec![pending_command.clone()],
1072 });
1073 }
1074 }
1075 })
1076 .ok();
1077 }
1078 });
1079
1080 if let Some(pending_command) = self.pending_command_for_position(command_range.start, cx) {
1081 pending_command.status = PendingSlashCommandStatus::Running {
1082 _task: insert_output_task.shared(),
1083 };
1084 cx.emit(ContextEvent::PendingSlashCommandsUpdated {
1085 removed: vec![pending_command.source_range.clone()],
1086 updated: vec![pending_command.clone()],
1087 });
1088 }
1089 }
1090
1091 pub fn completion_provider_changed(&mut self, cx: &mut ModelContext<Self>) {
1092 self.count_remaining_tokens(cx);
1093 }
1094
1095 pub fn assist(
1096 &mut self,
1097 selected_messages: HashSet<MessageId>,
1098 cx: &mut ModelContext<Self>,
1099 ) -> Vec<MessageAnchor> {
1100 let mut user_messages = Vec::new();
1101
1102 let last_message_id = if let Some(last_message_id) =
1103 self.message_anchors.iter().rev().find_map(|message| {
1104 message
1105 .start
1106 .is_valid(self.buffer.read(cx))
1107 .then_some(message.id)
1108 }) {
1109 last_message_id
1110 } else {
1111 return Default::default();
1112 };
1113
1114 let mut should_assist = false;
1115 for selected_message_id in selected_messages {
1116 let selected_message_role =
1117 if let Some(metadata) = self.messages_metadata.get(&selected_message_id) {
1118 metadata.role
1119 } else {
1120 continue;
1121 };
1122
1123 if selected_message_role == Role::Assistant {
1124 if let Some(user_message) = self.insert_message_after(
1125 selected_message_id,
1126 Role::User,
1127 MessageStatus::Done,
1128 cx,
1129 ) {
1130 user_messages.push(user_message);
1131 }
1132 } else {
1133 should_assist = true;
1134 }
1135 }
1136
1137 if should_assist {
1138 if !CompletionProvider::global(cx).is_authenticated() {
1139 log::info!("completion provider has no credentials");
1140 return Default::default();
1141 }
1142
1143 let request = self.to_completion_request(cx);
1144 let stream = CompletionProvider::global(cx).complete(request, cx);
1145 let assistant_message = self
1146 .insert_message_after(last_message_id, Role::Assistant, MessageStatus::Pending, cx)
1147 .unwrap();
1148
1149 // Queue up the user's next reply.
1150 let user_message = self
1151 .insert_message_after(assistant_message.id, Role::User, MessageStatus::Done, cx)
1152 .unwrap();
1153 user_messages.push(user_message);
1154
1155 let task = cx.spawn({
1156 |this, mut cx| async move {
1157 let assistant_message_id = assistant_message.id;
1158 let mut response_latency = None;
1159 let stream_completion = async {
1160 let request_start = Instant::now();
1161 let mut messages = stream.await.inner.await?;
1162
1163 while let Some(message) = messages.next().await {
1164 if response_latency.is_none() {
1165 response_latency = Some(request_start.elapsed());
1166 }
1167 let text = message?;
1168
1169 this.update(&mut cx, |this, cx| {
1170 let message_ix = this
1171 .message_anchors
1172 .iter()
1173 .position(|message| message.id == assistant_message_id)?;
1174 let message_range = this.buffer.update(cx, |buffer, cx| {
1175 let message_start_offset =
1176 this.message_anchors[message_ix].start.to_offset(buffer);
1177 let message_old_end_offset = this.message_anchors
1178 [message_ix + 1..]
1179 .iter()
1180 .find(|message| message.start.is_valid(buffer))
1181 .map_or(buffer.len(), |message| {
1182 message.start.to_offset(buffer).saturating_sub(1)
1183 });
1184 let message_new_end_offset =
1185 message_old_end_offset + text.len();
1186 buffer.edit(
1187 [(message_old_end_offset..message_old_end_offset, text)],
1188 None,
1189 cx,
1190 );
1191 message_start_offset..message_new_end_offset
1192 });
1193 this.reparse_edit_suggestions_in_range(message_range, cx);
1194 cx.emit(ContextEvent::StreamedCompletion);
1195
1196 Some(())
1197 })?;
1198 smol::future::yield_now().await;
1199 }
1200
1201 this.update(&mut cx, |this, cx| {
1202 this.pending_completions
1203 .retain(|completion| completion.id != this.completion_count);
1204 this.summarize(cx);
1205 })?;
1206
1207 anyhow::Ok(())
1208 };
1209
1210 let result = stream_completion.await;
1211
1212 this.update(&mut cx, |this, cx| {
1213 let error_message = result
1214 .err()
1215 .map(|error| error.to_string().trim().to_string());
1216
1217 this.update_metadata(assistant_message_id, cx, |metadata| {
1218 if let Some(error_message) = error_message.as_ref() {
1219 metadata.status =
1220 MessageStatus::Error(SharedString::from(error_message.clone()));
1221 } else {
1222 metadata.status = MessageStatus::Done;
1223 }
1224 });
1225
1226 if let Some(telemetry) = this.telemetry.as_ref() {
1227 let model = CompletionProvider::global(cx).model();
1228 telemetry.report_assistant_event(
1229 Some(this.id.0.clone()),
1230 AssistantKind::Panel,
1231 model.telemetry_id(),
1232 response_latency,
1233 error_message,
1234 );
1235 }
1236 })
1237 .ok();
1238 }
1239 });
1240
1241 self.pending_completions.push(PendingCompletion {
1242 id: post_inc(&mut self.completion_count),
1243 _task: task,
1244 });
1245 }
1246
1247 user_messages
1248 }
1249
1250 pub fn to_completion_request(&self, cx: &AppContext) -> LanguageModelRequest {
1251 let messages = self
1252 .messages(cx)
1253 .filter(|message| matches!(message.status, MessageStatus::Done))
1254 .map(|message| message.to_request_message(self.buffer.read(cx)));
1255
1256 LanguageModelRequest {
1257 model: CompletionProvider::global(cx).model(),
1258 messages: messages.collect(),
1259 stop: vec![],
1260 temperature: 1.0,
1261 }
1262 }
1263
1264 pub fn cancel_last_assist(&mut self) -> bool {
1265 self.pending_completions.pop().is_some()
1266 }
1267
1268 pub fn cycle_message_roles(&mut self, ids: HashSet<MessageId>, cx: &mut ModelContext<Self>) {
1269 for id in ids {
1270 if let Some(metadata) = self.messages_metadata.get(&id) {
1271 let role = metadata.role.cycle();
1272 self.update_metadata(id, cx, |metadata| metadata.role = role);
1273 }
1274 }
1275 }
1276
1277 pub fn update_metadata(
1278 &mut self,
1279 id: MessageId,
1280 cx: &mut ModelContext<Self>,
1281 f: impl FnOnce(&mut MessageMetadata),
1282 ) {
1283 let version = self.version.clone();
1284 let timestamp = self.next_timestamp();
1285 if let Some(metadata) = self.messages_metadata.get_mut(&id) {
1286 f(metadata);
1287 metadata.timestamp = timestamp;
1288 let operation = ContextOperation::UpdateMessage {
1289 message_id: id,
1290 metadata: metadata.clone(),
1291 version,
1292 };
1293 self.push_op(operation, cx);
1294 cx.emit(ContextEvent::MessagesEdited);
1295 cx.notify();
1296 }
1297 }
1298
1299 fn insert_message_after(
1300 &mut self,
1301 message_id: MessageId,
1302 role: Role,
1303 status: MessageStatus,
1304 cx: &mut ModelContext<Self>,
1305 ) -> Option<MessageAnchor> {
1306 if let Some(prev_message_ix) = self
1307 .message_anchors
1308 .iter()
1309 .position(|message| message.id == message_id)
1310 {
1311 // Find the next valid message after the one we were given.
1312 let mut next_message_ix = prev_message_ix + 1;
1313 while let Some(next_message) = self.message_anchors.get(next_message_ix) {
1314 if next_message.start.is_valid(self.buffer.read(cx)) {
1315 break;
1316 }
1317 next_message_ix += 1;
1318 }
1319
1320 let start = self.buffer.update(cx, |buffer, cx| {
1321 let offset = self
1322 .message_anchors
1323 .get(next_message_ix)
1324 .map_or(buffer.len(), |message| {
1325 buffer.clip_offset(message.start.to_offset(buffer) - 1, Bias::Left)
1326 });
1327 buffer.edit([(offset..offset, "\n")], None, cx);
1328 buffer.anchor_before(offset + 1)
1329 });
1330
1331 let version = self.version.clone();
1332 let anchor = MessageAnchor {
1333 id: MessageId(self.next_timestamp()),
1334 start,
1335 };
1336 let metadata = MessageMetadata {
1337 role,
1338 status,
1339 timestamp: anchor.id.0,
1340 };
1341 self.insert_message(anchor.clone(), metadata.clone(), cx);
1342 self.push_op(
1343 ContextOperation::InsertMessage {
1344 anchor: anchor.clone(),
1345 metadata,
1346 version,
1347 },
1348 cx,
1349 );
1350 Some(anchor)
1351 } else {
1352 None
1353 }
1354 }
1355
1356 pub fn split_message(
1357 &mut self,
1358 range: Range<usize>,
1359 cx: &mut ModelContext<Self>,
1360 ) -> (Option<MessageAnchor>, Option<MessageAnchor>) {
1361 let start_message = self.message_for_offset(range.start, cx);
1362 let end_message = self.message_for_offset(range.end, cx);
1363 if let Some((start_message, end_message)) = start_message.zip(end_message) {
1364 // Prevent splitting when range spans multiple messages.
1365 if start_message.id != end_message.id {
1366 return (None, None);
1367 }
1368
1369 let message = start_message;
1370 let role = message.role;
1371 let mut edited_buffer = false;
1372
1373 let mut suffix_start = None;
1374 if range.start > message.offset_range.start && range.end < message.offset_range.end - 1
1375 {
1376 if self.buffer.read(cx).chars_at(range.end).next() == Some('\n') {
1377 suffix_start = Some(range.end + 1);
1378 } else if self.buffer.read(cx).reversed_chars_at(range.end).next() == Some('\n') {
1379 suffix_start = Some(range.end);
1380 }
1381 }
1382
1383 let version = self.version.clone();
1384 let suffix = if let Some(suffix_start) = suffix_start {
1385 MessageAnchor {
1386 id: MessageId(self.next_timestamp()),
1387 start: self.buffer.read(cx).anchor_before(suffix_start),
1388 }
1389 } else {
1390 self.buffer.update(cx, |buffer, cx| {
1391 buffer.edit([(range.end..range.end, "\n")], None, cx);
1392 });
1393 edited_buffer = true;
1394 MessageAnchor {
1395 id: MessageId(self.next_timestamp()),
1396 start: self.buffer.read(cx).anchor_before(range.end + 1),
1397 }
1398 };
1399
1400 let suffix_metadata = MessageMetadata {
1401 role,
1402 status: MessageStatus::Done,
1403 timestamp: suffix.id.0,
1404 };
1405 self.insert_message(suffix.clone(), suffix_metadata.clone(), cx);
1406 self.push_op(
1407 ContextOperation::InsertMessage {
1408 anchor: suffix.clone(),
1409 metadata: suffix_metadata,
1410 version,
1411 },
1412 cx,
1413 );
1414
1415 let new_messages =
1416 if range.start == range.end || range.start == message.offset_range.start {
1417 (None, Some(suffix))
1418 } else {
1419 let mut prefix_end = None;
1420 if range.start > message.offset_range.start
1421 && range.end < message.offset_range.end - 1
1422 {
1423 if self.buffer.read(cx).chars_at(range.start).next() == Some('\n') {
1424 prefix_end = Some(range.start + 1);
1425 } else if self.buffer.read(cx).reversed_chars_at(range.start).next()
1426 == Some('\n')
1427 {
1428 prefix_end = Some(range.start);
1429 }
1430 }
1431
1432 let version = self.version.clone();
1433 let selection = if let Some(prefix_end) = prefix_end {
1434 MessageAnchor {
1435 id: MessageId(self.next_timestamp()),
1436 start: self.buffer.read(cx).anchor_before(prefix_end),
1437 }
1438 } else {
1439 self.buffer.update(cx, |buffer, cx| {
1440 buffer.edit([(range.start..range.start, "\n")], None, cx)
1441 });
1442 edited_buffer = true;
1443 MessageAnchor {
1444 id: MessageId(self.next_timestamp()),
1445 start: self.buffer.read(cx).anchor_before(range.end + 1),
1446 }
1447 };
1448
1449 let selection_metadata = MessageMetadata {
1450 role,
1451 status: MessageStatus::Done,
1452 timestamp: selection.id.0,
1453 };
1454 self.insert_message(selection.clone(), selection_metadata.clone(), cx);
1455 self.push_op(
1456 ContextOperation::InsertMessage {
1457 anchor: selection.clone(),
1458 metadata: selection_metadata,
1459 version,
1460 },
1461 cx,
1462 );
1463
1464 (Some(selection), Some(suffix))
1465 };
1466
1467 if !edited_buffer {
1468 cx.emit(ContextEvent::MessagesEdited);
1469 }
1470 new_messages
1471 } else {
1472 (None, None)
1473 }
1474 }
1475
1476 fn insert_message(
1477 &mut self,
1478 new_anchor: MessageAnchor,
1479 new_metadata: MessageMetadata,
1480 cx: &mut ModelContext<Self>,
1481 ) {
1482 cx.emit(ContextEvent::MessagesEdited);
1483
1484 self.messages_metadata.insert(new_anchor.id, new_metadata);
1485
1486 let buffer = self.buffer.read(cx);
1487 let insertion_ix = self
1488 .message_anchors
1489 .iter()
1490 .position(|anchor| {
1491 let comparison = new_anchor.start.cmp(&anchor.start, buffer);
1492 comparison.is_lt() || (comparison.is_eq() && new_anchor.id > anchor.id)
1493 })
1494 .unwrap_or(self.message_anchors.len());
1495 self.message_anchors.insert(insertion_ix, new_anchor);
1496 }
1497
1498 fn summarize(&mut self, cx: &mut ModelContext<Self>) {
1499 if self.message_anchors.len() >= 2 && self.summary.is_none() {
1500 if !CompletionProvider::global(cx).is_authenticated() {
1501 return;
1502 }
1503
1504 let messages = self
1505 .messages(cx)
1506 .map(|message| message.to_request_message(self.buffer.read(cx)))
1507 .chain(Some(LanguageModelRequestMessage {
1508 role: Role::User,
1509 content: "Summarize the context into a short title without punctuation.".into(),
1510 }));
1511 let request = LanguageModelRequest {
1512 model: CompletionProvider::global(cx).model(),
1513 messages: messages.collect(),
1514 stop: vec![],
1515 temperature: 1.0,
1516 };
1517
1518 let stream = CompletionProvider::global(cx).complete(request, cx);
1519 self.pending_summary = cx.spawn(|this, mut cx| {
1520 async move {
1521 let mut messages = stream.await.inner.await?;
1522
1523 while let Some(message) = messages.next().await {
1524 let text = message?;
1525 let mut lines = text.lines();
1526 this.update(&mut cx, |this, cx| {
1527 let version = this.version.clone();
1528 let timestamp = this.next_timestamp();
1529 let summary = this.summary.get_or_insert(Default::default());
1530 summary.text.extend(lines.next());
1531 summary.timestamp = timestamp;
1532 let operation = ContextOperation::UpdateSummary {
1533 summary: summary.clone(),
1534 version,
1535 };
1536 this.push_op(operation, cx);
1537 cx.emit(ContextEvent::SummaryChanged);
1538 })?;
1539
1540 // Stop if the LLM generated multiple lines.
1541 if lines.next().is_some() {
1542 break;
1543 }
1544 }
1545
1546 this.update(&mut cx, |this, cx| {
1547 let version = this.version.clone();
1548 let timestamp = this.next_timestamp();
1549 if let Some(summary) = this.summary.as_mut() {
1550 summary.done = true;
1551 summary.timestamp = timestamp;
1552 let operation = ContextOperation::UpdateSummary {
1553 summary: summary.clone(),
1554 version,
1555 };
1556 this.push_op(operation, cx);
1557 cx.emit(ContextEvent::SummaryChanged);
1558 }
1559 })?;
1560
1561 anyhow::Ok(())
1562 }
1563 .log_err()
1564 });
1565 }
1566 }
1567
1568 fn message_for_offset(&self, offset: usize, cx: &AppContext) -> Option<Message> {
1569 self.messages_for_offsets([offset], cx).pop()
1570 }
1571
1572 pub fn messages_for_offsets(
1573 &self,
1574 offsets: impl IntoIterator<Item = usize>,
1575 cx: &AppContext,
1576 ) -> Vec<Message> {
1577 let mut result = Vec::new();
1578
1579 let mut messages = self.messages(cx).peekable();
1580 let mut offsets = offsets.into_iter().peekable();
1581 let mut current_message = messages.next();
1582 while let Some(offset) = offsets.next() {
1583 // Locate the message that contains the offset.
1584 while current_message.as_ref().map_or(false, |message| {
1585 !message.offset_range.contains(&offset) && messages.peek().is_some()
1586 }) {
1587 current_message = messages.next();
1588 }
1589 let Some(message) = current_message.as_ref() else {
1590 break;
1591 };
1592
1593 // Skip offsets that are in the same message.
1594 while offsets.peek().map_or(false, |offset| {
1595 message.offset_range.contains(offset) || messages.peek().is_none()
1596 }) {
1597 offsets.next();
1598 }
1599
1600 result.push(message.clone());
1601 }
1602 result
1603 }
1604
1605 pub fn messages<'a>(&'a self, cx: &'a AppContext) -> impl 'a + Iterator<Item = Message> {
1606 let buffer = self.buffer.read(cx);
1607 let mut message_anchors = self.message_anchors.iter().enumerate().peekable();
1608 iter::from_fn(move || {
1609 if let Some((start_ix, message_anchor)) = message_anchors.next() {
1610 let metadata = self.messages_metadata.get(&message_anchor.id)?;
1611 let message_start = message_anchor.start.to_offset(buffer);
1612 let mut message_end = None;
1613 let mut end_ix = start_ix;
1614 while let Some((_, next_message)) = message_anchors.peek() {
1615 if next_message.start.is_valid(buffer) {
1616 message_end = Some(next_message.start);
1617 break;
1618 } else {
1619 end_ix += 1;
1620 message_anchors.next();
1621 }
1622 }
1623 let message_end = message_end
1624 .unwrap_or(language::Anchor::MAX)
1625 .to_offset(buffer);
1626
1627 return Some(Message {
1628 index_range: start_ix..end_ix,
1629 offset_range: message_start..message_end,
1630 id: message_anchor.id,
1631 anchor: message_anchor.start,
1632 role: metadata.role,
1633 status: metadata.status.clone(),
1634 });
1635 }
1636 None
1637 })
1638 }
1639
1640 pub fn save(
1641 &mut self,
1642 debounce: Option<Duration>,
1643 fs: Arc<dyn Fs>,
1644 cx: &mut ModelContext<Context>,
1645 ) {
1646 if self.replica_id() != ReplicaId::default() {
1647 // Prevent saving a remote context for now.
1648 return;
1649 }
1650
1651 self.pending_save = cx.spawn(|this, mut cx| async move {
1652 if let Some(debounce) = debounce {
1653 cx.background_executor().timer(debounce).await;
1654 }
1655
1656 let (old_path, summary) = this.read_with(&cx, |this, _| {
1657 let path = this.path.clone();
1658 let summary = if let Some(summary) = this.summary.as_ref() {
1659 if summary.done {
1660 Some(summary.text.clone())
1661 } else {
1662 None
1663 }
1664 } else {
1665 None
1666 };
1667 (path, summary)
1668 })?;
1669
1670 if let Some(summary) = summary {
1671 let context = this.read_with(&cx, |this, cx| this.serialize(cx))?;
1672 let path = if let Some(old_path) = old_path {
1673 old_path
1674 } else {
1675 let mut discriminant = 1;
1676 let mut new_path;
1677 loop {
1678 new_path = contexts_dir().join(&format!(
1679 "{} - {}.zed.json",
1680 summary.trim(),
1681 discriminant
1682 ));
1683 if fs.is_file(&new_path).await {
1684 discriminant += 1;
1685 } else {
1686 break;
1687 }
1688 }
1689 new_path
1690 };
1691
1692 fs.create_dir(contexts_dir().as_ref()).await?;
1693 fs.atomic_write(path.clone(), serde_json::to_string(&context).unwrap())
1694 .await?;
1695 this.update(&mut cx, |this, _| this.path = Some(path))?;
1696 }
1697
1698 Ok(())
1699 });
1700 }
1701}
1702
1703#[derive(Debug, Default)]
1704pub struct ContextVersion {
1705 context: clock::Global,
1706 buffer: clock::Global,
1707}
1708
1709impl ContextVersion {
1710 pub fn from_proto(proto: &proto::ContextVersion) -> Self {
1711 Self {
1712 context: language::proto::deserialize_version(&proto.context_version),
1713 buffer: language::proto::deserialize_version(&proto.buffer_version),
1714 }
1715 }
1716
1717 pub fn to_proto(&self, context_id: ContextId) -> proto::ContextVersion {
1718 proto::ContextVersion {
1719 context_id: context_id.to_proto(),
1720 context_version: language::proto::serialize_version(&self.context),
1721 buffer_version: language::proto::serialize_version(&self.buffer),
1722 }
1723 }
1724}
1725
1726#[derive(Debug)]
1727enum EditParsingState {
1728 None,
1729 InOldText {
1730 path: PathBuf,
1731 start_offset: usize,
1732 old_text_start_offset: usize,
1733 },
1734 InNewText {
1735 path: PathBuf,
1736 start_offset: usize,
1737 old_text_range: Range<usize>,
1738 new_text_start_offset: usize,
1739 },
1740}
1741
1742#[derive(Clone, Debug, PartialEq)]
1743pub struct EditSuggestion {
1744 pub source_range: Range<language::Anchor>,
1745 pub full_path: PathBuf,
1746}
1747
1748pub struct ParsedEditSuggestion {
1749 pub path: PathBuf,
1750 pub outer_range: Range<usize>,
1751 pub old_text_range: Range<usize>,
1752 pub new_text_range: Range<usize>,
1753}
1754
1755pub fn parse_next_edit_suggestion(lines: &mut rope::Lines) -> Option<ParsedEditSuggestion> {
1756 let mut state = EditParsingState::None;
1757 loop {
1758 let offset = lines.offset();
1759 let message_line = lines.next()?;
1760 match state {
1761 EditParsingState::None => {
1762 if let Some(rest) = message_line.strip_prefix("```edit ") {
1763 let path = rest.trim();
1764 if !path.is_empty() {
1765 state = EditParsingState::InOldText {
1766 path: PathBuf::from(path),
1767 start_offset: offset,
1768 old_text_start_offset: lines.offset(),
1769 };
1770 }
1771 }
1772 }
1773 EditParsingState::InOldText {
1774 path,
1775 start_offset,
1776 old_text_start_offset,
1777 } => {
1778 if message_line == "---" {
1779 state = EditParsingState::InNewText {
1780 path,
1781 start_offset,
1782 old_text_range: old_text_start_offset..offset,
1783 new_text_start_offset: lines.offset(),
1784 };
1785 } else {
1786 state = EditParsingState::InOldText {
1787 path,
1788 start_offset,
1789 old_text_start_offset,
1790 };
1791 }
1792 }
1793 EditParsingState::InNewText {
1794 path,
1795 start_offset,
1796 old_text_range,
1797 new_text_start_offset,
1798 } => {
1799 if message_line == "```" {
1800 return Some(ParsedEditSuggestion {
1801 path,
1802 outer_range: start_offset..offset + "```".len(),
1803 old_text_range,
1804 new_text_range: new_text_start_offset..offset,
1805 });
1806 } else {
1807 state = EditParsingState::InNewText {
1808 path,
1809 start_offset,
1810 old_text_range,
1811 new_text_start_offset,
1812 };
1813 }
1814 }
1815 }
1816 }
1817}
1818
1819#[derive(Clone)]
1820pub struct PendingSlashCommand {
1821 pub name: String,
1822 pub argument: Option<String>,
1823 pub status: PendingSlashCommandStatus,
1824 pub source_range: Range<language::Anchor>,
1825}
1826
1827#[derive(Clone)]
1828pub enum PendingSlashCommandStatus {
1829 Idle,
1830 Running { _task: Shared<Task<()>> },
1831 Error(String),
1832}
1833
1834#[derive(Serialize, Deserialize)]
1835pub struct SavedMessage {
1836 pub id: MessageId,
1837 pub start: usize,
1838 pub metadata: MessageMetadata,
1839}
1840
1841#[derive(Serialize, Deserialize)]
1842pub struct SavedContext {
1843 pub id: Option<ContextId>,
1844 pub zed: String,
1845 pub version: String,
1846 pub text: String,
1847 pub messages: Vec<SavedMessage>,
1848 pub summary: String,
1849 pub slash_command_output_sections:
1850 Vec<assistant_slash_command::SlashCommandOutputSection<usize>>,
1851}
1852
1853impl SavedContext {
1854 pub const VERSION: &'static str = "0.4.0";
1855
1856 pub fn from_json(json: &str) -> Result<Self> {
1857 let saved_context_json = serde_json::from_str::<serde_json::Value>(json)?;
1858 match saved_context_json
1859 .get("version")
1860 .ok_or_else(|| anyhow!("version not found"))?
1861 {
1862 serde_json::Value::String(version) => match version.as_str() {
1863 SavedContext::VERSION => {
1864 Ok(serde_json::from_value::<SavedContext>(saved_context_json)?)
1865 }
1866 SavedContextV0_3_0::VERSION => {
1867 let saved_context =
1868 serde_json::from_value::<SavedContextV0_3_0>(saved_context_json)?;
1869 Ok(saved_context.upgrade())
1870 }
1871 SavedContextV0_2_0::VERSION => {
1872 let saved_context =
1873 serde_json::from_value::<SavedContextV0_2_0>(saved_context_json)?;
1874 Ok(saved_context.upgrade())
1875 }
1876 SavedContextV0_1_0::VERSION => {
1877 let saved_context =
1878 serde_json::from_value::<SavedContextV0_1_0>(saved_context_json)?;
1879 Ok(saved_context.upgrade())
1880 }
1881 _ => Err(anyhow!("unrecognized saved context version: {}", version)),
1882 },
1883 _ => Err(anyhow!("version not found on saved context")),
1884 }
1885 }
1886
1887 fn into_ops(
1888 self,
1889 buffer: &Model<Buffer>,
1890 cx: &mut ModelContext<Context>,
1891 ) -> Vec<ContextOperation> {
1892 let mut operations = Vec::new();
1893 let mut version = clock::Global::new();
1894 let mut next_timestamp = clock::Lamport::new(ReplicaId::default());
1895
1896 let mut first_message_metadata = None;
1897 for message in self.messages {
1898 if message.id == MessageId(clock::Lamport::default()) {
1899 first_message_metadata = Some(message.metadata);
1900 } else {
1901 operations.push(ContextOperation::InsertMessage {
1902 anchor: MessageAnchor {
1903 id: message.id,
1904 start: buffer.read(cx).anchor_before(message.start),
1905 },
1906 metadata: MessageMetadata {
1907 role: message.metadata.role,
1908 status: message.metadata.status,
1909 timestamp: message.metadata.timestamp,
1910 },
1911 version: version.clone(),
1912 });
1913 version.observe(message.id.0);
1914 next_timestamp.observe(message.id.0);
1915 }
1916 }
1917
1918 if let Some(metadata) = first_message_metadata {
1919 let timestamp = next_timestamp.tick();
1920 operations.push(ContextOperation::UpdateMessage {
1921 message_id: MessageId(clock::Lamport::default()),
1922 metadata: MessageMetadata {
1923 role: metadata.role,
1924 status: metadata.status,
1925 timestamp,
1926 },
1927 version: version.clone(),
1928 });
1929 version.observe(timestamp);
1930 }
1931
1932 let timestamp = next_timestamp.tick();
1933 operations.push(ContextOperation::SlashCommandFinished {
1934 id: SlashCommandId(timestamp),
1935 output_range: language::Anchor::MIN..language::Anchor::MAX,
1936 sections: self
1937 .slash_command_output_sections
1938 .into_iter()
1939 .map(|section| {
1940 let buffer = buffer.read(cx);
1941 SlashCommandOutputSection {
1942 range: buffer.anchor_after(section.range.start)
1943 ..buffer.anchor_before(section.range.end),
1944 icon: section.icon,
1945 label: section.label,
1946 }
1947 })
1948 .collect(),
1949 version: version.clone(),
1950 });
1951 version.observe(timestamp);
1952
1953 let timestamp = next_timestamp.tick();
1954 operations.push(ContextOperation::UpdateSummary {
1955 summary: ContextSummary {
1956 text: self.summary,
1957 done: true,
1958 timestamp,
1959 },
1960 version: version.clone(),
1961 });
1962 version.observe(timestamp);
1963
1964 operations
1965 }
1966}
1967
1968#[derive(Copy, Clone, Debug, Eq, PartialEq, PartialOrd, Ord, Hash, Serialize, Deserialize)]
1969struct SavedMessageIdPreV0_4_0(usize);
1970
1971#[derive(Serialize, Deserialize)]
1972struct SavedMessagePreV0_4_0 {
1973 id: SavedMessageIdPreV0_4_0,
1974 start: usize,
1975}
1976
1977#[derive(Clone, Debug, Eq, PartialEq, Serialize, Deserialize)]
1978struct SavedMessageMetadataPreV0_4_0 {
1979 role: Role,
1980 status: MessageStatus,
1981}
1982
1983#[derive(Serialize, Deserialize)]
1984struct SavedContextV0_3_0 {
1985 id: Option<ContextId>,
1986 zed: String,
1987 version: String,
1988 text: String,
1989 messages: Vec<SavedMessagePreV0_4_0>,
1990 message_metadata: HashMap<SavedMessageIdPreV0_4_0, SavedMessageMetadataPreV0_4_0>,
1991 summary: String,
1992 slash_command_output_sections: Vec<assistant_slash_command::SlashCommandOutputSection<usize>>,
1993}
1994
1995impl SavedContextV0_3_0 {
1996 const VERSION: &'static str = "0.3.0";
1997
1998 fn upgrade(self) -> SavedContext {
1999 SavedContext {
2000 id: self.id,
2001 zed: self.zed,
2002 version: SavedContext::VERSION.into(),
2003 text: self.text,
2004 messages: self
2005 .messages
2006 .into_iter()
2007 .filter_map(|message| {
2008 let metadata = self.message_metadata.get(&message.id)?;
2009 let timestamp = clock::Lamport {
2010 replica_id: ReplicaId::default(),
2011 value: message.id.0 as u32,
2012 };
2013 Some(SavedMessage {
2014 id: MessageId(timestamp),
2015 start: message.start,
2016 metadata: MessageMetadata {
2017 role: metadata.role,
2018 status: metadata.status.clone(),
2019 timestamp,
2020 },
2021 })
2022 })
2023 .collect(),
2024 summary: self.summary,
2025 slash_command_output_sections: self.slash_command_output_sections,
2026 }
2027 }
2028}
2029
2030#[derive(Serialize, Deserialize)]
2031struct SavedContextV0_2_0 {
2032 id: Option<ContextId>,
2033 zed: String,
2034 version: String,
2035 text: String,
2036 messages: Vec<SavedMessagePreV0_4_0>,
2037 message_metadata: HashMap<SavedMessageIdPreV0_4_0, SavedMessageMetadataPreV0_4_0>,
2038 summary: String,
2039}
2040
2041impl SavedContextV0_2_0 {
2042 const VERSION: &'static str = "0.2.0";
2043
2044 fn upgrade(self) -> SavedContext {
2045 SavedContextV0_3_0 {
2046 id: self.id,
2047 zed: self.zed,
2048 version: SavedContextV0_3_0::VERSION.to_string(),
2049 text: self.text,
2050 messages: self.messages,
2051 message_metadata: self.message_metadata,
2052 summary: self.summary,
2053 slash_command_output_sections: Vec::new(),
2054 }
2055 .upgrade()
2056 }
2057}
2058
2059#[derive(Serialize, Deserialize)]
2060struct SavedContextV0_1_0 {
2061 id: Option<ContextId>,
2062 zed: String,
2063 version: String,
2064 text: String,
2065 messages: Vec<SavedMessagePreV0_4_0>,
2066 message_metadata: HashMap<SavedMessageIdPreV0_4_0, SavedMessageMetadataPreV0_4_0>,
2067 summary: String,
2068 api_url: Option<String>,
2069 model: OpenAiModel,
2070}
2071
2072impl SavedContextV0_1_0 {
2073 const VERSION: &'static str = "0.1.0";
2074
2075 fn upgrade(self) -> SavedContext {
2076 SavedContextV0_2_0 {
2077 id: self.id,
2078 zed: self.zed,
2079 version: SavedContextV0_2_0::VERSION.to_string(),
2080 text: self.text,
2081 messages: self.messages,
2082 message_metadata: self.message_metadata,
2083 summary: self.summary,
2084 }
2085 .upgrade()
2086 }
2087}
2088
2089#[derive(Clone)]
2090pub struct SavedContextMetadata {
2091 pub title: String,
2092 pub path: PathBuf,
2093 pub mtime: chrono::DateTime<chrono::Local>,
2094}
2095
2096#[cfg(test)]
2097mod tests {
2098 use super::*;
2099 use crate::{
2100 assistant_panel,
2101 slash_command::{active_command, file_command},
2102 FakeCompletionProvider, MessageId,
2103 };
2104 use assistant_slash_command::{ArgumentCompletion, SlashCommand};
2105 use fs::FakeFs;
2106 use gpui::{AppContext, TestAppContext, WeakView};
2107 use language::LspAdapterDelegate;
2108 use parking_lot::Mutex;
2109 use project::Project;
2110 use rand::prelude::*;
2111 use rope::Rope;
2112 use serde_json::json;
2113 use settings::SettingsStore;
2114 use std::{cell::RefCell, env, path::Path, rc::Rc, sync::atomic::AtomicBool};
2115 use text::network::Network;
2116 use ui::WindowContext;
2117 use unindent::Unindent;
2118 use util::{test::marked_text_ranges, RandomCharIter};
2119 use workspace::Workspace;
2120
2121 #[gpui::test]
2122 fn test_inserting_and_removing_messages(cx: &mut AppContext) {
2123 let settings_store = SettingsStore::test(cx);
2124 FakeCompletionProvider::setup_test(cx);
2125 cx.set_global(settings_store);
2126 assistant_panel::init(cx);
2127 let registry = Arc::new(LanguageRegistry::test(cx.background_executor().clone()));
2128
2129 let context = cx.new_model(|cx| Context::local(registry, None, cx));
2130 let buffer = context.read(cx).buffer.clone();
2131
2132 let message_1 = context.read(cx).message_anchors[0].clone();
2133 assert_eq!(
2134 messages(&context, cx),
2135 vec![(message_1.id, Role::User, 0..0)]
2136 );
2137
2138 let message_2 = context.update(cx, |context, cx| {
2139 context
2140 .insert_message_after(message_1.id, Role::Assistant, MessageStatus::Done, cx)
2141 .unwrap()
2142 });
2143 assert_eq!(
2144 messages(&context, cx),
2145 vec![
2146 (message_1.id, Role::User, 0..1),
2147 (message_2.id, Role::Assistant, 1..1)
2148 ]
2149 );
2150
2151 buffer.update(cx, |buffer, cx| {
2152 buffer.edit([(0..0, "1"), (1..1, "2")], None, cx)
2153 });
2154 assert_eq!(
2155 messages(&context, cx),
2156 vec![
2157 (message_1.id, Role::User, 0..2),
2158 (message_2.id, Role::Assistant, 2..3)
2159 ]
2160 );
2161
2162 let message_3 = context.update(cx, |context, cx| {
2163 context
2164 .insert_message_after(message_2.id, Role::User, MessageStatus::Done, cx)
2165 .unwrap()
2166 });
2167 assert_eq!(
2168 messages(&context, cx),
2169 vec![
2170 (message_1.id, Role::User, 0..2),
2171 (message_2.id, Role::Assistant, 2..4),
2172 (message_3.id, Role::User, 4..4)
2173 ]
2174 );
2175
2176 let message_4 = context.update(cx, |context, cx| {
2177 context
2178 .insert_message_after(message_2.id, Role::User, MessageStatus::Done, cx)
2179 .unwrap()
2180 });
2181 assert_eq!(
2182 messages(&context, cx),
2183 vec![
2184 (message_1.id, Role::User, 0..2),
2185 (message_2.id, Role::Assistant, 2..4),
2186 (message_4.id, Role::User, 4..5),
2187 (message_3.id, Role::User, 5..5),
2188 ]
2189 );
2190
2191 buffer.update(cx, |buffer, cx| {
2192 buffer.edit([(4..4, "C"), (5..5, "D")], None, cx)
2193 });
2194 assert_eq!(
2195 messages(&context, cx),
2196 vec![
2197 (message_1.id, Role::User, 0..2),
2198 (message_2.id, Role::Assistant, 2..4),
2199 (message_4.id, Role::User, 4..6),
2200 (message_3.id, Role::User, 6..7),
2201 ]
2202 );
2203
2204 // Deleting across message boundaries merges the messages.
2205 buffer.update(cx, |buffer, cx| buffer.edit([(1..4, "")], None, cx));
2206 assert_eq!(
2207 messages(&context, cx),
2208 vec![
2209 (message_1.id, Role::User, 0..3),
2210 (message_3.id, Role::User, 3..4),
2211 ]
2212 );
2213
2214 // Undoing the deletion should also undo the merge.
2215 buffer.update(cx, |buffer, cx| buffer.undo(cx));
2216 assert_eq!(
2217 messages(&context, cx),
2218 vec![
2219 (message_1.id, Role::User, 0..2),
2220 (message_2.id, Role::Assistant, 2..4),
2221 (message_4.id, Role::User, 4..6),
2222 (message_3.id, Role::User, 6..7),
2223 ]
2224 );
2225
2226 // Redoing the deletion should also redo the merge.
2227 buffer.update(cx, |buffer, cx| buffer.redo(cx));
2228 assert_eq!(
2229 messages(&context, cx),
2230 vec![
2231 (message_1.id, Role::User, 0..3),
2232 (message_3.id, Role::User, 3..4),
2233 ]
2234 );
2235
2236 // Ensure we can still insert after a merged message.
2237 let message_5 = context.update(cx, |context, cx| {
2238 context
2239 .insert_message_after(message_1.id, Role::System, MessageStatus::Done, cx)
2240 .unwrap()
2241 });
2242 assert_eq!(
2243 messages(&context, cx),
2244 vec![
2245 (message_1.id, Role::User, 0..3),
2246 (message_5.id, Role::System, 3..4),
2247 (message_3.id, Role::User, 4..5)
2248 ]
2249 );
2250 }
2251
2252 #[gpui::test]
2253 fn test_message_splitting(cx: &mut AppContext) {
2254 let settings_store = SettingsStore::test(cx);
2255 cx.set_global(settings_store);
2256 FakeCompletionProvider::setup_test(cx);
2257 assistant_panel::init(cx);
2258 let registry = Arc::new(LanguageRegistry::test(cx.background_executor().clone()));
2259
2260 let context = cx.new_model(|cx| Context::local(registry, None, cx));
2261 let buffer = context.read(cx).buffer.clone();
2262
2263 let message_1 = context.read(cx).message_anchors[0].clone();
2264 assert_eq!(
2265 messages(&context, cx),
2266 vec![(message_1.id, Role::User, 0..0)]
2267 );
2268
2269 buffer.update(cx, |buffer, cx| {
2270 buffer.edit([(0..0, "aaa\nbbb\nccc\nddd\n")], None, cx)
2271 });
2272
2273 let (_, message_2) = context.update(cx, |context, cx| context.split_message(3..3, cx));
2274 let message_2 = message_2.unwrap();
2275
2276 // We recycle newlines in the middle of a split message
2277 assert_eq!(buffer.read(cx).text(), "aaa\nbbb\nccc\nddd\n");
2278 assert_eq!(
2279 messages(&context, cx),
2280 vec![
2281 (message_1.id, Role::User, 0..4),
2282 (message_2.id, Role::User, 4..16),
2283 ]
2284 );
2285
2286 let (_, message_3) = context.update(cx, |context, cx| context.split_message(3..3, cx));
2287 let message_3 = message_3.unwrap();
2288
2289 // We don't recycle newlines at the end of a split message
2290 assert_eq!(buffer.read(cx).text(), "aaa\n\nbbb\nccc\nddd\n");
2291 assert_eq!(
2292 messages(&context, cx),
2293 vec![
2294 (message_1.id, Role::User, 0..4),
2295 (message_3.id, Role::User, 4..5),
2296 (message_2.id, Role::User, 5..17),
2297 ]
2298 );
2299
2300 let (_, message_4) = context.update(cx, |context, cx| context.split_message(9..9, cx));
2301 let message_4 = message_4.unwrap();
2302 assert_eq!(buffer.read(cx).text(), "aaa\n\nbbb\nccc\nddd\n");
2303 assert_eq!(
2304 messages(&context, cx),
2305 vec![
2306 (message_1.id, Role::User, 0..4),
2307 (message_3.id, Role::User, 4..5),
2308 (message_2.id, Role::User, 5..9),
2309 (message_4.id, Role::User, 9..17),
2310 ]
2311 );
2312
2313 let (_, message_5) = context.update(cx, |context, cx| context.split_message(9..9, cx));
2314 let message_5 = message_5.unwrap();
2315 assert_eq!(buffer.read(cx).text(), "aaa\n\nbbb\n\nccc\nddd\n");
2316 assert_eq!(
2317 messages(&context, cx),
2318 vec![
2319 (message_1.id, Role::User, 0..4),
2320 (message_3.id, Role::User, 4..5),
2321 (message_2.id, Role::User, 5..9),
2322 (message_4.id, Role::User, 9..10),
2323 (message_5.id, Role::User, 10..18),
2324 ]
2325 );
2326
2327 let (message_6, message_7) =
2328 context.update(cx, |context, cx| context.split_message(14..16, cx));
2329 let message_6 = message_6.unwrap();
2330 let message_7 = message_7.unwrap();
2331 assert_eq!(buffer.read(cx).text(), "aaa\n\nbbb\n\nccc\ndd\nd\n");
2332 assert_eq!(
2333 messages(&context, cx),
2334 vec![
2335 (message_1.id, Role::User, 0..4),
2336 (message_3.id, Role::User, 4..5),
2337 (message_2.id, Role::User, 5..9),
2338 (message_4.id, Role::User, 9..10),
2339 (message_5.id, Role::User, 10..14),
2340 (message_6.id, Role::User, 14..17),
2341 (message_7.id, Role::User, 17..19),
2342 ]
2343 );
2344 }
2345
2346 #[gpui::test]
2347 fn test_messages_for_offsets(cx: &mut AppContext) {
2348 let settings_store = SettingsStore::test(cx);
2349 FakeCompletionProvider::setup_test(cx);
2350 cx.set_global(settings_store);
2351 assistant_panel::init(cx);
2352 let registry = Arc::new(LanguageRegistry::test(cx.background_executor().clone()));
2353 let context = cx.new_model(|cx| Context::local(registry, None, cx));
2354 let buffer = context.read(cx).buffer.clone();
2355
2356 let message_1 = context.read(cx).message_anchors[0].clone();
2357 assert_eq!(
2358 messages(&context, cx),
2359 vec![(message_1.id, Role::User, 0..0)]
2360 );
2361
2362 buffer.update(cx, |buffer, cx| buffer.edit([(0..0, "aaa")], None, cx));
2363 let message_2 = context
2364 .update(cx, |context, cx| {
2365 context.insert_message_after(message_1.id, Role::User, MessageStatus::Done, cx)
2366 })
2367 .unwrap();
2368 buffer.update(cx, |buffer, cx| buffer.edit([(4..4, "bbb")], None, cx));
2369
2370 let message_3 = context
2371 .update(cx, |context, cx| {
2372 context.insert_message_after(message_2.id, Role::User, MessageStatus::Done, cx)
2373 })
2374 .unwrap();
2375 buffer.update(cx, |buffer, cx| buffer.edit([(8..8, "ccc")], None, cx));
2376
2377 assert_eq!(buffer.read(cx).text(), "aaa\nbbb\nccc");
2378 assert_eq!(
2379 messages(&context, cx),
2380 vec![
2381 (message_1.id, Role::User, 0..4),
2382 (message_2.id, Role::User, 4..8),
2383 (message_3.id, Role::User, 8..11)
2384 ]
2385 );
2386
2387 assert_eq!(
2388 message_ids_for_offsets(&context, &[0, 4, 9], cx),
2389 [message_1.id, message_2.id, message_3.id]
2390 );
2391 assert_eq!(
2392 message_ids_for_offsets(&context, &[0, 1, 11], cx),
2393 [message_1.id, message_3.id]
2394 );
2395
2396 let message_4 = context
2397 .update(cx, |context, cx| {
2398 context.insert_message_after(message_3.id, Role::User, MessageStatus::Done, cx)
2399 })
2400 .unwrap();
2401 assert_eq!(buffer.read(cx).text(), "aaa\nbbb\nccc\n");
2402 assert_eq!(
2403 messages(&context, cx),
2404 vec![
2405 (message_1.id, Role::User, 0..4),
2406 (message_2.id, Role::User, 4..8),
2407 (message_3.id, Role::User, 8..12),
2408 (message_4.id, Role::User, 12..12)
2409 ]
2410 );
2411 assert_eq!(
2412 message_ids_for_offsets(&context, &[0, 4, 8, 12], cx),
2413 [message_1.id, message_2.id, message_3.id, message_4.id]
2414 );
2415
2416 fn message_ids_for_offsets(
2417 context: &Model<Context>,
2418 offsets: &[usize],
2419 cx: &AppContext,
2420 ) -> Vec<MessageId> {
2421 context
2422 .read(cx)
2423 .messages_for_offsets(offsets.iter().copied(), cx)
2424 .into_iter()
2425 .map(|message| message.id)
2426 .collect()
2427 }
2428 }
2429
2430 #[gpui::test]
2431 async fn test_slash_commands(cx: &mut TestAppContext) {
2432 let settings_store = cx.update(SettingsStore::test);
2433 cx.set_global(settings_store);
2434 cx.update(FakeCompletionProvider::setup_test);
2435 cx.update(Project::init_settings);
2436 cx.update(assistant_panel::init);
2437 let fs = FakeFs::new(cx.background_executor.clone());
2438
2439 fs.insert_tree(
2440 "/test",
2441 json!({
2442 "src": {
2443 "lib.rs": "fn one() -> usize { 1 }",
2444 "main.rs": "
2445 use crate::one;
2446 fn main() { one(); }
2447 ".unindent(),
2448 }
2449 }),
2450 )
2451 .await;
2452
2453 let slash_command_registry = cx.update(SlashCommandRegistry::default_global);
2454 slash_command_registry.register_command(file_command::FileSlashCommand, false);
2455 slash_command_registry.register_command(active_command::ActiveSlashCommand, false);
2456
2457 let registry = Arc::new(LanguageRegistry::test(cx.executor()));
2458 let context = cx.new_model(|cx| Context::local(registry.clone(), None, cx));
2459
2460 let output_ranges = Rc::new(RefCell::new(HashSet::default()));
2461 context.update(cx, |_, cx| {
2462 cx.subscribe(&context, {
2463 let ranges = output_ranges.clone();
2464 move |_, _, event, _| match event {
2465 ContextEvent::PendingSlashCommandsUpdated { removed, updated } => {
2466 for range in removed {
2467 ranges.borrow_mut().remove(range);
2468 }
2469 for command in updated {
2470 ranges.borrow_mut().insert(command.source_range.clone());
2471 }
2472 }
2473 _ => {}
2474 }
2475 })
2476 .detach();
2477 });
2478
2479 let buffer = context.read_with(cx, |context, _| context.buffer.clone());
2480
2481 // Insert a slash command
2482 buffer.update(cx, |buffer, cx| {
2483 buffer.edit([(0..0, "/file src/lib.rs")], None, cx);
2484 });
2485 assert_text_and_output_ranges(
2486 &buffer,
2487 &output_ranges.borrow(),
2488 "
2489 «/file src/lib.rs»
2490 "
2491 .unindent()
2492 .trim_end(),
2493 cx,
2494 );
2495
2496 // Edit the argument of the slash command.
2497 buffer.update(cx, |buffer, cx| {
2498 let edit_offset = buffer.text().find("lib.rs").unwrap();
2499 buffer.edit([(edit_offset..edit_offset + "lib".len(), "main")], None, cx);
2500 });
2501 assert_text_and_output_ranges(
2502 &buffer,
2503 &output_ranges.borrow(),
2504 "
2505 «/file src/main.rs»
2506 "
2507 .unindent()
2508 .trim_end(),
2509 cx,
2510 );
2511
2512 // Edit the name of the slash command, using one that doesn't exist.
2513 buffer.update(cx, |buffer, cx| {
2514 let edit_offset = buffer.text().find("/file").unwrap();
2515 buffer.edit(
2516 [(edit_offset..edit_offset + "/file".len(), "/unknown")],
2517 None,
2518 cx,
2519 );
2520 });
2521 assert_text_and_output_ranges(
2522 &buffer,
2523 &output_ranges.borrow(),
2524 "
2525 /unknown src/main.rs
2526 "
2527 .unindent()
2528 .trim_end(),
2529 cx,
2530 );
2531
2532 #[track_caller]
2533 fn assert_text_and_output_ranges(
2534 buffer: &Model<Buffer>,
2535 ranges: &HashSet<Range<language::Anchor>>,
2536 expected_marked_text: &str,
2537 cx: &mut TestAppContext,
2538 ) {
2539 let (expected_text, expected_ranges) = marked_text_ranges(expected_marked_text, false);
2540 let (actual_text, actual_ranges) = buffer.update(cx, |buffer, _| {
2541 let mut ranges = ranges
2542 .iter()
2543 .map(|range| range.to_offset(buffer))
2544 .collect::<Vec<_>>();
2545 ranges.sort_by_key(|a| a.start);
2546 (buffer.text(), ranges)
2547 });
2548
2549 assert_eq!(actual_text, expected_text);
2550 assert_eq!(actual_ranges, expected_ranges);
2551 }
2552 }
2553
2554 #[test]
2555 fn test_parse_next_edit_suggestion() {
2556 let text = "
2557 some output:
2558
2559 ```edit src/foo.rs
2560 let a = 1;
2561 let b = 2;
2562 ---
2563 let w = 1;
2564 let x = 2;
2565 let y = 3;
2566 let z = 4;
2567 ```
2568
2569 some more output:
2570
2571 ```edit src/foo.rs
2572 let c = 1;
2573 ---
2574 ```
2575
2576 and the conclusion.
2577 "
2578 .unindent();
2579
2580 let rope = Rope::from(text.as_str());
2581 let mut lines = rope.chunks().lines();
2582 let mut suggestions = vec![];
2583 while let Some(suggestion) = parse_next_edit_suggestion(&mut lines) {
2584 suggestions.push((
2585 suggestion.path.clone(),
2586 text[suggestion.old_text_range].to_string(),
2587 text[suggestion.new_text_range].to_string(),
2588 ));
2589 }
2590
2591 assert_eq!(
2592 suggestions,
2593 vec![
2594 (
2595 Path::new("src/foo.rs").into(),
2596 [
2597 " let a = 1;", //
2598 " let b = 2;",
2599 "",
2600 ]
2601 .join("\n"),
2602 [
2603 " let w = 1;",
2604 " let x = 2;",
2605 " let y = 3;",
2606 " let z = 4;",
2607 "",
2608 ]
2609 .join("\n"),
2610 ),
2611 (
2612 Path::new("src/foo.rs").into(),
2613 [
2614 " let c = 1;", //
2615 "",
2616 ]
2617 .join("\n"),
2618 String::new(),
2619 )
2620 ]
2621 );
2622 }
2623
2624 #[gpui::test]
2625 async fn test_serialization(cx: &mut TestAppContext) {
2626 let settings_store = cx.update(SettingsStore::test);
2627 cx.set_global(settings_store);
2628 cx.update(FakeCompletionProvider::setup_test);
2629 cx.update(assistant_panel::init);
2630 let registry = Arc::new(LanguageRegistry::test(cx.executor()));
2631 let context = cx.new_model(|cx| Context::local(registry.clone(), None, cx));
2632 let buffer = context.read_with(cx, |context, _| context.buffer.clone());
2633 let message_0 = context.read_with(cx, |context, _| context.message_anchors[0].id);
2634 let message_1 = context.update(cx, |context, cx| {
2635 context
2636 .insert_message_after(message_0, Role::Assistant, MessageStatus::Done, cx)
2637 .unwrap()
2638 });
2639 let message_2 = context.update(cx, |context, cx| {
2640 context
2641 .insert_message_after(message_1.id, Role::System, MessageStatus::Done, cx)
2642 .unwrap()
2643 });
2644 buffer.update(cx, |buffer, cx| {
2645 buffer.edit([(0..0, "a"), (1..1, "b\nc")], None, cx);
2646 buffer.finalize_last_transaction();
2647 });
2648 let _message_3 = context.update(cx, |context, cx| {
2649 context
2650 .insert_message_after(message_2.id, Role::System, MessageStatus::Done, cx)
2651 .unwrap()
2652 });
2653 buffer.update(cx, |buffer, cx| buffer.undo(cx));
2654 assert_eq!(buffer.read_with(cx, |buffer, _| buffer.text()), "a\nb\nc\n");
2655 assert_eq!(
2656 cx.read(|cx| messages(&context, cx)),
2657 [
2658 (message_0, Role::User, 0..2),
2659 (message_1.id, Role::Assistant, 2..6),
2660 (message_2.id, Role::System, 6..6),
2661 ]
2662 );
2663
2664 let serialized_context = context.read_with(cx, |context, cx| context.serialize(cx));
2665 let deserialized_context = cx.new_model(|cx| {
2666 Context::deserialize(
2667 serialized_context,
2668 Default::default(),
2669 registry.clone(),
2670 None,
2671 cx,
2672 )
2673 });
2674 let deserialized_buffer =
2675 deserialized_context.read_with(cx, |context, _| context.buffer.clone());
2676 assert_eq!(
2677 deserialized_buffer.read_with(cx, |buffer, _| buffer.text()),
2678 "a\nb\nc\n"
2679 );
2680 assert_eq!(
2681 cx.read(|cx| messages(&deserialized_context, cx)),
2682 [
2683 (message_0, Role::User, 0..2),
2684 (message_1.id, Role::Assistant, 2..6),
2685 (message_2.id, Role::System, 6..6),
2686 ]
2687 );
2688 }
2689
2690 #[gpui::test(iterations = 100)]
2691 async fn test_random_context_collaboration(cx: &mut TestAppContext, mut rng: StdRng) {
2692 let min_peers = env::var("MIN_PEERS")
2693 .map(|i| i.parse().expect("invalid `MIN_PEERS` variable"))
2694 .unwrap_or(2);
2695 let max_peers = env::var("MAX_PEERS")
2696 .map(|i| i.parse().expect("invalid `MAX_PEERS` variable"))
2697 .unwrap_or(5);
2698 let operations = env::var("OPERATIONS")
2699 .map(|i| i.parse().expect("invalid `OPERATIONS` variable"))
2700 .unwrap_or(50);
2701
2702 let settings_store = cx.update(SettingsStore::test);
2703 cx.set_global(settings_store);
2704 cx.update(FakeCompletionProvider::setup_test);
2705 cx.update(assistant_panel::init);
2706 let slash_commands = cx.update(SlashCommandRegistry::default_global);
2707 slash_commands.register_command(FakeSlashCommand("cmd-1".into()), false);
2708 slash_commands.register_command(FakeSlashCommand("cmd-2".into()), false);
2709 slash_commands.register_command(FakeSlashCommand("cmd-3".into()), false);
2710
2711 let registry = Arc::new(LanguageRegistry::test(cx.background_executor.clone()));
2712 let network = Arc::new(Mutex::new(Network::new(rng.clone())));
2713 let mut contexts = Vec::new();
2714
2715 let num_peers = rng.gen_range(min_peers..=max_peers);
2716 let context_id = ContextId::new();
2717 for i in 0..num_peers {
2718 let context = cx.new_model(|cx| {
2719 Context::new(
2720 context_id.clone(),
2721 i as ReplicaId,
2722 language::Capability::ReadWrite,
2723 registry.clone(),
2724 None,
2725 cx,
2726 )
2727 });
2728
2729 cx.update(|cx| {
2730 cx.subscribe(&context, {
2731 let network = network.clone();
2732 move |_, event, _| {
2733 if let ContextEvent::Operation(op) = event {
2734 network
2735 .lock()
2736 .broadcast(i as ReplicaId, vec![op.to_proto()]);
2737 }
2738 }
2739 })
2740 .detach();
2741 });
2742
2743 contexts.push(context);
2744 network.lock().add_peer(i as ReplicaId);
2745 }
2746
2747 let mut mutation_count = operations;
2748
2749 while mutation_count > 0
2750 || !network.lock().is_idle()
2751 || network.lock().contains_disconnected_peers()
2752 {
2753 let context_index = rng.gen_range(0..contexts.len());
2754 let context = &contexts[context_index];
2755
2756 match rng.gen_range(0..100) {
2757 0..=29 if mutation_count > 0 => {
2758 log::info!("Context {}: edit buffer", context_index);
2759 context.update(cx, |context, cx| {
2760 context
2761 .buffer
2762 .update(cx, |buffer, cx| buffer.randomly_edit(&mut rng, 1, cx));
2763 });
2764 mutation_count -= 1;
2765 }
2766 30..=44 if mutation_count > 0 => {
2767 context.update(cx, |context, cx| {
2768 let range = context.buffer.read(cx).random_byte_range(0, &mut rng);
2769 log::info!("Context {}: split message at {:?}", context_index, range);
2770 context.split_message(range, cx);
2771 });
2772 mutation_count -= 1;
2773 }
2774 45..=59 if mutation_count > 0 => {
2775 context.update(cx, |context, cx| {
2776 if let Some(message) = context.messages(cx).choose(&mut rng) {
2777 let role = *[Role::User, Role::Assistant, Role::System]
2778 .choose(&mut rng)
2779 .unwrap();
2780 log::info!(
2781 "Context {}: insert message after {:?} with {:?}",
2782 context_index,
2783 message.id,
2784 role
2785 );
2786 context.insert_message_after(message.id, role, MessageStatus::Done, cx);
2787 }
2788 });
2789 mutation_count -= 1;
2790 }
2791 60..=74 if mutation_count > 0 => {
2792 context.update(cx, |context, cx| {
2793 let command_text = "/".to_string()
2794 + slash_commands
2795 .command_names()
2796 .choose(&mut rng)
2797 .unwrap()
2798 .clone()
2799 .as_ref();
2800
2801 let command_range = context.buffer.update(cx, |buffer, cx| {
2802 let offset = buffer.random_byte_range(0, &mut rng).start;
2803 buffer.edit(
2804 [(offset..offset, format!("\n{}\n", command_text))],
2805 None,
2806 cx,
2807 );
2808 offset + 1..offset + 1 + command_text.len()
2809 });
2810
2811 let output_len = rng.gen_range(1..=10);
2812 let output_text = RandomCharIter::new(&mut rng)
2813 .filter(|c| *c != '\r')
2814 .take(output_len)
2815 .collect::<String>();
2816
2817 let num_sections = rng.gen_range(0..=3);
2818 let mut sections = Vec::with_capacity(num_sections);
2819 for _ in 0..num_sections {
2820 let section_start = rng.gen_range(0..output_len);
2821 let section_end = rng.gen_range(section_start..=output_len);
2822 sections.push(SlashCommandOutputSection {
2823 range: section_start..section_end,
2824 icon: ui::IconName::Ai,
2825 label: "section".into(),
2826 });
2827 }
2828
2829 log::info!(
2830 "Context {}: insert slash command output at {:?} with {:?}",
2831 context_index,
2832 command_range,
2833 sections
2834 );
2835
2836 let command_range =
2837 context.buffer.read(cx).anchor_after(command_range.start)
2838 ..context.buffer.read(cx).anchor_after(command_range.end);
2839 context.insert_command_output(
2840 command_range,
2841 Task::ready(Ok(SlashCommandOutput {
2842 text: output_text,
2843 sections,
2844 run_commands_in_text: false,
2845 })),
2846 true,
2847 cx,
2848 );
2849 });
2850 cx.run_until_parked();
2851 mutation_count -= 1;
2852 }
2853 75..=84 if mutation_count > 0 => {
2854 context.update(cx, |context, cx| {
2855 if let Some(message) = context.messages(cx).choose(&mut rng) {
2856 let new_status = match rng.gen_range(0..3) {
2857 0 => MessageStatus::Done,
2858 1 => MessageStatus::Pending,
2859 _ => MessageStatus::Error(SharedString::from("Random error")),
2860 };
2861 log::info!(
2862 "Context {}: update message {:?} status to {:?}",
2863 context_index,
2864 message.id,
2865 new_status
2866 );
2867 context.update_metadata(message.id, cx, |metadata| {
2868 metadata.status = new_status;
2869 });
2870 }
2871 });
2872 mutation_count -= 1;
2873 }
2874 _ => {
2875 let replica_id = context_index as ReplicaId;
2876 if network.lock().is_disconnected(replica_id) {
2877 network.lock().reconnect_peer(replica_id, 0);
2878
2879 let (ops_to_send, ops_to_receive) = cx.read(|cx| {
2880 let host_context = &contexts[0].read(cx);
2881 let guest_context = context.read(cx);
2882 (
2883 guest_context.serialize_ops(&host_context.version(cx), cx),
2884 host_context.serialize_ops(&guest_context.version(cx), cx),
2885 )
2886 });
2887 let ops_to_send = ops_to_send.await;
2888 let ops_to_receive = ops_to_receive
2889 .await
2890 .into_iter()
2891 .map(ContextOperation::from_proto)
2892 .collect::<Result<Vec<_>>>()
2893 .unwrap();
2894 log::info!(
2895 "Context {}: reconnecting. Sent {} operations, received {} operations",
2896 context_index,
2897 ops_to_send.len(),
2898 ops_to_receive.len()
2899 );
2900
2901 network.lock().broadcast(replica_id, ops_to_send);
2902 context
2903 .update(cx, |context, cx| context.apply_ops(ops_to_receive, cx))
2904 .unwrap();
2905 } else if rng.gen_bool(0.1) && replica_id != 0 {
2906 log::info!("Context {}: disconnecting", context_index);
2907 network.lock().disconnect_peer(replica_id);
2908 } else if network.lock().has_unreceived(replica_id) {
2909 log::info!("Context {}: applying operations", context_index);
2910 let ops = network.lock().receive(replica_id);
2911 let ops = ops
2912 .into_iter()
2913 .map(ContextOperation::from_proto)
2914 .collect::<Result<Vec<_>>>()
2915 .unwrap();
2916 context
2917 .update(cx, |context, cx| context.apply_ops(ops, cx))
2918 .unwrap();
2919 }
2920 }
2921 }
2922 }
2923
2924 cx.read(|cx| {
2925 let first_context = contexts[0].read(cx);
2926 for context in &contexts[1..] {
2927 let context = context.read(cx);
2928 assert!(context.pending_ops.is_empty());
2929 assert_eq!(
2930 context.buffer.read(cx).text(),
2931 first_context.buffer.read(cx).text(),
2932 "Context {} text != Context 0 text",
2933 context.buffer.read(cx).replica_id()
2934 );
2935 assert_eq!(
2936 context.message_anchors,
2937 first_context.message_anchors,
2938 "Context {} messages != Context 0 messages",
2939 context.buffer.read(cx).replica_id()
2940 );
2941 assert_eq!(
2942 context.messages_metadata,
2943 first_context.messages_metadata,
2944 "Context {} message metadata != Context 0 message metadata",
2945 context.buffer.read(cx).replica_id()
2946 );
2947 assert_eq!(
2948 context.slash_command_output_sections,
2949 first_context.slash_command_output_sections,
2950 "Context {} slash command output sections != Context 0 slash command output sections",
2951 context.buffer.read(cx).replica_id()
2952 );
2953 }
2954 });
2955 }
2956
2957 fn messages(context: &Model<Context>, cx: &AppContext) -> Vec<(MessageId, Role, Range<usize>)> {
2958 context
2959 .read(cx)
2960 .messages(cx)
2961 .map(|message| (message.id, message.role, message.offset_range))
2962 .collect()
2963 }
2964
2965 #[derive(Clone)]
2966 struct FakeSlashCommand(String);
2967
2968 impl SlashCommand for FakeSlashCommand {
2969 fn name(&self) -> String {
2970 self.0.clone()
2971 }
2972
2973 fn description(&self) -> String {
2974 format!("Fake slash command: {}", self.0)
2975 }
2976
2977 fn menu_text(&self) -> String {
2978 format!("Run fake command: {}", self.0)
2979 }
2980
2981 fn complete_argument(
2982 self: Arc<Self>,
2983 _query: String,
2984 _cancel: Arc<AtomicBool>,
2985 _workspace: Option<WeakView<Workspace>>,
2986 _cx: &mut AppContext,
2987 ) -> Task<Result<Vec<ArgumentCompletion>>> {
2988 Task::ready(Ok(vec![]))
2989 }
2990
2991 fn requires_argument(&self) -> bool {
2992 false
2993 }
2994
2995 fn run(
2996 self: Arc<Self>,
2997 _argument: Option<&str>,
2998 _workspace: WeakView<Workspace>,
2999 _delegate: Arc<dyn LspAdapterDelegate>,
3000 _cx: &mut WindowContext,
3001 ) -> Task<Result<SlashCommandOutput>> {
3002 Task::ready(Ok(SlashCommandOutput {
3003 text: format!("Executed fake command: {}", self.0),
3004 sections: vec![],
3005 run_commands_in_text: false,
3006 }))
3007 }
3008 }
3009}