chore(wren-ai-service): Update retrieval prompt rule for GA4 struct t…

…ype column (#867)
Canner · Nov 6, 2024 · b207f28 · b207f28
1 parent 1a42964
commit b207f28
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/wren-ai-service/src/pipelines/common.py b/wren-ai-service/src/pipelines/common.py
@@ -287,6 +287,7 @@ async def _task(result: Dict[str, str]):
 
     SQL
     SELECT ApprovedTimestamp AS _timestamp FROM orders AS _orders;
+- DON'T USE '.' in column/table alias, replace '.' with '_' in column/table alias.
 """
 
 

diff --git a/wren-ai-service/src/pipelines/retrieval/retrieval.py b/wren-ai-service/src/pipelines/retrieval/retrieval.py
@@ -32,6 +32,7 @@
 4. If a "." is included in columns, put the name before the first dot into chosen columns.
 5. The number of columns chosen must match the number of reasoning.
 6. Final chosen columns must be only column names, don't prefix it with table names.
+7. If the chosen column is a child column of a STRUCT type column, choose the parent column instead of the child column.
 
 ### FINAL ANSWER FORMAT ###
 Please provide your response as a JSON object, structured as follows: