rag/
lib.rs

1pub mod keyword;
2pub mod hybrid;
3pub mod dedup;
4pub mod rerank;
5pub mod index_ivf;
6
7pub mod embeddings;
8pub mod vector_store;
9pub mod retriever;
10pub mod chunker;
11pub mod errors;
12pub mod mcp;
13pub mod index;
14pub mod ingestion;
15pub mod graph;
16pub mod graph_rag;
17
18pub use embeddings::{
19    EmbeddingModel, HttpEmbeddingModel, OpenAIEmbeddingModel, EmbeddingRequest, OllamaEmbeddingModel,
20};
21pub use vector_store::{
22    JsonPersistentVectorStore, VectorStore, InMemoryVectorStore, MinimalVectorDB, Document,
23    Similarity, MetadataFilter, load_all_documents,
24};
25pub use retriever::Retriever;
26pub use chunker::{TextChunker, FixedSizeChunker, ParagraphChunker, SentenceChunker};
27pub use errors::{RagError, Result};
28pub use mcp::RagMcpServer;
29pub use index::{DistanceMetric, FlatIndex, Index};
30pub use index_ivf::IvfflatIndex;
31pub use ingestion::{Source, ExtractedDocument, PdfSource, CodebaseSource, WikiSource};
32pub use graph::{GraphStore, GraphNode, GraphEdge, GraphPath, Community, GraphPersisted};
33pub use graph_rag::{
34    EntityExtractor, ExtractedEntity, GraphInfo, GraphRagEngine, GraphRagResult, GraphRagSnapshot,
35    SeedEntityExtractor, SimpleEntityExtractor, EntityInfo,
36};
37pub use keyword::{tokenize, Bm25Index};
38pub use hybrid::merge_hybrid;
39pub use dedup::{content_jaccard, dedup_similarities};
40pub use rerank::{SimilarityReranker, PassthroughReranker, rerank_similarities};
rag/lib.rs

rag/
lib.rs
