mobinln commited on
Commit
8419674
1 Parent(s): b037f61

change prompt a little

Browse files
Files changed (1) hide show
  1. llm.py +1 -1
llm.py CHANGED
@@ -43,7 +43,7 @@ def generate_system_prompt(table_name, table_schema):
43
  column_type = col[2]
44
  prompt += f"- {column_name} ({column_type})\n"
45
 
46
- prompt += "\nPlease generate a SQL query based on the following natural language question. ONLY return the SQL query."
47
 
48
  return prompt
49
 
 
43
  column_type = col[2]
44
  prompt += f"- {column_name} ({column_type})\n"
45
 
46
+ prompt += "\nGenerate a SQL query based on the following natural language question. ONLY return the SQL query and nothing else."
47
 
48
  return prompt
49