from pydantic import BaseModel


class ConvertRequest(BaseModel):
    text: str
    format: str  # e.g. "mol", "smi", "sdf", "inchi", etc.
    add_hydrogen: bool = False
    convert_3d: bool = False
    optimize_geometry: bool = False


class SmilesRequest(BaseModel):
    text: str


class MolFileModel(BaseModel):
    molfile: str


class BasicError(BaseModel):
    error: str