added 2 methods for parsing CoT result

queryHelperManagerCoT.py

@@ -41,14 +41,27 @@ class QueryHelperChainOfThought:
     prompt = self.getPromptForCot()
     self.gptInstanceForCoT.setSystemPrompt(prompt)
     gptResponse = self.gptInstanceForCoT.getResponseForUserInput(userInput)
-    try:
-      txt = gptResponse.split("```json")[-1].split('```')[0].replace('\n', '')
-      sqlResult = json.loads(txt)['finalResult']
+    tryParsing = True
+    parsedSql = False
+    if tryParsing:
+      try:
+        txt = gptResponse.split("```json")[-1].split('```')[0].replace('\n', '')
+        sqlResult = json.loads(txt)['finalResult']
+        parsedSql = True
+        tryParsing = False
+      except:
+        print("Couldn't parse desired result from gpt response using method 1.")
+    if tryParsing:
+      try:
+        sqlResult = json.loads(gptResponse)
+        parsedSql = True
+        tryParsing = False
+      except:
+        print("Couldn't parse desired result from gpt response using method 2")
+    if parsedSql:
       formattedSql = sqlparse.format(sqlResult, reindent=True)
-      print(formattedSql)
       responseToReturn = formattedSql
-    except:
-      print("Couldn't parse desired result from gpt response")
+    else:
       responseToReturn = gptResponse
     return responseToReturn