From e0ee13f37966d4d04852963e1a957cc38950c1df Mon Sep 17 00:00:00 2001 From: unintendedfraud Date: Sat, 6 May 2023 12:34:31 +0200 Subject: [PATCH] clean version --- handlers/message_created.go | 10 +++------- openai/client.go | 1 - 2 files changed, 3 insertions(+), 8 deletions(-) diff --git a/handlers/message_created.go b/handlers/message_created.go index 122a44b..0768693 100644 --- a/handlers/message_created.go +++ b/handlers/message_created.go @@ -1,7 +1,6 @@ package handlers import ( - "fmt" "hal/env" "hal/openai" "log" @@ -16,17 +15,14 @@ func OnMessageCreated(s *discordgo.Session, m *discordgo.MessageCreate) { if containUser(m.Mentions, s.State.User.ID) { aiclient := openai.NewClient(env.GetEnvVariables().OpenaiHalToken) - fmt.Println("calling completions with", m.Content) res, err := aiclient.Completions(m.Content) if err != nil { 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].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()) } return diff --git a/openai/client.go b/openai/client.go index 78f7345..9d5d93b 100644 --- a/openai/client.go +++ b/openai/client.go @@ -20,7 +20,6 @@ func NewClient(token string) *Client { } func (c Client) Completions(prompt string) (*CompletionResponse, error) { - fmt.Println("prompting: ", prompt) body := &CompletionPayload{ Model: "gpt-3.5-turbo", Prompt: cleanPrompt(prompt),