Merge pull request #1830 from sinaptik-ai/fix/result_not_found_error

fix: template not adding the response type

Author: ArslanSaleem

pandasai/core/code_execution/code_executor.py

@@ -39,7 +39,9 @@ def execute_and_return_result(self, code: str) -> Any:
 
         # Get the result
         if "result" not in self._environment:
-            raise NoResultFoundError("No result returned")
+            raise NoResultFoundError(
+                "No result was returned from the code execution. Please return the result in dictionary format, for example: result = {'type': ..., 'value': ...}"
+            )
 
         return self._environment.get("result", None)
 

pandasai/core/prompts/templates/generate_python_code_with_sql.tmpl

@@ -10,7 +10,8 @@ def execute_sql_query(sql_query: str) -> pd.Dataframe
     """This method connects to the database, executes the sql query and returns the dataframe"""
 </function>
 
-{% if last_code_generated != "" and context.memory.count() > 0 %}
+{% 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: