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

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

__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",
]