anumaurya114exp commited on
Commit
f1037ed
·
1 Parent(s): 9a04841

Update queryHelperManagerCoT.py

Browse files
Files changed (1) hide show
  1. queryHelperManagerCoT.py +1 -1
queryHelperManagerCoT.py CHANGED
@@ -88,7 +88,7 @@ ORDER BY state, category_rank"
88
  }"""
89
  promptTableInfo = self.getSystemPromptForTableCols()
90
  selectedTablesAndCols = self.metadataLayout.getSelectedTablesAndCols()
91
- promptColumnsInfo = getSystemPromptForQuery(selectedTablesAndCols)
92
 
93
  prompt = f"""You are a powerful text to sql model. Your task is to return sql query which answers
94
  user's input. Please follow subquery structure if the sql needs to have multiple subqueries.
 
88
  }"""
89
  promptTableInfo = self.getSystemPromptForTableCols()
90
  selectedTablesAndCols = self.metadataLayout.getSelectedTablesAndCols()
91
+ promptColumnsInfo = self.getSystemPromptForQuery(selectedTablesAndCols)
92
 
93
  prompt = f"""You are a powerful text to sql model. Your task is to return sql query which answers
94
  user's input. Please follow subquery structure if the sql needs to have multiple subqueries.