diff --git a/wren-ai-service/src/core/engine.py b/wren-ai-service/src/core/engine.py index 8d54b36f43..b9fd975229 100644 --- a/wren-ai-service/src/core/engine.py +++ b/wren-ai-service/src/core/engine.py @@ -108,7 +108,6 @@ def is_sql_keyword(text: str) -> bool: "TOP", "WITH", "RECURSIVE", - "CTE", # Data types "INTEGER", "INT", @@ -146,40 +145,12 @@ def is_sql_keyword(text: str) -> bool: "CURRENT_DATE", "CURRENT_TIME", "CURRENT_TIMESTAMP", - "EXTRACT", - "DATE_TRUNC", - "DATE_PART", - # Aggregate functions (common ones) - "COUNT", - "SUM", - "AVG", - "MIN", - "MAX", - "STDDEV", - "VARIANCE", - # Window functions - "OVER", - "PARTITION", - "ROW_NUMBER", - "RANK", - "DENSE_RANK", - "LAG", - "LEAD", - "FIRST_VALUE", - "LAST_VALUE", - "NTILE", # Other common keywords "CASE", "WHEN", "THEN", "ELSE", "END", - "CAST", - "CONVERT", - "COALESCE", - "NULLIF", - "IFNULL", - "ISNULL", "DESC", "ASC", "TRUE",