chore(wren-ai-service): minor updates (#1847)

Author: cyyeh

wren-ai-service/src/pipelines/generation/followup_sql_generation_reasoning.py

@@ -56,6 +56,7 @@
 ### QUESTION ###
 User's Question: {{ query }}
 Language: {{ language }}
+Current Time: {{ current_time }}
 
 Let's think step by step.
 """
@@ -81,6 +82,7 @@ def prompt(
             instructions=instructions,
         ),
         language=configuration.language,
+        current_time=configuration.show_current_time(),
     )
     return {"prompt": clean_up_new_lines(_prompt.get("prompt"))}
 

wren-ai-service/src/pipelines/generation/sql_answer.py

@@ -39,7 +39,7 @@
 """
 
 sql_to_answer_user_prompt_template = """
-### Input
+### Inputs ###
 User's question: {{ query }}
 SQL: {{ sql }}
 Data: 

wren-ai-service/src/pipelines/generation/sql_generation_reasoning.py

@@ -44,9 +44,10 @@
 {% endfor %}
 {% endif %}
 
-### QUESTION ###
+### INPUTS ###
 User's Question: {{ query }}
 Language: {{ language }}
+Current Time: {{ current_time }}
 
 Let's think step by step.
 """
@@ -70,6 +71,7 @@ def prompt(
             instructions=instructions,
         ),
         language=configuration.language,
+        current_time=configuration.show_current_time(),
     )
     return {"prompt": clean_up_new_lines(_prompt.get("prompt"))}
 

wren-ai-service/src/pipelines/generation/utils/sql.py

@@ -183,10 +183,10 @@ async def _classify_generation_result(
 ### INSTRUCTIONS ###
 1. Think deeply and reason about the user's question, the database schema, and the user's query history if provided.
 2. Explicitly state the following information in the reasoning plan: 
-if the user puts any specific timeframe(e.g. YYYY-MM-DD) in the user's question, you will put the absolute time frame in the SQL query; 
-Otherwise, you will put the relative timeframe in the SQL query.
-3. For the ranking problem, you must use the ranking function, `DENSE_RANK()` to rank the results and then use `WHERE` clause to filter the results.
-4. For the ranking problem, you must add the ranking column to the final SELECT clause.
+if the user puts any specific timeframe(e.g. YYYY-MM-DD) in the user's question(excluding the value of the current time), you will put the absolute time frame in the SQL query; 
+otherwise, you will put the relative timeframe in the SQL query.
+3. For the ranking problem(e.g. "top x", "bottom x", "first x", "last x"), you must use the ranking function, `DENSE_RANK()` to rank the results and then use `WHERE` clause to filter the results.
+4. For the ranking problem(e.g. "top x", "bottom x", "first x", "last x"), you must add the ranking column to the final SELECT clause.
 5. If USER INSTRUCTIONS section is provided, make sure to consider them in the reasoning plan.
 6. If SQL SAMPLES section is provided, make sure to consider them in the reasoning plan.
 7. Give a step by step reasoning plan in order to answer user's question.