Add chat message role const (#111)

2023-03-03 17:07:09 +08:00
parent 4e1be7d99f
commit d1a81205bf
1 changed files with 7 additions and 0 deletions
--- a/chat.go
+++ b/chat.go
@@ -8,6 +8,13 @@ import (
 	"net/http"
 )

+// Chat message role defined by the OpenAI API.
+const (
+	ChatMessageRoleSystem    = "system"
+	ChatMessageRoleUser      = "user"
+	ChatMessageRoleAssistant = "assistant"
+)
+
 var (
 	ErrChatCompletionInvalidModel = errors.New("currently, only gpt-3.5-turbo and gpt-3.5-turbo-0301 are supported")
 )