From 02813d646fece60f79eb3e62d41eba7fe1106cac Mon Sep 17 00:00:00 2001 From: Chih-Yu Yeh Date: Mon, 13 May 2024 14:39:53 +0800 Subject: [PATCH] fix indexing (#238) --- .../src/pipelines/ask/indexing_pipeline.py | 11 +++++------ 1 file changed, 5 insertions(+), 6 deletions(-) diff --git a/wren-ai-service/src/pipelines/ask/indexing_pipeline.py b/wren-ai-service/src/pipelines/ask/indexing_pipeline.py index a03b48a84..7954aa1ff 100644 --- a/wren-ai-service/src/pipelines/ask/indexing_pipeline.py +++ b/wren-ai-service/src/pipelines/ask/indexing_pipeline.py @@ -122,14 +122,13 @@ class DDLConverter: logger.debug(f"original mdl_json: {json.dumps(mdl_json, indent=2)}") - for i, _ in enumerate(mdl_json["relationships"]): - mdl_json["relationships"][i]["type"] = "relationship" - semantics = { "models": [], - "relationships": mdl_json["relationships"], - "views": mdl_json["views"], - "metrics": mdl_json["metrics"], + "relationships": mdl_json["relationships"] + if "relationships" in mdl_json + else [], + "views": mdl_json["views"] if "views" in mdl_json else [], + "metrics": mdl_json["metrics"] if "metrics" in mdl_json else [], } for model in mdl_json["models"]: