go_openai

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 模型