mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-12-03 09:53:45 +00:00
0a04e62a83
Rename AWS_BEDROCK_API_KEY to AWS_BEARER_TOKEN_BEDROCK to align with the naming convention used in AWS Bedrock documentation and the AWS web console UI. This reduces confusion when developers compare LLS docs with AWS docs. Also rename the Python field aws_bedrock_api_key to aws_bearer_token_bedrock throughout the codebase for consistency.
39 lines
1.5 KiB
Python
39 lines
1.5 KiB
Python
# Copyright (c) Meta Platforms, Inc. and affiliates.
|
|
# All rights reserved.
|
|
#
|
|
# This source code is licensed under the terms described in the LICENSE file in
|
|
# the root directory of this source tree.
|
|
|
|
from llama_stack.providers.remote.inference.bedrock.config import BedrockConfig
|
|
|
|
|
|
def test_bedrock_config_defaults_no_env(monkeypatch):
|
|
"""Test BedrockConfig defaults when env vars are not set"""
|
|
monkeypatch.delenv("AWS_BEARER_TOKEN_BEDROCK", raising=False)
|
|
monkeypatch.delenv("AWS_DEFAULT_REGION", raising=False)
|
|
config = BedrockConfig()
|
|
assert config.auth_credential is None
|
|
assert config.region_name == "us-east-2"
|
|
|
|
|
|
def test_bedrock_config_reads_from_env(monkeypatch):
|
|
"""Test BedrockConfig field initialization reads from environment variables"""
|
|
monkeypatch.setenv("AWS_DEFAULT_REGION", "eu-west-1")
|
|
config = BedrockConfig()
|
|
assert config.region_name == "eu-west-1"
|
|
|
|
|
|
def test_bedrock_config_with_values():
|
|
"""Test BedrockConfig accepts explicit values via alias"""
|
|
config = BedrockConfig(api_key="test-key", region_name="us-west-2")
|
|
assert config.auth_credential.get_secret_value() == "test-key"
|
|
assert config.region_name == "us-west-2"
|
|
|
|
|
|
def test_bedrock_config_sample():
|
|
"""Test BedrockConfig sample_run_config returns correct format"""
|
|
sample = BedrockConfig.sample_run_config()
|
|
assert "api_key" in sample
|
|
assert "region_name" in sample
|
|
assert sample["api_key"] == "${env.AWS_BEARER_TOKEN_BEDROCK:=}"
|
|
assert sample["region_name"] == "${env.AWS_DEFAULT_REGION:=us-east-2}"
|