Update Classes/Owiki_Class.py

Classes/Owiki_Class.py

@@ -33,7 +33,12 @@ class OWiki:
         return prompt_template
     
     def create_sql_prompt_template(self,schemas):
-        prompt =  """Write an SQL query for the following questions whose schemas are as follows.\nSQL Schema:"""
+        prompt =  """You are an expert in writing SQL commands as best as you can. 
+        Here are the rules you must follow to complete your task.
+        1. If you need more context about question don't hesitate to ask but don't keep on asking all the time.
+        2. Please make sure you answer correctly so that you will get a very high reward.
+        
+        For the below schemas write an SQL query. \nSQL Schema:"""
         for table_name,table_schema in schemas.items():
             prompt+= f"Table Name: {table_name} Schema:"
             for key,value in table_schema.items():