Struct MockDriver 

pub struct MockDriver { /* private fields */ }

Mock driver that returns pre-configured responses.

Implementations

impl MockDriver

pub fn new(responses: Vec<CompletionResponse>) -> Self

Create a mock driver with a sequence of responses.

Responses are returned in order. If exhausted, returns a default “end of mock responses” response.

pub fn single_response(text: &str) -> Self

Create a mock that returns a single text response.

pub fn tool_then_response( tool_name: &str, tool_input: Value, final_text: &str, ) -> Self

Create a mock that first requests a tool call, then responds.

pub fn with_context_window(self, size: usize) -> Self

Set context window size.

pub fn with_cost_per_token(self, cost: f64) -> Self

Set cost per token for testing cost budget enforcement.

Trait Implementations

impl LlmDriver for MockDriver

fn complete<'life0, 'async_trait>( &'life0 self, _request: CompletionRequest, ) -> Pin<Box<dyn Future<Output = Result<CompletionResponse, AgentError>> + Send + 'async_trait>>

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

Non-streaming completion.

fn context_window(&self) -> usize

Maximum context window in tokens.

fn privacy_tier(&self) -> PrivacyTier

Privacy tier this driver operates at.

fn estimate_cost(&self, usage: &TokenUsage) -> f64

Estimate cost in USD for a single completion’s token usage.

fn stream<'life0, 'async_trait>( &'life0 self, request: CompletionRequest, tx: Sender<StreamEvent>, ) -> Pin<Box<dyn Future<Output = Result<CompletionResponse, AgentError>> + Send + 'async_trait>>

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

Streaming completion with channel-based events.