在go语言中向OpenAI API发送请求实现chatgpt功能代码示例

Go  /  管理员 发布于 1年前   785

chatgpt功能就是聊天机器人程序，智能对话，下面是用go语言实现的请求对话；

实现功能：

使用了一个本地的SOCKS5代理来发送一个消息 "Hello!" 到 OpenAI 的 API，并打印 API 的响应。

示例代码：

package main

import (
    "bytes"
    "context"
    "encoding/json"
    "fmt"
    "golang.org/x/net/proxy"
    "io"
    "net"
    "net/http"
)

// ChatCompletionRequest 描述了向 OpenAI API 发送的请求格式。
type ChatCompletionRequest struct {
    Model    string     `json:"model"`
Messages[] ChatCompletionMessage `json:"messages"`
} 

// ChatCompletionMessage 描述了消息的格式，其中包含角色（如用户、AI）和内容。
type ChatCompletionMessage struct {
    Role    string `json:"role"`
    Content string `json:"content"`
}

func main() {
    socks5Addr := "127.0.0.1:10808" //本地代理
    // 创建SOCKS5拨号器
    dialer, err := proxy.SOCKS5("tcp", socks5Addr, nil, proxy.Direct)
    if err != nil {
        fmt.Println("Error creating dialer:", err)
        return
    }
    // 封装拨号器以支持DialContext
    dialerFunc := func(ctx context.Context, network, addr string) (conn net.Conn, err error) {
        return dialer.Dial(network, addr)
    }
    httpTransport := &http.Transport{
        DialContext: dialerFunc,
    }
    httpClient := &http.Client{
        Transport: httpTransport,
    }
    url := "https://api.openai.com/v1/chat/completions" // OpenAI completion endpoint
    apiKey := "your token" //chatgpt's api-key
    requestBody := ChatCompletionRequest{
        Model: "gpt-3.5-turbo",
        Messages: []ChatCompletionMessage{
            {
                Role:    "user",
                Content: "Hello!",
            },
        },
    }
    jsonBody, err := json.Marshal(requestBody)
    if err != nil {
        fmt.Println("Error marshalling JSON:", err)
        return
    }
    req, err := http.NewRequest("POST", url, bytes.NewBuffer(jsonBody))
    if err != nil {
        fmt.Println("Error creating request:", err)
        return
    }
    req.Header.Set("Content-Type", "application/json")
    req.Header.Set("Authorization", "Bearer "+apiKey)
    resp, err := httpClient.Do(req)
    if err != nil {
        fmt.Println("Error sending request:", err)
        return
    }
    defer resp.Body.Close()
    body, err := io.ReadAll(resp.Body)
    if err != nil {
        fmt.Println("Error reading response:", err)
        return
    }
    fmt.Println("Received:", string(body))
}

运行一下，看看效果：