xgpt3

README

xgpt3

go-gpt3 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"
	gogpt "github.com/sashabaranov/go-gpt3"

	_ "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)

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

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

	// 请求
	req := gogpt.CompletionRequest{
		Model:           gogpt.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 *gogpt.Client, ch conversation.Handler) *Client
  • func (c *Client) CloseConversation(ctx context.Context, userId string) error
  • func (c *Client) CreateConversationChatCompletion(ctx context.Context, request gogpt.ChatCompletionRequest) (gogpt.ChatCompletionResponse, error)
  • func (c *Client) CreateConversationChatCompletionWithChannel(ctx context.Context, request gogpt.ChatCompletionRequest, channel string) (gogpt.ChatCompletionResponse, error)
  • func (c *Client) CreateConversationCompletion(ctx context.Context, request gogpt.CompletionRequest) (gogpt.CompletionResponse, error)
  • func (c *Client) CreateConversationCompletionWithChannel(ctx context.Context, request gogpt.CompletionRequest, channel string) (gogpt.CompletionResponse, error)
  • func (c *Client) WithLogger(l zerolog.Logger) *Client
  • func (c *Client) WithMaxTurn(n int) *Client

Types

type Client

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

func NewClient

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

func (*Client) CloseConversation

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

func (*Client) CreateConversationChatCompletion

func (c *Client) CreateConversationChatCompletion(
	ctx context.Context,
	request gogpt.ChatCompletionRequest,
) (gogpt.ChatCompletionResponse, error)

func (*Client) CreateConversationChatCompletionWithChannel

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

func (*Client) CreateConversationCompletion

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

func (*Client) CreateConversationCompletionWithChannel

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

func (*Client) WithLogger

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

func (*Client) WithMaxTurn

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