fix: remove "do not import anything" from prompts

Sinaptik-AI · Oct 2, 2023 · 7872a8a · 7872a8a
1 parent 6f28eac
commit 7872a8a
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
diff --git a/assets/prompt-templates/correct_error_prompt.tmpl b/assets/prompt-templates/correct_error_prompt.tmpl
@@ -12,4 +12,4 @@ You generated this python code:
 It fails with the following error:
 {error_returned}
 
-Correct the python code and return a new python code (do not import anything) that fixes the above mentioned error. Do not generate the same code again.
+Correct the python code and return a new python code that fixes the above mentioned error. Do not generate the same code again.
diff --git a/pandasai/prompts/correct_error_prompt.py b/pandasai/prompts/correct_error_prompt.py
@@ -13,7 +13,7 @@
 It fails with the following error:
 {error_returned}
 
-Correct the python code and return a new python code (do not import anything) that fixes the above mentioned error. Do not generate the same code again.
+Correct the python code and return a new python code that fixes the above mentioned error. Do not generate the same code again.
 """  # noqa: E501
 
 from .base import FileBasedPrompt

diff --git a/tests/prompts/test_correct_error_prompt.py b/tests/prompts/test_correct_error_prompt.py
@@ -49,6 +49,6 @@ def test_str_with_args(self):
 It fails with the following error:
 Error message
 
-Correct the python code and return a new python code (do not import anything) that fixes the above mentioned error. Do not generate the same code again.
+Correct the python code and return a new python code that fixes the above mentioned error. Do not generate the same code again.
 """  # noqa: E501
         )
diff --git a/tests/test_smartdatalake.py b/tests/test_smartdatalake.py
@@ -175,7 +175,7 @@ def analyze_data(df):
 It fails with the following error:
 Test error
 
-Correct the python code and return a new python code (do not import anything) that fixes the above mentioned error. Do not generate the same code again.
+Correct the python code and return a new python code that fixes the above mentioned error. Do not generate the same code again.
 """  # noqa: E501
         )