Build Pydantic AI Variant Triage Agent

from pydantic import BaseModel, Field
from pydantic_ai import Agent
from typing import List, Literal

class VariantClassification(BaseModel):
    variant_id: str
    classification: Literal['Pathogenic', 'Likely Pathogenic', 'VUS', 'Likely Benign', 'Benign']
    acmg_criteria: List[str]
    confidence: float = Field(ge=0.0, le=1.0)
    rationale: str

agent = Agent(
    'anthropic:claude-3-5-sonnet-20241022',
    result_type=VariantClassification,
    system_prompt='You are an expert clinical geneticist classifying missense variants based on ACMG rules.'
)

# Write the runner function to process input variant annotations