# api/models.py
from pydantic import BaseModel
from typing import Optional

class TaskRequest(BaseModel):
    task: str
    session_id: Optional[str] = None
    predefined_task: Optional[str] = None

class TaskResponse(BaseModel):
    status: str
    result: Optional[str] = None
    error: Optional[str] = None