Format

examples/chat_agent_executor_with_function_calling/base.ipynb

@@ -242,9 +242,10 @@
     "import json\n",
     "from langchain_core.messages import FunctionMessage\n",
     "\n",
+    "\n",
     "# Define the function that determines whether to continue or not\n",
     "def should_continue(state):\n",
-    "    messages = state['messages']\n",
+    "    messages = state[\"messages\"]\n",
     "    last_message = messages[-1]\n",
     "    # If there is no function call, then we finish\n",
     "    if \"function_call\" not in last_message.additional_kwargs:\n",
@@ -253,23 +254,27 @@
     "    else:\n",
     "        return \"continue\"\n",
     "\n",
+    "\n",
     "# Define the function that calls the model\n",
     "def call_model(state):\n",
-    "    messages = state['messages']\n",
+    "    messages = state[\"messages\"]\n",
     "    response = model.invoke(messages)\n",
     "    # We return a list, because this will get added to the existing list\n",
     "    return {\"messages\": [response]}\n",
     "\n",
+    "\n",
     "# Define the function to execute tools\n",
     "def call_tool(state):\n",
-    "    messages = state['messages']\n",
+    "    messages = state[\"messages\"]\n",
     "    # Based on the continue condition\n",
     "    # we know the last message involves a function call\n",
     "    last_message = messages[-1]\n",
     "    # We construct an ToolInvocation from the function_call\n",
     "    action = ToolInvocation(\n",
     "        tool=last_message.additional_kwargs[\"function_call\"][\"name\"],\n",
-    "        tool_input=json.loads(last_message.additional_kwargs[\"function_call\"][\"arguments\"]),\n",
+    "        tool_input=json.loads(\n",
+    "            last_message.additional_kwargs[\"function_call\"][\"arguments\"]\n",
+    "        ),\n",
     "    )\n",
     "    # We call the tool_executor and get back a response\n",
     "    response = tool_executor.invoke(action)\n",
@@ -297,6 +302,7 @@
    "outputs": [],
    "source": [
     "from langgraph.graph import StateGraph, END\n",
+    "\n",
     "# Define a new graph\n",
     "workflow = StateGraph(AgentState)\n",
     "\n",
@@ -325,13 +331,13 @@
     "        # If `tools`, then we call the tool node.\n",
     "        \"continue\": \"action\",\n",
     "        # Otherwise we finish.\n",
-    "        \"end\": END\n",
-    "    }\n",
+    "        \"end\": END,\n",
+    "    },\n",
     ")\n",
     "\n",
     "# We now add a normal edge from `tools` to `agent`.\n",
     "# This means that after `tools` is called, `agent` node is called next.\n",
-    "workflow.add_edge('action', 'agent')\n",
+    "workflow.add_edge(\"action\", \"agent\")\n",
     "\n",
     "# Finally, we compile it!\n",
     "# This compiles it into a LangChain Runnable,\n",