chatGLM

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) OverrideGetUrlFromEndpoint() func(endpoint string) (string, error)
  • func (d *Driver) SetMaxAnswerLength(length int) error
  • func (d *Driver) SetTemperature(temperature float64) error
  • func (d *Driver) StartModel() error
  • func (d *Driver) StopModel() error
• type GLMRequest
• type GLMResponse

Types

type Driver

type Driver struct {
	*ArtisanCloud.BaseDriver
}

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) OverrideGetUrlFromEndpoint

func (d *Driver) OverrideGetUrlFromEndpoint() func(endpoint string) (string, error)

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

type GLMRequest

type GLMRequest struct {
	Prompt  string        `json:"prompt"`
	History []interface{} `json:"history"`
}

type GLMResponse

type GLMResponse struct {
	Response string     `json:"response"`
	History  [][]string `json:"history"`
	Status   int        `json:"status"`
	Time     string     `json:"time"`
}