diff --git a/examples/rag/langgraph_crag_local.ipynb b/examples/rag/langgraph_crag_local.ipynb
index dbac6f697..d6e791ae3 100644
--- a/examples/rag/langgraph_crag_local.ipynb
+++ b/examples/rag/langgraph_crag_local.ipynb
@@ -429,6 +429,14 @@
     "]\n",
     "\n",
     "\n",
+    "def find_tool_calls_react(messages):\n",
+    "    \"\"\"\n",
+    "    Find all tool calls in the messages returned\n",
+    "    \"\"\"\n",
+    "    tool_calls = [tc['name'] for m in messages['messages'] for tc in getattr(m, 'tool_calls', [])]\n",
+    "    return tool_calls\n",
+    "\n",
+    "\n",
     "def check_trajectory_react(root_run: Run, example: Example) -> dict:\n",
     "    \"\"\"\n",
     "    Check if all expected tools are called in exact order and without any additional tool calls.\n",