andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-09-19 17:46:14 +00:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Merge branch 'main' into viditostwal/platform-applications-catalog

Browse Source
This commit is contained in:
ViditOstwal
2026-09-11 17:07:32 +05:30
parent 7fbe938681 c5759ce854
commit f33df9809b
1 changed files with 24 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

41
lib/crewai/tests/llms/bedrock/test_bedrock.py
View File

@@ -602,24 +602,31 @@ def test_bedrock_tool_conversion():
assert "inputSchema" in bedrock_tools[0]["toolSpec"]
def test_bedrock_environment_variable_credentials(bedrock_mocks):
"""
Test that AWS credentials are properly loaded from environment
"""
mock_session_class, _ = bedrock_mocks
def test_bedrock_environment_variable_credentials():
"""Pass AWS credentials and region from the environment to boto3."""
with (
patch.dict(
os.environ,
{
"AWS_ACCESS_KEY_ID": "test-access-key-123",
"AWS_SECRET_ACCESS_KEY": "test-secret-key-456",
"AWS_DEFAULT_REGION": "eu-west-1",
},
clear=False,
),
patch(
"crewai.llms.providers.bedrock.completion.Session"
) as mock_session_class,
):
mock_session_class.return_value.client.return_value = MagicMock()
LLM(model="bedrock/anthropic.claude-3-5-sonnet-20241022-v2:0")
mock_session_class.reset_mock()
with patch.dict(os.environ, {
"AWS_ACCESS_KEY_ID": "test-access-key-123",
"AWS_SECRET_ACCESS_KEY": "test-secret-key-456"
}):
llm = LLM(model="bedrock/anthropic.claude-3-5-sonnet-20241022-v2:0")
assert mock_session_class.called
call_kwargs = mock_session_class.call_args[1] if mock_session_class.call_args else {}
assert call_kwargs.get('aws_access_key_id') == "test-access-key-123"
assert call_kwargs.get('aws_secret_access_key') == "test-secret-key-456"
mock_session_class.assert_called_once_with(
aws_access_key_id="test-access-key-123",
aws_secret_access_key="test-secret-key-456",
aws_session_token=None,
region_name="eu-west-1",
)
def test_bedrock_token_usage_tracking():