from typing import TypedDict, List
from langgraph.graph import START, END, StateGraph

class AgentState(TypedDict):
    messages: List[str]
    current_step: str

def hello_node(state: AgentState):
    state["messages"].append("hello")
    state["current_step"] = "hello"
    return state

def goodbye_node(state: AgentState):
    state["messages"].append("goodbye")
    state["current_step"] = "goodbye"
    return state

builder = StateGraph(AgentState)
builder.add_node("hello", hello_node)
builder.add_node("goodbye", goodbye_node)
builder.add_edge(START, "hello")
builder.add_edge("hello", "goodbye")
builder.add_edge("goodbye", END)

graph1 = builder.compile()

# 执行工作流
res = graph1.invoke({"messages": [], "current_step": ""})
print("工作流执行结果:", res["messages"])

# from IPython.display import Image
# Image(graph1.get_graph().draw_mermaid_png())

mermaid_text = graph1.get_graph().draw_mermaid()
# 生成HTML可视化文件
html_content = f"""
<!DOCTYPE html>
<html>
<head>
    <title>LangGraph工作流可视化</title>
    <script type="module">
        import mermaid from 'https://cdn.jsdelivr.net/npm/mermaid@10/dist/mermaid.esm.min.mjs';
        mermaid.initialize({{ startOnLoad: true }});
    </script>
</head>
<body>
    <h1>Customer Service Workflow</h1>
    <div class="mermaid">
{mermaid_text}
    </div>
</body>
</html>
"""

html_filename = "workflow_visualization1.html"
with open(html_filename, "w", encoding="utf-8") as f:
    f.write(html_content)
print(f"✅ HTML可视化文件已保存为: {html_filename}")

代码示例如上，核心生图代码：

mermaid_text = graph1.get_graph().draw_mermaid()
# 生成HTML可视化文件
html_content = f"""
<!DOCTYPE html>
<html>
<head>
    <title>LangGraph工作流可视化</title>
    <script type="module">
        import mermaid from 'https://cdn.jsdelivr.net/npm/mermaid@10/dist/mermaid.esm.min.mjs';
        mermaid.initialize({{ startOnLoad: true }});
    </script>
</head>
<body>
    <h1>Customer Service Workflow</h1>
    <div class="mermaid">
{mermaid_text}
    </div>
</body>
</html>
"""

html_filename = "workflow_visualization1.html"
with open(html_filename, "w", encoding="utf-8") as f:
    f.write(html_content)
print(f"✅ HTML可视化文件已保存为: {html_filename}")

然后打开html即可得到如下显示

# arcgis
Logo
AI Agent技术社区

Agent 垂直技术社区，欢迎活跃、内容共建。

更多推荐

告别扫描文档处理烦恼：ScanTailor Advanced的智能文档优化解决方案

ScanTailor Advanced是一款专业的扫描文档后期处理工具，专为解决老旧文档数字化中的各种问题而设计。通过智能内容识别、批量处理和多线程优化，它能将倾斜、模糊、杂乱的扫描文档转化为整洁的电子文件，显著提升文档数字化效率。## 文档数字化的常见痛点与解决方案### 页面歪斜校正：智能算法自动修正传统扫描文档经常出现页面歪斜问题，手动调整每页耗时耗力。ScanTailor Adv

avatar AI Agent技术社区

SeedVR2终极指南：如何用单步扩散模型实现专业级视频修复

还在为老旧视频的模糊画质而烦恼吗？面对抖动严重的监控录像或褪色的家庭影像，你是否曾希望有一种简单高效的修复方案？SeedVR2正是为解决这些痛点而生的一站式视频修复工具。作为字节跳动推出的先进视频修复模型，它能够在单步推理中实现专业级的视频质量提升，彻底改变了传统视频修复的工作流程。## 核心价值：为什么选择SeedVR2进行视频修复？**传统视频修复的三大瓶颈**：1. **计算成本高

avatar AI Agent技术社区

终极指南：如何用Python自动化工具告别大麦抢票焦虑，轻松获取心仪演出门票

还记得那些让你心跳加速的抢票时刻吗？热门演唱会开票瞬间，你手忙脚乱地刷新页面，却眼睁睁看着"立即购买"按钮变灰，票在几秒内售罄。这种"拼手速"的抢票时代已经过去，现在有了基于Python开发的大麦自动抢票工具，让你告别手速焦虑，轻松获取周杰伦、五月天、周深等热门演出的门票。## 🔥 你的抢票痛点，我们都有解决方案### 场景一：热门演唱会一票难求"周杰伦演唱会开票3秒售罄，我连页面都没

avatar AI Agent技术社区