guise/ai/
reasoning.rs

1//! `AIReasoning` — extended thinking, folded away.
2//!
3//! Reasoning output is usually longer than the answer it produced and is not
4//! what the reader came for, so it collapses by default. It is not hidden
5//! though: being able to open it is the point, and while it is still streaming
6//! the header says so, because a collapsed block that is quietly filling up is
7//! the one case where the user wants to know before they open it.
8//!
9//! The open state belongs to whatever owns the transcript — a turn scrolled
10//! out of view should not forget it was expanded — so this is a controlled
11//! component: pass `open` and an `on_toggle`.
12//!
13//! ```ignore
14//! AIReasoning::new(("reasoning", turn_index), text)
15//!     .open(turn.reasoning_open)
16//!     .streaming(turn.streaming)
17//!     .on_toggle(cx.listener(|this, _, _, cx| this.toggle_reasoning(cx)))
18//! ```
19
20use gpui::prelude::*;
21use gpui::{div, px, App, ClickEvent, ElementId, IntoElement, SharedString, Window};
22
23use crate::devtools::Probed;
24use crate::icon::{Icon, IconName};
25use crate::input::ClickHandler;
26use crate::markdown::Markdown;
27use crate::theme::{theme, ColorName, Size};
28
29/// A collapsible block of the model's reasoning.
30#[derive(IntoElement)]
31pub struct AIReasoning {
32    id: ElementId,
33    text: SharedString,
34    label: Option<SharedString>,
35    open: bool,
36    streaming: bool,
37    size: Size,
38    on_toggle: Option<ClickHandler>,
39}
40
41impl AIReasoning {
42    pub fn new(id: impl Into<ElementId>, text: impl Into<SharedString>) -> Self {
43        AIReasoning {
44            id: id.into(),
45            text: text.into(),
46            label: None,
47            open: false,
48            streaming: false,
49            size: Size::Sm,
50            on_toggle: None,
51        }
52    }
53
54    /// Override the header text. The default names the state.
55    pub fn label(mut self, label: impl Into<SharedString>) -> Self {
56        self.label = Some(label.into());
57        self
58    }
59
60    pub fn open(mut self, open: bool) -> Self {
61        self.open = open;
62        self
63    }
64
65    /// Still arriving, so the header says so even while collapsed.
66    pub fn streaming(mut self, streaming: bool) -> Self {
67        self.streaming = streaming;
68        self
69    }
70
71    pub fn size(mut self, size: Size) -> Self {
72        self.size = size;
73        self
74    }
75
76    pub fn on_toggle(
77        mut self,
78        handler: impl Fn(&ClickEvent, &mut Window, &mut App) + 'static,
79    ) -> Self {
80        self.on_toggle = Some(Box::new(handler));
81        self
82    }
83}
84
85impl RenderOnce for AIReasoning {
86    fn render(self, _window: &mut Window, cx: &mut App) -> impl IntoElement {
87        let t = theme(cx);
88        let dimmed = t.dimmed().hsla();
89        let border = t.border().hsla();
90        let font_xs = t.font_size(Size::Xs);
91
92        let label = self.label.unwrap_or_else(|| {
93            if self.streaming {
94                SharedString::new_static("Thinking\u{2026}")
95            } else {
96                SharedString::new_static("Reasoning")
97            }
98        });
99
100        let header = div()
101            .id(self.id)
102            .flex()
103            .items_center()
104            .gap(px(6.0))
105            .cursor_pointer()
106            .text_size(px(font_xs))
107            .text_color(dimmed)
108            .child(
109                Icon::new(if self.open {
110                    IconName::ChevronDown
111                } else {
112                    IconName::ChevronRight
113                })
114                .size(Size::Xs)
115                .color(ColorName::Gray),
116            )
117            .child(
118                Icon::new(IconName::Brain)
119                    .size(Size::Xs)
120                    .color(ColorName::Gray),
121            )
122            .child(label)
123            .when_some(self.on_toggle, |header, handler| {
124                header.on_click(move |event, window, cx| handler(event, window, cx))
125            });
126
127        div()
128            .flex()
129            .flex_col()
130            .gap(px(6.0))
131            .w_full()
132            .child(header)
133            .when(self.open, |column| {
134                column.child(
135                    div()
136                        .w_full()
137                        .pl(px(10.0))
138                        .border_l_2()
139                        .border_color(border)
140                        .text_color(dimmed)
141                        .child(Markdown::new(self.text).size(self.size)),
142                )
143            })
144            .probe("AIReasoning")
145    }
146}
guise/ai/reasoning.rs

guise/ai/
reasoning.rs
