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

Update embedding_configurator.py

Browse Source 
Modified  _configure_bedrock method to use user submitted model_name rather than default  amazon.titan-embed-text-v1.

Sending model_name in short_term_memory (embedder_config/config) was not working.


 # Passing model_name to use model_name provide by user than using default. Added if/else for backward compatibility
This commit is contained in:
rishi154
2025-01-23 20:35:27 +05:30
committed by GitHub
parent e27a15023c
commit 0ad4d5fb6d
1 changed files with 10 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -142,9 +142,16 @@ class EmbeddingConfigurator:
AmazonBedrockEmbeddingFunction,
)
return AmazonBedrockEmbeddingFunction(
session=config.get("session"),
)
# 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
)
@staticmethod
def _configure_huggingface(config, model_name):