anumaurya114exp commited on
Commit
cc58896
·
1 Parent(s): 7e7f372

Update queryHelperManagerCoT.py

Browse files
Files changed (1) hide show
  1. 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
- formattedSql = sqlparse.format(sqlResult, reindent=True)
63
- responseToReturn = formattedSql
 
 
 
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