fix indexing bug

Canner · Nov 16, 2024 · 86092d6 · 86092d6
1 parent e5cf965
commit 86092d6
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/wren-ai-service/src/pipelines/indexing/indexing.py b/wren-ai-service/src/pipelines/indexing/indexing.py
@@ -243,8 +243,8 @@ def _convert_models_and_relationships(
                 if "relationship" not in column:
                     if "properties" in column:
                         column_properties = {
-                            "alias": column["properties"].pop("displayName", ""),
-                            "description": column["properties"].pop("description", ""),
+                            "alias": column["properties"].get("displayName", ""),
+                            "description": column["properties"].get("description", ""),
                         }
                         nested_cols = {
                             k: v
@@ -314,8 +314,8 @@ def _convert_models_and_relationships(
 
             if "properties" in model:
                 model_properties = {
-                    "alias": model["properties"].pop("displayName", ""),
-                    "description": model["properties"].pop("description", ""),
+                    "alias": model["properties"].get("displayName", ""),
+                    "description": model["properties"].get("description", ""),
                 }
                 comment = f"\n/* {orjson.dumps(model_properties).decode("utf-8")} */\n"
             else: