from flask import Blueprint
from handlers.langgraph_handler import run_langgraph, get_task_status, resume_task, visualize_graph

# 创建蓝图
bp = Blueprint('langgraph', __name__, url_prefix='/api/v1/langgraph')

# 定义路由
bp.add_url_rule('/run', view_func=run_langgraph, methods=['POST'])
bp.add_url_rule('/status/<thread_id>', view_func=get_task_status, methods=['GET'])
bp.add_url_rule('/resume/<thread_id>', view_func=resume_task, methods=['POST'])
bp.add_url_rule('/visualize/<thread_id>', view_func=visualize_graph, methods=['GET'])