Agent/agent/execution/__init__.py

"""
Execution - 执行追踪系统

核心职责：
1. Trace/Message 模型定义（新架构）
2. 存储接口和实现（文件系统）
3. GoalTree 集成（计划管理）
4. RESTful API（可视化查询）
5. WebSocket 推送（实时更新，支持断线续传）
"""

# 模型（核心，无依赖）
from agent.execution.models import Trace, Message

# 向后兼容：保留 Step 导出（已废弃）
from agent.execution.models import Step, StepType, StepStatus

# 存储接口（核心，无依赖）
from agent.execution.protocols import TraceStore

# 文件系统存储实现（跨进程 + 持久化）
from agent.execution.fs_store import FileSystemTraceStore


# API 路由（可选，需要 FastAPI）
def _get_api_router():
    """延迟导入 API Router（避免强制依赖 FastAPI）"""
    from agent.execution.api import router
    return router


def _get_ws_router():
    """延迟导入 WebSocket Router（避免强制依赖 FastAPI）"""
    from agent.execution.websocket import router
    return router


# WebSocket 广播函数（可选，需要 FastAPI）
def _get_broadcast_functions():
    """延迟导入 WebSocket 广播函数"""
    from agent.execution.websocket import (
        broadcast_goal_added,
        broadcast_goal_updated,
        broadcast_branch_started,
        broadcast_branch_completed,
        broadcast_explore_completed,
        broadcast_trace_completed,
    )
    return (
        broadcast_goal_added,
        broadcast_goal_updated,
        broadcast_branch_started,
        broadcast_branch_completed,
        broadcast_explore_completed,
        broadcast_trace_completed,
    )


__all__ = [
    # 模型
    "Trace",
    "Message",
    # 向后兼容（已废弃）
    "Step",
    "StepType",
    "StepStatus",
    # 存储
    "TraceStore",
    "FileSystemTraceStore",
]