"""Structured schemas for tasks, messages, plans, self-improvement, and AGI."""

from fusionagi.schemas.task import Task, TaskState, TaskPriority
from fusionagi.schemas.messages import AgentMessage, AgentMessageEnvelope
from fusionagi.schemas.plan import Plan, PlanStep
from fusionagi.schemas.recommendation import (
    Recommendation,
    RecommendationKind,
    TrainingSuggestion,
    TrainingSuggestionKind,
)
from fusionagi.schemas.goal import Goal, GoalBudget, GoalStatus, Blocker, Checkpoint
from fusionagi.schemas.grounding import Citation, GroundedClaim
from fusionagi.schemas.skill import Skill, SkillKind, SkillVersionInfo
from fusionagi.schemas.audit import AuditEntry, AuditEventType
from fusionagi.schemas.policy import PolicyRule, PolicyEffect
from fusionagi.schemas.world_model import StateTransition, UncertaintyInfo
from fusionagi.schemas.head import HeadId, HeadClaim, HeadRisk, HeadOutput
from fusionagi.schemas.witness import AgreementMap, TransparencyReport, FinalResponse
from fusionagi.schemas.commands import UserIntent, ParsedCommand, parse_user_input
from fusionagi.schemas.atomic import (
    AtomicUnitType,
    RelationType,
    AtomicSemanticUnit,
    SemanticRelation,
    DecompositionResult,
)

__all__ = [
    "Task",
    "TaskState",
    "TaskPriority",
    "AgentMessage",
    "AgentMessageEnvelope",
    "Plan",
    "PlanStep",
    "Recommendation",
    "RecommendationKind",
    "TrainingSuggestion",
    "TrainingSuggestionKind",
    "Goal",
    "GoalBudget",
    "GoalStatus",
    "Blocker",
    "Checkpoint",
    "Citation",
    "GroundedClaim",
    "Skill",
    "SkillKind",
    "SkillVersionInfo",
    "AuditEntry",
    "AuditEventType",
    "PolicyRule",
    "PolicyEffect",
    "StateTransition",
    "UncertaintyInfo",
    "HeadId",
    "HeadClaim",
    "HeadRisk",
    "HeadOutput",
    "AgreementMap",
    "TransparencyReport",
    "FinalResponse",
    "UserIntent",
    "ParsedCommand",
    "parse_user_input",
    "AtomicUnitType",
    "RelationType",
    "AtomicSemanticUnit",
    "SemanticRelation",
    "DecompositionResult",
]