from enum import Enum
from pydantic import BaseModel


class SessionType(str, Enum):
    EPHEMERAL = "ephemeral"
    PERSISTENT = "persistent"
    SCHEDULED = "scheduled"


class TaskStatus(str, Enum):
    PENDING = "pending"
    RUNNING = "running"
    COMPLETED = "completed"
    FAILED = "failed"


class CreateSessionRequest(BaseModel):
    type: SessionType = SessionType.EPHEMERAL
    title: str | None = None
    prompt: str | None = None


class Task(BaseModel):
    id: str
    type: SessionType
    prompt: str
    status: TaskStatus = TaskStatus.PENDING
    session_id: str | None = None
    schedule: str | None = None