Agent/agent/goal/__init__.py

"""
Goal 模块 - 执行计划管理

提供 Goal 和 GoalTree 数据模型，以及 goal 工具。
"""

from agent.goal.models import (
    Goal,
    GoalTree,
    GoalStatus,
    GoalType,
    GoalStats,
    BranchContext,
    BranchStatus,
)
from agent.goal.tool import goal_tool, create_goal_tool_schema

__all__ = [
    # Models
    "Goal",
    "GoalTree",
    "GoalStatus",
    "GoalType",
    "GoalStats",
    "BranchContext",
    "BranchStatus",
    # Tool
    "goal_tool",
    "create_goal_tool_schema",
]