graph

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

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

type ExtractorOption func(*Extractor)

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

type GraphNode struct {
	ID        string
	NodeType  string // ObjectType name from triple metadata (empty = untyped)
	Predicate string // the edge that led here
	FromNode  string // the source node this was discovered from
	Depth     int
}

type GraphRAGResult struct {
	// VectorResults are the original vector-search results.
	VectorResults []VectorResult
	// GraphResults are additional results discovered via graph traversal.
	GraphResults []GraphNode
}

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

type GraphRequest struct {
	Triples []Triple
}

type Predicate string

type PredicateValidatorFunc func(name string) bool

type Store interface {
	// AddTriple adds a single triple to the graph.
	AddTriple(ctx context.Context, t Triple) error

	// AddTriples adds multiple triples atomically.
	AddTriples(ctx context.Context, triples []Triple) error

	// RemoveTriple removes a triple from the graph.
	RemoveTriple(ctx context.Context, t Triple) error

	// QueryBySubject returns all triples with the given subject.
	QueryBySubject(ctx context.Context, subject string) ([]Triple, error)

	// QueryByObject returns all triples with the given object.
	QueryByObject(ctx context.Context, object string) ([]Triple, error)

	// QueryBySubjectPredicate returns triples matching subject and predicate.
	QueryBySubjectPredicate(ctx context.Context, subject, predicate string) ([]Triple, error)

	// Traverse performs a breadth-first traversal from a start node.
	// predicates filters which edge types to follow (empty = all).
	Traverse(ctx context.Context, startNode string, maxDepth int, predicates []string) ([]Triple, error)

	// Count returns the total number of triples in the store.
	Count(ctx context.Context) (int, error)

	// PredicateStats returns the number of triples for each predicate type.
	PredicateStats(ctx context.Context) (map[string]int, error)

	// AllTriples returns every triple in the store.
	AllTriples(ctx context.Context) ([]Triple, error)

	// ClearAll removes all triples from the store.
	ClearAll(ctx context.Context) error

	// Close closes the underlying store.
	Close() error
}

type Triple struct {
	Subject     string
	Predicate   string
	Object      string
	SubjectType string // ObjectType name for subject (empty = untyped)
	ObjectType  string // ObjectType name for object (empty = untyped)
	Metadata    map[string]string
}

type VectorResult struct {
	Collection string
	SourceID   string
	Content    string
	Distance   float32
}

type VectorRetrieveOptions struct {
	Collections []string
	Limit       int
	SessionKey  string
	MaxDistance float32
}

type VectorRetriever interface {
	Retrieve(ctx context.Context, query string, opts VectorRetrieveOptions) ([]VectorResult, error)
}