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

2023-11-03 13:40:16 -07:00
parent 53037b5ed8
commit 9c4b023297
4 changed files with 7 additions and 7 deletions
@@ -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)