LoviBot can now answer questions

openai.go

@@ -0,0 +1,107 @@
+package main
+
+import (
+	"context"
+	"fmt"
+	"strings"
+
+	openai "github.com/sashabaranov/go-openai"
+)
+
+/*
+Prompt is the Discord message that the user sent to the bot.
+*/
+func GetGPT3Response(prompt string, author string) (string, error) {
+	openAIToken := config.OpenAIToken
+	if openAIToken == "" {
+		return "", fmt.Errorf("OPENAI_API_KEY is not set")
+	}
+
+	// Remove the mention from the prompt
+	prompt = strings.Replace(prompt, "@LoviBot ", "", -1)
+
+	// Remove the mention from the prompt
+	prompt = strings.Replace(prompt, "LoviBot ", "", -1)
+
+	// Remove the mention from the prompt
+	prompt = strings.Replace(prompt, "<@345000831499894795> ", "", -1)
+
+	// Print the prompt
+	fmt.Println("Prompt:", author, ":", prompt)
+
+	// Check if the prompt is too long
+	if len(prompt) > 2048 {
+		return "", fmt.Errorf("prompt is too long")
+	}
+
+	// Check if the prompt is too short
+	if len(prompt) < 1 {
+		return "", fmt.Errorf("prompt is too short")
+	}
+
+	// Add additional information to the system message
+	var additionalInfo string
+	switch author {
+	case "thelovinator":
+		additionalInfo = "User (TheLovinator) is a programmer. Wants to live in the woods. Real name is Joakim. He made the bot."
+	case "killyoy":
+		additionalInfo = "User (KillYoy) likes to play video games. Real name is Andreas. Good at CSS."
+	case "forgefilip":
+		additionalInfo = "User (ForgeFilip) likes watches. Real name is Filip."
+	case "plubplub":
+		additionalInfo = "User (Piplup) likes to play WoW and Path of Exile. Real name is Axel. Is also called Bambi."
+	case "nobot":
+		additionalInfo = "User (Nobot) likes to play WoW. Real name is Gustav. Really good at programming."
+	case "kao172":
+		additionalInfo = "User (kao172) likes cars. Real name is Fredrik."
+	}
+
+	// Create a new client
+	client := openai.NewClient(openAIToken)
+
+	// System message
+	var systemMessage string
+	systemMessage = `You are in a Discord server. You are a space communist.
+	You are Swedish. 
+	Use Markdown for formatting. 
+	Please respond with a short message.
+	You want members of the server to like you and join Rymdkommunisterna AB.
+	You want to be a good bot so you will answer the user's question.
+	`
+
+	// Add additional information to the system message
+	if additionalInfo != "" {
+		systemMessage = fmt.Sprintf("%s %s", systemMessage, additionalInfo)
+	}
+
+	// Print the system message
+	fmt.Println("System message:", systemMessage)
+
+	// Create a completion
+	resp, err := client.CreateChatCompletion(
+		context.Background(),
+		openai.ChatCompletionRequest{
+			Model: openai.GPT3Dot5Turbo,
+			Messages: []openai.ChatCompletionMessage{
+				{
+					Role:    openai.ChatMessageRoleSystem,
+					Content: systemMessage,
+				},
+				{
+					Role:    openai.ChatMessageRoleUser,
+					Content: prompt,
+				},
+			},
+		},
+	)
+
+	if err != nil {
+		return "", fmt.Errorf("failed to get response from GPT-3: %v", err)
+	}
+
+	ourResponse := resp.Choices[0].Message.Content
+	ourResponse = strings.Replace(ourResponse, "As a space communist, ", "", -1)
+
+	fmt.Println("Response:", ourResponse)
+	return ourResponse, nil
+}