[Docs] Add ruff linting to .ipynb files (#645)

examples/multi_agent/agent_supervisor.ipynb

@@ -73,10 +73,9 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "from typing import Annotated, List, Tuple, Union\n",
+    "from typing import Annotated\n",
     "\n",
     "from langchain_community.tools.tavily_search import TavilySearchResults\n",
-    "from langchain_core.tools import tool\n",
     "from langchain_experimental.tools import PythonREPLTool\n",
     "\n",
     "tavily_tool = TavilySearchResults(max_results=5)\n",
@@ -161,8 +160,8 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "from langchain_core.prompts import ChatPromptTemplate, MessagesPlaceholder\n",
     "from langchain_core.output_parsers.openai_functions import JsonOutputFunctionsParser\n",
+    "from langchain_core.prompts import ChatPromptTemplate, MessagesPlaceholder\n",
     "\n",
     "members = [\"Researcher\", \"Coder\"]\n",
     "system_prompt = (\n",
@@ -231,12 +230,13 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "import operator\n",
-    "from typing import Annotated, Any, Dict, List, Optional, Sequence, TypedDict\n",
     "import functools\n",
+    "import operator\n",
+    "from typing import Sequence, TypedDict\n",
     "\n",
     "from langchain_core.prompts import ChatPromptTemplate, MessagesPlaceholder\n",
-    "from langgraph.graph import StateGraph, END\n",
+    "\n",
+    "from langgraph.graph import END, StateGraph\n",
     "\n",
     "\n",
     "# The agent state is the input to each node in the graph\n",

examples/multi_agent/hierarchical_agent_teams.ipynb

@@ -59,7 +59,6 @@
    "source": [
     "import getpass\n",
     "import os\n",
-    "import uuid\n",
     "\n",
     "\n",
     "def _set_if_undefined(var: str):\n",
@@ -105,13 +104,11 @@
    },
    "outputs": [],
    "source": [
-    "from typing import Annotated, List, Tuple, Union\n",
+    "from typing import Annotated, List\n",
     "\n",
-    "import matplotlib.pyplot as plt\n",
     "from langchain_community.document_loaders import WebBaseLoader\n",
     "from langchain_community.tools.tavily_search import TavilySearchResults\n",
     "from langchain_core.tools import tool\n",
-    "from langsmith import trace\n",
     "\n",
     "tavily_tool = TavilySearchResults(max_results=5)\n",
     "\n",
@@ -236,7 +233,7 @@
     "\n",
     "@tool\n",
     "def python_repl(\n",
-    "    code: Annotated[str, \"The python code to execute to generate your chart.\"]\n",
+    "    code: Annotated[str, \"The python code to execute to generate your chart.\"],\n",
     "):\n",
     "    \"\"\"Use this to execute python code. If you want to see the output of a value,\n",
     "    you should print it out with `print(...)`. This is visible to the user.\"\"\"\n",
@@ -274,13 +271,11 @@
    },
    "outputs": [],
    "source": [
-    "from typing import Any, Callable, List, Optional, TypedDict, Union\n",
+    "from typing import List, Optional\n",
     "\n",
     "from langchain.agents import AgentExecutor, create_openai_functions_agent\n",
     "from langchain.output_parsers.openai_functions import JsonOutputFunctionsParser\n",
     "from langchain_core.prompts import ChatPromptTemplate, MessagesPlaceholder\n",
-    "from langchain_core.runnables import Runnable\n",
-    "from langchain_core.tools import BaseTool\n",
     "from langchain_openai import ChatOpenAI\n",
     "\n",
     "from langgraph.graph import END, StateGraph\n",
@@ -383,9 +378,8 @@
     "import functools\n",
     "import operator\n",
     "\n",
-    "from langchain_core.messages import AIMessage, BaseMessage, HumanMessage\n",
+    "from langchain_core.messages import BaseMessage, HumanMessage\n",
     "from langchain_openai.chat_models import ChatOpenAI\n",
-    "import functools\n",
     "\n",
     "\n",
     "# ResearchTeam graph state\n",
@@ -586,7 +580,7 @@
     "        written_files = [\n",
     "            f.relative_to(WORKING_DIRECTORY) for f in WORKING_DIRECTORY.rglob(\"*\")\n",
     "        ]\n",
-    "    except:\n",
+    "    except Exception:\n",
     "        pass\n",
     "    if not written_files:\n",
     "        return {**state, \"current_files\": \"No files written.\"}\n",
@@ -785,10 +779,9 @@
    },
    "outputs": [],
    "source": [
-    "from langchain_core.messages import AIMessage, BaseMessage, HumanMessage\n",
+    "from langchain_core.messages import BaseMessage\n",
     "from langchain_openai.chat_models import ChatOpenAI\n",
     "\n",
-    "\n",
     "llm = ChatOpenAI(model=\"gpt-4-1106-preview\")\n",
     "\n",
     "supervisor_node = create_team_supervisor(\n",

examples/multi_agent/multi-agent-collaboration.ipynb

@@ -77,10 +77,11 @@
    "source": [
     "from langchain_core.messages import (\n",
     "    BaseMessage,\n",
-    "    ToolMessage,\n",
     "    HumanMessage,\n",
+    "    ToolMessage,\n",
     ")\n",
     "from langchain_core.prompts import ChatPromptTemplate, MessagesPlaceholder\n",
+    "\n",
     "from langgraph.graph import END, StateGraph\n",
     "\n",
     "\n",
@@ -123,10 +124,11 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "from langchain_core.tools import tool\n",
     "from typing import Annotated\n",
-    "from langchain_experimental.utilities import PythonREPL\n",
+    "\n",
     "from langchain_community.tools.tavily_search import TavilySearchResults\n",
+    "from langchain_core.tools import tool\n",
+    "from langchain_experimental.utilities import PythonREPL\n",
     "\n",
     "tavily_tool = TavilySearchResults(max_results=5)\n",
     "\n",
@@ -137,7 +139,7 @@
     "\n",
     "@tool\n",
     "def python_repl(\n",
-    "    code: Annotated[str, \"The python code to execute to generate your chart.\"]\n",
+    "    code: Annotated[str, \"The python code to execute to generate your chart.\"],\n",
     "):\n",
     "    \"\"\"Use this to execute python code. If you want to see the output of a value,\n",
     "    you should print it out with `print(...)`. This is visible to the user.\"\"\"\n",
@@ -182,7 +184,6 @@
     "from typing import Annotated, Sequence, TypedDict\n",
     "\n",
     "from langchain_openai import ChatOpenAI\n",
-    "from typing_extensions import TypedDict\n",
     "\n",
     "\n",
     "# This defines the object that is passed between each node\n",
@@ -210,6 +211,7 @@
    "outputs": [],
    "source": [
     "import functools\n",
+    "\n",
     "from langchain_core.messages import AIMessage\n",
     "\n",
     "\n",
@@ -377,7 +379,7 @@
     "\n",
     "try:\n",
     "    display(Image(graph.get_graph(xray=True).draw_mermaid_png()))\n",
-    "except:\n",
+    "except Exception:\n",
     "    # This requires some extra dependencies and is optional\n",
     "    pass"
    ]