andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-01-09 08:08:32 +00:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

fix: Add type safety to embedding configurator

Browse Source 
Co-Authored-By: Joe Moura <joao@crewai.com>
This commit is contained in:
Devin AI
2025-02-05 10:44:59 +00:00
parent 1c7c4cb828
commit 85a13751ba
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/crewai/utilities/embedding_configurator.py
View File

@@ -38,12 +38,12 @@ class EmbeddingConfigurator:
except Exception as e: except Exception as e:
raise ValueError(f"Invalid custom embedding function: {str(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( raise Exception(
f"Unsupported embedding provider: {provider}, supported providers: {list(self.embedding_functions.keys())}" 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 @staticmethod
def _create_default_embedding_function(): def _create_default_embedding_function():