diff --git a/docs/docs/integrations/tools/awslambda.ipynb b/docs/docs/integrations/tools/awslambda.ipynb
index 08a7ed4e692..c7bfb5039f8 100644
--- a/docs/docs/integrations/tools/awslambda.ipynb
+++ b/docs/docs/integrations/tools/awslambda.ipynb
@@ -62,23 +62,25 @@
    },
    "outputs": [],
    "source": [
-    "from langchain.agents import AgentType, initialize_agent, load_tools\n",
+    "from langchain.agents import AgentExecutor, OpenAIFunctionsAgent\n",
+    "from langchain.tools import Tool\n",
     "from langchain_openai import OpenAI\n",
     "\n",
     "llm = OpenAI(temperature=0)\n",
     "\n",
-    "tools = load_tools(\n",
-    "    [\"awslambda\"],\n",
-    "    awslambda_tool_name=\"email-sender\",\n",
-    "    awslambda_tool_description=\"sends an email with the specified content to test@testing123.com\",\n",
-    "    function_name=\"testFunction1\",\n",
+    "tools = Tool(\n",
+    "    name=\"email-sender\",\n",
+    "    description=\"Sends an email with the specified content to test@testing123.com\",\n",
+    "    func=lambda input_text: f\"Email sent to test@testing123.com with content: {input_text}\",\n",
     ")\n",
     "\n",
-    "agent = initialize_agent(\n",
-    "    tools, llm, agent=AgentType.ZERO_SHOT_REACT_DESCRIPTION, verbose=True\n",
-    ")\n",
+    "agent = OpenAIFunctionsAgent(llm=llm, tools=[tools])\n",
     "\n",
-    "agent.run(\"Send an email to test@testing123.com saying hello world.\")"
+    "agent_executor = AgentExecutor(agent=agent, tools=[tools], verbose=True)\n",
+    "\n",
+    "agent_executor.invoke(\n",
+    "    {\"input\": \" Send an email to test@testing123.com saying hello world.\"}\n",
+    ")"
    ]
   },
   {