Update queryHelperManagerCoT.py

queryHelperManagerCoT.py

@@ -59,12 +59,20 @@ class QueryHelperChainOfThought:
       except:
         print("Couldn't parse desired result from gpt response using method 2")
     if parsedSql:
+      isFormatted = False
       try:
         formattedSql = sqlparse.format(sqlResult, reindent=True)
         responseToReturn = formattedSql
+        isFormatted = True
       except:
-        print("Error while formatting parsed sql result :", sqlResult)
-        responseToReturn = sqlResult
+        isFormatted = False
+      if not isFormatted:
+        try:
+          formattedSql = sqlparse.format(sqlResult['result'], reindent=True)
+          responseToReturn = formattedSql
+          print("gpt didn't give parsed result. So parsing again. the formatting.")
+        except:
+          responseToReturn = str(sqlResult)
     else:
       responseToReturn = gptResponse
     return responseToReturn