Struct CodeRetriever 

pub struct CodeRetriever { /* private fields */ }

Budget-aware code retriever with automatic query classification.

Wraps a CodeStore and an LLM provider (for embedding) and exposes a single high-level CodeRetriever::retrieve method.

Examples

use std::sync::Arc;
use zeph_index::retriever::{CodeRetriever, RetrievalConfig, format_as_context};
use zeph_index::store::CodeStore;

let retriever = CodeRetriever::new(store, provider, RetrievalConfig::default());
let result = retriever.retrieve("explain how authentication works", 8_000).await?;
let xml = format_as_context(&result);
println!("{xml}");

Implementations

impl CodeRetriever

pub fn new( store: CodeStore, provider: Arc<AnyProvider>, config: RetrievalConfig, ) -> Self

Create a new CodeRetriever.

store must have its Qdrant collection already created (see CodeStore::ensure_collection).

pub async fn retrieve( &self, query: &str, available_tokens: usize, ) -> Result<RetrievedCode>

Retrieve relevant code chunks for a free-text query.

Classifies query via classify_query, then:

• For RetrievalStrategy::Grep queries — returns an empty RetrievedCode so the agent falls back to its shell grep or symbol_definition tools.
• For RetrievalStrategy::Semantic / RetrievalStrategy::Hybrid — embeds the query, searches Qdrant, applies the score threshold, and packs results within available_tokens * budget_ratio.

Errors

Returns an error if the embedding call or Qdrant search fails.

pub async fn retrieve_filtered( &self, query: &str, available_tokens: usize, language: &str, ) -> Result<RetrievedCode>

Retrieve relevant code, restricting results to a single language.

Behaves like CodeRetriever::retrieve but adds a Qdrant payload filter so only chunks whose language field matches language are returned.

Useful when the user or agent has already established the relevant language (e.g. “show me the Python error handling” should not return Rust results).

Arguments

• language — the language identifier as returned by crate::languages::Lang::id (e.g. "rust", "python").

Errors

Returns an error if embedding or Qdrant search fails.