Add `low_tokens_remaining` case to the assistant

Nate Butler created 2 years ago

Change summary

crates/ai/src/assistant.rs | 2 ++
crates/theme/src/theme.rs  | 1 +
2 files changed, 3 insertions(+)

Detailed changes

crates/ai/src/assistant.rs

@@ -2060,6 +2060,8 @@ impl ConversationEditor {
         let remaining_tokens = self.conversation.read(cx).remaining_tokens()?;
         let remaining_tokens_style = if remaining_tokens <= 0 {
             &style.no_remaining_tokens
+        } else if remaining_tokens <= 500 {
+            &style.low_remaining_tokens
         } else {
             &style.remaining_tokens
         };

crates/theme/src/theme.rs

@@ -1027,6 +1027,7 @@ pub struct AssistantStyle {
     pub system_sender: Interactive<ContainedText>,
     pub model: Interactive<ContainedText>,
     pub remaining_tokens: ContainedText,
+    pub low_remaining_tokens: ContainedText,
     pub no_remaining_tokens: ContainedText,
     pub error_icon: Icon,
     pub api_key_editor: FieldEditor,