Merge branch 'main' into brandon/improve-llm-structured-output

This commit is contained in:
Lorenze Jay
2025-02-04 13:44:28 -08:00
committed by GitHub
15 changed files with 236 additions and 26 deletions

View File

@@ -141,9 +141,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):