graph

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

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

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

type GraphNode struct {
	ID        string
	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 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 TextGenerator interface {
	GenerateText(ctx context.Context, systemPrompt, userPrompt string) (string, error)
}

type Triple struct {
	Subject   string
	Predicate string
	Object    string
	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)
}