|
682 | 682 | "metadata": {},
|
683 | 683 | "outputs": [],
|
684 | 684 | "source": [
|
685 |
| - "from utils import visualize_graph\n", |
686 | 685 | "simple_rag_graph = graph_builder.compile()\n",
|
687 | 686 | "\n",
|
688 |
| - "visualize_graph(simple_rag_graph)" |
| 687 | + "simple_rag_graph" |
689 | 688 | ]
|
690 | 689 | },
|
691 | 690 | {
|
|
893 | 892 | "graph_builder.add_edge(\"generate_response\", END)\n",
|
894 | 893 | "\n",
|
895 | 894 | "document_grading_graph = graph_builder.compile()\n",
|
896 |
| - "visualize_graph(document_grading_graph)" |
| 895 | + "document_grading_graph" |
897 | 896 | ]
|
898 | 897 | },
|
899 | 898 | {
|
|
1093 | 1092 | "memory = MemorySaver()\n",
|
1094 | 1093 | "\n",
|
1095 | 1094 | "graph = graph_builder.compile(checkpointer=memory)\n",
|
1096 |
| - "visualize_graph(graph)" |
| 1095 | + "graph" |
1097 | 1096 | ]
|
1098 | 1097 | },
|
1099 | 1098 | {
|
|
1274 | 1273 | },
|
1275 | 1274 | {
|
1276 | 1275 | "cell_type": "code",
|
1277 |
| - "execution_count": 60, |
| 1276 | + "execution_count": null, |
1278 | 1277 | "metadata": {},
|
1279 | 1278 | "outputs": [],
|
1280 | 1279 | "source": [
|
|
1295 | 1294 | " })\n",
|
1296 | 1295 | "\n",
|
1297 | 1296 | "\n",
|
1298 |
| - "graph = graph_builder.compile(checkpointer=mongodb_saver)\n", |
1299 |
| - "# visualize_graph(graph)" |
| 1297 | + "graph = graph_builder.compile(checkpointer=mongodb_saver)" |
1300 | 1298 | ]
|
1301 | 1299 | },
|
1302 | 1300 | {
|
|
1426 | 1424 | "graph = graph_builder.compile(\n",
|
1427 | 1425 | " name=\"mongodb_rag_pipeline\",\n",
|
1428 | 1426 | " checkpointer=mongodb_saver)\n",
|
1429 |
| - "visualize_graph(graph)" |
| 1427 | + "graph" |
1430 | 1428 | ]
|
1431 | 1429 | },
|
1432 | 1430 | {
|
|
1601 | 1599 | ")\n",
|
1602 | 1600 | "\n",
|
1603 | 1601 | "# Visualize the graph\n",
|
1604 |
| - "visualize_graph(mongodb_server_subagent)\n" |
| 1602 | + "mongodb_server_subagent\n" |
1605 | 1603 | ]
|
1606 | 1604 | },
|
1607 | 1605 | {
|
|
1706 | 1704 | "supervisor_agent = supervisor_prebuilt_workflow.compile(name=\"mongodb_supervisor_agent\", checkpointer=mongodb_saver)\n",
|
1707 | 1705 | "\n",
|
1708 | 1706 | "# Visualize the graph\n",
|
1709 |
| - "visualize_graph(supervisor_agent)" |
| 1707 | + "supervisor_agent" |
1710 | 1708 | ]
|
1711 | 1709 | },
|
1712 | 1710 | {
|
|
0 commit comments