starweaver-model 0.0.5

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

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