diff --git a/recipes/use_cases/agents/langchain/langgraph-tool-calling-agent.ipynb b/recipes/use_cases/agents/langchain/langgraph-tool-calling-agent.ipynb
index 1deecb684..b23924e10 100644
--- a/recipes/use_cases/agents/langchain/langgraph-tool-calling-agent.ipynb
+++ b/recipes/use_cases/agents/langchain/langgraph-tool-calling-agent.ipynb
@@ -138,8 +138,11 @@
    "source": [
     "import replicate\n",
     "\n",
+    "from langchain_core.messages import ToolMessage\n",
+    "from langchain_core.runnables import RunnableLambda\n",
     "from langchain_core.tools import tool\n",
     "from langgraph.prebuilt import ToolNode\n",
+    "\n",
     "from langchain_community.tools.tavily_search import TavilySearchResults\n",
     "\n",
     "@tool\n",
@@ -375,7 +378,6 @@
     "from langgraph.checkpoint.sqlite import SqliteSaver\n",
     "from langgraph.graph import END, StateGraph\n",
     "from langgraph.prebuilt import ToolNode, tools_condition\n",
-    "from langchain_core.runnables import RunnableLambda\n",
     "\n",
     "# Graph\n",
     "builder = StateGraph(State)\n",
@@ -690,7 +692,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.10.14"
+   "version": "3.11.8"
   }
  },
  "nbformat": 4,