assistant2: Adjust role info colors on the thread header (#22318)

Danilo Leal created 1 year ago

<img width="800" alt="Screenshot 2024-12-20 at 9 01 29 PM"
src="https://github.com/user-attachments/assets/5e9e9bf2-c84e-4215-b658-9b668dd151b2"
/>

Release Notes:

- N/A

Change summary

crates/assistant2/src/active_thread.rs | 16 ++++++++++------
1 file changed, 10 insertions(+), 6 deletions(-)

Detailed changes

crates/assistant2/src/active_thread.rs

@@ -205,10 +205,10 @@ impl ActiveThread {
 
         let context = self.thread.read(cx).context_for_message(message_id);
 
-        let (role_icon, role_name) = match message.role {
-            Role::User => (IconName::Person, "You"),
-            Role::Assistant => (IconName::ZedAssistant, "Assistant"),
-            Role::System => (IconName::Settings, "System"),
+        let (role_icon, role_name, role_color) = match message.role {
+            Role::User => (IconName::Person, "You", Color::Muted),
+            Role::Assistant => (IconName::ZedAssistant, "Assistant", Color::Accent),
+            Role::System => (IconName::Settings, "System", Color::Default),
         };
 
         div()
@@ -234,9 +234,13 @@ impl ActiveThread {
                                     .child(
                                         Icon::new(role_icon)
                                             .size(IconSize::XSmall)
-                                            .color(Color::Muted),
+                                            .color(role_color),
                                     )
-                                    .child(Label::new(role_name).size(LabelSize::XSmall)),
+                                    .child(
+                                        Label::new(role_name)
+                                            .size(LabelSize::XSmall)
+                                            .color(role_color),
+                                    ),
                             ),
                     )
                     .child(v_flex().px_2().py_1().text_ui(cx).child(markdown.clone()))