Spaces:
Runtime error
Runtime error
Commit ·
cc58896
1
Parent(s): 7e7f372
Update queryHelperManagerCoT.py
Browse files- queryHelperManagerCoT.py +5 -2
queryHelperManagerCoT.py
CHANGED
|
@@ -59,8 +59,11 @@ class QueryHelperChainOfThought:
|
|
| 59 |
except:
|
| 60 |
print("Couldn't parse desired result from gpt response using method 2")
|
| 61 |
if parsedSql:
|
| 62 |
-
|
| 63 |
-
|
|
|
|
|
|
|
|
|
|
| 64 |
else:
|
| 65 |
responseToReturn = gptResponse
|
| 66 |
return responseToReturn
|
|
|
|
| 59 |
except:
|
| 60 |
print("Couldn't parse desired result from gpt response using method 2")
|
| 61 |
if parsedSql:
|
| 62 |
+
try:
|
| 63 |
+
formattedSql = sqlparse.format(sqlResult, reindent=True)
|
| 64 |
+
responseToReturn = formattedSql
|
| 65 |
+
except:
|
| 66 |
+
responseToReturn = sqlResult
|
| 67 |
else:
|
| 68 |
responseToReturn = gptResponse
|
| 69 |
return responseToReturn
|