KCH / docs /projects /DataCrew.md
bsamadi's picture
Update to pixi env
c032460

Course Project: DataCrew

A CLI tool that uses local LLMs and multi-agent systems to transform spreadsheets into intelligent PDF reports.


Overview

β”Œβ”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”
β”‚                        DataCrew CLI                              β”‚
β”œβ”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€
β”‚  $ datacrew ingest sales_2024.xlsx                              β”‚
β”‚  $ datacrew ask "What were the top 5 products by revenue?"      β”‚
β”‚  $ datacrew report "Q4 Executive Summary" --output report.pdf   β”‚
β””β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”˜

Architecture

CSV/XLSX ──► SQLite ──► MCP Server ──► Multi-Agent Crew ──► PDF Report
                              β”‚
                              β–Ό
                    Docker Model Runner
                      (Local LLM)

Data Flow

β”Œβ”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”     β”Œβ”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”     β”Œβ”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”
β”‚   CSV/XLSX   │────►│    SQLite    │────►│  MCP Server  β”‚
β”‚    Files     β”‚     β”‚   Database   β”‚     β”‚   (Tools)    β”‚
β””β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”˜     β””β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”˜     β””β”€β”€β”€β”€β”€β”€β”¬β”€β”€β”€β”€β”€β”€β”€β”˜
                                                  β”‚
                                                  β–Ό
β”Œβ”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”     β”Œβ”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”     β”Œβ”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”
β”‚  PDF Report  │◄────│  Agent Crew  │◄────│  Local LLM   β”‚
β”‚   Output     β”‚     β”‚  (CrewAI)    β”‚     β”‚   (Docker)   β”‚
β””β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”˜     β””β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”˜     β””β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”˜

Agent System

Agent Role Tools Output
Data Analyst Understands schema, writes SQL queries MCP Database Tools Query results, data summaries
Insights Agent Interprets results, finds patterns Python Analysis, Statistics Key findings, trends, anomalies
Report Writer Creates narrative sections LLM Generation Executive summary, section text
PDF Composer Formats and assembles final report ReportLab/WeasyPrint Formatted PDF document

Agent Workflow

User Request: "Generate Q4 Executive Summary"
                    β”‚
                    β–Ό
         β”Œβ”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”
         β”‚    Data Analyst     β”‚
         β”‚  "What data do we   β”‚
         β”‚   need for Q4?"     β”‚
         β””β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”¬β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”˜
                    β”‚ SQL Queries
                    β–Ό
         β”Œβ”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”
         β”‚   Insights Agent    β”‚
         β”‚  "What patterns     β”‚
         β”‚   emerge from       β”‚
         β”‚   this data?"       β”‚
         β””β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”¬β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”˜
                    β”‚ Key Findings
                    β–Ό
         β”Œβ”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”
         β”‚   Report Writer     β”‚
         β”‚  "Write narrative   β”‚
         β”‚   sections for      β”‚
         β”‚   each finding"     β”‚
         β””β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”¬β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”˜
                    β”‚ Text Sections
                    β–Ό
         β”Œβ”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”
         β”‚   PDF Composer      β”‚
         β”‚  "Assemble into     β”‚
         β”‚   formatted PDF"    β”‚
         β””β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”¬β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”˜
                    β”‚
                    β–Ό
              report.pdf

CLI Commands

datacrew ingest

Ingest CSV or XLSX files into the local SQLite database.

# Ingest a single file
datacrew ingest sales_2024.xlsx

# Ingest with custom table name
datacrew ingest sales_2024.xlsx --table quarterly_sales

