ai-space/goose
3
0
Fork 0
mirror of https://github.com/aaif-goose/goose.git synced 2026-07-03 14:10:03 +02:00
Code Issues Packages Projects Releases Wiki Activity

feat(providers): add vercel ai gateway (#9144)

Browse Source
This commit is contained in:
Alex Hancock
2026-05-11 20:41:16 -04:00
committed by GitHub
parent 71b9d575a0
commit 6e57d245d2
2 changed files with 182 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
crates/goose/src/config/declarative_providers.rs
+17
View File
@@ -688,6 +688,23 @@ mod tests {
assert_eq!(config.models[0].context_limit, 131072);
}
#[test]
fn test_vercel_ai_gateway_json_deserializes() {
let json = include_str!("../providers/declarative/vercel_ai_gateway.json");
let config: DeclarativeProviderConfig =
serde_json::from_str(json).expect("vercel_ai_gateway.json should parse");
assert_eq!(config.name, "vercel_ai_gateway");
assert_eq!(config.display_name, "Vercel AI Gateway");
assert!(matches!(config.engine, ProviderEngine::OpenAI));
assert_eq!(config.api_key_env, "AI_GATEWAY_API_KEY");
assert_eq!(
config.base_url,
"https://ai-gateway.vercel.sh/v1/chat/completions"
);
assert_eq!(config.supports_streaming, Some(true));
assert!(!config.models.is_empty());
}
#[test]
fn test_validate_provider_id_rejects_legacy_punctuation_for_new_ids() {
assert!(validate_provider_id("custom_z.ai").is_err());
crates/goose/src/providers/declarative/vercel_ai_gateway.json
+165
View File
@@ -0,0 +1,165 @@
{
"name": "vercel_ai_gateway",
"engine": "openai",
"display_name": "Vercel AI Gateway",
"description": "Unified gateway to OpenAI, Anthropic, Google, xAI, and many more frontier models via Vercel AI Gateway",
"api_key_env": "AI_GATEWAY_API_KEY",
"base_url": "https://ai-gateway.vercel.sh/v1/chat/completions",
"models": [
{
"name": "anthropic/claude-sonnet-4.6",
"context_limit": 1000000,
"input_token_cost": 0.000003,
"output_token_cost": 0.000015,
"currency": "USD"
},
{
"name": "anthropic/claude-sonnet-4.5",
"context_limit": 1000000,
"input_token_cost": 0.000003,
"output_token_cost": 0.000015,
"currency": "USD"
},
{
"name": "anthropic/claude-opus-4.6",
"context_limit": 1000000,
"input_token_cost": 0.000005,
"output_token_cost": 0.000025,
"currency": "USD"
},
{
"name": "anthropic/claude-opus-4.5",
"context_limit": 200000,
"input_token_cost": 0.000005,
"output_token_cost": 0.000025,
"currency": "USD"
},
{
"name": "anthropic/claude-haiku-4.5",
"context_limit": 200000,
"input_token_cost": 0.000001,
"output_token_cost": 0.000005,
"currency": "USD"
},
{
"name": "openai/gpt-5",
"context_limit": 400000,
"input_token_cost": 0.00000125,
"output_token_cost": 0.00001,
"currency": "USD"
},
{
"name": "openai/gpt-5-codex",
"context_limit": 400000,
"input_token_cost": 0.00000125,
"output_token_cost": 0.00001,
"currency": "USD"
},
{
"name": "openai/gpt-5-mini",
"context_limit": 400000,
"input_token_cost": 0.00000025,
"output_token_cost": 0.000002,
"currency": "USD"
},
{
"name": "openai/gpt-5.1-codex",
"context_limit": 400000,
"input_token_cost": 0.00000125,
"output_token_cost": 0.00001,
"currency": "USD"
},
{
"name": "openai/gpt-4.1",
"context_limit": 1047576,
"input_token_cost": 0.000002,
"output_token_cost": 0.000008,
"currency": "USD"
},
{
"name": "openai/gpt-4o",
"context_limit": 128000,
"input_token_cost": 0.0000025,
"output_token_cost": 0.00001,
"currency": "USD"
},
{
"name": "openai/o3",
"context_limit": 200000,
"input_token_cost": 0.000002,
"output_token_cost": 0.000008,
"currency": "USD"
},
{
"name": "openai/o4-mini",
"context_limit": 200000,
"input_token_cost": 0.0000011,
"output_token_cost": 0.0000044,
"currency": "USD"
},
{
"name": "google/gemini-3-pro-preview",
"context_limit": 1000000,
"input_token_cost": 0.000002,
"output_token_cost": 0.000012,
"currency": "USD"
},
{
"name": "google/gemini-2.5-pro",
"context_limit": 1048576,
"input_token_cost": 0.00000125,
"output_token_cost": 0.00001,
"currency": "USD"
},
{
"name": "google/gemini-2.5-flash",
"context_limit": 1000000,
"input_token_cost": 0.0000003,
"output_token_cost": 0.0000025,
"currency": "USD"
},
{
"name": "xai/grok-4",
"context_limit": 256000,
"input_token_cost": 0.000003,
"output_token_cost": 0.000015,
"currency": "USD"
},
{
"name": "xai/grok-4-fast-reasoning",
"context_limit": 2000000,
"input_token_cost": 0.0000002,
"output_token_cost": 0.0000005,
"currency": "USD"
},
{
"name": "xai/grok-code-fast-1",
"context_limit": 256000,
"input_token_cost": 0.0000002,
"output_token_cost": 0.0000015,
"currency": "USD"
},
{
"name": "deepseek/deepseek-v3.1",
"context_limit": 163840,
"input_token_cost": 0.00000056,
"output_token_cost": 0.00000168,
"currency": "USD"
},
{
"name": "moonshotai/kimi-k2-thinking",
"context_limit": 262114,
"input_token_cost": 0.0000006,
"output_token_cost": 0.0000025,
"currency": "USD"
},
{
"name": "zai/glm-4.6",
"context_limit": 200000,
"input_token_cost": 0.0000006,
"output_token_cost": 0.0000022,
"currency": "USD"
}
],
"supports_streaming": true
}