Struct MockChatModel

pub struct MockChatModel { /* private fields */ }

A simple mock chat model for testing

Implementations

impl MockChatModel

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

Create a new mock chat model

pub fn add_response(self, response: impl Into<String>) -> Self

Add a response to the mock

Trait Implementations

impl BaseChatModel for MockChatModel

fn generate_chat<'life0, 'async_trait>( &'life0 self, _messages: Vec<AnyMessage>, _config: Option<GenerationConfig>, _runnable_config: Option<RunnableConfig>, ) -> Pin<Box<dyn Future<Output = Result<AnyMessage>> + Send + 'async_trait>>

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

Generate a response from chat messages

fn generate_chat_batch<'life0, 'async_trait>( &'life0 self, conversations: Vec<Vec<AnyMessage>>, config: Option<GenerationConfig>, runnable_config: Option<RunnableConfig>, ) -> Pin<Box<dyn Future<Output = Result<Vec<AnyMessage>>> + Send + 'async_trait>>

where Self: Sync + 'async_trait, 'life0: 'async_trait,

Generate responses from multiple chat conversations

fn stream_chat<'life0, 'async_trait>( &'life0 self, messages: Vec<AnyMessage>, config: Option<GenerationConfig>, runnable_config: Option<RunnableConfig>, ) -> Pin<Box<dyn Future<Output = Result<Pin<Box<dyn Stream<Item = Result<AnyMessage>> + Send>>>> + Send + 'async_trait>>

where Self: Sync + 'async_trait, 'life0: 'async_trait,

Stream chat generation

impl BaseLanguageModel for MockChatModel

fn model_name(&self) -> &str

Get the model name

fn model_type(&self) -> &str

Get the model type

fn supports_streaming(&self) -> bool

Check if the model supports streaming

fn input_schema(&self) -> Option<Value>

Get the input schema for this model

fn output_schema(&self) -> Option<Value>

Get the output schema for this model