Struct VertexAiRagRetrievalTool 

pub struct VertexAiRagRetrievalTool { /* private fields */ }

Retrieval tool that searches via the Vertex AI RAG API.

Calls the Vertex AI RAG :retrieveContexts endpoint to retrieve relevant document chunks from the configured corpora.

Quick start

let tool = VertexAiRagRetrievalTool::new(
    VertexAiRagConfig::from_corpus(
        "projects/my-proj/locations/us-central1/ragCorpora/my-corpus",
    )
    .with_similarity_top_k(5),
)
.with_token("ya29.my-access-token");

let results = tool.retrieve("what is the refund policy?", 5).await?;

Implementations

impl VertexAiRagRetrievalTool

pub fn new(config: VertexAiRagConfig) -> Self

Create a new Vertex AI RAG retrieval tool.

No auth token is configured yet — call with_token or with_token_refresher before issuing requests.

pub fn with_token(self, token: impl Into<String>) -> Self

Set a static bearer token for all requests.

pub fn with_token_refresher( self, f: impl Fn() -> String + Send + Sync + 'static, ) -> Self

Set a dynamic token refresher closure (invoked before every request).

pub fn corpora(&self) -> &[String]

Returns the configured corpora resource names.

pub fn corpus(&self) -> &str

Returns the first configured corpus (convenience accessor).

Trait Implementations

impl BaseRetrievalTool for VertexAiRagRetrievalTool

fn name(&self) -> &str

The name of this retrieval tool.

fn retrieve<'life0, 'life1, 'async_trait>( &'life0 self, query: &'life1 str, top_k: usize, ) -> Pin<Box<dyn Future<Output = Result<Vec<RetrievalResult>, ToolError>> + Send + 'async_trait>>

where Self: 'async_trait, 'life0: 'async_trait, 'life1: 'async_trait,

Search the corpus with a query and return ranked results.