objectiveai-sdk 2.0.6

ObjectiveAI SDK, definitions, and utilities
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
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52

//! Upstream enumeration.

use schemars::JsonSchema;
use serde::{Deserialize, Serialize};

/// Supported agent upstreams.
#[derive(
    Debug, Clone, Copy, PartialEq, Eq, Hash, Serialize, Deserialize, Default, JsonSchema, arbitrary::Arbitrary,
)]
#[serde(rename_all = "snake_case")]
#[schemars(rename = "agent.Upstream")]
pub enum Upstream {
    /// Unknown Upstream.
    #[schemars(title = "Unknown")]
    #[default]
    Unknown,
    /// OpenRouter Upstream.
    #[schemars(title = "Openrouter")]
    Openrouter,
    /// Claude Agent SDK Upstream.
    #[schemars(title = "ClaudeAgentSdk")]
    ClaudeAgentSdk,
    /// Codex SDK Upstream.
    #[schemars(title = "CodexSdk")]
    CodexSdk,
    /// Mock Upstream.
    #[schemars(title = "Mock")]
    Mock,
}

pub mod validate {
    use super::Upstream;

    pub fn validate(upstreams: &[Upstream]) -> Result<(), String> {
        // Check for duplicates
        let mut seen = std::collections::HashSet::new();
        for upstream in upstreams {
            if !seen.insert(upstream) {
                return Err(
                    "`upstreams` contains duplicate entries".to_string()
                );
            }
        }

        // Check for Unknown
        if seen.contains(&Upstream::Unknown) {
            return Err("`upstreams` contains unknown upstream".to_string());
        }

        Ok(())
    }
}