starweaver-model 0.7.0

Provider-neutral model protocol and wire adapters for Starweaver
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
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131

//! HTTP config helpers for built-in provider presets.

use crate::{AuthConfig, HttpModelConfig};

fn google_cloud_model_resource(model_name: &str) -> String {
    if model_name.starts_with("projects/")
        || model_name.starts_with("models/")
        || model_name.starts_with("publishers/")
    {
        model_name.to_string()
    } else if let Some((publisher, model)) = model_name.split_once('/') {
        format!("publishers/{publisher}/models/{model}")
    } else {
        format!("publishers/google/models/{model_name}")
    }
}

fn google_cloud_base_url(location: &str) -> String {
    match location {
        "global" => "https://aiplatform.googleapis.com".to_string(),
        "us" | "eu" => format!("https://aiplatform.{location}.rep.googleapis.com"),
        location => format!("https://{location}-aiplatform.googleapis.com"),
    }
}

/// Build an Anthropic HTTP model config from an API key.
#[must_use]
pub fn anthropic_http_config(api_key: impl Into<String>) -> HttpModelConfig {
    let mut config =
        HttpModelConfig::provider_endpoint("https://api.anthropic.com/v1", "v1", "messages");
    config.auth = Some(AuthConfig::Header {
        name: "x-api-key".to_string(),
        value: api_key.into(),
    });
    config
        .headers
        .insert("anthropic-version".to_string(), "2023-06-01".to_string());
    config
}

/// Build an `OpenAI` Chat Completions HTTP model config from an API key.
#[must_use]
pub fn openai_chat_http_config(api_key: impl Into<String>) -> HttpModelConfig {
    let mut config =
        HttpModelConfig::provider_endpoint("https://api.openai.com/v1", "v1", "chat/completions");
    config.auth = Some(AuthConfig::Bearer {
        token: api_key.into(),
    });
    config
}

/// Build an `OpenAI` Responses HTTP model config from an API key.
#[must_use]
pub fn openai_responses_http_config(api_key: impl Into<String>) -> HttpModelConfig {
    let mut config =
        HttpModelConfig::provider_endpoint("https://api.openai.com/v1", "v1", "responses");
    config.auth = Some(AuthConfig::Bearer {
        token: api_key.into(),
    });
    config
}

/// Build an xAI Responses HTTP model config from an API key.
#[must_use]
pub fn xai_responses_http_config(api_key: impl Into<String>) -> HttpModelConfig {
    let mut config = HttpModelConfig::provider_endpoint("https://api.x.ai/v1", "v1", "responses");
    config.auth = Some(AuthConfig::Bearer {
        token: api_key.into(),
    });
    config
}

/// Build a Gemini HTTP model config from an API key and model name.
#[must_use]
pub fn gemini_http_config(
    api_key: impl Into<String>,
    model_name: impl Into<String>,
) -> HttpModelConfig {
    let model_name = model_name.into();
    HttpModelConfig::provider_endpoint(
        "https://generativelanguage.googleapis.com/v1beta",
        "v1beta",
        format!("models/{model_name}:generateContent?key={}", api_key.into()),
    )
}

/// Build a Google Cloud Gemini HTTP config for Vertex AI Express Mode API keys.
#[must_use]
pub fn google_cloud_http_config(
    api_key: impl Into<String>,
    model_name: impl Into<String>,
) -> HttpModelConfig {
    let model_name = model_name.into();
    let mut config = HttpModelConfig::provider_endpoint(
        "https://aiplatform.googleapis.com",
        "v1beta1",
        format!(
            "{}:generateContent",
            google_cloud_model_resource(&model_name)
        ),
    );
    config
        .headers
        .insert("x-goog-api-key".to_string(), api_key.into());
    config
}

/// Build a Google Cloud Gemini HTTP config for project/location scoped bearer auth.
#[must_use]
pub fn google_cloud_project_http_config(
    bearer_token: impl Into<String>,
    model_name: impl Into<String>,
    project: impl AsRef<str>,
    location: impl AsRef<str>,
) -> HttpModelConfig {
    let model_name = model_name.into();
    let project = project.as_ref();
    let location = location.as_ref();
    let model_resource = google_cloud_model_resource(&model_name);
    let endpoint = if model_resource.starts_with("projects/") {
        format!("{model_resource}:generateContent")
    } else {
        format!("projects/{project}/locations/{location}/{model_resource}:generateContent")
    };
    let mut config =
        HttpModelConfig::provider_endpoint(google_cloud_base_url(location), "v1beta1", endpoint);
    config.auth = Some(AuthConfig::Bearer {
        token: bearer_token.into(),
    });
    config
}