From 85a13751ba98b03ace8efff5e9c47e3e8b177ac2 Mon Sep 17 00:00:00 2001 From: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> Date: Wed, 5 Feb 2025 10:44:59 +0000 Subject: [PATCH] fix: Add type safety to embedding configurator Co-Authored-By: Joe Moura --- src/crewai/utilities/embedding_configurator.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/crewai/utilities/embedding_configurator.py b/src/crewai/utilities/embedding_configurator.py index b86541a38..c8888ea70 100644 --- a/src/crewai/utilities/embedding_configurator.py +++ b/src/crewai/utilities/embedding_configurator.py @@ -38,12 +38,12 @@ class EmbeddingConfigurator: except Exception as e: raise ValueError(f"Invalid custom embedding function: {str(e)}") - if provider not in self.embedding_functions: + if not provider or provider not in self.embedding_functions: raise Exception( f"Unsupported embedding provider: {provider}, supported providers: {list(self.embedding_functions.keys())}" ) - return self.embedding_functions[provider](config, model_name) + return self.embedding_functions[str(provider)](config, model_name) @staticmethod def _create_default_embedding_function():