diff --git a/src/crewai/utilities/embedding_configurator.py b/src/crewai/utilities/embedding_configurator.py index 657363967..b75cd2f33 100644 --- a/src/crewai/utilities/embedding_configurator.py +++ b/src/crewai/utilities/embedding_configurator.py @@ -180,9 +180,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):