Merge pull request #13 from togethercomputer/fede/pre-release

Fede/pre release

Author: shang-zhu

interpreter/README.md

@@ -16,7 +16,7 @@ A containerized Python code execution service that provides an API for running P
 
 2. **Verify it's running:**
    ```bash
-   curl http://localhost:8000/health
+   curl http://localhost:8123/health
    ```
 
 3. **Stop the service:**

open_data_scientist/__init__.py

@@ -0,0 +1,12 @@
+"""
+ReAct Data Science Agent - AI-powered data analysis assistant.
+"""
+
+__version__ = "0.1.0a1"
+__author__ = "TogetherAI"
+__email__ = "[email protected]"
+
+from .codeagent import *
+from .cli import main
+
+__all__ = ["main"] 

open_data_scientist/cli.py

@@ -54,7 +54,7 @@ def get_data_directory(data_dir: Optional[str]) -> Optional[str]:
             )
 
         use_current = Confirm.ask(
-            "\n[bold]Do you want to upload files from the current directory?[/bold]",
+            "\n[bold]Important: Do you want to upload files from the current directory?[/bold]",
             default=False,
         )
 

open_data_scientist/utils/executors.py

@@ -6,13 +6,14 @@
 import os
 from pathlib import Path
 from typing import Dict
+import sys
 
-
-
+assert os.getenv("TOGETHER_API_KEY"), "TOGETHER_API_KEY environment variable must be set"
 client = Client()
 code_interpreter = client.code_interpreter
 
 
+
 def collect_files(directory) -> list[Dict[str, str]]:
     """
     Collects all files from the specified directory and its subdirectories.
@@ -100,6 +101,15 @@ def execute_code_factory(type: str):
     Factory function to create an executor function based on the type of code to execute.
     """
     if type == "internal":
+        # Check if the internal service is healthy before returning the executor
+        base_url = os.getenv("CODE_INTERPRETER_URL", "http://localhost:8123")
+        try:
+            health_response = requests.get(f"{base_url}/health", timeout=5)
+            if health_response.status_code != 200:
+                raise requests.exceptions.RequestException("Health check failed")
+        except requests.exceptions.RequestException:
+            print("No docker container available. Use the option '--executor tci' if you don't want to build the container.")
+            sys.exit(1)
         return execute_code_internal
     elif type == "tci":
         return execute_code_tci

open_data_scientist/utils/strings.py

@@ -33,8 +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. 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.
+        - While you can generate plots and images, you cannot see them, you are not a vision model. Never generate plots and images unless you are asked to.
+        - Do not provide comments on the plots and images you generate in your final answer.
 
         WAIT FOR THE RESULT OF THE ACTION BEFORE PROCEEDING.
 

pyproject.toml

@@ -4,12 +4,12 @@ build-backend = "hatchling.build"
 
 [project]
 name = "open-data-scientist"
-version = "0.1.0"
+version = "0.1.0a1"
 description = "ReAct Data Science Agent - AI-powered data analysis assistant"
 readme = "README.md"
-requires-python = ">=3.12"
+requires-python = ">=3.10"
 authors = [
-    {name = "Your Name", email = "[email protected]"},
+    {name = "TogetherAI", email = "[email protected]"},
 ]
 keywords = ["data-science", "ai", "react", "analysis", "llm"]
 classifiers = [
@@ -18,6 +18,8 @@ classifiers = [
     "Intended Audience :: Science/Research",
     "License :: OSI Approved :: MIT License",
     "Programming Language :: Python :: 3",
+    "Programming Language :: Python :: 3.10",
+    "Programming Language :: Python :: 3.11",
     "Programming Language :: Python :: 3.12",
     "Topic :: Scientific/Engineering :: Artificial Intelligence",
     "Topic :: Scientific/Engineering :: Information Analysis",