memory

Documentation

Index

• type BufferMemory
  • func NewBufferedMemory(maxHistory int) *BufferMemory
  • func (memory *BufferMemory) Add(msg *engines.ChatMessage) error
  • func (memory *BufferMemory) AddPrompt(prompt *engines.ChatPrompt) error
  • func (memory *BufferMemory) PromptWithContext(nextMessages ...*engines.ChatMessage) (*engines.ChatPrompt, error)
• type Memory
• type SummarisedMemory
  • func NewSummarisedMemory(recentMessageLimit int, model engines.LLM) *SummarisedMemory
  • func (memory *SummarisedMemory) Add(msg *engines.ChatMessage) error
  • func (memory *SummarisedMemory) AddPrompt(prompt *engines.ChatPrompt) error
  • func (memory *SummarisedMemory) PromptWithContext(nextMessages ...*engines.ChatMessage) (*engines.ChatPrompt, error)
• type TextEmbedder
• type Vectorstore
• type VectorstoreMemory

Types

type BufferMemory

type BufferMemory struct {
	MaxHistory int
	Buffer     []*engines.ChatMessage
}

func NewBufferedMemory

func NewBufferedMemory(maxHistory int) *BufferMemory

func (*BufferMemory) Add

func (memory *BufferMemory) Add(msg *engines.ChatMessage) error

func (*BufferMemory) AddPrompt

func (memory *BufferMemory) AddPrompt(prompt *engines.ChatPrompt) error

func (*BufferMemory) PromptWithContext

func (memory *BufferMemory) PromptWithContext(nextMessages ...*engines.ChatMessage) (*engines.ChatPrompt, error)

type Memory

type Memory interface {
	Add(msg *engines.ChatMessage) error
	AddPrompt(prompt *engines.ChatPrompt) error
	PromptWithContext(nextMessages ...*engines.ChatMessage) (*engines.ChatPrompt, error)
}

type SummarisedMemory

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

func NewSummarisedMemory

func NewSummarisedMemory(recentMessageLimit int, model engines.LLM) *SummarisedMemory

func (*SummarisedMemory) Add

func (memory *SummarisedMemory) Add(msg *engines.ChatMessage) error

func (*SummarisedMemory) AddPrompt

func (memory *SummarisedMemory) AddPrompt(prompt *engines.ChatPrompt) error

func (*SummarisedMemory) PromptWithContext

func (memory *SummarisedMemory) PromptWithContext(nextMessages ...*engines.ChatMessage) (*engines.ChatPrompt, error)

type TextEmbedder

type TextEmbedder interface {
	Embed(text string) []float64
}

type Vectorstore

type Vectorstore interface {
	Store(key []float64, value string) error
	FindNearest(key []float64, k int) ([]string, error)
}

type VectorstoreMemory

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