coai/cli/filter.go
2023-10-28 22:31:02 +08:00

19 lines
519 B
Go

package cli
import (
"chat/adapter/chatgpt"
"fmt"
"github.com/spf13/viper"
"strings"
)
func FilterApiKeyCommand(args []string) {
data := strings.Trim(strings.TrimSpace(GetArgString(args, 0)), "\"")
endpoint := viper.GetString("openai.test")
keys := strings.Split(data, "|")
available := chatgpt.FilterKeysNative(endpoint, keys)
fmt.Println(fmt.Sprintf("[cli] filtered %d keys, %d available, %d unavailable", len(keys), len(available), len(keys)-len(available)))
fmt.Println(strings.Join(available, "|"))
}