go-gpt3: 强大的Go语言OpenAI GPT-3 API客户端

go-gpt3

go-gpt3: 强大的Go语言OpenAI GPT-3 API客户端

go-gpt3是一个专为Go语言开发的OpenAI GPT-3 API客户端库,旨在帮助开发者轻松地在Go程序中集成和使用GPT-3的强大功能。这个开源项目由PullRequestInc团队开发和维护,为Go开发者提供了一种简单而高效的方式来与OpenAI的GPT-3 API进行交互。

主要特性

go-gpt3库提供了以下主要功能和特性:

完整的API支持:支持GPT-3的所有主要API,包括完成(Completion)、聊天(Chat)、嵌入(Embedding)等。
流式传输:支持完成API的流式传输功能,允许实时获取生成结果。
灵活配置:提供多种客户端选项,如自定义基础URL、用户代理、超时设置等。
类型安全:利用Go的类型系统,提供类型安全的API调用方式。
错误处理:内置全面的错误处理机制,便于调试和问题排查。
文档完善:提供详细的Go文档和使用示例,方便开发者快速上手。

安装和使用

要使用go-gpt3,首先需要安装该库:

go get github.com/PullRequestInc/go-gpt3

安装完成后,可以通过以下方式创建一个客户端并使用:

import "github.com/PullRequestInc/go-gpt3"

client := gpt3.NewClient("your-api-key")

API调用示例

以下是一个使用go-gpt3进行文本完成的简单示例:

resp, err := client.Completion(ctx, gpt3.CompletionRequest{
    Prompt:    []string{"The first thing you should know about Go is"},
    MaxTokens: gpt3.IntPtr(30),
    Stop:      []string{"."},
    Echo:      true,
})
if err != nil {
    log.Fatalln(err)
}
fmt.Println(resp.Choices[0].Text)

这个示例展示了如何使用完成API来生成关于Go语言的文本。

GPT-3 Text Completion

高级功能

go-gpt3不仅支持基本的API调用,还提供了许多高级功能:

流式传输:

err := client.CompletionStream(ctx, gpt3.CompletionRequest{
    Prompt: []string{"Write a story about"},
    Stream: true,
}, func(resp *gpt3.CompletionResponse) {
    fmt.Print(resp.Choices[0].Text)
})

聊天completion:

resp, err := client.ChatCompletion(ctx, gpt3.ChatCompletionRequest{
    Model: gpt3.GPT3Dot5Turbo,
    Messages: []gpt3.ChatCompletionRequestMessage{
        {Role: "user", Content: "Tell me a joke"},
    },
})

文本嵌入:

resp, err := client.Embeddings(ctx, gpt3.EmbeddingsRequest{
    Input: []string{"The food was delicious and the waiter..."},
    Model: gpt3.AdaSearchQuery,
})

这些高级功能使得go-gpt3能够满足各种复杂的应用场景需求。

错误处理

go-gpt3提供了详细的错误信息,有助于开发者快速定位和解决问题:

if err != nil {
    if apiErr, ok := err.(gpt3.APIError); ok {
        fmt.Printf("API error: %s\n", apiErr.Message)
    } else {
        fmt.Printf("Error: %v\n", err)
    }
}

自定义选项

go-gpt3允许开发者通过多种选项来自定义客户端行为:

client := gpt3.NewClient(
    apiKey,
    gpt3.WithBaseURL("https://custom-openai-proxy.com"),
    gpt3.WithTimeout(30 * time.Second),
    gpt3.WithDefaultEngine("text-davinci-002"),
)