fix: change aws-bedrock to bedrock (#375)

Author: amadeusz-ds

CHANGELOG.md

@@ -1,7 +1,8 @@
-## 0.4.8-dev1
+## 0.5.0
 
 ### Fixes
 
+* **Change aws-bedrock to bedrock**
 * **Update Sharepoint tests**
 
 ### Enhancements

requirements/embed/aws-bedrock.in renamed to requirements/embed/bedrock.in

@@ -1,30 +1,30 @@
 # This file was autogenerated by uv via the following command:
-#    uv pip compile aws-bedrock.in --output-file aws-bedrock.txt --no-strip-extras --python-version 3.9
+#    uv pip compile ./embed/bedrock.in --output-file ./embed/bedrock.txt --no-strip-extras --python-version 3.9
 aioboto3==13.1.1
-    # via -r aws-bedrock.in
+    # via -r ./embed/bedrock.in
 aiobotocore[boto3]==2.13.1
     # via aioboto3
 aiofiles==24.1.0
     # via aioboto3
 aiohappyeyeballs==2.4.4
     # via aiohttp
-aiohttp==3.11.10
+aiohttp==3.11.12
     # via aiobotocore
 aioitertools==0.12.0
     # via aiobotocore
-aiosignal==1.3.1
+aiosignal==1.3.2
     # via aiohttp
 async-timeout==5.0.1
     # via aiohttp
-attrs==24.2.0
+attrs==25.1.0
     # via aiohttp
 boto3==1.34.131
     # via
-    #   -r aws-bedrock.in
+    #   -r ./embed/bedrock.in
     #   aiobotocore
 botocore==1.34.131
     # via
-    #   -c ../common/constraints.txt
+    #   -c ./embed/../common/constraints.txt
     #   aiobotocore
     #   boto3
     #   s3transfer
@@ -58,11 +58,11 @@ typing-extensions==4.12.2
     #   multidict
 urllib3==1.26.20
     # via
-    #   -c ../common/constraints.txt
+    #   -c ./embed/../common/constraints.txt
     #   botocore
-wrapt==1.17.0
+wrapt==1.17.2
     # via
-    #   -c ../common/constraints.txt
+    #   -c ./embed/../common/constraints.txt
     #   aiobotocore
 yarl==1.18.3
     # via aiohttp

requirements/embed/aws-bedrock.txt renamed to requirements/embed/bedrock.txt

@@ -142,7 +142,7 @@ def load_requirements(file: Union[str, Path]) -> List[str]:
     "embed-voyageai": load_requirements("requirements/embed/voyageai.in"),
     "embed-mixedbreadai": load_requirements("requirements/embed/mixedbreadai.in"),
     "openai": load_requirements("requirements/embed/openai.in"),
-    "bedrock": load_requirements("requirements/embed/aws-bedrock.in"),
+    "bedrock": load_requirements("requirements/embed/bedrock.in"),
     "togetherai": load_requirements("requirements/embed/togetherai.in"),
 }
 

setup.py

@@ -31,7 +31,7 @@ def get_aws_credentials() -> dict:
 def test_bedrock_embedder(embedder_file: Path):
     aws_credentials = get_aws_credentials()
     embedder_config = EmbedderConfig(
-        embedding_provider="aws-bedrock",
+        embedding_provider="bedrock",
         embedding_aws_access_key_id=aws_credentials["aws_access_key_id"],
         embedding_aws_secret_access_key=aws_credentials["aws_secret_access_key"],
     )

test/integration/embedders/test_bedrock.py

@@ -1 +1 @@
-__version__ = "0.4.8-dev1"  # pragma: no cover
+__version__ = "0.5.0"  # pragma: no cover

unstructured_ingest/__version__.py

@@ -417,7 +417,7 @@ def get_cli_options() -> t.List[click.Option]:
         embed_providers = [
             "openai",
             "huggingface",
-            "aws-bedrock",
+            "bedrock",
             "vertexai",
             "voyageai",
             "octoai",

unstructured_ingest/cli/interfaces.py

@@ -226,7 +226,7 @@ def get_embedder(self) -> "BaseEmbeddingEncoder":
             )
 
             return OctoAIEmbeddingEncoder(config=OctoAiEmbeddingConfig(**kwargs))
-        elif self.provider == "aws-bedrock":
+        elif self.provider == "bedrock":
             from unstructured_ingest.embed.bedrock import (
                 BedrockEmbeddingConfig,
                 BedrockEmbeddingEncoder,

unstructured_ingest/interfaces.py

@@ -18,7 +18,7 @@ class EmbedderConfig(BaseModel):
             "openai",
             "azure-openai",
             "huggingface",
-            "aws-bedrock",
+            "bedrock",
             "vertexai",
             "voyageai",
             "octoai",
@@ -162,7 +162,7 @@ def get_embedder(self) -> "BaseEmbeddingEncoder":
         if self.embedding_provider == "octoai":
             return self.get_octoai_embedder(embedding_kwargs=kwargs)
 
-        if self.embedding_provider == "aws-bedrock":
+        if self.embedding_provider == "bedrock":
             return self.get_bedrock_embedder()
 
         if self.embedding_provider == "vertexai":