Upload 4 files

nlp/sql_model.py

@@ -22,30 +22,18 @@ class SQLModel:
             "- Manager (VARCHAR)\n\n"
             "Here are a few examples:\n\n"
             "1. **Input**: \"Show me all employees in the Sales department.\"\n"
-            "**Output**:\n\n"
-            "        SELECT *\n"
-            "        FROM Employees\n"
-            "        WHERE Department = 'Sales';\n\n"
+            "**Output**:""SELECT * FROM Employees WHERE Department = 'Sales';"
             "2. **Input**: \"Who is the manager of the Engineering department?\"\n"
-            "**Output**:\n\n"
-            "        SELECT Manager\n"
-            "        FROM Departments\n"
-            "        WHERE Name = 'Engineering';\n\n"
+            "**Output**:""SELECT Manager FROM Departments WHERE Name = 'Engineering';"
             "3. **Input**: \"List all employees hired after 2021-01-01.\"\n"
-            "**Output**:\n\n"
-            "        SELECT *\n"
-            "        FROM Employees\n"
-            "        WHERE Hire_Date > '2021-01-01';\n\n"
+            "**Output**:""SELECT *""FROM Employees""WHERE Hire_Date > '2021-01-01';"
             "4. **Input**: \"What is the total salary expense for the Marketing department?\"\n"
-            "**Output**:\n\n"
-            "        SELECT SUM(Salary)\n"
-            "        FROM Employees\n"
-            "        WHERE Department = 'Marketing';\n\n"
+            "**Output**:""SELECT SUM(Salary) FROM Employees WHERE Department = 'Marketing';"
             "5. **Input**: \"Find the average salary of employees in each department.\"\n"
-            "**Output**:\n\n"
-            "        SELECT Department, AVG(Salary) AS average_salary\n"
-            "        FROM Employees\n"
-            "        GROUP BY Department;\n\n"
+            "**Output**:""SELECT Department, AVG(Salary) AS average_salary FROM Employees GROUP BY Department;"
+            "6. **Input**: \"Find the name of employee with Highest salary in HR department.\"\n"
+            "**Output**:""SELECT Name FROM Employees WHERE Department = 'HR' AND Salary = (SELECT MAX(Salary) FROM Employees WHERE Department = 'HR');"
+            
             "Please do not return additional text besides query.\n"
             "Please only answer queries which makes sense for the given schema. Else just return - \"No information found\""
         )