modify function list to uppercase

Canner · Oct 31, 2024 · b3d030b · b3d030b
1 parent 26192bb
commit b3d030b
Showing 1 changed file with 42 additions and 42 deletions.
diff --git a/wren-ai-service/src/pipelines/common.py b/wren-ai-service/src/pipelines/common.py
@@ -203,52 +203,52 @@ async def _task(result: Dict[str, str]):
 - ONLY USE the tables and columns mentioned in the database schema.
 - ONLY USE "*" if the user query asks for all the columns of a table.
 - ONLY CHOOSE columns belong to the tables mentioned in the database schema.
-- ONLY USE the following SQL functions:
+- ONLY USE the following SQL functions when generating answers:
   - Aggregation functions:
-    - avg
-    - count
-    - max
-    - min
-    - sum
-    - array_agg
-    - bool_or
+    - AVG
+    - COUNT
+    - MAX
+    - MIN
+    - SUM
+    - ARRAY_AGG
+    - BOOL_OR
   - Math functions:
-    - abs
-    - cbrt
-    - ceil
-    - exp
-    - floor
-    - ln
-    - round
-    - sign
-    - greatest
-    - least
-    - mod
-    - power
+    - ABS
+    - CBRT
+    - CEIL
+    - EXP
+    - FLOOR
+    - LN
+    - ROUND
+    - SIGN
+    - GREATEST
+    - LEAST
+    - MOD
+    - POWER
   - String functions:
-    - length
-    - reverse
-    - chr
-    - concat
-    - format
-    - lower
-    - lpad
-    - ltrim
-    - position
-    - replace
-    - rpad
-    - rtrim
-    - strpos
-    - substr
-    - substring
-    - translate
-    - trim
-    - upper
+    - LENGTH
+    - REVERSE
+    - CHR
+    - CONCAT
+    - FORMAT
+    - LOWER
+    - LPAD
+    - LTRIM
+    - POSITION
+    - REPLACE
+    - RPAD
+    - RTRIM
+    - STRPOS
+    - SUBSTR
+    - SUBSTRING
+    - TRANSLATE
+    - TRIM
+    - UPPER
   - Date and Time functions:
-    - current_date
-    - current_timestamp
-    - date_trunc
-    - extract
+    - CURRENT_DATE
+    - CURRENT_TIMESTAMP
+    - DATE_TRUNC
+    - EXTRACT
   - operators:
     - `+`
     - `-`