diff --git a/wren-ai-service/src/pipelines/generation/sql_generation.py b/wren-ai-service/src/pipelines/generation/sql_generation.py index 3417f308a9..c798bee5d3 100644 --- a/wren-ai-service/src/pipelines/generation/sql_generation.py +++ b/wren-ai-service/src/pipelines/generation/sql_generation.py @@ -42,14 +42,6 @@ {% endfor %} {% endif %} -{% if exclude %} -### EXCLUDED STATEMETS ### -Ensure that the following excluded statements are not used in the generated queries to maintain variety and avoid repetition. -{% for doc in exclude %} - {{ doc.statement }} -{% endfor %} -{% endif %} - {{ text_to_sql_rules }} {% if instructions %} {{ instructions }} @@ -77,7 +69,6 @@ def prompt( query: str, documents: List[str], - exclude: List[Dict], text_to_sql_rules: str, prompt_builder: PromptBuilder, configuration: Configuration | None = None, @@ -86,7 +77,6 @@ def prompt( return prompt_builder.run( query=query, documents=documents, - exclude=exclude, text_to_sql_rules=text_to_sql_rules, instructions=construct_instructions(configuration), sql_samples=sql_samples, @@ -162,7 +152,6 @@ async def run( self, query: str, contexts: List[str], - exclude: List[Dict], configuration: Configuration = Configuration(), sql_samples: List[Dict] | None = None, project_id: str | None = None, @@ -173,7 +162,6 @@ async def run( inputs={ "query": query, "documents": contexts, - "exclude": exclude, "sql_samples": sql_samples, "project_id": project_id, "configuration": configuration, @@ -191,5 +179,4 @@ async def run( "sql_generation", query="this is a test query", contexts=[], - exclude=[], ) diff --git a/wren-ai-service/src/web/v1/services/ask.py b/wren-ai-service/src/web/v1/services/ask.py index 6bf26e7faa..6870b82940 100644 --- a/wren-ai-service/src/web/v1/services/ask.py +++ b/wren-ai-service/src/web/v1/services/ask.py @@ -293,7 +293,6 @@ async def ask( ].run( query=user_query, contexts=documents, - exclude=historical_question_result, project_id=ask_request.project_id, configuration=ask_request.configurations, sql_samples=sql_samples,