baml-toolkit

BAML Toolkit Plugin for Claude Code

Comprehensive toolkit for BAML (Boundary ML) development - build type-safe LLM integrations faster with project scaffolding, code generation, testing, and framework integration.

What is BAML?

BAML is a domain-specific language for building LLM applications with:

• Type-safe function definitions - No more JSON parsing headaches
• Multi-language code generation - Python, TypeScript, Go, Ruby, Elixir
• Built-in error handling - Retry logic, broken JSON fixing
• IDE integration - VSCode autocomplete and playground

Learn more: BAML Documentation

Features

Slash Commands (User-Invoked)

• /baml-toolkit:init - Initialize new BAML project with structure and configuration
• /baml-toolkit:generate - Generate client code from BAML definitions (with watch mode)
• /baml-toolkit:test - Run BAML function tests and display results
• /baml-toolkit:create-function - Scaffold new BAML function with types and prompts
• /baml-toolkit:create-type - Create BAML type definitions (class, enum, alias)
• /baml-toolkit:create-client - Configure LLM client (OpenAI, Anthropic, Google, etc.)
• /baml-toolkit:integrate - Integrate with frameworks (React, FastAPI, Express, Django, etc.)
• /baml-toolkit:batch-gemini - NEW! Batch process multiple files through BAML functions using Gemini

Skills (AI-Invoked Autonomously)

Claude will automatically help with BAML when:

• baml-scaffolder: You describe wanting to extract/classify/generate with LLMs
• baml-generator: .baml files are modified (auto-regenerates client code)
• baml-integrator: You ask about using BAML in your app framework
• gemini-batch-processor: NEW! You need to process multiple files cost-effectively with Gemini

Auto-Generation Hook

The plugin automatically regenerates client code when you modify .baml files - no manual steps needed!

Installation

Prerequisites

• Claude Code installed
• BAML CLI installed:

  # Using npm (recommended)
  npm install -g @boundaryml/baml
  
  # Or using pip
  pip install baml-py
  

• LLM API key (OpenAI, Anthropic, or Google)

Install Plugin

# Install from directory
claude plugin install /Users/orlandobruno/Documents/Dev/baml-toolkit --scope user

# Restart Claude Code to load the plugin

Verify Installation

# Check plugin is loaded
claude plugin list

# Should see "baml-toolkit" in the list

✨ What's New: Gemini Batch Processing

Process thousands of documents cost-effectively!

The plugin now includes powerful batch processing capabilities using Google's Gemini API:

• 💰 17x cheaper than GPT-4 (with Gemini Flash)
• 📄 2M token context - process entire documents at once
• ⚡ Fast & efficient - batch hundreds of files in minutes
• 🎯 Type-safe - BAML schemas ensure consistent results

# Process 500 invoices for ~$0.50
/baml-toolkit:batch-gemini ExtractInvoice invoices/*.pdf --output results.jsonl

# Classify 1000 reviews for ~$0.08
/baml-toolkit:batch-gemini ClassifyReview reviews.csv --column "text" --output classified.csv

See Batch Processing Examples below!

Quick Start

1. Initialize a BAML Project

/baml-toolkit:init my-llm-app

This creates:

my-llm-app/
├── baml_src/
│   ├── main.baml           # Your function definitions
│   └── clients.baml        # LLM client configurations
├── baml_client/            # Generated code (auto-created)
└── baml.config.json        # BAML configuration

2. Configure Your LLM Client

# Set your API key
export OPENAI_API_KEY="sk-..."
# Or for Gemini batch processing
export GOOGLE_API_KEY="your-gemini-key"

# Create a client configuration
/baml-toolkit:create-client "GPT4 client for OpenAI"
# Or use Gemini for cost-effective batch processing
/baml-toolkit:create-client "Gemini for batch processing"

3. Create Your First Function

/baml-toolkit:create-function "ExtractReceipt that takes an image and returns vendor, items, and total"

Or just describe what you need:

User: "I need to extract contact info from emails"
Claude: [Uses baml-scaffolder skill to create function automatically]

4. Generate Client Code

/baml-toolkit:generate --watch

The plugin automatically regenerates code when you edit .baml files!

5. Test Your Function

/baml-toolkit:test ExtractReceipt

6. Use in Your App

/baml-toolkit:integrate react
# or fastapi, express, django, etc.

Usage Examples

Example 1: Receipt Extraction

Create function (automatically done by Claude):

class Receipt {
  vendor string
  date string
  items ReceiptItem[]
  total float
}

class ReceiptItem {
  name string
  quantity int
  price float
}