diff --git a/engine/baml-lib/llm-client/src/clients/aws_bedrock.rs b/engine/baml-lib/llm-client/src/clients/aws_bedrock.rs index db69e7ad0..c093e4f71 100644 --- a/engine/baml-lib/llm-client/src/clients/aws_bedrock.rs +++ b/engine/baml-lib/llm-client/src/clients/aws_bedrock.rs @@ -196,18 +196,14 @@ impl UnresolvedAwsBedrock { let session_token = match self.session_token.as_ref() { Some(session_token) => { let token = session_token.resolve(ctx)?; - if !token.is_empty() && !token.starts_with('$') { + if !token.is_empty() { Some(token) } else { None } } None => match ctx.get_env_var("AWS_SESSION_TOKEN") { - Ok(session_token) - if !session_token.is_empty() && !session_token.starts_with('$') => - { - Some(session_token) - } + Ok(session_token) if !session_token.is_empty() => Some(session_token), _ => None, }, }; diff --git a/integ-tests/typescript/test-report.html b/integ-tests/typescript/test-report.html index 9b2a7b338..3a65c8b39 100644 --- a/integ-tests/typescript/test-report.html +++ b/integ-tests/typescript/test-report.html @@ -257,4 +257,87 @@ font-size: 1rem; padding: 0 0.5rem; } -