docs: __start__/__end__ consistency (#1874)

* changes

* skip mermaid

* retries fix

* viz error

* changes
This commit is contained in:
Isaac Francisco
2024-09-30 13:55:53 -07:00
committed by GitHub
parent d29747906d
commit 17dc1108a3
18 changed files with 110 additions and 108 deletions
@@ -2545,7 +2545,7 @@
"builder.add_edge(\"fetch_user_info\", \"assistant\")\n",
"\n",
"\n",
"def route_tools(state: State) -> Literal[\"safe_tools\", \"sensitive_tools\", \"__end__\"]:\n",
"def route_tools(state: State):\n",
" next_node = tools_condition(state)\n",
" # If no tools are invoked, return to the user\n",
" if next_node == END:\n",
@@ -2562,6 +2562,7 @@
"builder.add_conditional_edges(\n",
" \"assistant\",\n",
" route_tools,\n",
" [\"safe_tools\", \"sensitive_tools\", END]\n",
")\n",
"builder.add_edge(\"safe_tools\", \"assistant\")\n",
"builder.add_edge(\"sensitive_tools\", \"assistant\")\n",
@@ -3523,12 +3524,7 @@
"\n",
"def route_update_flight(\n",
" state: State,\n",
") -> Literal[\n",
" \"update_flight_sensitive_tools\",\n",
" \"update_flight_safe_tools\",\n",
" \"leave_skill\",\n",
" \"__end__\",\n",
"]:\n",
"):\n",
" route = tools_condition(state)\n",
" if route == END:\n",
" return END\n",
@@ -3544,7 +3540,7 @@
"\n",
"builder.add_edge(\"update_flight_sensitive_tools\", \"update_flight\")\n",
"builder.add_edge(\"update_flight_safe_tools\", \"update_flight\")\n",
"builder.add_conditional_edges(\"update_flight\", route_update_flight)\n",
"builder.add_conditional_edges(\"update_flight\", route_update_flight, [\"update_flight_sensitive_tools\",\"update_flight_safe_tools\",\"leave_skill\",END])\n",
"\n",
"\n",
"# This node will be shared for exiting all specialized assistants\n",
@@ -3608,12 +3604,7 @@
"\n",
"def route_book_car_rental(\n",
" state: State,\n",
") -> Literal[\n",
" \"book_car_rental_safe_tools\",\n",
" \"book_car_rental_sensitive_tools\",\n",
" \"leave_skill\",\n",
" \"__end__\",\n",
"]:\n",
"):\n",
" route = tools_condition(state)\n",
" if route == END:\n",
" return END\n",
@@ -3629,7 +3620,7 @@
"\n",
"builder.add_edge(\"book_car_rental_sensitive_tools\", \"book_car_rental\")\n",
"builder.add_edge(\"book_car_rental_safe_tools\", \"book_car_rental\")\n",
"builder.add_conditional_edges(\"book_car_rental\", route_book_car_rental)"
"builder.add_conditional_edges(\"book_car_rental\", route_book_car_rental, [\"book_car_rental_safe_tools\",\"book_car_rental_sensitive_tools\",\"leave_skill\",END])"
]
},
{
@@ -3665,9 +3656,7 @@
"\n",
"def route_book_hotel(\n",
" state: State,\n",
") -> Literal[\n",
" \"leave_skill\", \"book_hotel_safe_tools\", \"book_hotel_sensitive_tools\", \"__end__\"\n",
"]:\n",
"):\n",
" route = tools_condition(state)\n",
" if route == END:\n",
" return END\n",
@@ -3683,7 +3672,7 @@
"\n",
"builder.add_edge(\"book_hotel_sensitive_tools\", \"book_hotel\")\n",
"builder.add_edge(\"book_hotel_safe_tools\", \"book_hotel\")\n",
"builder.add_conditional_edges(\"book_hotel\", route_book_hotel)"
"builder.add_conditional_edges(\"book_hotel\", route_book_hotel, [\"leave_skill\", \"book_hotel_safe_tools\", \"book_hotel_sensitive_tools\", END])"
]
},
{
@@ -3720,12 +3709,7 @@
"\n",
"def route_book_excursion(\n",
" state: State,\n",
") -> Literal[\n",
" \"book_excursion_safe_tools\",\n",
" \"book_excursion_sensitive_tools\",\n",
" \"leave_skill\",\n",
" \"__end__\",\n",
"]:\n",
"):\n",
" route = tools_condition(state)\n",
" if route == END:\n",
" return END\n",
@@ -3741,7 +3725,7 @@
"\n",
"builder.add_edge(\"book_excursion_sensitive_tools\", \"book_excursion\")\n",
"builder.add_edge(\"book_excursion_safe_tools\", \"book_excursion\")\n",
"builder.add_conditional_edges(\"book_excursion\", route_book_excursion)"
"builder.add_conditional_edges(\"book_excursion\", route_book_excursion, [\"book_excursion_safe_tools\",\"book_excursion_sensitive_tools\",\"leave_skill\",END])"
]
},
{
@@ -3768,13 +3752,7 @@
"\n",
"def route_primary_assistant(\n",
" state: State,\n",
") -> Literal[\n",
" \"primary_assistant_tools\",\n",
" \"enter_update_flight\",\n",
" \"enter_book_hotel\",\n",
" \"enter_book_excursion\",\n",
" \"__end__\",\n",
"]:\n",
"):\n",
" route = tools_condition(state)\n",
" if route == END:\n",
" return END\n",
@@ -3797,14 +3775,14 @@
"builder.add_conditional_edges(\n",
" \"primary_assistant\",\n",
" route_primary_assistant,\n",
" {\n",
" \"enter_update_flight\": \"enter_update_flight\",\n",
" \"enter_book_car_rental\": \"enter_book_car_rental\",\n",
" \"enter_book_hotel\": \"enter_book_hotel\",\n",
" \"enter_book_excursion\": \"enter_book_excursion\",\n",
" \"primary_assistant_tools\": \"primary_assistant_tools\",\n",
" END: END,\n",
" },\n",
" [\n",
" \"enter_update_flight\",\n",
" \"enter_book_car_rental\",\n",
" \"enter_book_hotel\",\n",
" \"enter_book_excursion\",\n",
" \"primary_assistant_tools\",\n",
" END,\n",
" ],\n",
")\n",
"builder.add_edge(\"primary_assistant_tools\", \"primary_assistant\")\n",
"\n",