Struct MockLanguageModelClient

pub struct MockLanguageModelClient<E> { /* private fields */ }

Implementations

impl<E> MockLanguageModelClient<E>

pub fn batches(&self) -> &RwLock<HashMap<String, Batch>>

pub fn files(&self) -> &RwLock<HashMap<String, Bytes>>

pub fn fail_on_file_create_openai_error(&self) -> &bool

pub fn fail_on_file_create_other_error(&self) -> &bool

pub fn mock_batch_config(&self) -> &RwLock<MockBatchConfig>

NEW FIELD: stores extra toggles and attempt counters for advanced mock behaviors

impl<E> MockLanguageModelClient<E>

pub fn set_batches( &mut self, val: RwLock<HashMap<String, Batch>>, ) -> &mut MockLanguageModelClient<E>

pub fn set_files( &mut self, val: RwLock<HashMap<String, Bytes>>, ) -> &mut MockLanguageModelClient<E>

pub fn set_fail_on_file_create_openai_error( &mut self, val: bool, ) -> &mut MockLanguageModelClient<E>

pub fn set_fail_on_file_create_other_error( &mut self, val: bool, ) -> &mut MockLanguageModelClient<E>

pub fn set_mock_batch_config( &mut self, val: RwLock<MockBatchConfig>, ) -> &mut MockLanguageModelClient<E>

NEW FIELD: stores extra toggles and attempt counters for advanced mock behaviors

impl<E> MockLanguageModelClient<E>

where E: From<OpenAIClientError> + From<Error> + Debug + Send + Sync,

pub fn configure_inprogress_then_complete_with( &self, batch_id: &str, want_output: bool, want_error: bool, )

Configure the mock so that the next time we retrieve batch_id, it will show status = InProgress, then on retrieve => flip to Completed with the specified (want_output, want_error).

This is used by “success” scenarios where we want the batch to end up Completed with optional output_file_id and/or error_file_id.

pub fn configure_failure(&self, batch_id: &str, is_immediate: bool)

impl<E> MockLanguageModelClient<E>

where E: From<OpenAIClientError> + From<Error> + Debug + Send + Sync,

pub fn new() -> MockLanguageModelClient<E>

pub fn set_batch_to_inprogress_then_complete_with( &self, batch_id: &str, want_output: bool, want_error: bool, )

Helper to forcibly mark the given batch_id as “InProgress” in the batches map with a note in our mock_batch_config so that on the very next retrieve, we flip it to “Completed” with an output_file_id and/or error_file_id.

This ensures that when wait_for_batch_completion sees it become “Completed,” it already has the relevant file IDs.

Trait Implementations

impl<E> CreateBatch for MockLanguageModelClient<E>

where E: From<OpenAIClientError> + From<Error> + Debug + Send + Sync,

type Error = E

fn create_batch<'life0, 'life1, 'async_trait>( &'life0 self, input_file_id: &'life1 str, ) -> Pin<Box<dyn Future<Output = Result<Batch, <MockLanguageModelClient<E> as CreateBatch>::Error>> + Send + 'async_trait>>

where 'life0: 'async_trait, 'life1: 'async_trait, MockLanguageModelClient<E>: 'async_trait,

impl<E> Debug for MockLanguageModelClient<E>

where E: Debug,

fn fmt(&self, f: &mut Formatter<'_>) -> Result<(), Error>

Formats the value using the given formatter.

impl<E> GetBatchFileContent for MockLanguageModelClient<E>

where E: From<OpenAIClientError> + From<Error> + Debug + Send + Sync,

type Error = E

fn file_content<'life0, 'life1, 'async_trait>( &'life0 self, file_id: &'life1 str, ) -> Pin<Box<dyn Future<Output = Result<Bytes, <MockLanguageModelClient<E> as GetBatchFileContent>::Error>> + Send + 'async_trait>>

where 'life0: 'async_trait, 'life1: 'async_trait, MockLanguageModelClient<E>: 'async_trait,

impl<E> RetrieveBatchById for MockLanguageModelClient<E>

where E: From<OpenAIClientError> + From<Error> + Debug + Send + Sync,

type Error = E

fn retrieve_batch<'life0, 'life1, 'async_trait>( &'life0 self, batch_id: &'life1 str, ) -> Pin<Box<dyn Future<Output = Result<Batch, <MockLanguageModelClient<E> as RetrieveBatchById>::Error>> + Send + 'async_trait>>

where 'life0: 'async_trait, 'life1: 'async_trait, MockLanguageModelClient<E>: 'async_trait,

impl<E> UploadBatchFileCore for MockLanguageModelClient<E>

where E: From<OpenAIClientError> + From<Error> + Debug + Send + Sync,

type Error = E

fn upload_batch_file_path<'life0, 'life1, 'async_trait>( &'life0 self, file_path: &'life1 Path, ) -> Pin<Box<dyn Future<Output = Result<OpenAIFile, <MockLanguageModelClient<E> as UploadBatchFileCore>::Error>> + Send + 'async_trait>>

where 'life0: 'async_trait, 'life1: 'async_trait, MockLanguageModelClient<E>: 'async_trait,

impl<E> WaitForBatchCompletion for MockLanguageModelClient<E>

where E: From<OpenAIClientError> + From<Error> + Debug + Send + Sync,

type Error = E

fn wait_for_batch_completion<'life0, 'life1, 'async_trait>( &'life0 self, batch_id: &'life1 str, ) -> Pin<Box<dyn Future<Output = Result<Batch, <MockLanguageModelClient<E> as WaitForBatchCompletion>::Error>> + Send + 'async_trait>>

where 'life0: 'async_trait, 'life1: 'async_trait, MockLanguageModelClient<E>: 'async_trait,

impl<E> LanguageModelClientInterface<E> for MockLanguageModelClient<E>

where E: From<OpenAIClientError> + From<BatchDownloadError> + From<Error> + From<BatchMetadataError> + Debug + Send + Sync,