update

Author: cyyeh

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

@@ -49,8 +49,8 @@
 {% for history in histories %}
 Question:
 {{ history.question }}
-SQL:
-{{ history.sql }}
+Response:
+{{ history.response }}
 {% endfor %}
 
 ### QUESTION ###

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

@@ -170,8 +170,8 @@
 {% for history in histories %}
 Question:
 {{ history.question }}
-SQL:
-{{ history.sql }}
+Response:
+{{ history.response }}
 {% endfor %}
 {% endif %}
 

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

@@ -515,7 +515,9 @@ def construct_ask_history_messages(
         )
         messages.append(
             ChatMessage.from_assistant(
-                history.sql if hasattr(history, "sql") else history["sql"]
+                history.response
+                if hasattr(history, "response")
+                else history["response"]
             )
         )
     return messages

wren-ai-service/src/web/v1/services/ask.py

@@ -14,7 +14,7 @@
 
 
 class AskHistory(BaseModel):
-    sql: str
+    response: str = Field(alias=AliasChoices("response", "sql"))
     question: str
 
 
@@ -229,7 +229,7 @@ async def ask(
                     if self._allow_intent_classification:
                         last_sql_data = None
                         if histories:
-                            if last_sql := histories[-1].sql:
+                            if last_sql := histories[-1].response:
                                 last_sql_data = (
                                     await self._pipelines["sql_executor"].run(
                                         sql=last_sql,