gemini_adk_rs/tools/retrieval/mod.rs
1//! Retrieval tools — provide RAG (Retrieval-Augmented Generation) capabilities.
2//!
3//! Mirrors ADK-Python's `tools/retrieval` module. Provides base traits
4//! and implementations for retrieving relevant documents to augment
5//! LLM context.
6
7mod base;
8mod files;
9#[cfg(feature = "vertex-ai-rag")]
10mod vertex_ai_rag;
11
12pub use base::{BaseRetrievalTool, RetrievalResult};
13pub use files::FilesRetrievalTool;
14#[cfg(feature = "vertex-ai-rag")]
15pub use vertex_ai_rag::{VertexAiRagConfig, VertexAiRagRetrievalTool};