Skip to main content

Open-source Python SDK that automates the complete ML lifecycle using a modular multi-agent architecture orchestrated by LangGraph.

Project description


Table of Contents

Navigation
  1. Why PhronesisML
  2. Core Principles
  3. Key Features
  4. Architecture Overview
  5. How It Works
  6. Engine Selection Logic
  7. Technology Stack
  8. Performance
  9. Installation
  10. Quick Start
  11. Examples
  12. SDK Interfaces
  13. Project Structure
  14. Offline-First Philosophy
  15. Security
  16. Roadmap
  17. Contributing
  18. Community
  19. FAQ
  20. License
  21. Acknowledgements

Why PhronesisML

Notebooks AutoML Tools PhronesisML
Structure Ad hoc, cell-by-cell Fixed, opaque Modular agents on typed WorkflowState
Transparency High, but unorganized Low — black box High — every decision is inspectable
Overridable N/A Rarely Yes — imputation, encoding, model choice
Reusable Low Low High — same pipeline, swap the data
Production-ready No Partially Yes — versioned, reportable artifacts
Works offline Yes Rarely Yes — by design

In short: PhronesisML recommends; it does not obscure. Every stage of the pipeline is a discrete, testable, reusable unit of code operating on a shared, typed WorkflowState.

PhronesisML is SDK-first — the CLI, the FastAPI service, and any future GUI are thin clients built on the same SDK a data scientist would import directly. There is exactly one source of truth for ML logic.


Core Principles

PhronesisML is built around five commitments that shape every design decision in the codebase:

Principle What It Means in Practice
Simple API Phronesis("data.csv").run() is a complete, working pipeline. Complexity is opt-in, not mandatory.
Advanced Power The same SDK exposes stage-by-stage control, custom configuration, and low-level workflow access when you need it.
Offline-First The core pipeline runs with no network dependency — your data never has to leave your machine to be analyzed.
Production-Ready Every run produces versioned, structured artifacts, not a throwaway notebook cell.
Intelligent Automation Engine selection, target detection, and model recommendation are automated but always inspectable and overridable.

Key Features

Agents
Multi-Agent
LangGraph
Orchestration
Engines
Auto Engine
ETL
Cleaning
EDA
Analysis
Features
Engineering
Target
Detection
Models
Selection
Evaluation
Metrics
Explain
Explainability
Reports
Reporting
Plugins
Extensible
Feature Description Status
Multi-Agent Workflow Each pipeline stage is an independent agent with a single responsibility
LangGraph Orchestration Agents are nodes in a directed graph; state passing, retries
Automatic Engine Selection Dataset size determines Pandas, Polars, or PySpark
ETL Declarative extraction, cleaning, and transformation
Validation Schema, type, and quality validation before downstream processing
EDA Automated statistical profiling and structured dataset summaries
Feature Engineering Automated and configurable transformation, encoding, derivation
Target Detection Heuristic, overridable identification of prediction target
Model Recommendation Rule- and metric-driven suggestion of candidate model families
Explainability Post-training feature importance and model-behavior summaries
Reporting Structured, versionable output artifacts for every stage
FastAPI Interface REST API with file upload, background jobs, OpenAPI docs
Offline-First Core pipeline stages run without network access
SDK-First Every interface is a client of the SDK
Plugin System Extension points for custom agents, models, engines, storage 🔜 Planned

Architecture Overview

graph TD
    A["Python SDK<br/>Phronesis class"] --> B
    B["LangGraph Workflow<br/>Orchestrates agents"] --> C
    C["Agents<br/>One responsibility each"] --> D
    D["Services<br/>Reusable domain logic"] --> E
    E["Data Engines<br/>Pandas / Polars / PySpark"] --> F
    F["Reports / Storage<br/>Persisted artifacts"]

    style A fill:#6366F1,stroke:#4F46E5,color:#fff,stroke-width:2px
    style B fill:#A855F7,stroke:#9333EA,color:#fff,stroke-width:2px
    style C fill:#EC4899,stroke:#DB2777,color:#fff,stroke-width:2px
    style D fill:#14B8A6,stroke:#0D9488,color:#fff,stroke-width:2px
    style E fill:#F97316,stroke:#EA580C,color:#fff,stroke-width:2px
    style F fill:#06B6D4,stroke:#0891B2,color:#fff,stroke-width:2px
