fix groupByKeys bug

Canner · Jul 10, 2024 · 68c9f5d · 68c9f5d
1 parent 6eca8a9
commit 68c9f5d
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/wren-ai-service/src/pipelines/sql_explanation/generation.py b/wren-ai-service/src/pipelines/sql_explanation/generation.py
@@ -41,8 +41,13 @@ def _compose_sql_expression_of_filter_type(filter_analysis: Dict) -> str:
     return ""
 
 
-def _compose_sql_expression_of_groupby_type(groupby_keys: List[List[str]]) -> List[str]:
-    return [f"{','.join(groupby_key)}" for groupby_key in groupby_keys]
+def _compose_sql_expression_of_groupby_type(
+    groupby_keys: List[List[dict]],
+) -> List[str]:
+    return [
+        ", ".join([expression["expression"] for expression in groupby_key])
+        for groupby_key in groupby_keys
+    ]
 
 
 def _compose_sql_expression_of_relation_type(relation: Dict) -> List[str]: