rig-core 0.37.0

An opinionated library for building LLM powered applications.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35

//! Model-listing helpers for deterministic tests.

use crate::{
    client::ModelLister,
    model::{Model, ModelList, ModelListingError},
    wasm_compat::WasmCompatSend,
};

/// A [`ModelLister`] that returns a preconfigured list of models.
pub struct MockModelLister {
    models: Vec<Model>,
}

impl MockModelLister {
    /// Create a model lister from a fixed model list.
    pub fn new(models: Vec<Model>) -> Self {
        Self { models }
    }
}

impl ModelLister for MockModelLister {
    type Client = Vec<Model>;

    fn new(client: Self::Client) -> Self {
        Self { models: client }
    }

    fn list_all(
        &self,
    ) -> impl std::future::Future<Output = Result<ModelList, ModelListingError>> + WasmCompatSend
    {
        let models = self.models.clone();
        async move { Ok(ModelList::new(models)) }
    }
}