Layer Responsibility Depends On
Python SDK Single public entry point (Phronesis class)
LangGraph Workflow Pipeline as graph; owns WorkflowState Called by SDK
Agents One pipeline responsibility each; read/write state Graph nodes
Services Stateless, reusable domain logic Called by agents
Data Engines Pandas, Polars, PySpark implementations Called by services
Reports / Storage Persists run artifacts Written to by agents

Design Principles

Principle What It Means
SDK-first One source of truth for ML logic
Offline-first Core pipeline runs without network
Deterministic ML Same input + config = same output
Dependency Injection Agents receive dependencies rather than construct them
Clean Architecture Layers depend inward, never outward
Single Responsibility One agent, one job
Strategy Pattern Interchangeable engine selection

How It Works

graph TD
    A["1. Dataset Upload"] --> B["2. Validation"]
    B --> C["3. Profiling"]
    C --> D["4. ETL"]
    D --> E["5. EDA"]
    E --> F["6. Feature Engineering"]
    F --> G["7. Target Detection"]
    G --> H["8. Model Recommendation"]
    H --> I["9. Training"]
    I --> J["10. Evaluation"]
    J --> K["11. Explainability"]
    K --> L["12. Reporting"]

    style A fill:#6366F1,color:#fff
    style B fill:#EF4444,color:#fff
    style C fill:#F97316,color:#fff
    style D fill:#0EA5E9,color:#fff
    style E fill:#F97316,color:#fff
    style F fill:#EAB308,color:#000
    style G fill:#EF4444,color:#fff
    style H fill:#8B5CF6,color:#fff
    style I fill:#8B5CF6,color:#fff
    style J fill:#14B8A6,color:#fff
    style K fill:#EC4899,color:#fff
    style L fill:#06B6D4,color:#fff

Each numbered stage is implemented as its own agent — see Project Structure for where each one lives in the codebase.


Engine Selection Logic

PhronesisML supports three interchangeable dataframe backends behind a single DataEngine interface, so pipeline code never imports Pandas, Polars, or PySpark directly.

graph LR
    A["Dataset"] --> B{"Profile size<br/>& shape"}
    B -->|"Small–medium,<br/>fits in memory"| C["Pandas"]
    B -->|"Larger,<br/>single machine"| D["Polars"]
    B -->|"Larger than<br/>memory / distributed"| E["PySpark"]
    C --> F["Same DataEngine<br/>interface downstream"]
    D --> F
    E --> F

    style A fill:#6366F1,color:#fff
    style B fill:#A855F7,color:#fff
    style C fill:#150458,color:#fff
    style D fill:#CD792C,color:#fff
    style E fill:#E25A1C,color:#fff
    style F fill:#06B6D4,color:#fff
Engine Best For Why
Pandas Small-to-medium, in-memory Ubiquitous, well understood
Polars Larger single-machine workloads Rust-based, multi-threaded query engine
PySpark Distributed / larger-than-memory data Industry standard at scale

Selection is automatic by default but can always be forced explicitly — see Quick Start.


Technology Stack

Python LangGraph Pandas Polars PySpark scikit-learn Pydantic FastAPI pytest


Performance

PhronesisML's engine abstraction exists specifically so the pipeline can move from Pandas to Polars to PySpark as data size grows, without changing pipeline code. Formal, reproducible benchmarks across dataset sizes and engines are not yet published — this section will be filled in with measured numbers (not estimates) once a benchmark suite lands. Track progress in Roadmap.


Installation

Standard (recommended)

pip install phronesisml

Supported file formats

