chat

Documentation

Overview

Package chat provides a chat prompt template. Sometimes you need to define a chat prompt, this package provides a way to do that.

Index

• type Chat
  • func New(promptMessages PromptMessages) *Chat
  • func (p *Chat) AddMessagePromptTemplate(message PromptMessage)
  • func (p *Chat) ToMessages() (Messages, error)
• type Message
• type MessageType
• type Messages
• type Prompt
• type PromptMessage
• type PromptMessages

Types

type Chat

type Chat struct {
	PromptMessages PromptMessages
}

func New

func New(promptMessages PromptMessages) *Chat

func (*Chat) AddMessagePromptTemplate

func (p *Chat) AddMessagePromptTemplate(message PromptMessage)

func (*Chat) ToMessages

func (p *Chat) ToMessages() (Messages, error)

ToMessages converts the chat prompt template to a list of messages.

type Message

type Message struct {
	Type    MessageType
	Content string
}

type MessageType

type MessageType string

const (
	MessageTypeSystem    MessageType = "system"
	MessageTypeUser      MessageType = "user"
	MessageTypeAssistant MessageType = "assistant"
)

type Messages

type Messages []Message

type Prompt

type Prompt interface {
	Prompt() string
	Format(input interface{}) error
}

type PromptMessage

type PromptMessage struct {
	Type   MessageType
	Prompt Prompt
}

type PromptMessages

type PromptMessages []PromptMessage