mirror of
https://github.com/openai/codex.git
synced 2026-10-28 18:58:46 +00:00
Celia Chen
d5e256ceb2
## What changed - Add the built-in `amazon-bedrock-runtime` provider for regional `bedrock-runtime` OpenAI-compatible endpoints. - Use endpoint-specific SigV4 service configuration while preserving bearer token auth and per-provider AWS profile, region, and transport overrides. - Provide global and US cross-region GPT-5.6 model variants, prefer global routing for fallback and background tasks, and disable unsupported web search. ## Testing - Cover provider configuration and capabilities, endpoint and auth selection, runtime catalog ordering, model fallback, and web search omission. GitOrigin-RevId: d158ab74a5768e1b4c61063365a0c06735313322
47 lines
1.2 KiB
Rust
47 lines
1.2 KiB
Rust
use codex_model_provider_info::AMAZON_BEDROCK_RUNTIME_PROVIDER_ID;
|
|
use codex_model_provider_info::ModelProviderAwsAuthInfo;
|
|
use pretty_assertions::assert_eq;
|
|
|
|
use super::ConfigToml;
|
|
|
|
#[test]
|
|
fn runtime_provider_accepts_aws_profile_and_region_overrides() {
|
|
let config = toml::from_str::<ConfigToml>(
|
|
r#"
|
|
[model_providers.amazon-bedrock-runtime.aws]
|
|
profile = "runtime-profile"
|
|
region = "us-west-2"
|
|
"#,
|
|
)
|
|
.expect("Bedrock Runtime AWS overrides should deserialize");
|
|
|
|
assert_eq!(
|
|
config
|
|
.model_providers
|
|
.get(AMAZON_BEDROCK_RUNTIME_PROVIDER_ID)
|
|
.and_then(|provider| provider.aws.clone()),
|
|
Some(ModelProviderAwsAuthInfo {
|
|
profile: Some("runtime-profile".to_string()),
|
|
region: Some("us-west-2".to_string()),
|
|
})
|
|
);
|
|
}
|
|
|
|
#[test]
|
|
fn custom_provider_still_rejects_aws_auth() {
|
|
let error = toml::from_str::<ConfigToml>(
|
|
r#"
|
|
[model_providers.custom]
|
|
name = "Custom"
|
|
|
|
[model_providers.custom.aws]
|
|
region = "us-west-2"
|
|
"#,
|
|
)
|
|
.expect_err("custom providers must not accept AWS auth");
|
|
|
|
assert!(error.to_string().contains(
|
|
"provider aws is only supported for `amazon-bedrock` or `amazon-bedrock-runtime`"
|
|
));
|
|
}
|