xgpt3

README

xgpt3

go-openai API wrapper that supports conversation

Installation

go get github.com/fanchunke/xgpt3

Usage

package main

import (
	"context"
	"fmt"
	"log"

	"github.com/fanchunke/xgpt3"
	"github.com/fanchunke/xgpt3/conversation/ent"
	"github.com/fanchunke/xgpt3/conversation/ent/chatent"
	"github.com/sashabaranov/go-openai"

	_ "github.com/go-sql-driver/mysql"
)

func main() {
	// 连接数据库
	entClient, err := chatent.Open("mysql", "root:12345678@tcp(127.0.0.1:3306)/chatgpt?parseTime=True")
	if err != nil {
		log.Fatalf("Open database failed: %s", err)
	}

	// 生成数据库表
	if err := entClient.Schema.Create(context.Background()); err != nil {
		log.Fatalf("Create database schema failed: %s", err)
	}

	// conversation handler
	handler := ent.New(entClient)

	// openai client
	gptClient := openai.NewClient("authToken")

	// xgpt3 client
	xgpt3Client := xgpt3.NewClient(gptClient, handler)

	// 请求
	req := openai.CompletionRequest{
		Model:           openai.GPT3TextDavinci003,
		MaxTokens:       100,
		Prompt:          "Lorem ipsum",
		TopP:            1,
		Temperature:     0.9,
		PresencePenalty: 0.6,
		User:            "fanchunke",
	}
	resp, err := xgpt3Client.CreateConversationCompletion(context.Background(), req)
	if err != nil {
		log.Fatalf("CreateConversationCompletion failed: %s", err)
	}
	fmt.Println(resp.Choices[0].Text)
}

Documentation

Index

• type Client
  • func NewClient(client *openai.Client, ch conversation.Handler) *Client
  • func (c *Client) CloseConversation(ctx context.Context, userId string) error
  • func (c *Client) CreateChatCompletion(ctx context.Context, request openai.ChatCompletionRequest) (openai.ChatCompletionResponse, error)
  • func (c *Client) CreateChatCompletionWithChannel(ctx context.Context, request openai.ChatCompletionRequest, channel string) (openai.ChatCompletionResponse, error)
  • func (c *Client) CreateConversationCompletion(ctx context.Context, request openai.CompletionRequest) (openai.CompletionResponse, error)
  • func (c *Client) CreateConversationCompletionWithChannel(ctx context.Context, request openai.CompletionRequest, channel string) (openai.CompletionResponse, error)
  • func (c *Client) WithLogger(l zerolog.Logger) *Client
  • func (c *Client) WithMaxTurn(n int) *Client

Types

type Client

type Client struct {
	*openai.Client
	// contains filtered or unexported fields
}

func NewClient

func NewClient(client *openai.Client, ch conversation.Handler) *Client

func (*Client) CloseConversation

func (c *Client) CloseConversation(ctx context.Context, userId string) error

func (*Client) CreateChatCompletion

func (c *Client) CreateChatCompletion(ctx context.Context, request openai.ChatCompletionRequest) (openai.ChatCompletionResponse, error)

func (*Client) CreateChatCompletionWithChannel

func (c *Client) CreateChatCompletionWithChannel(ctx context.Context, request openai.ChatCompletionRequest, channel string) (openai.ChatCompletionResponse, error)

func (*Client) CreateConversationCompletion

func (c *Client) CreateConversationCompletion(
	ctx context.Context,
	request openai.CompletionRequest,
) (openai.CompletionResponse, error)

func (*Client) CreateConversationCompletionWithChannel

func (c *Client) CreateConversationCompletionWithChannel(
	ctx context.Context,
	request openai.CompletionRequest,
	channel string,
) (openai.CompletionResponse, error)

func (*Client) WithLogger

func (c *Client) WithLogger(l zerolog.Logger) *Client

func (*Client) WithMaxTurn

func (c *Client) WithMaxTurn(n int) *Client