initialize/auth.go

package initialize

import (
	"bufio"
	groq "github.com/learnLi/groq_client"
	"groqai2api/global"
	"groqai2api/pkg/accountpool"
	"os"
)

func InitAuth() {
	var Secrets []*groq.Account
	// Read accounts.txt and create a list of accounts
	if _, err := os.Stat("session_tokens.txt"); err == nil {
		// Each line is a proxy, put in proxies array
		file, _ := os.Open("session_tokens.txt")
		defer file.Close()
		scanner := bufio.NewScanner(file)
		for scanner.Scan() {
			// Split by :
			token := scanner.Text()
			if len(token) == 0 {
				continue
			}
			// Append to accounts
			Secrets = append(Secrets, groq.NewAccount(token, ""))
		}
	}

	global.AccountPool = accountpool.NewAccounts(Secrets)
}