This commit is contained in:
Chester Curme
2025-01-17 11:05:55 -05:00
parent a54689affd
commit 5e3aa495d9
+6 -2
View File
@@ -1523,7 +1523,9 @@
"\n",
"\n",
"def chatbot(state: State):\n",
" return {\"messages\": [llm_with_tools.invoke(state[\"messages\"])]}\n",
" message = llm_with_tools.invoke(state[\"messages\"])\n",
" assert(len(message.tool_calls) <= 1)\n",
" return {\"messages\": [message]}\n",
"\n",
"\n",
"graph_builder.add_node(\"chatbot\", chatbot)\n",
@@ -1991,7 +1993,9 @@
"\n",
"\n",
"def chatbot(state: State):\n",
" return {\"messages\": [llm_with_tools.invoke(state[\"messages\"])]}\n",
" message = llm_with_tools.invoke(state[\"messages\"])\n",
" assert(len(message.tool_calls) <= 1)\n",
" return {\"messages\": [message]}\n",
"\n",
"\n",
"graph_builder = StateGraph(State)\n",