memory

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

type Context struct {
	Facts         []Fact
	Conversations []Conversation
}

type Conversation struct {
	ID          int64
	SessionID   string
	Timestamp   time.Time
	Query       string
	Response    string
	Metadata    map[string]string
	ExtractedAt *time.Time // Tracks when facts were extracted to prevent duplicates
}

type ConversationTurn struct {
	SessionID string
	Timestamp time.Time
	Query     string
	Response  string
}

type Extractor interface {
	Extract(ctx context.Context, conversations []Conversation) ([]Fact, error)
}

type Fact struct {
	ID         int64
	Category   FactCategory
	Text       string
	Confidence float64 // 0.0-1.0
	SourceIDs  []int64 // conversation IDs
	CreatedAt  time.Time
	UpdatedAt  time.Time
}

type FactCategory string

type Filter struct {
	SessionID         string
	Since             time.Time
	NotExtractedSince time.Time // Load conversations not extracted since this time
	Categories        []FactCategory
	Limit             int
}

type MemoryService interface {
	// Remember stores a conversation turn
	Remember(ctx context.Context, turn ConversationTurn) error

	// Recall retrieves relevant context for a query
	Recall(ctx context.Context, query string, opts RecallOptions) (Context, error)

	// ExtractFacts extracts facts from recent conversations
	ExtractFacts(ctx context.Context) error

	// StartBackgroundExtraction starts periodic fact extraction
	StartBackgroundExtraction(interval time.Duration)

	// Close cleans up resources
	Close() error
}

type RecallOptions struct {
	IncludeFacts         bool
	IncludeConversations bool
	FactLimit            int
	ConversationLimit    int
}

type Retriever interface {
	GetRelevantFacts(ctx context.Context, query string, limit int) ([]Fact, error)
}

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

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

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

type Storage interface {
	SaveConversation(ctx context.Context, conv Conversation) error
	LoadConversations(ctx context.Context, filter Filter) ([]Conversation, error)
	MarkExtracted(ctx context.Context, convIDs []int64, extractedAt time.Time) error
	SaveFact(ctx context.Context, fact Fact) error
	LoadFacts(ctx context.Context, filter Filter) ([]Fact, error)
	Close() error
}