18 lines
359 B
Go
18 lines
359 B
Go
|
package cli
|
||
|
|
|
||
|
|
// FriendlyRole returns a human friendly signifier for the message's role.
|
||
|
|
func (m *Message) FriendlyRole() string {
|
||
|
|
var friendlyRole string
|
||
|
|
switch m.Role {
|
||
|
|
case "user":
|
||
|
|
friendlyRole = "You"
|
||
|
|
case "system":
|
||
|
|
friendlyRole = "System"
|
||
|
|
case "assistant":
|
||
|
|
friendlyRole = "Assistant"
|
||
|
|
default:
|
||
|
|
friendlyRole = m.Role
|
||
|
|
}
|
||
|
|
return friendlyRole
|
||
|
|
}
|