RAG Config • Rust AI Agent SDK

RAGConfig: Configuration for RAG pipeline.

RAGConfig

Defined in the RAG module.

Rust AI Agent SDK

Configuration for RAG pipeline.

Fields

Name	Type	Description
`top_k`	`usize`	Maximum number of chunks to retrieve
`score_threshold`	`f32`	Minimum relevance score threshold
`max_context_tokens`	`usize`	Maximum context tokens
`strategy`	`RetrievalStrategy`	Retrieval strategy
`citations_mode`	`CitationsMode`	Citations mode
`rerank`	`bool`	Enable reranking
`compress`	`bool`	Enable context compression
`chunk_overlap`	`usize`	Chunk overlap for splitting
`chunk_size`	`usize`	Chunk size for splitting

Methods

`new`

rust

fn new() -> Self

Create a new RAGConfig

`top_k`

rust

fn top_k(mut self, k: usize) -> Self

Set top_k

Parameters:

Name	Type
`k`	`usize`

`score_threshold`

rust

fn score_threshold(mut self, threshold: f32) -> Self

Set score threshold

Parameters:

Name	Type
`threshold`	`f32`

`max_context_tokens`

rust

fn max_context_tokens(mut self, tokens: usize) -> Self

Set max context tokens

Parameters:

Name	Type
`tokens`	`usize`

`strategy`

rust

fn strategy(mut self, strategy: RetrievalStrategy) -> Self

Set retrieval strategy

Parameters:

Name	Type
`strategy`	`RetrievalStrategy`

`citations_mode`

rust

fn citations_mode(mut self, mode: CitationsMode) -> Self

Set citations mode

Parameters:

Name	Type
`mode`	`CitationsMode`

`rerank`

rust

fn rerank(mut self, enable: bool) -> Self

Enable reranking

Parameters:

Name	Type
`enable`	`bool`

`compress`

rust

fn compress(mut self, enable: bool) -> Self

Enable compression

Parameters:

Name	Type
`enable`	`bool`

Source

View on GitHub

praisonai/src/rag/mod.rs at line 210

RAG Config • Rust AI Agent SDK

RAGConfig

Fields

Methods

new

top_k

score_threshold

max_context_tokens

strategy

citations_mode

rerank

compress

Source

Related Documentation

`new`

`top_k`

`score_threshold`

`max_context_tokens`

`strategy`

`citations_mode`

`rerank`

`compress`