memory

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

type CitationsMode string

type EmbeddingProvider interface {
	// Embed generates a single embedding
	Embed(text string) ([]float32, error)
	// EmbedBatch generates multiple embeddings in one call
	EmbedBatch(texts []string) ([][]float32, error)
	// Dimension returns the dimension of embeddings
	Dimension() int
	// MaxBatchSize returns the maximum batch size
	MaxBatchSize() int
}

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

type HybridKeywordResult struct {
	ID        string
	Path      string
	StartLine int
	EndLine   int
	Source    string
	Snippet   string
	TextScore float64
}

type HybridVectorResult struct {
	ID          string
	Path        string
	StartLine   int
	EndLine     int
	Source      string
	Snippet     string
	VectorScore float64
}

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

type MMRConfig struct {
	// Enabled enables MMR re-ranking
	Enabled bool `json:"enabled"`
	// Lambda balances relevance vs diversity: 0=max diversity, 1=max relevance
	Lambda float64 `json:"lambda"`
}

type ManagerConfig struct {
	Store        Store
	Provider     EmbeddingProvider
	CacheMaxSize int
}

type MemoryItem struct {
	Text     string
	Source   MemorySource
	Type     MemoryType
	Metadata MemoryMetadata
}

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

type MemoryMetadata struct {
	// FilePath is the source file path (if from file)
	FilePath string `json:"file_path,omitempty"`
	// LineNumber is the line in the source file
	LineNumber int `json:"line_number,omitempty"`
	// EndLineNumber is the end line for multi-line snippets
	EndLineNumber int `json:"end_line_number,omitempty"`
	// SessionKey is the session identifier (if from session)
	SessionKey string `json:"session_key,omitempty"`
	// Tags are user-defined tags
	Tags []string `json:"tags,omitempty"`
	// Importance is a user-assigned importance score (0-1)
	Importance float64 `json:"importance,omitempty"`
	// AccessCount tracks how often this memory was accessed
	AccessCount int `json:"access_count,omitempty"`
	// LastAccessed is when this memory was last retrieved
	LastAccessed time.Time `json:"last_accessed,omitempty"`
	// Timestamp is the original timestamp for session/daily memories
	Timestamp *time.Time `json:"timestamp,omitempty"`
}

type MemorySearchManager interface {
	// Search 执行语义搜索
	Search(ctx context.Context, query string, opts SearchOptions) ([]*SearchResult, error)
	// Add 添加记忆（仅 builtin 支持）
	Add(ctx context.Context, text string, source MemorySource, memType MemoryType, metadata MemoryMetadata) error
	// GetStatus 获取状态
	GetStatus() map[string]interface{}
	// Close 关闭
	Close() error
}

type MemorySource string

type MemoryStats struct {
	TotalCount int `json:"total_count"`
	CacheSize  int `json:"cache_size"`
}

type MemoryType string

type OpenAIConfig struct {
	APIKey     string
	BaseURL    string
	Model      string
	Timeout    time.Duration
	MaxRetries int
}

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

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

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

type SearchOptions struct {
	// Limit is the maximum number of results to return
	Limit int `json:"limit"`
	// MinScore is the minimum similarity score (0-1)
	MinScore float64 `json:"min_score"`
	// Sources filters which memory sources to search
	Sources []MemorySource `json:"sources,omitempty"`
	// Types filters which memory types to search
	Types []MemoryType `json:"types,omitempty"`
	// Hybrid enables hybrid vector+keyword search
	Hybrid bool `json:"hybrid"`
	// VectorWeight is the weight for vector similarity in hybrid search (0-1)
	VectorWeight float64 `json:"vector_weight"`
	// TextWeight is the weight for keyword match in hybrid search (0-1)
	TextWeight float64 `json:"text_weight"`
	// MMR enables Maximal Marginal Relevance re-ranking for diversity
	MMR *MMRConfig `json:"mmr,omitempty"`
	// TemporalDecay enables time-aware scoring
	TemporalDecay *TemporalDecayConfig `json:"temporal_decay,omitempty"`
	// IncludeCitations adds citation strings to results
	IncludeCitations bool `json:"include_citations,omitempty"`
}

type SearchResult struct {
	VectorEmbedding
	Score       float64 `json:"score"`
	MatchedText string  `json:"matched_text"`
	Highlight   string  `json:"highlight,omitempty"`
	// VectorScore is the pure vector similarity score (before hybrid/decay)
	VectorScore float64 `json:"vector_score,omitempty"`
	// TextScore is the keyword match score (for hybrid search)
	TextScore float64 `json:"text_score,omitempty"`
	// Citation is the formatted citation string (e.g., "memory/2024-02-15.md#L10-L20")
	Citation string `json:"citation,omitempty"`
	// AgeInDays is the age of the memory in days (for temporal decay)
	AgeInDays float64 `json:"age_in_days,omitempty"`
}

type Store interface {
	// Add adds a memory to the store
	Add(embedding *VectorEmbedding) error
	// AddBatch adds multiple memories in one transaction
	AddBatch(embeddings []*VectorEmbedding) error
	// Search performs similarity search
	Search(query []float32, opts SearchOptions) ([]*SearchResult, error)
	// Get retrieves a memory by ID
	Get(id string) (*VectorEmbedding, error)
	// Delete removes a memory by ID
	Delete(id string) error
	// Update updates an existing memory
	Update(embedding *VectorEmbedding) error
	// List lists all memories with optional filtering
	List(filter func(*VectorEmbedding) bool) ([]*VectorEmbedding, error)
	// Close closes the store
	Close() error
}

type StoreConfig struct {
	// DBPath is the path to the SQLite database file
	DBPath string
	// Provider is the embedding provider to use
	Provider EmbeddingProvider
	// EnableVectorSearch enables vector similarity search (requires sqlite-vec)
	EnableVectorSearch bool
	// EnableFTS enables full-text search
	EnableFTS bool
	// VectorExtensionPath is the path to the sqlite-vec extension
	VectorExtensionPath string
}

type TemporalDecayConfig struct {
	// Enabled enables temporal decay
	Enabled bool `json:"enabled"`
	// HalfLifeDays is the half-life for exponential decay (default: 30 days)
	HalfLifeDays float64 `json:"half_life_days"`
}

type VectorEmbedding struct {
	ID        string         `json:"id"`
	Vector    []float32      `json:"vector"`
	Dimension int            `json:"dimension"`
	Text      string         `json:"text"`
	Source    MemorySource   `json:"source"`
	Type      MemoryType     `json:"type"`
	CreatedAt time.Time      `json:"created_at"`
	UpdatedAt time.Time      `json:"updated_at"`
	Metadata  MemoryMetadata `json:"metadata"`
}