added a how-to for new retry dictionary

docs/docs/how-tos/node-retries.ipynb

@@ -31,7 +31,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 1,
+   "execution_count": null,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -41,7 +41,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 2,
+   "execution_count": 14,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -78,20 +78,9 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 3,
+   "execution_count": null,
    "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "RetryPolicy(initial_interval=0.5, backoff_factor=2.0, max_interval=128.0, max_attempts=3, jitter=True, retry_on=<function default_retry_on at 0x78b964b89940>)"
-      ]
-     },
-     "execution_count": 3,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
+   "outputs": [],
    "source": [
     "from langgraph.pregel import RetryPolicy\n",
     "\n",
@@ -131,7 +120,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 5,
+   "execution_count": 16,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -180,6 +169,68 @@
     "\n",
     "graph = builder.compile()"
    ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "## State-Modifying Retry Policies\n",
+    "\n",
+    "Sometimes you might want to modify the state before retrying after an error occurs. LangGraph now supports this through a dictionary mapping of exceptions to state-modifying functions in the retry policy.\n"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 19,
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Exception occured: Error occurred in processing\n",
+      "Counter: 0\n",
+      "Exception occured: Error occurred in processing\n",
+      "Counter: 1\n",
+      "Exception occured: Error occurred in processing\n",
+      "Counter: 2\n",
+      "{'counter': 3, 'result': 'Success!'}\n"
+     ]
+    }
+   ],
+   "source": [
+    "class CounterState(TypedDict):\n",
+    "    counter: int\n",
+    "    result: str\n",
+    "\n",
+    "\n",
+    "def increment_counter(state: CounterState, exception: Exception):\n",
+    "    print(f\"Exception occured: {exception}\")\n",
+    "    print(f\"Counter: {state['counter']}\")\n",
+    "    state['counter'] += 1\n",
+    "\n",
+    "def processing_node(state: CounterState) -> CounterState:\n",
+    "    if state['counter'] < 3:\n",
+    "        raise ValueError(\"Error occurred in processing\")\n",
+    "    state['result'] = \"Success!\"\n",
+    "    return state\n",
+    "\n",
+    "workflow = StateGraph(CounterState)\n",
+    "\n",
+    "retry_policy = RetryPolicy(\n",
+    "    max_attempts=5,\n",
+    "    retry_on={ValueError: increment_counter}\n",
+    ")\n",
+    "\n",
+    "\n",
+    "workflow.add_node(\"process\", processing_node, retry=retry_policy)\n",
+    "workflow.add_edge(START, \"process\")\n",
+    "workflow.add_edge(\"process\", END)\n",
+    "\n",
+    "app = workflow.compile()\n",
+    "final_state = app.invoke({\"counter\": 0, \"result\": \"\"})\n",
+    "print(final_state)"
+   ]
   }
  ],
  "metadata": {