chatBot

Documentation

Index

• type ChatBot
  • func NewChatBot(client contract.ChatBotClientInterface) (*ChatBot, error)
  • func (bot *ChatBot) CreateChatCompletion(ctx context.Context, message *model.Message, role model.Role) (*model.Message, error)
  • func (bot *ChatBot) CreateChatCompletionStream(ctx context.Context, message *model.Message, role model.Role, ...) (*model.Message, error)
  • func (bot *ChatBot) CreateCompletion(ctx context.Context, prompt *model.Message) (*model.Message, error)
  • func (bot *ChatBot) CreateCompletionStream(ctx context.Context, message *model.Message, role model.Role, ...) (*model.Message, error)

Types

type ChatBot

type ChatBot struct {
	*robots.Robot
	Client contract.ChatBotClientInterface
}

func NewChatBot

func NewChatBot(client contract.ChatBotClientInterface) (*ChatBot, error)

func (*ChatBot) CreateChatCompletion

func (bot *ChatBot) CreateChatCompletion(ctx context.Context, message *model.Message, role model.Role) (*model.Message, error)

SendMessage 向指定对话发送消息

func (*ChatBot) CreateChatCompletionStream

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

func (*ChatBot) CreateCompletion

func (bot *ChatBot) CreateCompletion(ctx context.Context, prompt *model.Message) (*model.Message, error)

GenerateAnswer 生成无上下文回答

func (*ChatBot) CreateCompletionStream

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