From 4beb5d644185c0fce4cad59cb22d0a32f294ae11 Mon Sep 17 00:00:00 2001
From: Nuno Campos <nuno@boringbits.io>
Date: Mon, 23 Oct 2023 14:17:16 +0100
Subject: [PATCH] Update notebook

---
 examples/combine_docs.ipynb | 38 ++++++++++++++++++-------------------
 1 file changed, 19 insertions(+), 19 deletions(-)

diff --git a/examples/combine_docs.ipynb b/examples/combine_docs.ipynb
index ffd417a0f..4657ed08c 100644
--- a/examples/combine_docs.ipynb
+++ b/examples/combine_docs.ipynb
@@ -25,7 +25,7 @@
     "from langchain.schema.document import Document\n",
     "from langchain.schema import format_document\n",
     "\n",
-    "from permchain import Pregel, PregelRead, channels"
+    "from permchain import Channels, Pregel, PregelRead\n"
    ]
   },
   {
@@ -45,7 +45,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "from langchain.schema.runnable import RunnableLambda"
+    "from langchain.schema.runnable import RunnableLambda\n"
    ]
   },
   {
@@ -59,7 +59,7 @@
     "\n",
     "_combine_documents = RunnableLambda(\n",
     "    lambda x: format_document(x, DEFAULT_DOCUMENT_PROMPT)\n",
-    ").map() | (lambda x: \"\\n\\n\".join(x))"
+    ").map() | (lambda x: \"\\n\\n\".join(x))\n"
    ]
   },
   {
@@ -72,7 +72,7 @@
     "docs = [\n",
     "    Document(page_content=\"Harrison used to work at Kensho\"),\n",
     "    Document(page_content=\"Ankush worked at Facebook\"),\n",
-    "]"
+    "]\n"
    ]
   },
   {
@@ -98,7 +98,7 @@
     "    )\n",
     "    | ChatOpenAI()\n",
     "    | StrOutputParser()\n",
-    ")"
+    ")\n"
    ]
   },
   {
@@ -119,7 +119,7 @@
     }
    ],
    "source": [
-    "stuff_chain.invoke({\"question\": \"where did harrison work\", \"docs\": docs})"
+    "stuff_chain.invoke({\"question\": \"where did harrison work\", \"docs\": docs})\n"
    ]
   },
   {
@@ -139,7 +139,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "many_docs = docs * 5"
+    "many_docs = docs * 5\n"
    ]
   },
   {
@@ -164,7 +164,7 @@
     "            new_result_doc_list.append(_sub_result_docs[:-1])\n",
     "            _sub_result_docs = _sub_result_docs[-1:]\n",
     "    new_result_doc_list.append(_sub_result_docs)\n",
-    "    return new_result_doc_list"
+    "    return new_result_doc_list\n"
    ]
   },
   {
@@ -196,7 +196,7 @@
    "source": [
     "# Just to show what its like split\n",
     "split_docs = _split_list_of_docs(many_docs)\n",
-    "split_docs"
+    "split_docs\n"
    ]
   },
   {
@@ -206,15 +206,15 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "chans = {\n",
+    "channels = {\n",
     "    # input\n",
-    "    \"question\": channels.LastValue(str),\n",
-    "    \"docs\": channels.Inbox(Document),\n",
+    "    \"question\": Channels.LastValue(str),\n",
+    "    \"docs\": Channels.Inbox(Document),\n",
     "    # intermediate\n",
-    "    \"docs_to_finalize\": channels.Inbox(Document),\n",
+    "    \"docs_to_finalize\": Channels.Inbox(Document),\n",
     "    # output\n",
-    "    \"answer\": channels.LastValue(str),\n",
-    "}"
+    "    \"answer\": Channels.LastValue(str),\n",
+    "}\n"
    ]
   },
   {
@@ -249,7 +249,7 @@
     "    Pregel.subscribe_to(\"docs_to_finalize\", key=\"docs\").join([\"question\"])\n",
     "    | stuff_chain\n",
     "    | Pregel.write_to(\"answer\")\n",
-    ")"
+    ")\n"
    ]
   },
   {
@@ -264,11 +264,11 @@
     "        \"collapse\": collapse,\n",
     "        \"finalize\": finalize,\n",
     "    },\n",
-    "    channels=chans,\n",
+    "    channels=channels,\n",
     "    input=[\"question\", \"docs\"],\n",
     "    output=\"answer\",\n",
     "    debug=True,\n",
-    ")"
+    ")\n"
    ]
   },
   {
@@ -332,7 +332,7 @@
     }
    ],
    "source": [
-    "reduce_chain.invoke({\"question\": \"where did harrison work\", \"docs\": many_docs})"
+    "reduce_chain.invoke({\"question\": \"where did harrison work\", \"docs\": many_docs})\n"
    ]
   },
   {