stakai 0.3.80

A provider-agnostic Rust SDK for AI completions with streaming support - Built by Stakpak
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

//! Shared OpenAI runtime profile types.

#[derive(Debug, Clone, PartialEq, Eq)]
pub struct OfficialBackendProfile {
    pub base_url: String,
}

#[derive(Debug, Clone, PartialEq, Eq)]
pub struct CompatibleBackendProfile {
    pub base_url: String,
}

#[derive(Debug, Clone, PartialEq, Eq)]
pub struct CodexBackendProfile {
    pub base_url: String,
    pub originator: String,
    pub chatgpt_account_id: String,
}

#[derive(Debug, Clone, PartialEq, Eq)]
pub enum OpenAIBackendProfile {
    Official(OfficialBackendProfile),
    Codex(CodexBackendProfile),
    Compatible(CompatibleBackendProfile),
}

impl OpenAIBackendProfile {
    pub fn base_url(&self) -> &str {
        match self {
            Self::Official(profile) => &profile.base_url,
            Self::Codex(profile) => &profile.base_url,
            Self::Compatible(profile) => &profile.base_url,
        }
    }
}