Format Supported Dependency
CSV / TSV pandas (core)
Excel (.xlsx) openpyxl (core)
Parquet pyarrow (core)
JSON / JSONL pandas (core)
Feather / Arrow pyarrow (core)

Optional extras

Extra Install What it adds
all pip install phronesisml[all] Everything below
api pip install phronesisml[api] FastAPI REST endpoints
cli pip install phronesisml[cli] CLI commands
explain pip install phronesisml[explain] SHAP explanations
boost pip install phronesisml[boost] XGBoost models
mlflow pip install phronesisml[mlflow] MLflow tracking
spark pip install phronesisml[spark] PySpark engine
parquet pip install phronesisml[parquet] Parquet support
dev pip install phronesisml[dev] pytest, ruff, mypy

From source

git clone https://github.com/kartik00052/PhronesisML.git
cd PhronesisML
python -m venv .venv
source .venv/bin/activate   # Windows: .venv\Scripts\activate
pip install -e ".[dev]"

Quick Start

SDK (Python)

from phronesisml import Phronesis

ml = Phronesis("data/customers.csv")
ml.run()
print(ml.report())

CLI

phronesisml run data/customers.csv
phronesisml run data/customers.csv --engine polars
phronesisml info

FastAPI

pip install phronesisml[api]
uvicorn phronesisml.interfaces.api.app:app --reload

Examples

Incremental Usage
from phronesisml import Phronesis

ml = Phronesis("data/customers.csv")
ml.load()
print(ml.summary())

ml.clean(null_strategy="fill")
ml.validate()
ml.eda()

ml.detect_target()
result = ml.train()
print(f"Best model: {result.model_type} ({result.score:.4f})")

print(ml.evaluate())
print(ml.explain())
Simple API (one-liner functions)
from phronesisml import analyze, train

profile = analyze("data/customers.csv")
print(f"{profile.shape[0]} rows, {profile.shape[1]} columns")

result = train("data/customers.csv")
print(f"Best model: {result.best_model_type} ({result.best_score:.4f})")
Async variants
from phronesisml import analyze_async, train_async
import asyncio

async def main():
    profile = await analyze_async("data/customers.csv")
    result = await train_async("data/customers.csv")

asyncio.run(main())
Error handling
from phronesisml import train
from phronesisml.exceptions import DataValidationError, EngineSelectionError, WorkflowError

try:
    result = train("data/customers.csv")
except DataValidationError as e:
    print(f"Dataset failed validation: {e}")
except EngineSelectionError as e:
    print(f"Could not select a data engine: {e}")
except WorkflowError as e:
    print(f"Pipeline failed: {e}")
Advanced — low-level workflow API
import asyncio
from phronesisml import run_pipeline

async def main():
    result = await run_pipeline(data_path="data/customers.csv")
    print(result)

asyncio.run(main())

SDK Interfaces

PhronesisML is SDK-first: the CLI and FastAPI service are thin clients that call the same SDK you import directly.

Interface Install Description
Python SDK pip install phronesisml from phronesisml import Phronesis
CLI pip install phronesisml[cli] phronesisml run data.csv
FastAPI pip install phronesisml[api] uvicorn phronesisml.interfaces.api.app:app

Project Structure

phronesisml/
  __init__.py          # Public SDK surface
  exceptions.py        # Exception hierarchy
  agents/              # Pipeline agents (11 total)
  configs/             # Pydantic configuration
  data/                # Data loading, validation, profiling
  engines/             # Pandas/Polars/Spark abstraction
  interfaces/           # CLI (Typer) + FastAPI
  ml/                   # Model definitions, training, metrics
  rag/                   # RAG infrastructure
  workflow/             # LangGraph workflow orchestration

Offline-First Philosophy

PhronesisML's core pipeline — validation, profiling, ETL, EDA, feature engineering, target detection, model recommendation, training, evaluation, explainability, and reporting — runs entirely on your machine, with no network dependency and no requirement to send data to a hosted service. Optional extras (mlflow for remote experiment tracking, future cloud storage backends) are opt-in additions on top of an offline-capable core, not a requirement for it.


