使用openaigo库集成OpenAI GPT-3/GPT-3.5 API

openaigo是一个简单易用的Go语言客户端库，用于与OpenAI的GPT-3、GPT-3.5和ChatGPT API进行交互。下面我将详细介绍如何使用这个库。

安装openaigo

首先，使用go get安装openaigo库：

go get github.com/otiai10/openaigo

基本使用示例

1. 初始化客户端

package main

import (
	"context"
	"fmt"
	"github.com/otiai10/openaigo"
)

func main() {
	// 使用你的OpenAI API密钥初始化客户端
	client := openaigo.NewClient("your-api-key-here")
	
	// 可选: 设置组织ID(如果你属于多个组织)
	// client.Organization = "org-your-organization-id"
}

2. 发送简单的文本补全请求

func simpleCompletion(client *openaigo.Client) {
	request := openaigo.CompletionRequest{
		Model:     "text-davinci-003",  // GPT-3模型
		Prompt:    "Go语言的特点是",
		MaxTokens: 100,
		Temperature: 0.7,
	}
	
	response, err := client.Completion(context.Background(), request)
	if err != nil {
		fmt.Printf("Completion error: %v\n", err)
		return
	}
	
	fmt.Println("Completion result:")
	for _, choice := range response.Choices {
		fmt.Println(choice.Text)
	}
}

3. 使用ChatGPT API (GPT-3.5-turbo)

func chatCompletion(client *openaigo.Client) {
	request := openaigo.ChatRequest{
		Model: "gpt-3.5-turbo",
		Messages: []openaigo.Message{
			{Role: "system", Content: "你是一个有帮助的助手。"},
			{Role: "user", Content: "Go语言的主要特性是什么？"},
		},
		MaxTokens: 200,
		Temperature: 0.7,
	}
	
	response, err := client.Chat(context.Background(), request)
	if err != nil {
		fmt.Printf("Chat completion error: %v\n", err)
		return
	}
	
	fmt.Println("Chat completion result:")
	for _, choice := range response.Choices {
		fmt.Printf("Role: %s\nContent: %s\n", choice.Message.Role, choice.Message.Content)
	}
}

高级功能

1. 流式响应

func streamCompletion(client *openaigo.Client) {
	request := openaigo.ChatRequest{
		Model: "gpt-3.5-turbo",
		Messages: []openaigo.Message{
			{Role: "user", Content: "用Go语言实现一个简单的HTTP服务器"},
		},
		Stream: true,
	}
	
	// 创建上下文以便可以取消流
	ctx, cancel := context.WithCancel(context.Background())
	defer cancel()
	
	// 创建通道接收流事件
	stream, err := client.ChatStream(ctx, request)
	if err != nil {
		fmt.Printf("Stream error: %v\n", err)
		return
	}
	
	fmt.Println("Streaming response:")
	for event := range stream {
		if event.Error != nil {
			fmt.Printf("Stream error: %v\n", event.Error)
			break
		}
		for _, choice := range event.Choices {
			fmt.Print(choice.Delta.Content)
		}
	}
	fmt.Println()
}

2. 自定义HTTP客户端

func customHTTPClient() {
	client := openaigo.NewClient("your-api-key-here")
	
	// 自定义HTTP客户端配置
	client.HTTPClient = &http.Client{
		Timeout: 30 * time.Second,
		Transport: &http.Transport{
			MaxIdleConns:        10,
			IdleConnTimeout:     30 * time.Second,
			DisableCompression:  true,
		},
	}
	
	// 现在可以使用自定义HTTP客户端的client进行API调用
}

错误处理

func handleErrors(client *openaigo.Client) {
	_, err := client.Chat(context.Background(), openaigo.ChatRequest{
		Model: "invalid-model",
		Messages: []openaigo.Message{
			{Role: "user", Content: "测试错误处理"},
		},
	})
	
	if err != nil {
		// 检查是否是API错误
		if apiErr, ok := err.(*openaigo.APIError); ok {
			fmt.Printf("API Error: %s (Status: %d)\n", apiErr.Message, apiErr.StatusCode)
		} else {
			fmt.Printf("Other error: %v\n", err)
		}
		return
	}
}

最佳实践

1. 环境变量管理API密钥：不要将API密钥硬编码在代码中

   apiKey := os.Getenv("OPENAI_API_KEY")
   client := openaigo.NewClient(apiKey)
   

2. 设置合理的超时：为API调用设置适当的超时

   ctx, cancel := context.WithTimeout(context.Background(), 30*time.Second)
   defer cancel()
   response, err := client.Chat(ctx, request)
   

3. 限制最大token数：根据你的需求设置合理的max_tokens值，避免不必要的费用

4. 处理速率限制：实现重试逻辑处理速率限制错误

openaigo库提供了简洁的接口来与OpenAI的各种API交互，通过上述示例你可以快速集成GPT-3/GPT-3.5到你的Go应用程序中。