chatmodel

Documentation

Index

• type ChatModel
  • func NewChatModel(generateFunc GenerateFunc) *ChatModel
  • func (b *ChatModel) Call(ctx context.Context, messages []golc.ChatMessage) (golc.ChatMessage, error)
  • func (b *ChatModel) CallPrompt(ctx context.Context, promptValue golc.PromptValue) (golc.ChatMessage, error)
  • func (b *ChatModel) Generate(ctx context.Context, messages [][]golc.ChatMessage) (*golc.LLMResult, error)
  • func (b *ChatModel) GeneratePrompt(ctx context.Context, promptValues []golc.PromptValue) (*golc.LLMResult, error)
  • func (b *ChatModel) Predict(ctx context.Context, text string) (string, error)
  • func (b *ChatModel) PredictMessages(ctx context.Context, messages []golc.ChatMessage) (golc.ChatMessage, error)
• type GenerateFunc

Types

type ChatModel

type ChatModel struct {
	// contains filtered or unexported fields
}

func NewChatModel

func NewChatModel(generateFunc GenerateFunc) *ChatModel

func (*ChatModel) Call

func (b *ChatModel) Call(ctx context.Context, messages []golc.ChatMessage) (golc.ChatMessage, error)

func (*ChatModel) CallPrompt

func (b *ChatModel) CallPrompt(ctx context.Context, promptValue golc.PromptValue) (golc.ChatMessage, error)

func (*ChatModel) Generate

func (b *ChatModel) Generate(ctx context.Context, messages [][]golc.ChatMessage) (*golc.LLMResult, error)

func (*ChatModel) GeneratePrompt

func (b *ChatModel) GeneratePrompt(ctx context.Context, promptValues []golc.PromptValue) (*golc.LLMResult, error)

func (*ChatModel) Predict

func (b *ChatModel) Predict(ctx context.Context, text string) (string, error)

func (*ChatModel) PredictMessages

func (b *ChatModel) PredictMessages(ctx context.Context, messages []golc.ChatMessage) (golc.ChatMessage, error)

type GenerateFunc

type GenerateFunc func(ctx context.Context, messages []golc.ChatMessage) (*golc.LLMResult, error)