Struct MockCompletionModel 

pub struct MockCompletionModel { /* private fields */ }

Available on crate feature test-utils only.

A cloneable scripted CompletionModel for tests.

Each completion or stream call consumes exactly one scripted turn. If no turn is available, the model returns CompletionError::ProviderError with a clear message instead of repeating previous responses.

Implementations

impl MockCompletionModel

pub fn new(turns: impl IntoIterator<Item = MockTurn>) -> MockCompletionModel

Available on crate features test-utils only.

Create a mock model from scripted non-streaming turns.

pub fn text(text: impl Into<String>) -> MockCompletionModel

Available on crate features test-utils only.

Create a mock model that returns one text completion.

pub fn from_turns( turns: impl IntoIterator<Item = MockTurn>, ) -> MockCompletionModel

Available on crate features test-utils only.

Create a mock model from scripted non-streaming turns.

pub fn from_stream_turns( stream_turns: impl IntoIterator<Item = impl IntoIterator<Item = MockStreamEvent>>, ) -> MockCompletionModel

Available on crate features test-utils only.

Create a mock model from scripted streaming turns.

pub fn requests(&self) -> Vec<CompletionRequest>

Available on crate features test-utils only.

Return cloned requests received by this model.

pub fn request_count(&self) -> usize

Available on crate features test-utils only.

Return the number of requests received by this model.

Trait Implementations

impl Clone for MockCompletionModel

fn clone(&self) -> MockCompletionModel

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl CompletionModel for MockCompletionModel

async fn completion( &self, request: CompletionRequest, ) -> Result<CompletionResponse, CompletionError>

Generates a completion response for the given completion request.

async fn stream( &self, request: CompletionRequest, ) -> Result<StreamingCompletionResponse, CompletionError>

Streams a completion response for the given completion request.

fn completion_request( &self, prompt: impl Into<Message>, ) -> CompletionRequestBuilder<Self>

where Self: Sized + Clone,

Generates a completion request builder for the given prompt.

fn capabilities(&self) -> ProviderCapabilities

Provider behavior a runtime should account for when preparing requests.

impl Default for MockCompletionModel

fn default() -> MockCompletionModel

Returns the “default value” for a type.