From 4769122828da2ea1293fb3430d7f667e2caee663 Mon Sep 17 00:00:00 2001
From: Vadym Barda <vadym@langchain.dev>
Date: Wed, 28 Aug 2024 16:40:06 -0400
Subject: [PATCH] docs: fix map-reduce issue (#1523)

---
 examples/map-reduce.ipynb | 40 +++++++++++++++++++++++----------------
 1 file changed, 24 insertions(+), 16 deletions(-)

diff --git a/examples/map-reduce.ipynb b/examples/map-reduce.ipynb
index cefecbc96..055ca43b7 100644
--- a/examples/map-reduce.ipynb
+++ b/examples/map-reduce.ipynb
@@ -25,7 +25,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": null,
+   "execution_count": 1,
    "id": "3eb04cd1",
    "metadata": {},
    "outputs": [],
@@ -36,10 +36,18 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 1,
+   "execution_count": 2,
    "id": "dc292321",
    "metadata": {},
-   "outputs": [],
+   "outputs": [
+    {
+     "name": "stdin",
+     "output_type": "stream",
+     "text": [
+      "ANTHROPIC_API_KEY:  ········\n"
+     ]
+    }
+   ],
    "source": [
     "import os\n",
     "import getpass\n",
@@ -55,7 +63,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 10,
+   "execution_count": 3,
    "id": "0f0f78e4-423d-4e2d-aa1a-01efaec4715f",
    "metadata": {},
    "outputs": [],
@@ -63,7 +71,7 @@
     "import operator\n",
     "from typing import Annotated, TypedDict\n",
     "\n",
-    "from langchain_core.pydantic_v1 import BaseModel\n",
+    "from langchain_core.pydantic_v1 import BaseModel, Field\n",
     "from langchain_anthropic import ChatAnthropic\n",
     "\n",
     "from langgraph.constants import Send\n",
@@ -87,7 +95,7 @@
     "\n",
     "\n",
     "class BestJoke(BaseModel):\n",
-    "    id: int\n",
+    "    id: int = Field(description=\"Index of the best joke, starting with 0\")\n",
     "\n",
     "\n",
     "model = ChatAnthropic(model=\"claude-3-5-sonnet-20240620\")\n",
@@ -161,7 +169,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 11,
+   "execution_count": 4,
    "id": "37ed1f71-63db-416f-b715-4617b33d4b7f",
    "metadata": {},
    "outputs": [
@@ -172,7 +180,7 @@
        "<IPython.core.display.Image object>"
       ]
      },
-     "execution_count": 11,
+     "execution_count": 4,
      "metadata": {},
      "output_type": "execute_result"
     }
@@ -185,7 +193,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 9,
+   "execution_count": 5,
    "id": "fd90cace",
    "metadata": {},
    "outputs": [
@@ -193,12 +201,12 @@
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "{'generate_topics': {'subjects': ['lion', 'elephant', 'penguin', 'dolphin']}}\n",
+      "{'generate_topics': {'subjects': ['Lions', 'Elephants', 'Penguins', 'Dolphins']}}\n",
       "{'generate_joke': {'jokes': [\"Why don't elephants use computers? They're afraid of the mouse!\"]}}\n",
-      "{'generate_joke': {'jokes': [\"Why don't dolphins use smartphones? They're afraid of phishing!\"]}}\n",
-      "{'generate_joke': {'jokes': [\"Why don't lions like fast food? Because they can't catch it!\"]}}\n",
+      "{'generate_joke': {'jokes': [\"Why don't dolphins use smartphones? Because they're afraid of phishing!\"]}}\n",
       "{'generate_joke': {'jokes': [\"Why don't you see penguins in Britain? Because they're afraid of Wales!\"]}}\n",
-      "{'best_joke': {'best_selected_joke': \"Why don't you see penguins in Britain? Because they're afraid of Wales!\"}}\n"
+      "{'generate_joke': {'jokes': [\"Why don't lions like fast food? Because they can't catch it!\"]}}\n",
+      "{'best_joke': {'best_selected_joke': \"Why don't dolphins use smartphones? Because they're afraid of phishing!\"}}\n"
      ]
     }
    ],
@@ -211,9 +219,9 @@
  ],
  "metadata": {
   "kernelspec": {
-   "display_name": "Python 3 (ipykernel)",
+   "display_name": "langgraph",
    "language": "python",
-   "name": "python3"
+   "name": "langgraph"
   },
   "language_info": {
    "codemirror_mode": {
@@ -225,7 +233,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.11.8"
+   "version": "3.11.9"
   }
  },
  "nbformat": 4,