chore: rename 'aws' to 'bedrock' (#403)

## Description
<!-- What does this PR do? -->
Prep for #183 by first
renaming aws to bedrock.

## PR Type

<!-- Delete the types that don't apply --!>

💅 Refactor

## Relevant issues

<!-- e.g. "Fixes #123" -->

## Checklist
- [x] I have added unit tests that prove my fix/feature works
- [ ] New and existing tests pass locally
- [x] Documentation was updated where necessary
- [ ] I have read and followed the [contribution
guidelines](https://github.com/mozilla-ai/any-llm/blob/main/CONTRIBUTING.md)```

Author: njbrake

pyproject.toml

@@ -19,7 +19,7 @@ dependencies = [
 [project.optional-dependencies]
 
 all = [
-  "any-llm-sdk[mistral,anthropic,huggingface,gemini,vertexai,cohere,cerebras,fireworks,groq,aws,azure,azureopenai,watsonx,together,sambanova,ollama,moonshot,nebius,xai,databricks,deepseek,inception,openai,openrouter,portkey,lmstudio,llama,voyage,perplexity,llamafile,llamacpp]"
+  "any-llm-sdk[mistral,anthropic,huggingface,gemini,vertexai,cohere,cerebras,fireworks,groq,bedrock,azure,azureopenai,watsonx,together,sambanova,ollama,moonshot,nebius,xai,databricks,deepseek,inception,openai,openrouter,portkey,lmstudio,llama,voyage,perplexity,llamafile,llamacpp]"
 ]
 
 perplexity = []
@@ -56,7 +56,7 @@ groq = [
   "groq"
 ]
 
-aws = [
+bedrock = [
   "boto3",
   "instructor",
 ]

src/any_llm/provider.py

@@ -35,7 +35,7 @@ class ProviderName(StrEnum):
     """String enum for supported providers."""
 
     ANTHROPIC = "anthropic"
-    AWS = "aws"
+    BEDROCK = "bedrock"
     AZURE = "azure"
     AZUREOPENAI = "azureopenai"
     CEREBRAS = "cerebras"

src/any_llm/providers/aws/__init__.py

@@ -0,0 +1,3 @@
+from .bedrock import BedrockProvider
+
+__all__ = ["BedrockProvider"]

src/any_llm/providers/bedrock/__init__.py

@@ -29,7 +29,7 @@
     MISSING_PACKAGES_ERROR = e
 
 
-class AwsProvider(Provider):
+class BedrockProvider(Provider):
     """AWS Bedrock Provider using boto3 and instructor for structured output."""
 
     PROVIDER_NAME = "aws"

src/any_llm/providers/aws/aws.py renamed to src/any_llm/providers/bedrock/bedrock.py

@@ -49,7 +49,7 @@ def provider_model_map() -> dict[ProviderName, str]:
         ProviderName.COHERE: "command-a-03-2025",
         ProviderName.CEREBRAS: "llama-3.3-70b",
         ProviderName.HUGGINGFACE: "huggingface/tgi",  # This is the syntax used in `litellm` when using HF Inference Endpoints (https://docs.litellm.ai/docs/providers/huggingface#dedicated-inference-endpoints)
-        ProviderName.AWS: "amazon.nova-lite-v1:0",
+        ProviderName.BEDROCK: "amazon.nova-lite-v1:0",
         ProviderName.WATSONX: "ibm/granite-3-8b-instruct",
         ProviderName.FIREWORKS: "accounts/fireworks/models/llama4-scout-instruct-basic",
         ProviderName.GROQ: "llama-3.1-8b-instant",
@@ -73,7 +73,7 @@ def embedding_provider_model_map() -> dict[ProviderName, str]:
         ProviderName.NEBIUS: "Qwen/Qwen3-Embedding-8B",
         ProviderName.SAMBANOVA: "E5-Mistral-7B-Instruct",
         ProviderName.MISTRAL: "mistral-embed",
-        ProviderName.AWS: "amazon.titan-embed-text-v2:0",
+        ProviderName.BEDROCK: "amazon.titan-embed-text-v2:0",
         ProviderName.OLLAMA: "gpt-oss:20b",
         ProviderName.LLAMAFILE: "N/A",
         ProviderName.LMSTUDIO: "text-embedding-nomic-embed-text-v1.5",

src/any_llm/providers/aws/utils.py renamed to src/any_llm/providers/bedrock/utils.py

@@ -4,16 +4,16 @@
 from unittest.mock import Mock, patch
 
 from any_llm.provider import ClientConfig
-from any_llm.providers.aws.aws import AwsProvider
+from any_llm.providers.bedrock import BedrockProvider
 from any_llm.types.completion import CompletionParams
 
 
 @contextmanager
 def mock_aws_provider(region: str):  # type: ignore[no-untyped-def]
     with (
         patch.dict(os.environ, {"AWS_REGION": region}),
-        patch("any_llm.providers.aws.aws.AwsProvider._check_aws_credentials"),
-        patch("any_llm.providers.aws.aws._convert_response"),
+        patch("any_llm.providers.bedrock.bedrock.BedrockProvider._check_aws_credentials"),
+        patch("any_llm.providers.bedrock.bedrock._convert_response"),
         patch("boto3.client") as mock_boto3_client,
     ):
         mock_client = Mock()
@@ -28,7 +28,7 @@ def test_boto3_client_created_with_api_base() -> None:
     region = "us-east-1"
 
     with mock_aws_provider(region) as mock_boto3_client:
-        provider = AwsProvider(ClientConfig(api_base=custom_endpoint, api_key="test_key"))
+        provider = BedrockProvider(ClientConfig(api_base=custom_endpoint, api_key="test_key"))
         provider.completion(CompletionParams(model_id="model-id", messages=[{"role": "user", "content": "Hello"}]))
 
         mock_boto3_client.assert_called_once_with("bedrock-runtime", endpoint_url=custom_endpoint, region_name=region)
@@ -39,7 +39,7 @@ def test_boto3_client_created_without_api_base() -> None:
     region = "us-west-2"
 
     with mock_aws_provider(region) as mock_boto3_client:
-        provider = AwsProvider(ClientConfig(api_key="test_key"))
+        provider = BedrockProvider(ClientConfig(api_key="test_key"))
         provider.completion(CompletionParams(model_id="model-id", messages=[{"role": "user", "content": "Hello"}]))
 
         mock_boto3_client.assert_called_once_with("bedrock-runtime", endpoint_url=None, region_name=region)
@@ -52,7 +52,7 @@ def test_completion_with_kwargs() -> None:
     messages = [{"role": "user", "content": "Hello"}]
 
     with mock_aws_provider(region) as mock_boto3_client:
-        provider = AwsProvider(ClientConfig(api_key="test_key"))
+        provider = BedrockProvider(ClientConfig(api_key="test_key"))
         provider.completion(
             CompletionParams(
                 model_id=model_id,
@@ -85,7 +85,7 @@ def mock_aws_embedding_provider(region: str):  # type: ignore[no-untyped-def]
     """Mock AWS provider specifically for embedding tests."""
     with (
         patch.dict(os.environ, {"AWS_REGION": region}),
-        patch("any_llm.providers.aws.aws.AwsProvider._check_aws_credentials"),
+        patch("any_llm.providers.bedrock.bedrock.BedrockProvider._check_aws_credentials"),
         patch("boto3.client") as mock_boto3_client,
     ):
         mock_client = Mock()
@@ -104,7 +104,7 @@ def test_embedding_single_string() -> None:
     with mock_aws_embedding_provider(region) as (mock_boto3_client, mock_client):
         mock_client.invoke_model.return_value = {"body": Mock(read=Mock(return_value=json.dumps(mock_response_body)))}
 
-        provider = AwsProvider(ClientConfig(api_key="test_key"))
+        provider = BedrockProvider(ClientConfig(api_key="test_key"))
         response = provider.embedding(model_id, input_text)
 
         mock_boto3_client.assert_called_once_with("bedrock-runtime", endpoint_url=None, region_name=region)
@@ -138,7 +138,7 @@ def test_embedding_list_of_strings() -> None:
             {"body": Mock(read=Mock(return_value=json.dumps(mock_response_bodies[1])))},
         ]
 
-        provider = AwsProvider(ClientConfig(api_key="test_key"))
+        provider = BedrockProvider(ClientConfig(api_key="test_key"))
         response = provider.embedding(model_id, input_texts)
 
         mock_boto3_client.assert_called_once_with("bedrock-runtime", endpoint_url=None, region_name=region)

tests/conftest.py

@@ -129,7 +129,7 @@ def test_all_providers_have_required_attributes(provider: ProviderName) -> None:
 
 
 def test_providers_raise_MissingApiKeyError(provider: ProviderName) -> None:
-    if provider.value in ("aws", "ollama", "lmstudio", "llamafile"):
+    if provider in (ProviderName.BEDROCK, ProviderName.OLLAMA, ProviderName.LMSTUDIO, ProviderName.LLAMAFILE):
         pytest.skip("This provider handles `api_key` differently.")
     with patch.dict(os.environ, {}, clear=True):
         with pytest.raises(MissingApiKeyError):
@@ -140,7 +140,7 @@ def test_providers_raise_MissingApiKeyError(provider: ProviderName) -> None:
     ("provider_name", "module_name"),
     [
         ("anthropic", "anthropic"),
-        ("aws", "boto3"),
+        ("bedrock", "boto3"),
         ("azure", "azure"),
         ("cerebras", "cerebras"),
         ("cohere", "cohere"),