Spaces:

Express-Analytics
/

QueryHelper

Runtime error

anumaurya114exp commited on Dec 13, 2023

Commit

36ea358

1 Parent(s): 3d2478f

bug fix

Files changed (1) hide show

queryHelperManagerCoT.py CHANGED Viewed

@@ -42,7 +42,7 @@ class QueryHelperChainOfThought:
     self.gptInstanceForCoT.setSystemPrompt(prompt)
     gptResponse = self.gptInstanceForCoT.getResponseForUserInput(userInput)
     try:
-      txt = response.split("```json")[-1].split('```')[0].replace('\n', '')
       sqlResult = json.loads(txt)['finalResult']
       formattedSql = sqlparse.format(sqlResult, reindent=True)
       print(formattedSql)

     self.gptInstanceForCoT.setSystemPrompt(prompt)
     gptResponse = self.gptInstanceForCoT.getResponseForUserInput(userInput)
     try:
+      txt = gptResponse.split("```json")[-1].split('```')[0].replace('\n', '')
       sqlResult = json.loads(txt)['finalResult']
       formattedSql = sqlparse.format(sqlResult, reindent=True)
       print(formattedSql)