Struct MockProvider 

pub struct MockProvider { /* private fields */ }

Implementations

impl MockProvider

pub fn with_usage(scripted: Vec<Completion>) -> Self

Alias for new — both accept completions that may have usage set. Provided for API symmetry when callers want to be explicit about usage.

pub fn new(scripted: Vec<Completion>) -> Self

Create a new MockProvider with the given scripted completions.

pub fn with_on_complete(self, notify: Arc<Notify>) -> Self

Attach a notifier that fires once per complete() call, after the completion is dequeued and immediately before it is returned. Tests use this to deterministically synchronise external side-effects (e.g. cancellation) with agent progress, avoiding wall-clock sleeps.

pub fn with_on_complete_fn<F>(self, hook: F) -> Self

where F: Fn() + Send + Sync + 'static,

Attach a synchronous hook that runs once per complete() call, after the completion is dequeued and immediately before it is returned.

pub fn calls(&self) -> Vec<Vec<Message>>

Snapshot of the transcripts the agent has sent to this provider.

pub fn with_structured_responses(self, responses: Vec<Result<Value>>) -> Self

Set the queue of structured responses for complete_structured. Each call to complete_structured pops the next response. If the queue is empty, it returns an error (fallback path).

Trait Implementations

impl Default for MockProvider

fn default() -> MockProvider

Returns the “default value” for a type.

impl LlmProvider for MockProvider

fn complete<'life0, 'life1, 'life2, 'async_trait>( &'life0 self, messages: &'life1 [Message], _tools: &'life2 [ToolSpec], ) -> Pin<Box<dyn Future<Output = Result<Completion>> + Send + 'async_trait>>

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

fn complete_structured<'life0, 'async_trait>( &'life0 self, _req: StructuredRequest, ) -> Pin<Box<dyn Future<Output = Result<Value>> + Send + 'async_trait>>

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

Request a JSON response conforming to a caller-supplied schema. Default impl returns an error. Providers that support structured output (e.g. OpenAI-compatible) override this.

fn stream<'life0, 'life1, 'life2, 'async_trait>( &'life0 self, messages: &'life1 [Message], tools: &'life2 [ToolSpec], stream_tx: Option<StreamSender>, ) -> Pin<Box<dyn Future<Output = Result<Completion>> + Send + 'async_trait>>

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

Stream a completion token-by-token.

fn complete_with_search<'life0, 'life1, 'life2, 'life3, 'async_trait>( &'life0 self, messages: &'life1 [Message], eager_tools: &'life2 [(ToolSpec, Option<String>)], deferred_tools: &'life3 [(ToolSpec, Option<String>)], ) -> Pin<Box<dyn Future<Output = Result<Completion>> + Send + 'async_trait>>

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

Variant that accepts a partition between eager and deferred tools. Providers that support native deferred loading (e.g. Anthropic via defer_loading: true + tool_reference content blocks) override this. The default implementation concatenates the two lists (dropping the hints) and calls complete() — i.e., it ignores the partition and behaves identically to the legacy interface.

fn complete_simple<'life0, 'life1, 'async_trait>( &'life0 self, prompt: &'life1 str, _temperature: f32, ) -> Pin<Box<dyn Future<Output = Result<String>> + Send + 'async_trait>>

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

Simple completion with a single user prompt.