from pydantic import BaseModel
from enum import Enum
from typing import Optional

FunctionStatus = Enum("FunctionStatus", ["SUCCESS", "FAILED"])

class FunctionResponse(BaseModel):
    name: str
    status: FunctionStatus
    message: str = ""
    args: Optional[dict] = None