Merge pull request #1159 from akshseh/fix_visualization_example

fix: update the function for node colors
This commit is contained in:
Nuno Campos
2024-07-30 10:04:05 -07:00
committed by GitHub
+2 -2
View File
@@ -268,7 +268,7 @@
],
"source": [
"from IPython.display import Image, display\n",
"from langchain_core.runnables.graph import CurveStyle, MermaidDrawMethod, NodeColors\n",
"from langchain_core.runnables.graph import CurveStyle, MermaidDrawMethod, NodeStyles\n",
"\n",
"display(\n",
" Image(\n",
@@ -340,7 +340,7 @@
" Image(\n",
" app.get_graph().draw_mermaid_png(\n",
" curve_style=CurveStyle.LINEAR,\n",
" node_colors=NodeColors(start=\"#ffdfba\", end=\"#baffc9\", other=\"#fad7de\"),\n",
" node_colors=NodeStyles(first=\"#ffdfba\", last=\"#baffc9\", default=\"#fad7de\"),\n",
" wrap_label_n_words=9,\n",
" output_file_path=None,\n",
" draw_method=MermaidDrawMethod.PYPPETEER,\n",