Security

  • No data leaves your machine unless you explicitly configure a remote storage or tracking backend.
  • No hidden network calls in the core pipeline — installation extras that do require network access (e.g. mlflow) are opt-in.
  • Dependency-pinned releases to reduce supply-chain surface area.
  • Found a security issue? Please open a private security advisory on GitHub rather than a public issue.

Roadmap

Status Version

Completed

  • Core WorkflowState + LangGraph orchestration
  • All 11 pipeline agents
  • Pandas, Polars, PySpark engines with auto-selection
  • Local filesystem storage
  • CLI and FastAPI interfaces
  • HTML report generation
  • Full test suites

Planned

  • Plugin system with entry-points discovery
  • S3, GCS, Azure Blob storage backends
  • DuckDB engine
  • PDF report rendering
  • Parallel/branching agent execution
  • Reproducible benchmark suite (see Performance)
  • Desktop GUI client
  • Human-in-the-loop checkpoints

Contributing

PRs Welcome

make check       # lint + typecheck + test
make format      # auto-fix formatting
make build        # build wheel + sdist
make docker      # build and run Docker image

New to the project? Look for issues labeled good first issue. Please open an issue before starting on a large change, so the design can be discussed first.


Community


FAQ

Why not just use AutoML?

AutoML tools optimize for a leaderboard metric and hide their reasoning. PhronesisML makes every decision inspectable and overridable.

Why LangGraph?

LangGraph models workflows as a graph of stateful nodes with conditional edges — a direct map onto PhronesisML's pipeline shape.

Why Polars in addition to Pandas?

Polars' Rust-based, multi-threaded engine handles larger workloads significantly faster. PhronesisML upgrades automatically when warranted — see Engine Selection Logic.

Can I run only part of the pipeline?

Yes — run() and run_pipeline() accept a stages parameter so you can run, for example, only validation and EDA.

Does PhronesisML require an internet connection?

No. The core pipeline is offline-first — see Offline-First Philosophy. Only optional extras like remote MLflow tracking require network access.


License

Licensed under the MIT License — see LICENSE for the full text.


Acknowledgements

scikit-learn FastAPI LangGraph MLflow Polars

PhronesisML's design draws inspiration from the architectural patterns and developer experience established by these projects, without any affiliation with or endorsement from them.


Built with a commitment to transparent, inspectable ML pipelines.

GitHub PyPI

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

phronesisml-0.2.1.tar.gz (137.2 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

phronesisml-0.2.1-py3-none-any.whl (136.0 kB view details)

Uploaded Python 3

File details

Details for the file phronesisml-0.2.1.tar.gz.

File metadata

  • Download URL: phronesisml-0.2.1.tar.gz
  • Upload date:
  • Size: 137.2 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.2.0 CPython/3.11.9

File hashes

Hashes for phronesisml-0.2.1.tar.gz
Algorithm Hash digest
SHA256 1d6b238ac40409b725517350a07eaac3a047421d644db9b1388f2fd02a604a48
MD5 93d21bcbc441a8f8198efd4907d7a0c7
BLAKE2b-256 bf244745aa0ec953c343b39558ab063879794c624729a74650a21ca9e459a3b9

See more details on using hashes here.

File details

Details for the file phronesisml-0.2.1-py3-none-any.whl.

File metadata

  • Download URL: phronesisml-0.2.1-py3-none-any.whl
  • Upload date:
  • Size: 136.0 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.2.0 CPython/3.11.9

File hashes

Hashes for phronesisml-0.2.1-py3-none-any.whl
Algorithm Hash digest
SHA256 74203a774278b44ceeecd143ebfb6dae810c3f8a9798d7eef2ff65432ca754ac
MD5 44b877e64e58dde6d0d7d0906c7dfc1b
BLAKE2b-256 23b4c67dc77aa6a3968249ab757179afd6073c5d63eba9655cba7faacd595138

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page