Struct MockLanguageModel 

pub struct MockLanguageModel { /* private fields */ }

A mock language model for testing that tracks inputs and yields predefined outputs.

Implementations

impl MockLanguageModel

pub fn new() -> Self

Construct a new mock language model instance.

pub fn set_provider(&mut self, provider: &'static str)

Override the provider identifier returned by the mock.

pub fn set_model_id<S: Into<String>>(&mut self, model_id: S)

Override the model identifier returned by the mock.

pub fn set_metadata(&mut self, metadata: Option<LanguageModelMetadata>)

Override the metadata returned by the mock.

pub fn enqueue_generate_results<I>(&self, results: I) -> &Self

where I: IntoIterator<Item = MockGenerateResult>,

Enqueue one or more mocked generate results.

Panics

Panics if the internal state mutex is poisoned.

pub fn enqueue_generate<R>(&self, result: R) -> &Self

where R: Into<MockGenerateResult>,

Convenience to enqueue a single mocked generate result.

pub fn enqueue_stream_results<I>(&self, results: I) -> &Self

where I: IntoIterator<Item = MockStreamResult>,

Enqueue one or more mocked stream results.

Panics

Panics if the internal state mutex is poisoned.

pub fn enqueue_stream<R>(&self, result: R) -> &Self

where R: Into<MockStreamResult>,

Convenience to enqueue a single mocked stream result.

pub fn tracked_generate_inputs(&self) -> Vec<LanguageModelInput>

Retrieve the tracked generate inputs accumulated so far.

Panics

Panics if the internal state mutex is poisoned.

pub fn tracked_stream_inputs(&self) -> Vec<LanguageModelInput>

Retrieve the tracked stream inputs accumulated so far.

Panics

Panics if the internal state mutex is poisoned.

pub fn reset(&self)

Reset tracked inputs without touching enqueued results.

Panics

Panics if the internal state mutex is poisoned.

pub fn restore(&self)

Clear both tracked inputs and enqueued results.

Panics

Panics if the internal state mutex is poisoned.

Trait Implementations

impl Default for MockLanguageModel

fn default() -> Self

Returns the “default value” for a type.

impl LanguageModel for MockLanguageModel

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

fn model_id(&self) -> String

fn metadata(&self) -> Option<&LanguageModelMetadata>

fn generate( &self, input: LanguageModelInput, ) -> BoxFuture<'_, LanguageModelResult<ModelResponse>>

fn stream( &self, input: LanguageModelInput, ) -> BoxFuture<'_, LanguageModelResult<LanguageModelStream>>