Spaces:
Runtime error
Runtime error
Commit ·
4397c21
1
Parent(s): be452a0
added logging.
Browse files
app.py
CHANGED
|
@@ -7,6 +7,7 @@ import sqlparse
|
|
| 7 |
import re
|
| 8 |
import os
|
| 9 |
import warnings
|
|
|
|
| 10 |
|
| 11 |
|
| 12 |
from config import *
|
|
@@ -156,7 +157,14 @@ def checkAuth(username, password):
|
|
| 156 |
def respond(message, chatHistory):
|
| 157 |
"""gpt response handler for gradio ui"""
|
| 158 |
global queryHelper
|
| 159 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 160 |
chatHistory.append((message, botMessage))
|
| 161 |
time.sleep(2)
|
| 162 |
return "", chatHistory
|
|
@@ -189,6 +197,9 @@ def testSQL(sql):
|
|
| 189 |
df = pd.read_sql_query(sql, con=conn)
|
| 190 |
return pd.DataFrame(df)
|
| 191 |
except Exception as e:
|
|
|
|
|
|
|
|
|
|
| 192 |
print(f"Error occured during running the query {sql}.\n and the error is {str(e)}")
|
| 193 |
|
| 194 |
prompt = f"Please correct the following sql query, also it has to be run on {PLATFORM}. sql query is \n {sql}. the error occured is {str(e)}."
|
|
|
|
| 7 |
import re
|
| 8 |
import os
|
| 9 |
import warnings
|
| 10 |
+
from persistStorage import saveLog
|
| 11 |
|
| 12 |
|
| 13 |
from config import *
|
|
|
|
| 157 |
def respond(message, chatHistory):
|
| 158 |
"""gpt response handler for gradio ui"""
|
| 159 |
global queryHelper
|
| 160 |
+
try:
|
| 161 |
+
botMessage = queryHelper.getQueryForUserInput(message)
|
| 162 |
+
except Exception as e:
|
| 163 |
+
errorMessage = {"function":"queryHelper.getQueryForUserInput","error":str(e), "userInput":message}
|
| 164 |
+
saveLog(errorMessage, 'error')
|
| 165 |
+
|
| 166 |
+
logMessage = {"userInput":message, "queryGenerated":botMessage}
|
| 167 |
+
saveLog(logMessage)
|
| 168 |
chatHistory.append((message, botMessage))
|
| 169 |
time.sleep(2)
|
| 170 |
return "", chatHistory
|
|
|
|
| 197 |
df = pd.read_sql_query(sql, con=conn)
|
| 198 |
return pd.DataFrame(df)
|
| 199 |
except Exception as e:
|
| 200 |
+
errorMessage = {"function":"testSQL","error":str(e), "userInput":sql}
|
| 201 |
+
saveLog(errorMessage, 'error')
|
| 202 |
+
|
| 203 |
print(f"Error occured during running the query {sql}.\n and the error is {str(e)}")
|
| 204 |
|
| 205 |
prompt = f"Please correct the following sql query, also it has to be run on {PLATFORM}. sql query is \n {sql}. the error occured is {str(e)}."
|