Update queryHelperManagerCoT.py

queryHelperManagerCoT.py

@@ -133,7 +133,7 @@ ORDER BY chandelier_count DESC"""
     question = "top 5 customers who bought most chandeliers in nov 2023"
     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"""
     for idx, tableName in enumerate(prospectTablesAndCols.keys(), start=1):
-        promptForQuery += f"table name is {tableName}, table data is {self.sampleData[tableName][prospectTablesAndCols[tableName]].head(gptSampleRows)}"   
+        promptForQuery += f"table name is {tableName}, table data is {self.sampleData[tableName][prospectTablesAndCols[tableName]].head(self.gptSampleRows)}"   
     promptForQuery += f"and table Relations are {TABLE_RELATIONS}"
     return promptForQuery.replace("\\"," ").replace("  "," ").replace("XXXX", "    ")