Fix syntax for multiple alternate responses #44

Author: pramitchoudhary

sidekick/configs/prompt_template.py

@@ -99,7 +99,7 @@
 
 
 ### Input:
-For SQL TABLE '{table_name}' with sample question/answer pairs,\n({sample_queries}), create a SQL (dialect:SQLite) query to answer the following question:\n{question_txt}.
+For SQL TABLE '{table_name}' with sample question/answer pairs,\n({sample_queries}), create a valid SQL (dialect:SQLite) query to answer the following question:\n{question_txt}.
 This query will run on a database whose schema is represented in this string:
 CREATE TABLE '{table_name}' ({column_info}
 );

sidekick/query.py

@@ -584,6 +584,10 @@ def generate_sql(
                         _out = output_re.sequences[sorted_idx]
                         res = tokenizer.decode(_out[input_length:], skip_special_tokens=True)
                         result = res.replace("table_name", _table_name)
+                        # Remove the last semi-colon if exists at the end
+                        # we will add it later
+                        if result.endswith(";"):
+                            result = result.replace(";", "")
                         if "LIMIT".lower() not in result.lower():
                             res = "SELECT " + result.strip() + " LIMIT 100;"
                         else:
@@ -602,6 +606,8 @@ def generate_sql(
                 # COLLATE NOCASE is used to ignore case sensitivity, this might be specific to sqlite
                 _temp = _res.replace("table_name", table_name).split(";")[0]
 
+                if _temp.endswith(";"):
+                    _temp = _temp.replace(";", "")
                 if "LIMIT".lower() not in _temp.lower():
                     res = "SELECT " + _temp.strip() + " LIMIT 100;"
                 else: