ai-lib-core 1.0.0

AI-Protocol execution runtime core (protocol, client, pipeline, transport)
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

//! Fallback Operator
//!
//! This operator handles failover to alternative models/providers.

pub struct FallbackOperator {
    pub candidates: Vec<String>, // List of model IDs
}

impl FallbackOperator {
    pub fn new(candidates: Vec<String>) -> Self {
        Self { candidates }
    }

    pub fn next(&self, current_failed_model: &str) -> Option<&str> {
        // Find current model index and return next
        let idx = self
            .candidates
            .iter()
            .position(|r| r == current_failed_model);
        match idx {
            Some(i) => {
                if i + 1 < self.candidates.len() {
                    Some(&self.candidates[i + 1])
                } else {
                    None // End of list
                }
            }
            None => self.candidates.first().map(|s| s.as_str()),
        }
    }
}