Struct MockProvider 

pub struct MockProvider { /* private fields */ }

Deterministic LLM provider for tests.

Queue responses with the builder API, then call complete() to consume them in FIFO order.

Example

use pe_core::mock_provider::MockProvider;
use pe_core::llm::LlmProvider;

let provider = MockProvider::new()
    .respond_with("Hello!")
    .respond_with("Goodbye!");

let r1 = provider.complete(&[], &[]).await.unwrap();
assert_eq!(r1.message.content.as_text(), Some("Hello!"));

let r2 = provider.complete(&[], &[]).await.unwrap();
assert_eq!(r2.message.content.as_text(), Some("Goodbye!"));

Implementations

impl MockProvider

pub fn new() -> Self

Create a new MockProvider with an empty response queue.

pub fn respond_with(self, text: impl Into<String>) -> Self

Queue a plain text response.

pub fn respond_with_tool_call( self, tool_name: impl Into<String>, args: Value, ) -> Self

Queue a tool call response.

pub fn respond_with_error(self, err: PeError) -> Self

Queue an error response.

pub fn with_embedding(self, embedding: Vec<f32>) -> Self

Set the embedding to return for all embed() calls.

pub fn remaining(&self) -> usize

Number of responses remaining in the queue.

Trait Implementations

impl Default for MockProvider

fn default() -> Self

Returns the “default value” for a type.

impl LlmProvider for MockProvider

fn complete( &self, _messages: &[Message], _tools: &[ToolSchema], ) -> Pin<Box<dyn Future<Output = Result<LlmResponse, PeError>> + Send + '_>>

Non-streaming completion. Returns the full response.

fn stream( &self, _messages: &[Message], _tools: &[ToolSchema], ) -> StreamFuture<'_>

Streaming completion. Yields tokens (and tool call deltas) as they arrive.

fn embed( &self, _text: &str, ) -> Pin<Box<dyn Future<Output = Result<Vec<f32>, PeError>> + Send + '_>>

Embed text into a vector. Used for semantic routing and memory search.

fn provider_name(&self) -> &'static str

Human-readable provider name for logging (“openai”, “anthropic”, “mock”).