anumaurya114exp commited on
Commit
9a04841
·
1 Parent(s): 714fdb8
Files changed (1) hide show
  1. queryHelperManager.py +2 -1
queryHelperManager.py CHANGED
@@ -50,7 +50,8 @@ class QueryHelper:
50
  for col in selectedTablesAndCols[table]:
51
  if col in prospectTablesAndColsText:
52
  prospectTablesAndCols[table].append(col)
53
- promptForQuery = getSystemPromptForQuery(prospectTablesAndCols)
 
54
  self.gptInstanceForQuery.setSystemPrompt(promptForQuery)
55
  gptResponse = self.gptInstanceForQuery.getResponseForUserInput(userInput)
56
  return gptResponse
 
50
  for col in selectedTablesAndCols[table]:
51
  if col in prospectTablesAndColsText:
52
  prospectTablesAndCols[table].append(col)
53
+ print("tables and cols select by gpt", prospectTablesAndCols)
54
+ promptForQuery = self.getSystemPromptForQuery(prospectTablesAndCols)
55
  self.gptInstanceForQuery.setSystemPrompt(promptForQuery)
56
  gptResponse = self.gptInstanceForQuery.getResponseForUserInput(userInput)
57
  return gptResponse