diff --git a/src/crewai/utilities/embedding_configurator.py b/src/crewai/utilities/embedding_configurator.py index 71965bf53..df9eda399 100644 --- a/src/crewai/utilities/embedding_configurator.py +++ b/src/crewai/utilities/embedding_configurator.py @@ -142,9 +142,16 @@ class EmbeddingConfigurator: AmazonBedrockEmbeddingFunction, ) - return AmazonBedrockEmbeddingFunction( - session=config.get("session"), - ) + # Passing model_name to use model_name provide by user than using default. Added if/else for backward compatibility + if model_name==None: + return AmazonBedrockEmbeddingFunction( + session=config.get("session"), + ) + else: + return AmazonBedrockEmbeddingFunction( + session=config.get("session"), + model_name=model_name + ) @staticmethod def _configure_huggingface(config, model_name):