objectiveai-sdk 2.1.1

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
53
54
55
56
57
58
59
60
61
62

//! 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(())
    }
}