update retrieval prompt rule

Canner · Nov 4, 2024 · 3bf0267 · 3bf0267
1 parent b445eca
commit 3bf0267
Showing 1 changed file with 1 addition and 0 deletions.
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: