memory

type AddRequest struct {
	AgentID  string
	Content  string
	Metadata map[string]any
	Embed    bool // whether to generate embedding
}

type Backend interface {
	MemoryStore
	Initialize(ctx context.Context, config *BackendConfig) error
	Close() error
	Health() error
}

type BackendConfig struct {
	Type    string         `json:"type"` // sqlite, postgres, mem0, chroma
	DSN     string         `json:"dsn,omitempty"`
	Options map[string]any `json:"options,omitempty"`
}

type ListOptions struct {
	Limit  int
	Offset int
}

type Memory struct {
	ID        string
	AgentID   string
	Content   string
	Embedding []float64
	Metadata  map[string]any
	CreatedAt time.Time
	UpdatedAt time.Time
	Score     float64 // relevance score for search results
}

type MemoryStore interface {
	Add(ctx context.Context, req *AddRequest) (*Memory, error)
	Get(ctx context.Context, id string) (*Memory, error)
	Update(ctx context.Context, id string, req *UpdateRequest) error
	Delete(ctx context.Context, id string) error
	Search(ctx context.Context, query string, opts *SearchOptions) ([]*Memory, error)
	SearchSimilar(ctx context.Context, embedding []float64, opts *SearchOptions) ([]*Memory, error)
	ListByAgent(ctx context.Context, agentID string, opts *ListOptions) ([]*Memory, error)
	ClearAgentMemory(ctx context.Context, agentID string) error
	BatchAdd(ctx context.Context, reqs []*AddRequest) ([]*Memory, error)
}

type SearchOptions struct {
	Limit     int
	Offset    int
	Threshold float64
	Filters   map[string]any
}

type SearchResult struct {
	ID       string
	Score    float64
	Metadata map[string]any
}

type UpdateRequest struct {
	Content  string
	Metadata map[string]any
}

type VectorStore interface {
	Insert(ctx context.Context, id string, embedding []float64, metadata map[string]any) error
	Search(ctx context.Context, embedding []float64, k int) ([]*SearchResult, error)
	Delete(ctx context.Context, id string) error
}