From f206334e72a7f15b2069a1a32dd932bfed0b10b4 Mon Sep 17 00:00:00 2001
From: Matt Low <matt@mlow.ca>
Date: Wed, 29 Nov 2023 05:57:38 +0000
Subject: [PATCH] Use MessageRole constants elsewhere

---
 pkg/cli/cmd.go | 10 +++++-----
 pkg/cli/tty.go |  6 +++---
 2 files changed, 8 insertions(+), 8 deletions(-)

diff --git a/pkg/cli/cmd.go b/pkg/cli/cmd.go
index c09ad02..06f89dc 100644
--- a/pkg/cli/cmd.go
+++ b/pkg/cli/cmd.go
@@ -322,7 +322,7 @@ var replyCmd = &cobra.Command{
 
 		userReply := Message{
 			ConversationID:  conversation.ID,
-			Role:            "user",
+			Role:            MessageRoleUser,
 			OriginalContent: messageContents,
 		}
 
@@ -366,12 +366,12 @@ var newCmd = &cobra.Command{
 		messages := []Message{
 			{
 				ConversationID:  conversation.ID,
-				Role:            "system",
+				Role:            MessageRoleSystem,
 				OriginalContent: SystemPrompt(),
 			},
 			{
 				ConversationID:  conversation.ID,
-				Role:            "user",
+				Role:            MessageRoleUser,
 				OriginalContent: messageContents,
 			},
 		}
@@ -411,11 +411,11 @@ var promptCmd = &cobra.Command{
 
 		messages := []Message{
 			{
-				Role:            "system",
+				Role:            MessageRoleSystem,
 				OriginalContent: SystemPrompt(),
 			},
 			{
-				Role:            "user",
+				Role:            MessageRoleUser,
 				OriginalContent: message,
 			},
 		}
diff --git a/pkg/cli/tty.go b/pkg/cli/tty.go
index 4ab6f38..2a1e13e 100644
--- a/pkg/cli/tty.go
+++ b/pkg/cli/tty.go
@@ -89,11 +89,11 @@ func (m *Message) RenderTTY() {
 
 	var roleStyle color.Style
 	switch m.Role {
-	case "system":
+	case MessageRoleSystem:
 		roleStyle = color.Style{color.HiRed}
-	case "user":
+	case MessageRoleUser:
 		roleStyle = color.Style{color.HiGreen}
-	case "assistant":
+	case MessageRoleAssistant:
 		roleStyle = color.Style{color.HiBlue}
 	default:
 		roleStyle = color.Style{color.FgWhite}