Fix

app-ngrok.py

@@ -67,8 +67,6 @@ def format(text):
 
     return final_query_markdown
 
-
-
 def generate(input_message: str, db_info="", temperature=0.1, top_p=0.9, top_k=0, repetition_penalty=1.08, format_sql=True, stop_sequence="###", log=False):
     # Format the user's input message
     messages = f"Below is an instruction that describes a task, paired with an input that provides further context. Write a response that appropriately completes the request.\n\n### Instruction:\n\nConvert text to sql: {input_message} {db_info}\n\n### Response:\n\n"
@@ -102,9 +100,10 @@ def generate(input_message: str, db_info="", temperature=0.1, top_p=0.9, top_k=0
             if response_text and response_text[-1] == ".":
                 response_text = response_text[:-1]
 
-            # Log the request to Firestore
-            formatted_query = format(response_text)
-            log_to_firestore(input_message, db_info, temperature, formatted_query if format_sql else response_text)
+            if log:
+                # Log the request to Firestore
+                formatted_query = format(response_text)
+                log_to_firestore(input_message, db_info, temperature, formatted_query if format_sql else response_text)
 
             if format_sql:
                 return formatted_query