Haiku for classification and routing tasks

import anthropic
import json

client = anthropic.Anthropic()

def classify_support_ticket(ticket_text: str) -> dict:
    """Route support tickets to category using Haiku."""
    message = client.messages.create(
        model="claude-3-5-haiku-20241022",
        max_tokens=100,
        system="You are a support ticket classifier. Respond with only valid JSON (no markdown). Categories: billing, technical, product_feedback, account.",
        messages=[
            {
                "role": "user",
                "content": f"Classify this ticket:\n\n{ticket_text}"
            }
        ]
    )
    
    response_text = message.content[0].text
    try:
        classification = json.loads(response_text)
    except json.JSONDecodeError:
        classification = {"category": "unknown", "confidence": 0.0}
    
    return {
        "category": classification.get("category"),
        "confidence": classification.get("confidence", 0.5),
        "input_tokens": message.usage.input_tokens,
        "output_tokens": message.usage.output_tokens
    }

ticket = "I was charged twice this month. My invoice shows two $99 charges on the same day."
result = classify_support_ticket(ticket)
print(json.dumps(result, indent=2))