This commit is contained in:
Chester Curme
2025-01-17 11:02:58 -05:00
parent 02ea523897
commit a54689affd
+14 -6
View File
@@ -958,7 +958,9 @@
"\n",
"# The config is the **second positional argument** to stream() or invoke()!\n",
"events = graph.stream(\n",
" {\"messages\": [{\"role\": \"user\", \"content\": user_input}]}, config, stream_mode=\"values\"\n",
" {\"messages\": [{\"role\": \"user\", \"content\": user_input}]},\n",
" config,\n",
" stream_mode=\"values\",\n",
")\n",
"for event in events:\n",
" event[\"messages\"][-1].pretty_print()"
@@ -998,7 +1000,9 @@
"\n",
"# The config is the **second positional argument** to stream() or invoke()!\n",
"events = graph.stream(\n",
" {\"messages\": [{\"role\": \"user\", \"content\": user_input}]}, config, stream_mode=\"values\"\n",
" {\"messages\": [{\"role\": \"user\", \"content\": user_input}]},\n",
" config,\n",
" stream_mode=\"values\",\n",
")\n",
"for event in events:\n",
" event[\"messages\"][-1].pretty_print()"
@@ -1231,7 +1235,7 @@
" # Because we will be interrupting during tool execution,\n",
" # we disable parallel tool calling to avoid repeating any\n",
" # tool invocations when we resume.\n",
" assert(len(message.tool_calls) <= 1)\n",
" assert len(message.tool_calls) <= 1\n",
" return {\"messages\": [message]}\n",
"\n",
"\n",
@@ -1348,7 +1352,9 @@
"config = {\"configurable\": {\"thread_id\": \"1\"}}\n",
"\n",
"events = graph.stream(\n",
" {\"messages\": [{\"role\": \"user\", \"content\": user_input}]}, config, stream_mode=\"values\"\n",
" {\"messages\": [{\"role\": \"user\", \"content\": user_input}]},\n",
" config,\n",
" stream_mode=\"values\",\n",
")\n",
"for event in events:\n",
" if \"messages\" in event:\n",
@@ -1663,7 +1669,7 @@
"\n",
"def chatbot(state: State):\n",
" message = llm_with_tools.invoke(state[\"messages\"])\n",
" assert(len(message.tool_calls) <= 1)\n",
" assert len(message.tool_calls) <= 1\n",
" return {\"messages\": [message]}\n",
"\n",
"\n",
@@ -1737,7 +1743,9 @@
"config = {\"configurable\": {\"thread_id\": \"1\"}}\n",
"\n",
"events = graph.stream(\n",
" {\"messages\": [{\"role\": \"user\", \"content\": user_input}]}, config, stream_mode=\"values\"\n",
" {\"messages\": [{\"role\": \"user\", \"content\": user_input}]},\n",
" config,\n",
" stream_mode=\"values\",\n",
")\n",
"for event in events:\n",
" if \"messages\" in event:\n",