search

Documentation

Index

• type ChunkResult
• type GlobalSearchResult
  • func GlobalSearch(ctx context.Context, st *store.Store, emb *embedder.Embedder, ...) (*GlobalSearchResult, error)
• type LocalSearchResult
  • func LocalSearch(ctx context.Context, st *store.Store, emb *embedder.Embedder, query string, ...) (*LocalSearchResult, error)

Types

type ChunkResult

type ChunkResult struct {
	Chunk     store.Chunk
	Score     float32
	EntityIDs []string
}

ChunkResult is a search result from chunk/vector search.

type GlobalSearchResult

type GlobalSearchResult struct {
	Answer      string
	Communities []*store.Community
}

GlobalSearchResult aggregates community summaries.

func GlobalSearch

func GlobalSearch(ctx context.Context, st *store.Store, emb *embedder.Embedder, prov llm.Provider, query string, communityLevel int) (*GlobalSearchResult, error)

GlobalSearch performs community summary aggregation (GraphRAG global search).

type LocalSearchResult

type LocalSearchResult struct {
	Chunks   []ChunkResult
	Entities []*store.Entity
	Rels     []*store.Relationship
}

LocalSearchResult combines vector + graph results.

func LocalSearch

func LocalSearch(ctx context.Context, st *store.Store, emb *embedder.Embedder, query string, topK, graphDepth int) (*LocalSearchResult, error)

LocalSearch performs vector similarity search + graph walk.