models.py

"""
Data models for the Tool Architect system.
"""
from dataclasses import dataclass, field
from typing import List, Optional, Literal


@dataclass
class Tool:
    """Represents an AI tool in the registry."""
    name: str
    slug: str
    categories: List[str]
    best_for: List[str]
    io: Optional[str] = None


@dataclass
class ToolAssignment:
    """A tool assigned to a task with routing metadata."""
    tool_slug: str
    fit_reason: str
    confidence: float
    best_prompt: str


@dataclass
class Task:
    """A concrete task within a phase."""
    id: str
    description: str
    outcome: str
    tools: List[ToolAssignment] = field(default_factory=list)


@dataclass
class Phase:
    """A phase in the project roadmap."""
    id: str
    name: str
    goal: str
    phase_type: Literal["research", "creative", "build", "launch", "ops"]
    tasks: List[Task] = field(default_factory=list)


@dataclass
class ProjectProfile:
    """Structured profile extracted from user idea."""
    idea_summary: str
    project_type: str
    domain: str
    primary_goal: str
    target_user: str
    constraints: List[str]
    channels: List[str]


@dataclass
class Roadmap:
    """Complete project roadmap."""
    project_profile: ProjectProfile
    phases: List[Phase]