diff --git a/wren-ai-service/src/pipelines/generation/semantics_description.py b/wren-ai-service/src/pipelines/generation/semantics_description.py index 640f9b9868..80bb0f2d16 100644 --- a/wren-ai-service/src/pipelines/generation/semantics_description.py +++ b/wren-ai-service/src/pipelines/generation/semantics_description.py @@ -29,6 +29,7 @@ def column_formatter(columns: list[dict]) -> list[dict]: "type": column["type"], "properties": { "description": column["properties"].get("description", ""), + "alias": column["properties"].get("displayName", ""), }, } for column in columns @@ -41,6 +42,7 @@ def extract(model: dict) -> dict: "columns": column_formatter(model["columns"]), "properties": { "description": model["properties"].get("description", ""), + "alias": model["properties"].get("displayName", ""), }, }