upgrade to gtp-4-turbo

handlers/message_created.go

@@ -9,32 +9,66 @@ import (
 	"github.com/bwmarrin/discordgo"
 )
 
-func OnMessageCreated(s *discordgo.Session, m *discordgo.MessageCreate) {
-	fmt.Printf("\nmessage received: [%s] [%s]", m.ID, m.Message.Content)
+const MAX_HISTORY = 50
 
-	if m.Author.ID == s.State.User.ID {
+var messagesHistory = []*openai.ChatMessage{}
+
+func OnMessageCreated(s *discordgo.Session, m *discordgo.MessageCreate) {
+	isHal := m.Author.ID == s.State.User.ID
+
+	addMessageToHistory(m.Message, isHal)
+
+	if isHal {
 		return
 	}
 
 	if containUser(m.Mentions, s.State.User.ID) {
 		aiclient := openai.NewClient(env.GetEnvVariables().OpenaiHalToken)
 
-		res, err := aiclient.Completions(m.Content)
+		res, err := aiclient.Chat(messagesHistory)
 		if err != nil {
+			sendResponse(
+				s,
+				m.ChannelID,
+				fmt.Sprintf("X_X: %s", err.Error()),
+			)
+
 			log.Panicf("failed to query open ai with the following prompt [%s]. Error: %s", m.Content, err.Error())
 		}
 
 		if len(res.Choices) > 0 {
-			aiResponse := res.Choices[0].Text
-			if _, err = s.ChannelMessageSend(m.ChannelID, aiResponse); err != nil {
-				log.Panicf("failed to send the response [%s] to the discord channel [%s]", aiResponse, err.Error())
-			}
+			aiResponse := res.Choices[0].Message.Content
+			sendResponse(s, m.ChannelID, aiResponse)
 		}
 
 		return
 	}
 }
 
+func addMessageToHistory(m *discordgo.Message, isHal bool) {
+	var role string
+	if isHal {
+		role = "system"
+	} else {
+		role = "user"
+	}
+
+	messagesHistory = append(messagesHistory, &openai.ChatMessage{
+		Role:    role,
+		Content: m.Content,
+	})
+
+	if len(messagesHistory) > MAX_HISTORY {
+		messagesHistory = messagesHistory[1:]
+	}
+}
+
+func sendResponse(s *discordgo.Session, channelID string, response string) {
+	if _, err := s.ChannelMessageSend(channelID, response); err != nil {
+		log.Panicf("failed to send the response [%s] to the discord channel [%s]", response, err.Error())
+	}
+}
+
 func containUser(users []*discordgo.User, userID string) bool {
 	for _, u := range users {
 		if u.ID == userID {