cleaned duckdbinst template

app.py

@@ -105,11 +105,11 @@ with gr.Blocks(gr.themes.Soft()) as demo:
     )
 
     # Update format dropdown when template is edited
-    # custom_prompt.change(
-    #    fn=handle_template_edit,
-    #    inputs=[prompt_format, custom_prompt],
-    #    outputs=[prompt_format]
-    #)
+    custom_prompt.change(
+        fn=handle_template_edit,
+        inputs=[prompt_format, custom_prompt],
+        outputs=[prompt_format]
+    )
 
     start_btn.click(
         fn=gradio_run_evaluation,

duckdb-nsql/eval/prompt_formatters.py

@@ -115,8 +115,17 @@ class DuckDBFormatter(RajkumarFormatter):
 class DuckDBInstFormatter(RajkumarFormatter):
     """DuckDB Inst class."""
 
-    PROMPT_TEMPLATE = """### Instruction:\n{instruction}\n\n### Input:\n{input}{context}\n### Question:\n{question}\n\n### Response (use duckdb shorthand if possible):\n"""
-    INSTRUCTION_TEMPLATE = """Your task is to generate valid duckdb SQL to answer the following question{has_schema}"""  # noqa: E501
+    PROMPT_TEMPLATE = """### Instruction:
+Your task is to generate valid duckdb SQL to answer the following question, given a duckdb database schema.
+
+Here is the database schema that the SQL query will run on:
+{schema}
+
+### Question:
+{question}
+
+### Response (use duckdb shorthand if possible):
+"""
 
     @classmethod
     def format_retrieved_context(
@@ -135,19 +144,8 @@ class DuckDBInstFormatter(RajkumarFormatter):
         context_text: str,
     ) -> str | list[str]:
         """Get prompt format."""
-        input = ""
-        if table_text:
-            input = """Here is the database schema that the SQL query will run on:\n{schema}\n""".format(  # noqa: E501
-                schema=table_text
-            )
         instruction = cls.PROMPT_TEMPLATE.format(
-            instruction=cls.INSTRUCTION_TEMPLATE.format(
-                has_schema="."
-                if table_text == ""
-                else ", given a duckdb database schema."
-            ),
-            context=context_text,
-            input=input,
+            schema=table_text,
             question=instruction,
         )
         return instruction