inject

Documentation

Index

• type MemoryInjector
  • func NewMemoryInjector(recallEngine recall.Recaller, logger *zap.Logger, alwaysIncludeUserPrompt bool) *MemoryInjector
  • func (mi *MemoryInjector) FilterRecallResults(ctx context.Context, results []recall.RecallResult, query string) []recall.RecallResult
  • func (mi *MemoryInjector) FormatMemoriesForSystemMessage(memories []*memory.Memory) string
  • func (mi *MemoryInjector) InjectMemoriesIntoPrompt(prompt string, projectID string, maxItems int, maxTokens int) (string, error)
  • func (mi *MemoryInjector) SetCoVeVerifier(verifier *cove.Verifier)

Types

type MemoryInjector

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

MemoryInjector handles injecting memories into prompts

func NewMemoryInjector

func NewMemoryInjector(recallEngine recall.Recaller, logger *zap.Logger, alwaysIncludeUserPrompt bool) *MemoryInjector

NewMemoryInjector creates a new memory injector

func (*MemoryInjector) FilterRecallResults

func (mi *MemoryInjector) FilterRecallResults(ctx context.Context, results []recall.RecallResult, query string) []recall.RecallResult

FilterRecallResults applies optional CoVe recall filtering (fail-safe).

func (*MemoryInjector) FormatMemoriesForSystemMessage

func (mi *MemoryInjector) FormatMemoriesForSystemMessage(memories []*memory.Memory) string

FormatMemoriesForSystemMessage formats memories in a structured way for system messages

func (*MemoryInjector) InjectMemoriesIntoPrompt

func (mi *MemoryInjector) InjectMemoriesIntoPrompt(prompt string, projectID string, maxItems int, maxTokens int) (string, error)

InjectMemoriesIntoPrompt injects relevant memories into a prompt

func (*MemoryInjector) SetCoVeVerifier

func (mi *MemoryInjector) SetCoVeVerifier(verifier *cove.Verifier)

SetCoVeVerifier sets the CoVe verifier for recall filtering