muna 0.0.9

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

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

use std::sync::Arc;
use crate::client::MunaClient;
use crate::services::{PredictionService, PredictorService};
use super::openai::OpenAIClient;
use super::remote::{BetaPredictionService, RemotePredictionService};

/// Client for incubating features.
#[derive(Clone)]
pub struct BetaClient {
    /// Make remote predictions.
    pub predictions: BetaPredictionService,
    /// OpenAI-compatible client.
    pub openai: OpenAIClient,
}

impl BetaClient {

    pub fn new(
        client: Arc<MunaClient>,
        predictors: PredictorService,
        predictions: PredictionService,
    ) -> Self {
        let beta_predictions = BetaPredictionService::new(client.clone());
        let remote_predictions = RemotePredictionService::new(client);
        let openai = OpenAIClient::new(predictors, predictions, remote_predictions);
        Self { predictions: beta_predictions, openai }
    }
}