memory

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

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

type DocumentStore interface {
	// SaveDocuments 批量保存文档
	SaveDocuments(ctx context.Context, collection string, docs []*schema.Document) error

	// RecallDocuments 根据查询召回相关文档
	RecallDocuments(ctx context.Context, collection string, query string, topK int) ([]*schema.Document, error)

	// GetDocuments 获取集合中的所有文档
	GetDocuments(ctx context.Context, collection string) ([]*schema.Document, error)

	// DeleteCollection 删除整个集合
	DeleteCollection(ctx context.Context, collection string) error

	// Close 关闭存储
	Close() error
}

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

type EmbeddingFunc func(ctx context.Context, text string) ([]float32, error)

type Indexer interface {
	// AddToIndex 将消息的嵌入向量添加到索引
	AddToIndex(ctx context.Context, sessionID string, msgs []*schema.Message) error

	// RemoveFromIndex 从索引中移除指定会话的所有向量
	RemoveFromIndex(sessionID string) error

	// RebuildIndex 从存储中重建完整索引
	RebuildIndex()

	// IndexReady 返回索引是否就绪
	IndexReady() bool
}

type LongTermStore interface {
	Store
	Retriever
}

type MessageRecord struct {
	ID        string            `json:"id"`
	SessionID string            `json:"session_id"`
	Role      string            `json:"role"`
	Content   string            `json:"content"`
	Embedding []float32         `json:"embedding,omitempty"`
	Timestamp time.Time         `json:"timestamp"`
	Metadata  map[string]string `json:"metadata,omitempty"`
}

type Option func(*Controller)

type Retriever interface {
	// Recall 根据查询召回相关消息
	Recall(ctx context.Context, sessionID string, query string, topK int) ([]*schema.Message, error)

	// Close 关闭检索器
	Close() error
}

type ShortMemoryManager interface {
	// GetRecent 返回当前窗口内的完整消息（不进行压缩）
	GetRecent(sessionID string) []*schema.Message

	// GetContextMessages 返回构建上下文所需的短期记忆消息
	// 内部可能整合：窗口内的完整消息 + 超出部分的摘要
	GetContextMessages(sessionID string) []*schema.Message

	// AddTurn 添加一轮对话，触发窗口维护、摘要生成等
	AddTurn(sessionID string, msgs []*schema.Message)

	// Clear 清空会话短期记忆
	Clear(sessionID string)
}

type Store interface {
	// Save 保存消息
	Save(ctx context.Context, sessionID string, msgs []*schema.Message) error

	// GetSession 获取完整会话历史
	GetSession(ctx context.Context, sessionID string) ([]*schema.Message, error)

	// ClearSession 清空会话
	ClearSession(ctx context.Context, sessionID string) error

	// Close 关闭存储
	Close() error
}

type StoreEvent struct {
	Type      StoreEventType
	SessionID string
	Messages  []*schema.Message // Save 事件时有值
}

type StoreEventType int

type StoreHook func(event StoreEvent)