/**
 * @author: dn-jinmin/dn-jinmin
 * @doc:
 */

package main

import (
	"context"
	"fmt"
	"github.com/sashabaranov/go-openai"
)

var (
	apikey = "sk-6wWCfobHHEX9n2QuGA4yi0jouBQ46DkQpL9XPgaAbuEd71c3"
	url    = "https://api.aiproxy.io/v1"
)

func main() {
	gptCfg := openai.DefaultConfig(apikey)
	gptCfg.BaseURL = url

	client := openai.NewClientWithConfig(gptCfg)

	res, err := client.CreateCompletion(context.Background(), openai.CompletionRequest{
		Model:  openai.GPT3Dot5TurboInstruct,
		Prompt: "你好",
	})

	if err != nil {
		panic(err)
	}
	fmt.Println(res.Choices)
}