black

docs/docs/how_to/migrate_agent.ipynb

@@ -24,7 +24,7 @@
    "outputs": [],
    "source": [
     "%%capture --no-stderr\n",
-    "%pip install -U langchain-openai langchain langgraph "
+    "%pip install -U langchain-openai langchain langgraph"
    ]
   },
   {
@@ -255,9 +255,7 @@
     "# This could also be a SystemMessage object\n",
     "# system_message = SystemMessage(content=\"You are a helpful assistant. Respond only in Spanish.\")\n",
     "\n",
-    "app = create_react_agent(\n",
-    "    model, tools, messages_modifier=system_message\n",
-    ")\n",
+    "app = create_react_agent(model, tools, messages_modifier=system_message)\n",
     "\n",
     "\n",
     "messages = app.invoke({\"messages\": [(\"user\", query)]})"
@@ -303,7 +301,9 @@
     "\n",
     "\n",
     "def _modify_messages(messages: list[AnyMessage]):\n",
-    "    return prompt.invoke({\"messages\": messages}).to_messages() + [(\"user\", \"Also say 'Pandamonium!' after the answer.\")]\n",
+    "    return prompt.invoke({\"messages\": messages}).to_messages() + [\n",
+    "        (\"user\", \"Also say 'Pandamonium!' after the answer.\")\n",
+    "    ]\n",
     "\n",
     "\n",
     "app = create_react_agent(model, tools, messages_modifier=_modify_messages)\n",
@@ -375,7 +375,7 @@
     }
    ],
    "source": [
-    "from langgraph.prebuilt import create_react_agent \n",
+    "from langgraph.prebuilt import create_react_agent\n",
     "\n",
     "app = create_react_agent(model, tools=tools)\n",
     "\n",
@@ -501,9 +501,11 @@
     "\n",
     "try:\n",
     "    for chunk in app.stream(\n",
-    "        {\"messages\": [(\"human\", query)]}, {\"recursion_limit\": RECURSION_LIMIT}, stream_mode=\"values\"\n",
+    "        {\"messages\": [(\"human\", query)]},\n",
+    "        {\"recursion_limit\": RECURSION_LIMIT},\n",
+    "        stream_mode=\"values\",\n",
     "    ):\n",
-    "        print(chunk['messages'][-1])\n",
+    "        print(chunk[\"messages\"][-1])\n",
     "except GraphRecursionError:\n",
     "    print({\"input\": query, \"output\": \"Agent stopped due to max iterations.\"})"
    ]