Agent/agent/trace/__init__.py

"""
Trace module - 执行追踪与计划管理

包含：
- Trace, Goal, Message 数据模型
- TraceStore 存储接口和实现
- goal 工具（计划管理）
- Context 压缩
- REST/WebSocket API
"""

from .models import Trace, Message
from .goal_models import Goal, GoalTree, GoalStatus, GoalType, GoalStats
from .protocols import TraceStore
from .store import FileSystemTraceStore
from .trace_id import generate_trace_id, generate_sub_trace_id, parse_parent_trace_id
from .goal_tool import set_goal_tree, get_goal_tree

__all__ = [
    # Models
    "Trace",
    "Message",
    "Goal",
    "GoalTree",
    "GoalStatus",
    "GoalType",
    "GoalStats",
    # Store
    "TraceStore",
    "FileSystemTraceStore",
    # Utils
    "generate_trace_id",
    "generate_sub_trace_id",
    "parse_parent_trace_id",
    # Goal tool
    "set_goal_tree",
    "get_goal_tree",
]