add groq cloud instead of opnai for chatbot

fix uppercase only for crypto
2024-11-07 16:05:02 +01:00 · 2024-11-07 16:05:02 +01:00 · d3f6deb6df
commit d3f6deb6df
parent 5db6a70961
5 changed files with 122 additions and 151 deletions
--- a/bot/bot.go
+++ b/bot/bot.go
@ -11,7 +11,6 @@ import (

 	"github.com/bwmarrin/discordgo"
 	"github.com/charmbracelet/log"
-	"github.com/sashabaranov/go-openai"
 )

 var (
@ -44,7 +43,7 @@ func Run() {
 	<-c
 }

-var conversationHistoryMap = make(map[string][]openai.ChatCompletionMessage)
+var conversationHistoryMap = make(map[string][]map[string]interface{})
 var sshConnections = make(map[string]*SSHConnection)

 func hasAdminRole(roles []string) bool {
@ -69,13 +68,11 @@ func newMessage(discord *discordgo.Session, message *discordgo.MessageCreate) {
 	channelID := message.ChannelID
 	conversationHistory = populateConversationHistory(discord, channelID, conversationHistory)

-	log.Info("Conversation history from bot.go:", conversationHistory)
-
 	if strings.HasPrefix(message.Content, "!cry") {
 		currentCryptoPrice := getCurrentCryptoPrice(message.Content)
 		discord.ChannelMessageSendComplex(message.ChannelID, currentCryptoPrice)
 	} else if strings.HasPrefix(message.Content, "!bit") || isPrivateChannel {
-		chatGPT(discord, message.ChannelID, message.Content, conversationHistory)
+		chatGPT(discord, message.ChannelID, conversationHistory)
 	} else if strings.HasPrefix(message.Content, "!genkey") {
 		if hasAdminRole(message.Member.Roles) {
 			err := GenerateAndSaveSSHKeyPairIfNotExist()
--- a/bot/chat.go
+++ b/bot/chat.go
@ -1,182 +1,160 @@
 package bot

 import (
-	"context"
-	"fmt"
+	"bytes"
+	"encoding/json"
+	"io"
+	"net/http"
+	"time"

 	"github.com/bwmarrin/discordgo"
 	"github.com/charmbracelet/log"
-	openai "github.com/sashabaranov/go-openai"
 )

 const (
 	maxTokens         = 2000
-	maxContextTokens  = 4000
+	maxContextTokens  = 2000
 	maxMessageTokens  = 2000
-	systemMessageText = "your name is !bit you are a discord bot"
+	systemMessageText = "your name is !bit you are a discord bot, you use brief answers untill asked to elaborate or explain"
 )

-func populateConversationHistory(session *discordgo.Session, channelID string, conversationHistory []openai.ChatCompletionMessage) []openai.ChatCompletionMessage {
+func populateConversationHistory(session *discordgo.Session, channelID string, conversationHistory []map[string]interface{}) []map[string]interface{} {
+	// Retrieve recent messages from the Discord channel
 	messages, err := session.ChannelMessages(channelID, 20, "", "", "")
 	if err != nil {
 		log.Error("Error retrieving channel history:", err)
 		return conversationHistory
 	}

+	// Define max tokens for the conversation history
+	maxTokens := 2000
 	totalTokens := 0
-	maxHistoryTokens := maxTokens

-	// Calculate total tokens without removing any messages
+	// Calculate current token count in conversation history
 	for _, msg := range conversationHistory {
-		totalTokens += len(msg.Content) + len(msg.Role) + 2
-	}
-
-	log.Info("Total Tokens Before Trimming:", totalTokens)
-
-	// Iterate from the beginning of conversationHistory (oldest messages)
-	for i := 0; i < len(conversationHistory); i++ {
-		msg := conversationHistory[i]
-		tokens := len(msg.Content) + len(msg.Role) + 2 // Account for role and content tokens
-
-		if totalTokens-tokens >= maxHistoryTokens {
-			// Remove the oldest message
-			log.Info("Removing Oldest Message:", msg.Content)
-			conversationHistory = conversationHistory[i+1:]
-			i-- // Adjust index after removal
-		} else {
-			totalTokens -= tokens
+		content, okContent := msg["Content"].(string)
+		role, okRole := msg["Role"].(string)
+		if okContent && okRole {
+			tokens := len(content) + len(role) + 2 // Account for tokens in content and role
+			totalTokens += tokens
+			log.Infof("Existing message tokens: %d", tokens)
 		}
 	}

-	log.Info("Total Tokens After Trimming:", totalTokens)
-
-	// Add new messages from the channel
+	// Process messages in reverse order (newest to oldest)
 	for i := len(messages) - 1; i >= 0; i-- {
 		message := messages[i]
-		if len(message.Content) > 0 {
-			tokens := len(message.Content) + 2 // Account for role and content tokens
-			if totalTokens+tokens <= maxContextTokens {
-				conversationHistory = append(conversationHistory, openai.ChatCompletionMessage{
-					Role:    openai.ChatMessageRoleUser,
-					Content: message.Content,
+
+		// Check if the message is older than 30 minutes
+		if time.Since(message.Timestamp) < 30*time.Minute {
+			tokens := len(message.Content) + 2
+			if totalTokens+tokens <= maxTokens {
+				// Append as map[string]interface{} instead of map[string]string
+				conversationHistory = append(conversationHistory, map[string]interface{}{
+					"role":    "user",
+					"content": message.Content,
 				})
 				totalTokens += tokens
-				log.Info("Adding New Message:", message.Content)
+				log.Infof("Adding message with tokens: %d", tokens)
 			} else {
-				if totalTokens+tokens > maxContextTokens {
-					log.Warn("Message token count exceeds maxContextTokens:", len(message.Content), len(message.Content)+2)
+				log.Warnf("Skipping message with tokens: %d", tokens)
+			}
 		} else {
-					log.Warn("Conversation history length exceeds maxContextTokens:", len(conversationHistory), maxHistoryTokens)
+			log.Infof("Skipping message, older than 30 minutes: %s", message.Content)
 		}
-				break
+
+		// Ensure the current message is included (regardless of token limit)
+		conversationHistory = append(conversationHistory, map[string]interface{}{
+			"role":    "user",
+			"content": message.Content,
+		})
+		totalTokens += len(message.Content) + 2
+		log.Infof("Adding message with tokens: %d", totalTokens)
+
+		// Now check if the token limit is exceeded and trim older messages
+		if totalTokens > maxTokens && len(conversationHistory) > 1 {
+			// Remove the oldest message from the history
+			conversationHistory = conversationHistory[1:]
+			content, okContent := conversationHistory[0]["Content"].(string)
+			role, okRole := conversationHistory[0]["Role"].(string)
+			if okContent && okRole {
+				tokens := len(content) + len(role) + 2
+				totalTokens -= tokens
+				log.Infof("Trimming message with tokens: %d", tokens)
 			}
+			log.Info("Trimming oldest message to maintain token limit")
 		}
 	}

-	// Log the final order of conversation history
-	log.Info("Final Conversation History Order:", conversationHistory)
-
+	log.Info("Final Conversation History Order: %s", conversationHistory)
 	return conversationHistory
 }

-func chatGPT(session *discordgo.Session, channelID string, message string, conversationHistory []openai.ChatCompletionMessage) {
-	client := openai.NewClient(OpenAIToken)
+// Function to handle Groq API requests and pagination
+func chatGPT(session *discordgo.Session, channelID string, conversationHistory []map[string]interface{}) {
+	OpenAIToken := OpenAIToken
+	GroqBaseURL := "https://api.groq.com/openai/v1"
+	GroqModel := "llama-3.1-70b-versatile"

-	// Perform GPT-4 completion
-	log.Info("Starting completion...", conversationHistory)
-	resp, err := client.CreateChatCompletion(
-		context.Background(),
-		openai.ChatCompletionRequest{
-			MaxTokens:        maxTokens,
-			FrequencyPenalty: 0.3,
-			PresencePenalty:  0.6,
-			Model:            openai.GPT3Dot5Turbo,
-			Messages:         conversationHistory, // Use trimmed conversation history
-		},
-	)
-	log.Info("completion done.")
+	// Add system message at the start of conversation history
+	conversationHistory = append([]map[string]interface{}{
+		{"role": "system", "content": systemMessageText},
+	}, conversationHistory...)

-	// Handle API errors
-	if err != nil {
-		log.Error("Error connecting to the OpenAI API:", err)
-		return
-	}
-
-	// Paginate the response and send as separate messages with clickable emojis
-	gptResponse := resp.Choices[0].Message.Content
-	pageSize := maxMessageTokens
-
-	// Split the response into pages
-	var pages []string
-	for i := 0; i < len(gptResponse); i += pageSize {
-		end := i + pageSize
-		if end > len(gptResponse) {
-			end = len(gptResponse)
-		}
-		pages = append(pages, gptResponse[i:end])
-	}
-
-	// Send the first page
-	currentPage := 0
-	totalPages := len(pages)
-	embed := &discordgo.MessageEmbed{
-		Title:       fmt.Sprintf("Page %d of %d", currentPage+1, totalPages),
-		Description: pages[currentPage],
-		Color:       0x00ff00, // Green color
-	}
-	msg, err := session.ChannelMessageSendEmbed(channelID, embed)
-	if err != nil {
-		log.Error("Error sending embed message:", err)
-		return
-	}
-
-	// Add reaction emojis for pagination if there are multiple pages
-	if totalPages > 1 { // Only add reactions if there are multiple pages
-		err = session.MessageReactionAdd(channelID, msg.ID, "⬅️")
-		if err != nil {
-			log.Error("Error adding reaction emoji:", err)
-			return
-		}
-		err = session.MessageReactionAdd(channelID, msg.ID, "➡️")
-		if err != nil {
-			log.Error("Error adding reaction emoji:", err)
-			return
-		}
-	}
-
-	// Create a reaction handler function
-	session.AddHandler(func(s *discordgo.Session, r *discordgo.MessageReactionAdd) {
-		// Call the reactionHandler function and pass totalPages
-		reactionHandler(s, r, currentPage, msg, pages, totalPages)
+	client := http.Client{}
+	requestBody, err := json.Marshal(map[string]interface{}{
+		"model":             GroqModel,
+		"messages":          conversationHistory,
+		"max_tokens":        maxTokens,
+		"frequency_penalty": 0.3,
+		"presence_penalty":  0.6,
 	})
-
-}
-
-func reactionHandler(session *discordgo.Session, r *discordgo.MessageReactionAdd, currentPage int, msg *discordgo.Message, pages []string, totalPages int) {
-	// Check if the reaction is from the same user and message
-	if r.UserID == session.State.User.ID || r.MessageID != msg.ID {
+	if err != nil {
+		log.Errorf("Failed to marshal request body: %v", err)
 		return
 	}

-	// Handle pagination based on reaction
-	if r.Emoji.Name == "⬅️" {
-		if currentPage > 0 {
-			currentPage--
+	req, err := http.NewRequest("POST", GroqBaseURL+"/chat/completions", bytes.NewBuffer(requestBody))
+	if err != nil {
+		log.Errorf("Failed to create request: %v", err)
+		return
 	}
-	} else if r.Emoji.Name == "➡️" {
-		if currentPage < len(pages)-1 {
-			currentPage++
+	req.Header.Set("Content-Type", "application/json")
+	req.Header.Set("Authorization", "Bearer "+OpenAIToken)
+
+	resp, err := client.Do(req)
+	if err != nil {
+		log.Errorf("Failed to make request: %v", err)
+		return
 	}
+	defer resp.Body.Close()
+
+	var groqResp struct {
+		Choices []struct {
+			Message struct {
+				Content string `json:"content"`
+			} `json:"message"`
+		} `json:"choices"`
 	}

-	// Update the message with the new page
-	updatedEmbed := &discordgo.MessageEmbed{
-		Title:       fmt.Sprintf("Page %d of %d", currentPage+1, len(pages)),
-		Description: pages[currentPage],
-		Color:       0x00ff00, // Green color
-	}
-	_, err := session.ChannelMessageEditEmbed(r.ChannelID, r.MessageID, updatedEmbed)
+	body, err := io.ReadAll(resp.Body)
 	if err != nil {
-		log.Error("Error editing embed message:", err)
+		log.Errorf("Failed to read response body: %v", err)
+		return
+	}
+
+	err = json.Unmarshal(body, &groqResp)
+	if err != nil {
+		log.Errorf("Failed to decode response: %v", err)
+		return
+	}
+
+	if len(groqResp.Choices) > 0 {
+		gptResponse := groqResp.Choices[0].Message.Content
+		_, err := session.ChannelMessageSend(channelID, gptResponse)
+		if err != nil {
+			log.Error("Error sending message:", err)
+			return
+		}
 	}
 }
--- a/bot/command-crypto.go
+++ b/bot/command-crypto.go
@ -5,7 +5,6 @@ import (
 	"fmt"
 	"io"
 	"net/http"
-	"regexp"
 	"strconv"
 	"strings"
 	"time"
@ -20,17 +19,17 @@ type CryptoData struct {
 }

 func getCurrentCryptoPrice(message string) *discordgo.MessageSend {
-	r, _ := regexp.Compile(`\s?[A-Z]{3,5}\s?`)
-	currency := r.FindString(message)
-	curr := strings.ReplaceAll(currency, " ", "")
+	message = strings.ToUpper(strings.TrimSpace(message)) // convert to uppercase and remove leading/trailing spaces
+	message = strings.ReplaceAll(message, "!CRY ", "")    // remove "!cry" prefix
+	currency := message[:]

-	if curr == "" {
+	if len(currency) < 3 || len(currency) > 5 { // check length of currency code
 		return &discordgo.MessageSend{
-			Content: "Sorry, cant recognize crypto currency shortcode try uppercase and with spaces around",
+			Content: "Sorry, cant recognize crypto currency shortcode try uppercase and with length between 3 to 5",
 		}
 	}

-	cryptoURL := fmt.Sprintf("%s?fsym=%s&tsyms=USD&api_key=%s", URL, curr, CryptoToken)
+	cryptoURL := fmt.Sprintf("%s?fsym=%s&tsyms=USD&api_key=%s", URL, currency, CryptoToken)
 	fmt.Println(cryptoURL)
 	client := http.Client{Timeout: 5 * time.Second}

@ -54,10 +53,10 @@ func getCurrentCryptoPrice(message string) *discordgo.MessageSend {
 		Embeds: []*discordgo.MessageEmbed{{
 			Type:        discordgo.EmbedTypeRich,
 			Title:       "Current Price",
-			Description: "Price for " + curr,
+			Description: "Price for " + currency,
 			Fields: []*discordgo.MessageEmbedField{
 				{
-					Name:   "1 " + curr,
+					Name:   "1 " + currency,
 					Value:  usd + " USD",
 					Inline: true,
 				},
--- a/go.mod
+++ b/go.mod
@ -101,7 +101,6 @@ require (
 require (
 	github.com/gorilla/websocket v1.5.0 // indirect
 	github.com/joho/godotenv v1.5.1
-	github.com/sashabaranov/go-openai v1.15.4
 	golang.org/x/crypto v0.14.0
 	golang.org/x/sys v0.13.0 // indirect
 )
--- a/go.sum
+++ b/go.sum
@ -194,8 +194,6 @@ github.com/rivo/uniseg v0.4.4 h1:8TfxU8dW6PdqD27gjM8MVNuicgxIjxpm4K7x4jp8sis=
 github.com/rivo/uniseg v0.4.4/go.mod h1:FN3SvrM+Zdj16jyLfmOkMNblXMcoc8DfTHruCPUcx88=
 github.com/rogpeppe/go-internal v1.9.0 h1:73kH8U+JUqXU8lRuOHeVHaa/SZPifC7BkcraZVejAe8=
 github.com/russross/blackfriday/v2 v2.1.0/go.mod h1:+Rmxgy9KzJVeS9/2gXHxylqXiyQDYRxCVz55jmeOWTM=
-github.com/sashabaranov/go-openai v1.15.4 h1:BXCR0Uxk5RipeY4yBC7g6pBVfcjh8jwrMNOYdie6yuk=
-github.com/sashabaranov/go-openai v1.15.4/go.mod h1:lj5b/K+zjTSFxVLijLSTDZuP7adOgerWeFyZLUhAKRg=
 github.com/spf13/cast v1.5.1 h1:R+kOtfhWQE6TVQzY+4D7wJLBgkdVasCEFxSUBYBYIlA=
 github.com/spf13/cast v1.5.1/go.mod h1:b9PdjNptOpzXr7Rq1q9gJML/2cdGQAo69NKzQ10KN48=
 github.com/spf13/cobra v1.7.0 h1:hyqWnYt1ZQShIddO5kBpj3vu05/++x6tJ6dg8EC572I=