muna 0.0.31

Run prediction functions in your Rust apps.
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
36
37
38
39
40
41
42
43
44
45
46
47
48
49

/*
*   Muna
*   Copyright © 2026 NatML Inc. All Rights Reserved.
*/

mod chat;
mod completions;
mod embeddings;
mod images;
mod inputs;
mod models;
mod schema;

pub use chat::*;
pub use completions::*;
pub use embeddings::*;
pub use images::*;
pub use inputs::*;
pub use models::*;
pub use schema::*;

use crate::services::{PredictionService, PredictorService};

/// Experimental OpenAI client.
#[derive(Clone)]
pub struct OpenAIClient {
    /// Chat service.
    pub chat: ChatService,
    /// Embeddings service.
    pub embeddings: EmbeddingService,
    /// Image generation service.
    pub images: ImageService,
    /// Models service.
    pub models: ModelService,
}

impl OpenAIClient {

    pub fn new(
        predictors: PredictorService,
        predictions: PredictionService
    ) -> Self {
        let chat = ChatService::new(predictors.clone(), predictions.clone());
        let embeddings = EmbeddingService::new(predictors.clone(), predictions.clone());
        let images = ImageService::new(predictors.clone(), predictions);
        let models = ModelService::new(predictors);
        Self { chat, embeddings, images, models }
    }
}