First commit

openai/client.go

@@ -0,0 +1,38 @@
+package openai
+
+import (
+	"context"
+
+	"git.ewnix.net/phlux/g/discord-gpt-bot/config"
+	"github.com/sashabaranov/go-openai"
+)
+
+type Message struct {
+	Role    string `json:"role"`
+	Content string `json:"content"`
+}
+
+func Ask(messages []Message) (string, error) {
+	client := openai.NewClient(config.OpenAIKey)
+
+	var chatHistory []openai.ChatCompletionMessage
+	for _, m := range messages {
+		chatHistory = append(chatHistory, openai.ChatCompletionMessage{
+			Role:    m.Role,
+			Content: m.Content,
+		})
+	}
+
+	resp, err := client.CreateChatCompletion(
+		context.Background(),
+		openai.ChatCompletionRequest{
+			Model:    openai.GPT4,
+			Messages: chatHistory,
+		},
+	)
+	if err != nil {
+		return "", err
+	}
+
+	return resp.Choices[0].Message.Content, nil
+}