Merge pull request #9 from togethercomputer/fede/maybe-solves

Fede/maybe solves

Author: vinid

interpreter/code_executor.py

@@ -77,9 +77,9 @@ async def execute(self, code: str, session: Session) -> Any:
             session.namespace.update(self.safe_builtins)
 
             original_cwd = os.getcwd()
+            os.chdir("/app/custom_data")
             try:
-                os.chdir("/app/custom_data")
-
+            
                 with redirect_stdout(stdout_capture), redirect_stderr(stderr_capture):
                     result = eval(
                         compile(code, "<string>", "eval"),

open_data_scientist/utils/executors.py

@@ -142,6 +142,9 @@ def execute_code_internal(
     outputs_list: list[dict[str, Any]] = []
     errors_list: list[str] = []
 
+    # NOTE: that here i am trying to reconstruct the output of TCI, but it is not perfect.
+    # Seems to work for most cases, but not all.
+
     if raw_response.get("success"):
         execution_summary_input["status"] = "success"
         result_data = raw_response.get("result")

open_data_scientist/utils/strings.py

@@ -301,7 +301,7 @@ def save_image_to_disk(b64_image, filename=None):
         console.print(f"[dim]💾 Image saved as: {filename}[/dim]")
         return filename
     except Exception as e:
-        console.print(f"[red]Error saving image: {e}[/red]")
+        console.print(f"⚠️ [yellow]Warning/Issue Detected - Error saving image: {e}[/yellow]")
         return None
 
 
@@ -322,11 +322,11 @@ def print_rich_execution_result(
     if parsed.has_errors:
         error_text = "\n".join(parsed.errors)
         if text_output:
-            text_output = f"{text_output}\n\n[bold red]Errors:[/bold red]\n{error_text}"
+            text_output = f"{text_output}\n\n⚠️ [bold yellow]Warning/Issue Detected:[/bold yellow]\n{error_text}"
         else:
-            text_output = f"[bold red]Error:[/bold red]\n{error_text}"
-        emoji = "❌"
-        border_style = "red"
+            text_output = f"⚠️ [bold yellow]Warning/Issue Detected:[/bold yellow]\n{error_text}"
+        emoji = "⚠️"
+        border_style = "yellow"
     else:
         # If we have images, mention them in the text
         if parsed.has_images: