Fix to Chat call to use correct documents

Author: jlinsdell-cohere

notebooks/Cohere_Elastic_Guide.ipynb

@@ -469,8 +469,8 @@
         "ranked_documents = []\n",
         "for document in response.body[\"rerank\"]:\n",
         "  ranked_documents.append({\n",
-        "      \"title\": documents[document[\"index\"]][\"title\"],\n",
-        "      \"text\": documents[document[\"index\"]][\"text\"]\n",
+        "      \"title\": raw_documents[int(document[\"index\"])][\"_source\"][\"title\"],\n",
+        "      \"text\": raw_documents[int(document[\"index\"])][\"_source\"][\"text\"]\n",
         "  })\n",
         "\n",
         "# Print the top 10 results\n",
@@ -528,7 +528,7 @@
       "source": [
         "response = co.chat(\n",
         "    message=query,\n",
-        "    documents=documents,\n",
+        "    documents=ranked_documents,\n",
         "    model='command-r-plus'\n",
         ")\n",
         "\n",