Struct MockProvider 

pub struct MockProvider {
    pub responses: HashMap<String, String>,
}

A mock provider for testing.

Fields

responses: HashMap<String, String>

Responses to return (slot_name -> code).

Implementations

impl MockProvider

pub fn new() -> Self

Create a new mock provider.

pub fn with_response( self, slot: impl Into<String>, code: impl Into<String>, ) -> Self

Add a mock response.

Trait Implementations

impl AiProvider for MockProvider

fn name(&self) -> &str

Get the provider name.

fn generate<'life0, 'async_trait>( &'life0 self, request: GenerationRequest, ) -> Pin<Box<dyn Future<Output = Result<GenerationResponse>> + Send + 'async_trait>>

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

Generate code for a slot.

fn generate_stream( &self, request: GenerationRequest, ) -> BoxStream<'static, Result<StreamResponse>>

Generate a stream of code for a slot.

fn generate_batch<'life0, 'async_trait>( &'life0 self, requests: Vec<GenerationRequest>, ) -> Pin<Box<dyn Future<Output = Result<Vec<GenerationResponse>>> + Send + 'async_trait>>

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

Generate code for multiple slots in batch.

fn health_check<'life0, 'async_trait>( &'life0 self, ) -> Pin<Box<dyn Future<Output = Result<bool>> + Send + 'async_trait>>

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

Check if the provider is available and configured correctly.

impl Debug for MockProvider

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

Formats the value using the given formatter.

impl Default for MockProvider

fn default() -> MockProvider

Returns the “default value” for a type.