Merge pull request #5 from togethercomputer/fede/mini-update

mini

Author: vinid

.gitignore

@@ -15,7 +15,8 @@ scripts/*
 dist/
 build/
 *.egg-info/
-
+*.cast
+*.gif
 # Virtual environment
 venv/
 env/

eval/test_env/auto-mpg.csv renamed to examples/base/auto-mpg.csv

@@ -29,7 +29,7 @@ def get_data_directory(data_dir: Optional[str]) -> Optional[str]:
         current_dir = os.getcwd()
 
         console.print("\n[yellow]No data directory specified.[/yellow]")
-        console.print(f"[blue]Current directory:[/blue] {current_dir}")
+        console.print(f"[blue]Current directory:[/blue] {Path(current_dir).name}")
 
         # Show files in current directory
         files = list(Path(current_dir).iterdir())
@@ -187,7 +187,7 @@ def main():
 
     # Show configuration
     # Update args for display
-    args.data_dir = data_dir or "None (no files will be uploaded)"
+    args.data_dir = (Path(data_dir).name if data_dir else "None (no files will be uploaded)")
     show_configuration(args)
 
     # Ask for confirmation
@@ -198,7 +198,7 @@ def main():
     # Welcome message
     welcome_text = "🚀 Starting ReAct Data Science Agent"
     if data_dir:
-        welcome_text += f"\n📁 Data from: {data_dir}"
+        welcome_text += f"\n📁 Data from: {Path(data_dir).name}"
     welcome_text += f"\n🧠 Model: {args.model}"
     welcome_text += f"\n⚡ Executor: {args.executor.upper()}"
 

open_data_scientist/cli.py

@@ -33,7 +33,8 @@
         - Plots and visualizations are automatically displayed to the user
         - Build on previous successful steps rather than starting over
         - If you don't print outputs, you will not get a result.
-        - While you can generate plots and images, you cannot see them, you are not a vision model. 
+        - While you can generate plots and images, you cannot see them, you are not a vision model. Don't generate plots and images unless you are asked to.
+          Do not provide comments on the plots and images you generate, you are not a vision model.
 
         WAIT FOR THE RESULT OF THE ACTION BEFORE PROCEEDING.