What is MLC Bakery?

MLC Bakery is a Python-based service for managing ML model provenance and lineage, built with FastAPI and SQLAlchemy. It helps track datasets, entities, activities, agents, and relationships in your ML workflows.

What are the key features of MLC Bakery?

Key features include dataset management with collection support, entity tracking, activity logging, agent management, provenance relationships tracking, and RESTful API endpoints.

What are the prerequisites for setting up MLC Bakery?

You need Python 3.12+, uv (Python package manager), and PostgreSQL (running locally or via Docker).

How do I install the dependencies for MLC Bakery?

Use uv pip install -e .[dev,webclient] to install main, dev, and webclient dependencies in editable mode.

How do I set up the environment variables?

Create a .env file in the project root by copying the example and edit it with your local PostgreSQL connection details. The key variable is DATABASE_URL.

How do I run database migrations?

Use uv run alembic upgrade heads to apply the latest database schema.

How do I start the MLC Bakery server locally?

Use uv run uvicorn mlcbakery.main:app --reload --host 0.0.0.0 --port 8000 to start the FastAPI application.

Where can I access the API documentation?

The Swagger UI is available at http://localhost:8000/docs, and the ReDoc documentation is at http://localhost:8000/redoc.

How do I run tests?

Use uv run pytest to run all tests, or uv run pytest tests/test_activities.py -v to run specific tests.

What is the project structure of MLC Bakery?

The project structure includes directories for Alembic migrations, GitHub Actions workflows, the main application package (with models, schemas, API routes, and the main entry point), tests, example environment variables, the Alembic configuration file, and the pyproject.toml file.

How do I reset the database for local development?

Use dropdb mlcbakery_dev and createdb mlcbakery_dev (using psql) to drop and recreate the database, then run uv run alembic upgrade heads to re-run migrations. Warning: This deletes all data in the development database.

How can I contribute to MLC Bakery?

Create a new branch for your feature, make your changes, run tests, commit your changes, push to the branch, and submit a pull request.

What license is MLC Bakery released under?

MLC Bakery is licensed under the MIT License.

How can I deploy MLC Bakery using Docker Compose?

Configure the ADMIN_AUTH_TOKEN in a .env file or directly in the docker-compose.yml, then run docker-compose up --build -d to build and run the services. Apply database migrations with docker-compose exec api alembic upgrade head.

What is the ADMIN_AUTH_TOKEN and why is it needed?

The ADMIN_AUTH_TOKEN is required for any mutable API operations (POST, PUT, PATCH, DELETE). Include it in requests as a Bearer token in the Authorization header.

What is the relationship between MLC Bakery and the UBOS platform?

MLC Bakery integrates with the UBOS platform, a full-stack AI Agent Development Platform, to ensure AI Agents are built on a foundation of trust, transparency, and accountability.