From 77b120323b9deb05d40904aaab26ccfc6342409f Mon Sep 17 00:00:00 2001 From: Nate Butler Date: Thu, 29 Jun 2023 17:44:47 -0400 Subject: [PATCH] Add `low_tokens_remaining` case to the assistant --- crates/ai/src/assistant.rs | 2 ++ crates/theme/src/theme.rs | 1 + 2 files changed, 3 insertions(+) diff --git a/crates/ai/src/assistant.rs b/crates/ai/src/assistant.rs index 9ca54e661a74d935356bf52378613e9ffb1dcb0e..6375c2fe4d089ae8db44c9138a33fe845334fe38 100644 --- a/crates/ai/src/assistant.rs +++ b/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 }; diff --git a/crates/theme/src/theme.rs b/crates/theme/src/theme.rs index e54dcdfd1e987eaf24656bc735079db54d37f0bc..01da555e1edf1d33486c0684270dc325ec97f20f 100644 --- a/crates/theme/src/theme.rs +++ b/crates/theme/src/theme.rs @@ -1027,6 +1027,7 @@ pub struct AssistantStyle { pub system_sender: Interactive, pub model: Interactive, pub remaining_tokens: ContainedText, + pub low_remaining_tokens: ContainedText, pub no_remaining_tokens: ContainedText, pub error_icon: Icon, pub api_key_editor: FieldEditor,