anumaurya114exp commited on
Commit
5b99a4b
·
1 Parent(s): a76a91d

Update queryHelperManager.py

Browse files
Files changed (1) hide show
  1. queryHelperManager.py +1 -1
queryHelperManager.py CHANGED
@@ -86,7 +86,7 @@ GROUP BY a.customer_id
86
  ORDER BY chandelier_count DESC"""
87
 
88
  question = "top 5 customers who bought most chandeliers in nov 2023"
89
- promptForQuery = f"""You are a powerful text to sql model. Answer user input with sql query. And the query needs to run on {platform}. and schemaName is {schemaName}. There is example user input and desired generated sql query. Follow similar patterns as example. eg case insensitive, explicit variable declaration etc. user input : {question}, query : {exampleQuery}. and table's data is \n"""
90
  for idx, tableName in enumerate(prospectTablesAndCols.keys(), start=1):
91
  promptForQuery += f"table name is {tableName}, table data is {self.sampleData[tableName][prospectTablesAndCols[tableName]].head(self.gptSampleRows)}"
92
  promptForQuery += f"and table Relations are {TABLE_RELATIONS}"
 
86
  ORDER BY chandelier_count DESC"""
87
 
88
  question = "top 5 customers who bought most chandeliers in nov 2023"
89
+ promptForQuery = f"""You are a powerful text to sql model. Answer user input with sql query. Answer only sql query. And the query needs to run on {platform}. and schemaName is {schemaName}. There is example user input and desired generated sql query. Follow similar patterns as example. eg case insensitive, explicit variable declaration etc. user input : {question}, query : {exampleQuery}. and table's data is \n"""
90
  for idx, tableName in enumerate(prospectTablesAndCols.keys(), start=1):
91
  promptForQuery += f"table name is {tableName}, table data is {self.sampleData[tableName][prospectTablesAndCols[tableName]].head(self.gptSampleRows)}"
92
  promptForQuery += f"and table Relations are {TABLE_RELATIONS}"