Struct MockLlmProvider 

pub struct MockLlmProvider { /* private fields */ }

A mock LLM provider for testing and development.

Implementations

impl MockLlmProvider

pub fn new() -> Self

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

Create a MockLlmProvider that always returns the given text.

Queues multiple copies of the response so it can handle multiple calls.

pub fn queue_response(&self, response: CompletionResponse)

Queue a response to be returned by the next complete call.

pub fn text_response(text: &str) -> CompletionResponse

Create a simple text response for testing.

pub fn tool_call_response( tool_name: &str, arguments: Value, ) -> CompletionResponse

Create a tool call response for testing.

pub fn multipart_response( text: &str, tool_name: &str, arguments: Value, ) -> CompletionResponse

Create a multipart response (text + tool call) for testing.

Trait Implementations

impl Default for MockLlmProvider

fn default() -> Self

Returns the “default value” for a type.

impl LlmProvider for MockLlmProvider

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

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

Perform a full completion and return the response.

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

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

Perform a streaming completion, sending events to the channel.

fn estimate_tokens(&self, messages: &[Message]) -> usize

Estimate the token count for a set of messages.

fn context_window(&self) -> usize

Return the context window size for this provider/model.

fn supports_tools(&self) -> bool

Return whether this provider supports tool/function calling.

fn cost_per_token(&self) -> (f64, f64)

Return the cost per token (input, output) in USD.

fn model_name(&self) -> &str

Return the model name.