diff --git a/llama-index-integrations/embeddings/llama-index-embeddings-bedrock/llama_index/embeddings/bedrock/base.py b/llama-index-integrations/embeddings/llama-index-embeddings-bedrock/llama_index/embeddings/bedrock/base.py index 0971640b2e757c90be6842ed96f5714a07f2e75e..ac5faa6d1c1b2affdbc579e69487c839bcc95a7a 100644 --- a/llama-index-integrations/embeddings/llama-index-embeddings-bedrock/llama_index/embeddings/bedrock/base.py +++ b/llama-index-integrations/embeddings/llama-index-embeddings-bedrock/llama_index/embeddings/bedrock/base.py @@ -311,7 +311,7 @@ class BedrockEmbedding(BaseEmbedding): client = session.client("bedrock") return cls( client=client, - model_name=model_name, + model=model_name, embed_batch_size=embed_batch_size, callback_manager=callback_manager, verbose=verbose, diff --git a/llama-index-legacy/llama_index/legacy/embeddings/bedrock.py b/llama-index-legacy/llama_index/legacy/embeddings/bedrock.py index 7244fad66551cf494aa0d8213f76b2f3f054b070..001a93ac07acb6f2dad81ee3327ee4b270728387 100644 --- a/llama-index-legacy/llama_index/legacy/embeddings/bedrock.py +++ b/llama-index-legacy/llama_index/legacy/embeddings/bedrock.py @@ -312,7 +312,7 @@ class BedrockEmbedding(BaseEmbedding): client = session.client("bedrock") return cls( client=client, - model_name=model_name, + model=model_name, embed_batch_size=embed_batch_size, callback_manager=callback_manager, verbose=verbose,