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

Update embedding_configurator.py

Browse Source 
Incorporated review comments
This commit is contained in:
rishi154
2025-01-23 21:43:56 +05:30
committed by GitHub
parent 0ad4d5fb6d
commit 1890e6e111
1 changed files with 5 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -142,16 +142,11 @@ class EmbeddingConfigurator:
AmazonBedrockEmbeddingFunction,
)
# 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
)
# 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):