fix: fix code newline character issue introduced by astor.to_source (#…

…345) * use ast.unparse instead of astor.to_source() * Revert to astor.to_source for unparsing and override pretty_source to avoid adding newline * fix formatting
Sinaptik-AI · Jul 8, 2023 · daf7336 · daf7336
1 parent 5e3f311
commit daf7336
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/pandasai/__init__.py b/pandasai/__init__.py
@@ -554,7 +554,7 @@ def _clean_code(self, code: str) -> str:
             new_body.append(node)
 
         new_tree = ast.Module(body=new_body)
-        return astor.to_source(new_tree).strip()
+        return astor.to_source(new_tree, pretty_source=lambda x: "".join(x)).strip()
 
     def _get_environment(self) -> dict:
         """

diff --git a/pandasai/helpers/save_chart.py b/pandasai/helpers/save_chart.py
@@ -113,4 +113,4 @@ def add_save_chart(
     logging.info(chart_save_msg)
 
     new_tree = ast.Module(body=new_body)
-    return astor.to_source(new_tree).strip()
+    return astor.to_source(new_tree, pretty_source=lambda x: "".join(x)).strip()