graph

type Client interface {
	// EnsureSchema creates indexes and constraints if they don't exist.
	// vectorDim is injected into the vector index DDL (e.g. 768 for nomic-embed-text).
	EnsureSchema(ctx context.Context, vectorDim int) error

	// UpsertEntity creates or updates an entity node.
	UpsertEntity(ctx context.Context, entity models.Entity) error

	// SearchEntities finds entities by fulltext + embedding similarity.
	SearchEntities(ctx context.Context, query string, embedding []float32, project string, limit int) ([]EntityResult, error)

	// GetEntity retrieves a single entity by ID.
	GetEntity(ctx context.Context, id string) (*models.Entity, error)

	// UpsertFact creates a RELATES_TO edge between two entities.
	UpsertFact(ctx context.Context, fact models.Fact) error

	// SearchFacts finds facts by hybrid search (BM25 + cosine + BFS).
	SearchFacts(ctx context.Context, query string, embedding []float32, limit int) ([]FactResult, error)

	// InvalidateFact sets ExpiredAt and InvalidAt on a fact (never deletes).
	InvalidateFact(ctx context.Context, id string, expiredAt, invalidAt time.Time) error

	// GetFactsBetween returns all active facts between two entities.
	GetFactsBetween(ctx context.Context, sourceID, targetID string) ([]models.Fact, error)

	// GetFactsForEntity returns all active facts involving an entity.
	GetFactsForEntity(ctx context.Context, entityID string) ([]models.Fact, error)

	// AppendEpisode adds an episode/session ID to a fact's episodes list.
	AppendEpisode(ctx context.Context, factID, episodeID string) error

	// AppendMemoryToFact adds a memory ID to a fact's source_memory_ids.
	AppendMemoryToFact(ctx context.Context, factID, memoryID string) error

	// GetMemoryFacts returns all facts derived from a given memory.
	GetMemoryFacts(ctx context.Context, memoryID string) ([]models.Fact, error)

	// RecallByGraph returns memory IDs relevant to a query via graph traversal.
	RecallByGraph(ctx context.Context, query string, embedding []float32, limit int) ([]string, error)

	// CreateEpisode stores an episode node in the graph.
	CreateEpisode(ctx context.Context, episode models.Episode) error

	// GetEpisodesForMemory returns all episodes linked to a given memory ID.
	GetEpisodesForMemory(ctx context.Context, memoryID string) ([]models.Episode, error)

	// Healthy returns true if the graph database is reachable.
	Healthy(ctx context.Context) bool

	// Close releases resources.
	Close() error
}

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

type EntityResult struct {
	ID    string  `json:"id"`
	Name  string  `json:"name"`
	Type  string  `json:"type"`
	Score float64 `json:"score"`
}

type FactAction string

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

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

type FactResult struct {
	ID              string   `json:"id"`
	Fact            string   `json:"fact"`
	SourceEntityID  string   `json:"source_entity_id"`
	TargetEntityID  string   `json:"target_entity_id"`
	SourceMemoryIDs []string `json:"source_memory_ids"`
	Score           float64  `json:"score"`
}

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

type SearchFunc func(ctx context.Context, limit int) ([]FactResult, error)