From 13145cad3146b7a3429cf5903bff00e9da03ccba Mon Sep 17 00:00:00 2001 From: unintendedfraud Date: Sat, 6 May 2023 11:38:30 +0200 Subject: [PATCH] log --- handlers/message_created.go | 12 +++++++----- openai/client.go | 6 +----- 2 files changed, 8 insertions(+), 10 deletions(-) diff --git a/handlers/message_created.go b/handlers/message_created.go index 1046455..e92692d 100644 --- a/handlers/message_created.go +++ b/handlers/message_created.go @@ -10,22 +10,23 @@ import ( ) func OnMessageCreated(s *discordgo.Session, m *discordgo.MessageCreate) { - log.Printf("new message posted: [%s], par [%s], authorID [%s], state user id [%s]", m.Content, m.Author.Username, m.Author.ID, s.State.User.ID) - - log.Printf("author [%+v], state user [%+v]", m.Author, s.State.User) if m.Author.ID == s.State.User.ID { return } if containUser(m.Mentions, s.State.User.ID) { - fmt.Println("going through openai stuff") + fmt.Println("create ai client") aiclient := openai.NewClient(env.GetEnvVariables().OpenaiHalToken) + fmt.Println("ai client created") + res, err := aiclient.ChatCompletions(m.Content) if err != nil { log.Panicf("failed to query open ai with the following prompt [%s]. Error: %s", m.Content, err.Error()) } + fmt.Println("ai client replied") + if len(res.Choices) > 0 { aiResponse := res.Choices[0].Message.Content if _, err = s.ChannelMessageSend(m.ChannelID, aiResponse); err != nil { @@ -33,13 +34,14 @@ func OnMessageCreated(s *discordgo.Session, m *discordgo.MessageCreate) { } } + fmt.Println("### ai did not return any response ###", res.Usage, res.Choices) + return } } func containUser(users []*discordgo.User, userID string) bool { for _, u := range users { - fmt.Println("## mentions ##", u.ID, u.Username, "-----", userID) if u.ID == userID { return true } diff --git a/openai/client.go b/openai/client.go index 5c919d1..2ad3600 100644 --- a/openai/client.go +++ b/openai/client.go @@ -8,7 +8,6 @@ import ( ) type Client struct { - Token string HttpClient *resty.Client } @@ -16,10 +15,7 @@ func NewClient(token string) *Client { httpclient := resty.New() httpclient.SetAuthToken(token) - return &Client{ - Token: token, - HttpClient: httpclient, - } + return &Client{HttpClient: httpclient} } func (c Client) ChatCompletions(prompt string) (*CompletionResponse, error) {