【langchain学习】解决langgraph在pycharm中无法可视化graph；出现＜IPython.core.display.Image object＞的问题

在jupyter等笔记本中可视化graph的方法可参考：https://langchain-ai.github.io/langgraph/how-tos/visualization/

这是官方文档中快速开始的例子：

graph = graph_builder.compile()  # 编译图

在jupyter notebook中可以这样表示：

from IPython.display import Image, display

try:
    display(Image(graph.get_graph().draw_mermaid_png()))
except Exception:
    # This requires some extra dependencies and is optional
    pass

在pycharm中可以这样表示：用matlpotlib替换IPython,生成jpg格式的流程图

import matplotlib.pyplot as plt
import matplotlib.image as mpimg  # 导入matplotlib.image用于读取图像
try:
    # 使用 Mermaid 生成图表并保存为文件
    mermaid_code = graph.get_graph().draw_mermaid_png()
    with open("graph.jpg", "wb") as f:
        f.write(mermaid_code)

    # 使用 matplotlib 显示图像
    img = mpimg.imread("graph.jpg")
    plt.imshow(img)
    plt.axis('off')  # 关闭坐标轴
    plt.show()
except Exception as e:
    print(f"An error occurred: {e}")