Use either embedder or embedding_model as YAML key (#905)

This commit is contained in:
Sidharth Mohanty
2023-11-04 02:10:16 +05:30
committed by GitHub
parent 53037b5ed8
commit 9c4b023297
4 changed files with 7 additions and 7 deletions

View File

@@ -359,7 +359,7 @@ class Pipeline(EmbedChain):
pipeline_config_data = config_data.get("app", {}).get("config", {})
db_config_data = config_data.get("vectordb", {})
embedding_model_config_data = config_data.get("embedding_model", {})
embedding_model_config_data = config_data.get("embedding_model", config_data.get("embedder", {}))
llm_config_data = config_data.get("llm", {})
pipeline_config = PipelineConfig(**pipeline_config_data)