src/models/request_models.py

from pydantic import BaseModel, Field

class ProviderNotesRequest(BaseModel):
    provider_notes: str = Field(
        ..., 
        description="The medical provider notes to analyze",
        min_length=10,
        example="Patient presents with acute bronchitis. Performed comprehensive examination and prescribed antibiotics."
    )
    
    class Config:
        json_schema_extra = {
            "example": {
                "provider_notes": "Patient presents with acute bronchitis. Cough for 5 days, productive with yellow sputum. Lung exam reveals diffuse wheezing. Prescribed azithromycin 500mg."
            }
        }

class ProviderNote(BaseModel):
    note: str

class CodingResponse(BaseModel):
    cpt_codes: list
    cpt_explanation: str
    icd_codes: list
    icd_explanation: str