"""
数据库模块
提供工作流和工作流运行的数据库操作接口
"""

from .models import Workflow, WorkflowRun, WorkflowRunNode, ComfyUIServer
from .api import (
    init_db,
    save_workflow,
    get_all_workflows,
    get_latest_workflow_by_base_name,
    get_workflow_by_version,
    get_workflow,
    delete_workflow,
    create_workflow_run,
    update_workflow_run_status,
    create_workflow_run_nodes,
    update_workflow_run_node_status,
    get_workflow_run,
    get_workflow_run_nodes,
    get_pending_workflow_runs,
    get_running_workflow_runs,
)

__all__ = [
    # 模型
    "Workflow",
    "WorkflowRun", 
    "WorkflowRunNode",
    "ComfyUIServer",
    
    # API函数
    "init_db",
    "save_workflow",
    "get_all_workflows",
    "get_latest_workflow_by_base_name",
    "get_workflow_by_version",
    "get_workflow",
    "delete_workflow",
    "create_workflow_run",
    "update_workflow_run_status",
    "create_workflow_run_nodes",
    "update_workflow_run_node_status",
    "get_workflow_run",
    "get_workflow_run_nodes",
    "get_pending_workflow_runs",
    "get_running_workflow_runs",
]