memory

type Embedder interface {
	Embed(ctx context.Context, texts []string) ([][]float32, error)
}

type Entry struct {
	ID        string            `json:"id"`
	Content   string            `json:"content"`
	Tags      []string          `json:"tags,omitempty"`
	Scope     string            `json:"scope,omitempty"`
	Metadata  map[string]string `json:"metadata,omitempty"`
	Embedding []float32         `json:"embedding,omitempty"`
	CreatedAt time.Time         `json:"created_at"`
	UpdatedAt time.Time         `json:"updated_at"`
}

type Filter struct {
	// Scope matches entries with exactly this scope when non-empty.
	Scope string
	// Tags matches entries that carry all of these tags.
	Tags []string
}

type Hit struct {
	Entry *Entry
	Score float64
}

type Hybrid struct {
	Retrievers []Retriever
	// K is the RRF constant. Defaults to 60.
	K float64
	// OnError receives the error of each retriever that fails while at
	// least one other succeeds. If nil, the errors are logged as warnings
	// with slog.Default, so a retriever that is permanently broken does not
	// go unnoticed.
	OnError func(err error)
}

type Keyword struct {
	Store Store
	// K1 and B are the BM25 parameters. Zero selects the defaults, 1.2 and
	// 0.75; a negative value selects 0, so B < 0 turns off document length
	// normalization.
	K1, B float64
}

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

type Memory interface {
	Remember(ctx context.Context, content string, opts ...RememberOption) (*Entry, error)
	Recall(ctx context.Context, query string, opts ...RecallOption) ([]Hit, error)
	Forget(ctx context.Context, id string) error
}

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

type Option func(*Manager)

type Query struct {
	Text   string
	Limit  int
	Filter Filter
	// MinScore drops hits scoring below it. Scores are retriever specific.
	MinScore float64
}

type RecallOption func(*Query)

type RememberOption func(*Entry)

type Retriever interface {
	Retrieve(ctx context.Context, q Query) ([]Hit, error)
}

type ScopeFunc func(ctx context.Context) (string, error)

type ScopedDeleter interface {
	DeleteInScope(ctx context.Context, id, scope string) error
}

type ScopedForgetter interface {
	ForgetInScope(ctx context.Context, id, scope string) error
}

type Store interface {
	// Put inserts or replaces an entry by ID.
	Put(ctx context.Context, e *Entry) error
	// Get returns an entry or ErrNotFound.
	Get(ctx context.Context, id string) (*Entry, error)
	Delete(ctx context.Context, id string) error
	// List returns matching entries, oldest first.
	List(ctx context.Context, f Filter) ([]*Entry, error)
}

type Vector struct {
	Store    Store
	Embedder Embedder
}