guise/ai/
thinking.rs

1//! `AIThinking` — the gap between sending and the first token.
2//!
3//! That gap can run to several seconds, and an unchanged screen during it
4//! reads as a hang. Three dots cycling is the smallest thing that says the
5//! request is alive without claiming to know how long it will take, which a
6//! progress bar would.
7
8use gpui::prelude::*;
9use gpui::{div, px, App, IntoElement, SharedString, Window};
10
11use crate::devtools::Probed;
12use crate::feedback::{Loader, LoaderVariant};
13use crate::theme::{theme, ColorName, Size};
14
15/// A "still working" indicator with an optional label.
16#[derive(IntoElement)]
17pub struct AIThinking {
18  label: Option<SharedString>,
19  size: Size,
20  color: Option<ColorName>,
21}
22
23impl AIThinking {
24  pub fn new() -> Self {
25    AIThinking {
26      label: None,
27      size: Size::Sm,
28      color: None,
29    }
30  }
31
32  /// Say what is happening — "Thinking", "Searching the web", "Running
33  /// tests". A specific label is worth far more than a generic one.
34  pub fn label(mut self, label: impl Into<SharedString>) -> Self {
35    self.label = Some(label.into());
36    self
37  }
38
39  pub fn size(mut self, size: Size) -> Self {
40    self.size = size;
41    self
42  }
43
44  pub fn color(mut self, color: ColorName) -> Self {
45    self.color = Some(color);
46    self
47  }
48}
49
50impl Default for AIThinking {
51  fn default() -> Self {
52    AIThinking::new()
53  }
54}
55
56impl RenderOnce for AIThinking {
57  fn render(self, _window: &mut Window, cx: &mut App) -> impl IntoElement {
58    let t = theme(cx);
59    let font = t.font_size(self.size);
60    let dot_color = self
61      .color
62      .map_or_else(|| t.dimmed(), |name| t.color(name, 6))
63      .hsla();
64    let dimmed = t.dimmed().hsla();
65    div()
66      .flex()
67      .items_center()
68      .gap(px(8.0))
69      .text_size(px(font))
70      .text_color(dimmed)
71      .child(
72        Loader::new()
73          .variant(LoaderVariant::Dots)
74          .size(self.size)
75          .color(dot_color),
76      )
77      .children(self.label)
78      .probe("AIThinking")
79  }
80}
guise/ai/thinking.rs

guise/ai/
thinking.rs
