mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-12-03 18:00:36 +00:00
dc4665af17
Some checks failed
Integration Tests (Replay) / generate-matrix (push) Successful in 4s
Test External Providers Installed via Module / test-external-providers-from-module (venv) (push) Has been skipped
Integration Auth Tests / test-matrix (oauth2_token) (push) Failing after 2s
SqlStore Integration Tests / test-postgres (3.13) (push) Failing after 5s
SqlStore Integration Tests / test-postgres (3.12) (push) Failing after 5s
Test Llama Stack Build / generate-matrix (push) Successful in 3s
API Conformance Tests / check-schema-compatibility (push) Successful in 10s
Python Package Build Test / build (3.12) (push) Failing after 6s
Python Package Build Test / build (3.13) (push) Failing after 6s
Test Llama Stack Build / build-single-provider (push) Successful in 50s
Vector IO Integration Tests / test-matrix (push) Failing after 56s
Test Llama Stack Build / build (push) Successful in 49s
UI Tests / ui-tests (22) (push) Successful in 1m1s
Test External API and Providers / test-external (venv) (push) Failing after 1m18s
Unit Tests / unit-tests (3.13) (push) Failing after 1m58s
Unit Tests / unit-tests (3.12) (push) Failing after 2m5s
Test Llama Stack Build / build-ubi9-container-distribution (push) Successful in 2m28s
Integration Tests (Replay) / Integration Tests (, , , client=, ) (push) Failing after 2m20s
Test Llama Stack Build / build-custom-container-distribution (push) Successful in 2m37s
Pre-commit / pre-commit (push) Successful in 3m50s
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. Closes #4147
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}"
|