# Ingest multiple files
datacrew ingest data/*.csv

# Ingest with schema inference options
datacrew ingest sales.csv --infer-types --date-columns "order_date,ship_date"

Options:

Flag Description Default
--table Custom table name Filename (sanitized)
--if-exists Behavior if table exists: fail, replace, append fail
--infer-types Automatically infer column types true
--date-columns Comma-separated list of date columns Auto-detect
--db Database file path ./data/datacrew.db

datacrew ask

Query the database using natural language.

# Simple query
datacrew ask "What were the top 5 products by revenue?"

# Query with output format
datacrew ask "Show monthly sales trends" --format table

# Query with export
datacrew ask "List all customers from California" --export customers_ca.csv

# Interactive mode
datacrew ask --interactive

Options:

Flag Description Default
--format Output format: table, json, csv, markdown table
--export Export results to file None
--explain Show generated SQL query false
--interactive Enter interactive query mode false
--limit Maximum rows to return 100

datacrew report

Generate PDF reports using the multi-agent system.

# Generate a report
datacrew report "Q4 Executive Summary"

# Specify output file
datacrew report "Q4 Executive Summary" --output reports/q4_summary.pdf

# Use a template
datacrew report "Monthly Sales" --template executive

# Include specific analyses
datacrew report "Product Analysis" --include trends,comparisons,recommendations

Options:

Flag Description Default
--output, -o Output PDF file path ./report.pdf
--template Report template: executive, detailed, minimal executive
--include Analyses to include All
--date-range Date range for analysis All data
--verbose, -v Show agent reasoning false

datacrew config

Manage configuration settings.

# Show current config
datacrew config show

# Set LLM model
datacrew config set llm.model "llama3.2:3b"

# Set database path
datacrew config set database.path "./data/mydata.db"

# Reset to defaults
datacrew config reset

datacrew schema

Inspect database schema.

# List all tables
datacrew schema list

# Show table details
datacrew schema describe sales

# Show sample data
datacrew schema sample sales --rows 5

Configuration

Configuration is stored in ~/.config/datacrew/config.toml or ./datacrew.toml in the project directory.

[datacrew]
version = "1.0.0"

[database]
path = "./data/datacrew.db"
echo = false

[llm]
provider = "docker"           # docker, ollama, openai
model = "llama3.2:3b"
temperature = 0.7
max_tokens = 4096
base_url = "http://localhost:11434"

[llm.docker]
runtime = "nvidia"            # nvidia, cpu
memory_limit = "8g"

[agents]
verbose = false
max_iterations = 10

[agents.analyst]
role = "Data Analyst"
goal = "Analyze data and write accurate SQL queries"

[agents.insights]
role = "Insights Specialist"
goal = "Find meaningful patterns and trends in data"

[agents.writer]
role = "Report Writer"
goal = "Create clear, compelling narrative content"

[agents.composer]
role = "PDF Composer"
goal = "Assemble professional PDF reports"

[reports]
output_dir = "./reports"
default_template = "executive"

[reports.templates.executive]
include_charts = true
include_recommendations = true
max_pages = 10

[reports.templates.detailed]
include_charts = true
include_recommendations = true
include_raw_data = true
max_pages = 50

[observability]
enabled = true
provider = "langfuse"         # langfuse, langsmith, console
trace_agents = true
log_tokens = true

Docker Stack

docker-compose.yml

version: "3.9"

services:
  # Local LLM via Docker Model Runner
  llm:
    image: ollama/ollama:latest
    runtime: nvidia
    environment:
      - OLLAMA_HOST=0.0.0.0
    volumes:
      - ollama_data:/root/.ollama
    ports:
      - "11434:11434"
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: 1
              capabilities: [gpu]
    healthcheck:
      test: ["CMD", "curl", "-f", "http://localhost:11434/api/tags"]
      interval: 30s
      timeout: 10s
      retries: 3

  # MCP Server for database access
  mcp-server:
    build:
      context: ./src/datacrew/mcp
      dockerfile: Dockerfile
    environment:
      - DATABASE_PATH=/data/datacrew.db
      - MCP_PORT=3000
    volumes:
      - ./data:/data
    ports:
      - "3000:3000"
    depends_on:
      - llm

  # Main application (for containerized usage)
  datacrew:
    build:
      context: .
      dockerfile: Dockerfile
    environment:
      - LLM_BASE_URL=http://llm:11434
      - MCP_SERVER_URL=http://mcp-server:3000
      - DATABASE_PATH=/data/datacrew.db
    volumes:
      - ./data:/data
      - ./reports:/reports
      - ./input:/input:ro
    depends_on:
      llm:
        condition: service_healthy
      mcp-server:
        condition: service_started
    profiles:
      - cli

volumes:
  ollama_data:

Running the Stack

# Start LLM and MCP server
docker compose up -d llm mcp-server

# Pull the model (first time only)
docker compose exec llm ollama pull llama3.2:3b

# Run DataCrew commands
docker compose run --rm datacrew ingest /input/sales.xlsx
docker compose run --rm datacrew ask "What is total revenue?"
docker compose run --rm datacrew report "Sales Summary" -o /reports/summary.pdf

# Or run locally with Docker backend
datacrew ingest sales.xlsx
datacrew ask "What is total revenue?"
datacrew report "Sales Summary"

Project Structure

datacrew/
β”œβ”€β”€ pyproject.toml              # pixi/uv project config
β”œβ”€β”€ pixi.lock
β”œβ”€β”€ docker-compose.yml          # Full stack orchestration
β”œβ”€β”€ Dockerfile
β”œβ”€β”€ datacrew.toml               # Default configuration
β”œβ”€β”€ README.md
β”‚
β”œβ”€β”€ src/
β”‚   └── datacrew/
β”‚       β”œβ”€β”€ __init__.py
β”‚       β”œβ”€β”€ __main__.py         # Entry point
β”‚       β”œβ”€β”€ cli.py              # Typer CLI commands
β”‚       β”œβ”€β”€ config.py           # TOML configuration loader
β”‚       β”‚
β”‚       β”œβ”€β”€ ingestion/          # CSV/XLSX β†’ SQLite
β”‚       β”‚   β”œβ”€β”€ __init__.py
β”‚       β”‚   β”œβ”€β”€ readers.py      # File readers (pandas, openpyxl)
β”‚       β”‚   β”œβ”€β”€ schema.py       # Schema inference
β”‚       β”‚   └── database.py     # SQLite operations
β”‚       β”‚
β”‚       β”œβ”€β”€ query/              # Natural language queries
β”‚       β”‚   β”œβ”€β”€ __init__.py
β”‚       β”‚   β”œβ”€β”€ nl2sql.py       # NL to SQL conversion
β”‚       β”‚   β”œβ”€β”€ executor.py     # Query execution
β”‚       β”‚   └── formatter.py    # Result formatting
β”‚       β”‚
β”‚       β”œβ”€β”€ agents/             # CrewAI agents
β”‚       β”‚   β”œβ”€β”€ __init__.py
β”‚       β”‚   β”œβ”€β”€ crew.py         # Crew orchestration
β”‚       β”‚   β”œβ”€β”€ analyst.py      # Data Analyst agent
β”‚       β”‚   β”œβ”€β”€ insights.py     # Insights Specialist agent
β”‚       β”‚   β”œβ”€β”€ writer.py       # Report Writer agent
β”‚       β”‚   └── composer.py     # PDF Composer agent
β”‚       β”‚
β”‚       β”œβ”€β”€ tools/              # Agent tools
β”‚       β”‚   β”œβ”€β”€ __init__.py
β”‚       β”‚   β”œβ”€β”€ sql_tools.py    # SQL execution tools
β”‚       β”‚   β”œβ”€β”€ analysis.py     # Statistical analysis tools
β”‚       β”‚   └── charts.py       # Chart generation tools
β”‚       β”‚
β”‚       β”œβ”€β”€ mcp/                # MCP server
β”‚       β”‚   β”œβ”€β”€ __init__.py
β”‚       β”‚   β”œβ”€β”€ server.py       # MCP server implementation
β”‚       β”‚   β”œβ”€β”€ tools.py        # MCP tool definitions
β”‚       β”‚   └── Dockerfile      # MCP server container
β”‚       β”‚
β”‚       β”œβ”€β”€ reports/            # PDF generation
β”‚       β”‚   β”œβ”€β”€ __init__.py
β”‚       β”‚   β”œβ”€β”€ generator.py    # Report generation orchestrator
β”‚       β”‚   β”œβ”€β”€ pdf.py          # PDF creation (WeasyPrint)
β”‚       β”‚   β”œβ”€β”€ charts.py       # Chart rendering
β”‚       β”‚   └── templates/      # HTML/CSS templates
β”‚       β”‚       β”œβ”€β”€ executive.html
β”‚       β”‚       β”œβ”€β”€ detailed.html
β”‚       β”‚       β”œβ”€β”€ minimal.html
β”‚       β”‚       └── styles.css
β”‚       β”‚
β”‚       β”œβ”€β”€ llm/                # LLM integration
β”‚       β”‚   β”œβ”€β”€ __init__.py
β”‚       β”‚   β”œβ”€β”€ client.py       # LLM client (Docker/Ollama/OpenAI)
β”‚       β”‚   └── prompts.py      # Prompt templates
β”‚       β”‚
β”‚       └── observability/      # Logging & tracing
β”‚           β”œβ”€β”€ __init__.py
β”‚           β”œβ”€β”€ tracing.py      # Distributed tracing
β”‚           └── metrics.py      # Token/cost tracking
β”‚
β”œβ”€β”€ tests/
β”‚   β”œβ”€β”€ __init__.py
β”‚   β”œβ”€β”€ conftest.py             # Pytest fixtures
β”‚   β”œβ”€β”€ test_cli.py
β”‚   β”œβ”€β”€ test_ingestion.py
β”‚   β”œβ”€β”€ test_query.py
β”‚   β”œβ”€β”€ test_agents.py
β”‚   β”œβ”€β”€ test_reports.py
β”‚   └── fixtures/
β”‚       β”œβ”€β”€ sample_sales.csv
β”‚       β”œβ”€β”€ sample_products.xlsx
β”‚       └── expected_outputs/
β”‚
β”œβ”€β”€ data/                       # Local data directory
β”‚   └── .gitkeep
β”‚
β”œβ”€β”€ reports/                    # Generated reports
β”‚   └── .gitkeep
β”‚
└── docs/                       # Documentation (Quarto)
    β”œβ”€β”€ _quarto.yml
    β”œβ”€β”€ index.qmd
    └── chapters/

Technology Stack

Category Tools
Package Management pixi, uv
CLI Framework Typer, Rich
Local LLM Docker Model Runner, Ollama
LLM Framework LangChain
Multi-Agent CrewAI
MCP Docker MCP Toolkit
Database SQLite
Data Processing pandas, openpyxl
PDF Generation WeasyPrint
Charts matplotlib, plotly
Observability Langfuse, OpenTelemetry
Testing pytest, DeepEval
Containerization Docker, Docker Compose

Example Usage

End-to-End Workflow

# 1. Start the Docker stack
docker compose up -d

# 2. Ingest your data
datacrew ingest quarterly_sales_2024.xlsx
datacrew ingest product_catalog.csv
datacrew ingest customer_data.csv

# 3. Explore with natural language queries
datacrew ask "How many records are in each table?"
datacrew ask "What are the top 10 products by revenue in Q4?"
datacrew ask "Show me the monthly sales trend for 2024"

# 4. Generate a comprehensive report
datacrew report "2024 Annual Sales Analysis" \
  --template detailed \
  --output reports/annual_2024.pdf \
  --include trends,top_products,regional_breakdown,recommendations \
  --verbose

# 5. View agent reasoning (verbose mode)
# [Data Analyst] Analyzing schema... found 3 tables
# [Data Analyst] Executing: SELECT strftime('%Y-%m', order_date) as month, SUM(revenue) ...
# [Insights Agent] Identified trend: 23% YoY growth in Q4
# [Insights Agent] Anomaly detected: December spike in electronics category
# [Report Writer] Generating executive summary...
# [PDF Composer] Assembling 12-page report...
# βœ“ Report saved to reports/annual_2024.pdf

Sample Report Output

β”Œβ”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”
β”‚                    2024 Annual Sales Analysis                  β”‚
β”‚                      Executive Summary                         β”‚
β”œβ”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€
β”‚                                                                β”‚
β”‚  Key Findings:                                                 β”‚
β”‚  β€’ Total revenue: $4.2M (+23% YoY)                            β”‚
β”‚  β€’ Top product category: Electronics (38% of revenue)         β”‚
β”‚  β€’ Strongest region: West Coast (42% of sales)                β”‚
β”‚  β€’ Customer retention rate: 78%                                β”‚
β”‚                                                                β”‚
β”‚  [Monthly Revenue Trend Chart]                                 β”‚
β”‚                                                                β”‚
β”‚  Recommendations:                                              β”‚
β”‚  1. Expand electronics inventory for Q1 2025                  β”‚
β”‚  2. Increase marketing spend in Midwest region                β”‚
β”‚  3. Launch loyalty program to improve retention               β”‚
β”‚                                                                β”‚
β””β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”˜

Learning Outcomes

By building DataCrew, learners will be able to:

  1. βœ… Set up modern Python projects with pixi and reproducible environments
  2. βœ… Build professional CLI tools with Typer and Rich
  3. βœ… Run local LLMs using Docker Model Runner
  4. βœ… Ingest and query data from spreadsheets using natural language
  5. βœ… Build MCP servers to connect AI agents to data sources
  6. βœ… Design multi-agent systems with CrewAI
  7. βœ… Generate PDF reports programmatically
  8. βœ… Implement observability for AI applications
  9. βœ… Test non-deterministic systems effectively
  10. βœ… Deploy self-hosted AI applications with Docker Compose