chore(wren-ai-service): remove unused code (#1115)

Canner · Jan 16, 2025 · e479e14 · e479e14
1 parent c451a19
commit e479e14
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 14 deletions.
diff --git 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
@@ -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,