Struct TestLLMProvider

pub struct TestLLMProvider { /* private fields */ }

Implementations

impl TestLLMProvider

pub fn new(config: LLMProviderConfig) -> Self

Creates a new instance of TestLLMProvider with the given configuration

pub fn set_fail_count(&self, count: usize)

pub fn set_delay(&self, delay_ms: u64)

pub fn get_total_calls(&self) -> usize

pub fn set_response(&mut self, response: String)

pub fn set_bad_response(&mut self, bad_response: String)

pub fn set_json_validation_failures(&self, count: usize)

pub fn reset(&self)

Trait Implementations

impl Clone for TestLLMProvider

fn clone(&self) -> TestLLMProvider

Returns a copy of the value.

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

Performs copy-assignment from source.

impl LLMProvider for TestLLMProvider

fn generate_message<'life0, 'life1, 'life2, 'async_trait>( &'life0 self, system_prompt: &'life1 str, user_prompt: &'life2 str, ) -> Pin<Box<dyn Future<Output = Result<String>> + Send + 'async_trait>>

where Self: 'async_trait, 'life0: 'async_trait, 'life1: 'async_trait, 'life2: 'async_trait,

Generates a message using the Test provider (returns model name + it’s own prompts as the message)