From a92613adcedb3ae52a6936c7ba157d3a4614dabe Mon Sep 17 00:00:00 2001 From: ChihYu Yeh Date: Thu, 23 Jan 2025 16:39:15 +0800 Subject: [PATCH] update --- .../src/pipelines/generation/sql_generation_reasoning.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/wren-ai-service/src/pipelines/generation/sql_generation_reasoning.py b/wren-ai-service/src/pipelines/generation/sql_generation_reasoning.py index 3f001dc845..063d59b9d8 100644 --- a/wren-ai-service/src/pipelines/generation/sql_generation_reasoning.py +++ b/wren-ai-service/src/pipelines/generation/sql_generation_reasoning.py @@ -18,13 +18,13 @@ sql_generation_reasoning_system_prompt = """ ### TASK ### -You are a helpful data analyst that great at thinking deeply and reasoning about the user's question and the database schema and give a step by step reasoning plan in order to answer user's question. +You are a helpful data analyst who is great at thinking deeply and reasoning about the user's question and the database schema, and you provide a step-by-step reasoning plan in order to answer the user's question. ### INSTRUCTIONS ### 1. Think deeply and reason about the user's question and the database schema. 2. Give a step by step reasoning plan in order to answer user's question. 3. The reasoning plan should be in the language same as the language user provided in the input. -4. Make sure considering current time provided in the input. +4. Make sure to consider the current time provided in the input if the user's question is related to the date/time. """ sql_generation_reasoning_user_prompt_template = """ @@ -48,7 +48,7 @@ def prompt( query: str, documents: List[str], prompt_builder: PromptBuilder, - configuration: Configuration | None = None, + configuration: Configuration | None = Configuration(), ) -> dict: return prompt_builder.run( query=query,