OpenAIGo: 轻量级Go语言OpenAI API客户端库

openaigo

OpenAIGo: 让OpenAI API在Go中大放异彩 🚀

在人工智能迅速发展的今天,OpenAI的GPT系列模型以其强大的自然语言处理能力引起了广泛关注。为了让Go语言开发者能够方便地使用这些先进的AI能力,GitHub用户otiai10开发了OpenAIGo这个优秀的开源项目。OpenAIGo是一个轻量级的Go语言OpenAI API客户端库,它简洁易用、依赖少、测试覆盖全面,为Go开发者提供了一种便捷的方式来集成OpenAI的强大功能。

特色功能与优势

OpenAIGo的主要特点包括:

简洁轻量: 代码结构清晰,依赖少,易于理解和使用。
全面支持: 支持OpenAI的多种API,包括GPT-3.5和GPT-4等最新模型。
高度可靠: 拥有完善的测试覆盖,确保库的稳定性。
社区维护: 由社区维护,持续更新以支持OpenAI的最新特性。

OpenAIGo Logo

快速上手

使用OpenAIGo非常简单。首先,你需要安装这个库:

go get github.com/otiai10/openaigo

然后,你可以像这样使用它:

package main

import (
    "context"
    "fmt"
    "os"

    "github.com/otiai10/openaigo"
)

func main() {
    client := openaigo.NewClient(os.Getenv("OPENAI_API_KEY"))
    request := openaigo.ChatRequest{
        Model: "gpt-4",
        Messages: []openaigo.Message{
            {Role: "user", Content: "Hello!"},
        },
    }
    ctx := context.Background()
    response, err := client.Chat(ctx, request)
    fmt.Println(response, err)
}

这个简单的例子展示了如何初始化客户端,发送一个聊天请求,并获取响应。

支持的API端点

OpenAIGo支持多种OpenAI API端点,包括但不限于:

模型列表和检索
文本补全
聊天补全
图像生成和编辑
嵌入
文件操作
微调
内容审核

这些全面的支持使得开发者可以轻松地在Go应用中集成OpenAI的各种功能。

高级功能

Function Calling

OpenAIGo支持OpenAI的function calling功能,这允许模型调用预定义的函数。使用方法如下:

request := openaigo.ChatRequest{
  Messages: []openaigo.Message{
    {Role: "user", Content: "How's the weather today in Tokyo?"},
  },
  Functions: []openaigo.Function{
    {
      Name: "get_weather",
      Parameters: openaigo.Parameters{
        Type: "object",
        Properties: map[string]map[string]any{
          "location": {"type": "string"},
          "date": {"type": "string", "description": "ISO 8601 date string"},
        },
        Required: []string{"location"},
      },
    },
  },
}

流式响应

对于需要实时响应的应用,OpenAIGo提供了流式API支持:

client := openaigo.NewClient(OPENAI_API_KEY)
request := openaigo.ChatRequest{
  Stream: true,
  StreamCallback: func(res ChatCompletionResponse, done bool, err error) {
    // 处理流式响应
  },
}

代理支持

在某些网络环境下,你可能需要使用代理。OpenAIGo允许你轻松配置HTTP客户端:

client := openaigo.NewClient(OPENAI_API_KEY)
transport := &http.Transport{ Proxy: http.ProxyFromEnvironment }
client.HTTPClient = &http.Client{ Transport: transport }