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>) -> Self

Create a mock model from scripted non-streaming turns.

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

Create a mock model that returns one text completion.

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

Create a mock model from scripted non-streaming turns.

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

Create a mock model from scripted streaming turns.

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

Return cloned requests received by this model.

pub fn request_count(&self) -> usize

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

type Response = MockResponse

The raw response type returned by the underlying completion model.

type StreamingResponse = MockResponse

The raw response type returned by the underlying completion model when streaming.

type Client = ()

Provider client type used to construct this model.

fn make(_: &Self::Client, _: impl Into<String>) -> Self

Construct a model handle from a provider client and model identifier.

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

Generates a completion response for the given completion request.

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

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

Generates a completion request builder for the given prompt.

impl Default for MockCompletionModel

fn default() -> MockCompletionModel

Returns the “default value” for a type.