clean version

This commit is contained in:
unintendedfraud
2023-05-06 12:34:31 +02:00
parent 801e51fe49
commit e0ee13f379
2 changed files with 3 additions and 8 deletions
+3 -7
View File
@@ -1,7 +1,6 @@
package handlers package handlers
import ( import (
"fmt"
"hal/env" "hal/env"
"hal/openai" "hal/openai"
"log" "log"
@@ -16,17 +15,14 @@ func OnMessageCreated(s *discordgo.Session, m *discordgo.MessageCreate) {
if containUser(m.Mentions, s.State.User.ID) { if containUser(m.Mentions, s.State.User.ID) {
aiclient := openai.NewClient(env.GetEnvVariables().OpenaiHalToken) aiclient := openai.NewClient(env.GetEnvVariables().OpenaiHalToken)
fmt.Println("calling completions with", m.Content)
res, err := aiclient.Completions(m.Content) res, err := aiclient.Completions(m.Content)
if err != nil { if err != nil {
log.Panicf("failed to query open ai with the following prompt [%s]. Error: %s", m.Content, 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
aiResponse := res.Choices[0].Text if _, err = s.ChannelMessageSend(m.ChannelID, aiResponse); err != nil {
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())
log.Panicf("failed to send the response [%s] to the discord channel [%s]", aiResponse, err.Error())
}
} }
return return
-1
View File
@@ -20,7 +20,6 @@ func NewClient(token string) *Client {
} }
func (c Client) Completions(prompt string) (*CompletionResponse, error) { func (c Client) Completions(prompt string) (*CompletionResponse, error) {
fmt.Println("prompting: ", prompt)
body := &CompletionPayload{ body := &CompletionPayload{
Model: "gpt-3.5-turbo", Model: "gpt-3.5-turbo",
Prompt: cleanPrompt(prompt), Prompt: cleanPrompt(prompt),