Error checking

app.py

@@ -74,7 +74,10 @@ def bot(input_message: str, db_info="", temperature=0.1, top_p=0.9, top_k=0, rep
         partial_text += new_text
 
     # Split the text by "|", and get the last element in the list which should be the final query
-    final_query = partial_text.split("|")[1].strip()
+    try:
+        final_query = partial_text.split("|")[1].strip()
+    except Exception:
+        final_query = partial_text
 
     try:
         # Attempt to format SQL query using sqlparse