mengbin92 / openai

use OpenAI API with golang

OpenAI

Overview

This is a library implemented in Go for OpenAI API. It supports:

• Models
• Completions
• Chat
• Edits
• Images
• Embeddings
• Audio
• Files
• Fine-tunes
• Moderations

Install

go get github.com/mengbin92/openai

ChatGPT example code

package main

import (
	"context"
	"fmt"
	"os"

	"github.com/mengbin92/openai"
)

func main() {
	client := openai.NewClient("your token", "your org", "proxy")

	resp, err := client.CreateChatCompletion(
		context.Background(),
		&openai.ChatCompletionRequset{
			Model: openai.GPT3Dot5Turbo,
			Messages: []openai.Message{
				{Role: openai.ChatMessageRoleUser, Content: "hi!"},
			},
		},
	)
	if err != nil {
		fmt.Printf("CreateChatCompletion error: %s\n", err.Error())
		os.Exit(-1)
	}
	fmt.Println(resp.Choices[0].Message.Content)
}