business-science/ai-data-science-team

Provides specialized agents for data loading, cleaning, visualization, and ML modeling that integrate with H2O and MLflow, orchestrated through a Streamlit-based visual pipeline editor. Supports both OpenAI and local LLMs via Ollama, enabling reproducible workflows with lineage tracking, multi-dataset handling, and automated script generation. The library exposes composable agent building blocks—including supervisor agents and multi-agent workflows—allowing custom data science task automation beyond the included Studio app.

5,036 stars. Available on PyPI.