richardr1126 commited on
Commit
55fe6af
1 Parent(s): c3f3890
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -294,6 +294,8 @@ def generate(input_message: str, db_info="", temperature=0.2, top_p=0.9, top_k=0
294
  query = format(query) if format_sql else query
295
  responses = [query]
296
 
 
 
297
  if log:
298
  # Log the request to Firestore
299
  log_message_to_firestore(input_message, db_info, temperature, output)
 
294
  query = format(query) if format_sql else query
295
  responses = [query]
296
 
297
+ output = responses[0]
298
+
299
  if log:
300
  # Log the request to Firestore
301
  log_message_to_firestore(input_message, db_info, temperature, output)