Merge pull request #2 from mongodb-developer/marco/langchain-workshop

Langchain workshop

Author: sis0k0

multi_agent.ipynb

@@ -568,12 +568,9 @@
       "metadata": {},
       "outputs": [],
       "source": [
+        "from utils import visualize_graph\n",
         "# Visualize the graph\n",
-        "try:\n",
-        "    display(Image(agent.get_graph().draw_mermaid_png()))\n",
-        "except Exception:\n",
-        "    # This requires some extra dependencies and is optional\n",
-        "    pass"
+        "visualize_graph(agent)"
       ]
     },
     {