BAML

pip install rdk baml-py --extra-index-url https://pypi.fury.io/021labs/

class CustomerInquiry {
  category "billing" | "technical" | "general"
  priority int @description("1-5, where 5 is highest")
  summary string
}

function ClassifyInquiry(message: string) -> CustomerInquiry {
  client Claude
  prompt #"
    Classify this customer inquiry:
    {{ message }}

    Return the classification as JSON.
  "#
}

generator target {
  output_type python/pydantic
  output_dir ../
  version "0.219.0"
}

client<llm> Claude {
  provider anthropic
  options {
    model "claude-sonnet-4-6"
  }
}

baml-cli generate

from baml_client import b
from baml_client.types import CustomerInquiry
from rdk import observe, instrument_baml, shutdown

# Wrap the BAML client — required for tracing
b = instrument_baml(b)

@observe(name="classify-inquiry")
def classify(message: str) -> CustomerInquiry:
    return b.ClassifyInquiry(message)

result = classify("I can't log into my account!")
print(f"Category: {result.category}")
print(f"Priority: {result.priority}")
print(f"Summary: {result.summary}")

shutdown()

import asyncio
from baml_client import b
from rdk import observe, instrument_baml, shutdown

b = instrument_baml(b)

@observe(name="async-classify")
async def async_classify(message: str):
    return await b.ClassifyInquiry(message)

result = asyncio.run(async_classify("Payment failed"))
shutdown()

class Address {
  street string
  city string
  country string
  postal_code string | null
}

class Contact {
  name string
  email string
  phone string | null
  address Address
}

function ExtractContact(text: string) -> Contact {
  client Claude
  prompt #"
    Extract contact information from:
    {{ text }}
  "#
}

from baml_client import b
from rdk import observe, instrument_baml, shutdown

b = instrument_baml(b)

@observe(name="extract-contact")
def extract_contact(text: str):
    return b.ExtractContact(text)

result = extract_contact("""
    John Smith
    john@example.com
    123 Main St, New York, NY 10001
""")
print(f"Name: {result.name}")
print(f"City: {result.address.city}")
shutdown()

from baml_client import b
from baml_client.errors import BamlValidationError
from rdk import observe, instrument_baml, shutdown

b = instrument_baml(b)

@observe(name="safe-classify")
def safe_classify(message: str):
    try:
        return b.ClassifyInquiry(message)
    except BamlValidationError as e:
        print(f"Validation failed: {e}")
        return None

result = safe_classify("Hello")
shutdown()