Agent/agent/execution/__init__.py

"""
Execution - 执行追踪系统

核心职责：
1. Trace/Step 模型定义
2. 存储接口和实现（文件系统）
3. Step 树可视化（文本/markdown/JSON）
4. RESTful API（可视化查询，支持 compact/full 视图）
5. WebSocket 推送（实时更新，支持断线续传）
"""

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

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

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

# Debug 工具（可视化）
from agent.execution.tree_dump import StepTreeDumper, dump_tree, dump_markdown, dump_json


# 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_step_added,
        broadcast_step_updated,
        broadcast_trace_completed,
    )
    return broadcast_step_added, broadcast_step_updated, broadcast_trace_completed


# 便捷属性（仅在访问时导入）
@property
def api_router():
    return _get_api_router()


@property
def ws_router():
    return _get_ws_router()


__all__ = [
    # 模型
    "Trace",
    "Step",
    "StepType",
    "StepStatus",
    # 存储
    "TraceStore",
    "FileSystemTraceStore",
    # Debug/可视化
    "StepTreeDumper",
    "dump_tree",
    "dump_markdown",
    "dump_json",
]