Updated Document Index Notebook to use MultipleChunkRetrieverQa inste…

…ad of RetrieverBasedQa (#835) * Updated document_index notebook to use MultipleChunkRetrieverQa --------- Co-authored-by: Ben Beinke <[email protected]>
Aleph-Alpha · May 15, 2024 · 0d25e19 · 0d25e19
1 parent b50a4cd
commit 0d25e19
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/documentation/document_index.ipynb b/src/documentation/document_index.ipynb
@@ -22,7 +22,7 @@
     "    LimitedConcurrencyClient,\n",
     ")\n",
     "from intelligence_layer.core import InMemoryTracer\n",
-    "from intelligence_layer.examples import RetrieverBasedQa, RetrieverBasedQaInput\n",
+    "from intelligence_layer.examples import MultipleChunkRetrieverQa, RetrieverBasedQaInput\n",
     "\n",
     "load_dotenv()"
    ]
@@ -266,7 +266,7 @@
     "\n",
     "Finally, since we have a ready-to use 'Retriever', we can employ it to do something more complicated than just search; it can serve as the basis for a question-answering system.\n",
     "\n",
-    "To do so, let's run a `RetrieverBasedQa` task:"
+    "To do so, let's run a `MultipleChunkRetrieverQa` task:"
    ]
   },
   {
@@ -276,7 +276,7 @@
    "outputs": [],
    "source": [
     "client = LimitedConcurrencyClient.from_env()\n",
-    "retriever_qa = RetrieverBasedQa(document_index_retriever)\n",
+    "retriever_qa = MultipleChunkRetrieverQa(document_index_retriever, insert_chunk_number=3)\n",
     "\n",
     "\n",
     "input = RetrieverBasedQaInput(\n",