coai/cli/filter.go
2023-11-22 12:29:08 +08:00

20 lines
523 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)
outputInfo("filter", fmt.Sprintf("filtered %d keys, %d available, %d unavailable", len(keys), len(available), len(keys)-len(available)))
fmt.Println(strings.Join(available, "|"))
}