This commit is contained in:
unintendedfraud
2023-11-07 09:22:41 +01:00
parent 280f7a1f2d
commit a20ee7c581
+11 -2
View File
@@ -16,15 +16,24 @@ var messagesHistory = []*openai.ChatMessage{}
func OnMessageCreated(s *discordgo.Session, m *discordgo.MessageCreate) { func OnMessageCreated(s *discordgo.Session, m *discordgo.MessageCreate) {
isHal := m.Author.ID == s.State.User.ID isHal := m.Author.ID == s.State.User.ID
fmt.Println("message received: ", m.Message.Content)
addMessageToHistory(m.Message, isHal) addMessageToHistory(m.Message, isHal)
fmt.Println("added message to history", len(messagesHistory))
if isHal { if isHal {
return return
} }
fmt.Println("check contains user")
if containUser(m.Mentions, s.State.User.ID) { if containUser(m.Mentions, s.State.User.ID) {
fmt.Println("lets go")
aiclient := openai.NewClient(env.GetEnvVariables().OpenaiHalToken) aiclient := openai.NewClient(env.GetEnvVariables().OpenaiHalToken)
fmt.Println("client created")
res, err := aiclient.Chat(messagesHistory) res, err := aiclient.Chat(messagesHistory)
if err != nil { if err != nil {
sendResponse( sendResponse(
@@ -40,8 +49,8 @@ func OnMessageCreated(s *discordgo.Session, m *discordgo.MessageCreate) {
aiResponse := res.Choices[0].Message.Content aiResponse := res.Choices[0].Message.Content
sendResponse(s, m.ChannelID, aiResponse) sendResponse(s, m.ChannelID, aiResponse)
} }
} else {
return fmt.Println("hal not mentioned")
} }
} }