improve prompt to always display result format

Author: ArslanSaleem

pandasai/core/prompts/templates/generate_python_code_with_sql.tmpl

@@ -11,6 +11,7 @@ def execute_sql_query(sql_query: str) -> pd.Dataframe
 </function>
 
 {% if last_code_generated and context.memory.count() > 0 %}
+Last code generated:
 {{ last_code_generated }}
 {% else %}
 Update this initial code:
@@ -26,7 +27,8 @@ import pandas as pd
 {% include 'shared/vectordb_docs.tmpl' with context %}
 {{ context.memory.get_last_message() }}
 
-At the end, declare "result" variable as a dictionary of type and value.
+At the end, declare "result" variable as a dictionary of type and value in the following format:
+{% include 'shared/output_type_template.tmpl' with context %}
 
 
 Generate python code and return full updated code:

tests/unit_tests/prompts/test_sql_prompt.py

@@ -94,7 +94,10 @@ def execute_sql_query(sql_query: str) -> pd.Dataframe
 
 
 
-At the end, declare "result" variable as a dictionary of type and value.
+At the end, declare "result" variable as a dictionary of type and value in the following format:
+
+{output_type_template}
+
 
 
 Generate python code and return full updated code: