OpenAI Golang SDK

go get -u github.com/artisancloud/openaiCopy

func main() {
    // 创建一个OpenAI API客户端
    config  := openai.V1Config{
        OpenAPIKey: "your-openai-key",     // 必填, 你的OpenAI API密钥
        Organization: "your-organization", // 可选, 你的组织id
        HttpDebug:  true,                  // 可选, 是否开启调试模式(使用默认log输出HTTP请求信息)
        ProxyURL: "proxy-url",             // 可选, 代理地址, 例如 http://xxxxxx:port, 如果包含基本鉴权, 格式为http://username:password@xxxxxx:port
    }
    client, err := openai.NewClient(&config)
    if err != nil {
        panic(err)
    }

    // 创建 ChatCompletion
    req1 := v1.CreateChatCompletionRequest{
        Model: "gpt-3.5-turbo",
        Messages: []v1.Message{
            {
                Role:    "user",
                Content: "Hello!",
            },
        },
    }
    result1, err1 := client.V1.Chat.CreateChatCompletion(&req1)
    if err1 != nil {
        if e, ok := err1.(*v1.Error); ok {
            fmt.Printf("open ai error: %v", e.Detail.Message)
        }
        fmt.Printf("error: %v", err1)
        return
    }

    // 输出结果
    fmt.Println(result1)

    // 创建 Completion
    req2 := v1.CreateCompletionRequest{
        Model:       "text-davinci-003",
        Prompt:      openaitype.StringOrArray{"Say this is a test", "Say Hello World"},
        MaxTokens:   openaitype.PointInt(150),
        Temperature: openaitype.PointFloat64(0),
        TopP:        openaitype.PointFloat64(1),
        N:           openaitype.PointInt(1),
        Stream:      openaitype.PointBool(false),
        Stop:        openaitype.StringOrArray{"\n"},
    }

    result2, err2 := client.V1.Completion.CreateCompletion(&req2)
    if err2 != nil {
        if e, ok := err2.(*v1.Error); ok {
            fmt.Printf("open ai error: %v", e.Detail.Message)
        }
        fmt.Printf("error: %v", err2)
        return
    }

    // 输出结果
    fmt.Println(result2)
}Copy