diff --git a/src/crewai/utilities/embedding_configurator.py b/src/crewai/utilities/embedding_configurator.py index 875dab977..ef07c8ebf 100644 --- a/src/crewai/utilities/embedding_configurator.py +++ b/src/crewai/utilities/embedding_configurator.py @@ -141,9 +141,11 @@ class EmbeddingConfigurator: AmazonBedrockEmbeddingFunction, ) - return AmazonBedrockEmbeddingFunction( - session=config.get("session"), - ) + # Allow custom model_name override with backwards compatibility + kwargs = {"session": config.get("session")} + if model_name is not None: + kwargs["model_name"] = model_name + return AmazonBedrockEmbeddingFunction(**kwargs) @staticmethod def _configure_huggingface(config, model_name):