go_openai

Documentation

Index

• type Driver
  • func NewDriver(config *rcconfig.ChatBot) *Driver
  • func (d *Driver) CreateChatCompletion(ctx context.Context, message *model2.Message, role model2.Role) (*model2.Message, error)
  • func (d *Driver) CreateChatCompletionStream(ctx context.Context, message *model2.Message, role model2.Role, ...) (*model2.Message, error)
  • func (d *Driver) CreateCompletion(ctx context.Context, message *model2.Message) (*model2.Message, error)
  • func (d *Driver) CreateCompletionStream(ctx context.Context, message *model2.Message, role model2.Role, ...) (*model2.Message, error)
  • func (d *Driver) GetConfig() *rcconfig.ChatBot
  • func (d *Driver) GetConversationHistory(conversation *model2.Conversation) []*model2.Message
  • func (d *Driver) SetConfig(config *rcconfig.ChatBot)
  • func (d *Driver) SetMaxAnswerLength(length int) error
  • func (d *Driver) SetTemperature(temperature float64) error
  • func (d *Driver) StartModel() error
  • func (d *Driver) StopModel() error

Types

type Driver

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

func NewDriver

func NewDriver(config *rcconfig.ChatBot) *Driver

func (*Driver) CreateChatCompletion

func (d *Driver) CreateChatCompletion(ctx context.Context, message *model2.Message, role model2.Role) (*model2.Message, error)

SendMessage 向指定对话发送消息

func (*Driver) CreateChatCompletionStream

func (d *Driver) CreateChatCompletionStream(ctx context.Context, message *model2.Message, role model2.Role, processStreamData func(data string, status model.ChatStatus) error) (*model2.Message, error)

func (*Driver) CreateCompletion

func (d *Driver) CreateCompletion(ctx context.Context, message *model2.Message) (*model2.Message, error)

GenerateAnswer 生成无上下文回答

func (*Driver) CreateCompletionStream

func (d *Driver) CreateCompletionStream(ctx context.Context, message *model2.Message, role model2.Role, processStreamData func(data string, status model.ChatStatus) error) (*model2.Message, error)

func (*Driver) GetConfig

func (d *Driver) GetConfig() *rcconfig.ChatBot

GetConfig 获取基本配置

func (*Driver) GetConversationHistory

func (d *Driver) GetConversationHistory(conversation *model2.Conversation) []*model2.Message

GetConversationHistory 获取指定对话的对话历史记录

func (*Driver) SetConfig

func (d *Driver) SetConfig(config *rcconfig.ChatBot)

SetConfig 设置基本配置

func (*Driver) SetMaxAnswerLength

func (d *Driver) SetMaxAnswerLength(length int) error

SetMaxAnswerLength 设置回答的最大长度

func (*Driver) SetTemperature

func (d *Driver) SetTemperature(temperature float64) error

SetTemperature 设置模型温度

func (*Driver) StartModel

func (d *Driver) StartModel() error

StartModel 启动 ChatGPT 模型

func (*Driver) StopModel

func (d *Driver) StopModel() error

StopModel 停止 ChatGPT 模型