From a2588d6c5759e0a3c95ba32d2daea688d6ea3774 Mon Sep 17 00:00:00 2001
From: Charles Lanahan <charles.lanahan@gmail.com>
Date: Thu, 10 Aug 2023 22:02:10 -0400
Subject: [PATCH] Update openai embeddings notebook with correct embedding
 model in section 2 (#5831)

In second section it looks like a copy/paste from the first section and
doesn't include the specific embedding model mentioned in the example so
I added it for clarity.
---------

Co-authored-by: Bagatur <baskaryan@gmail.com>
---
 .../integrations/text_embedding/openai.ipynb  | 126 ++++++++++++++++--
 1 file changed, 113 insertions(+), 13 deletions(-)

diff --git a/docs/extras/integrations/text_embedding/openai.ipynb b/docs/extras/integrations/text_embedding/openai.ipynb
index 9cb9c62502..7dbf174ceb 100644
--- a/docs/extras/integrations/text_embedding/openai.ipynb
+++ b/docs/extras/integrations/text_embedding/openai.ipynb
@@ -12,7 +12,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 1,
+   "execution_count": 6,
    "id": "0be1af71",
    "metadata": {},
    "outputs": [],
@@ -22,7 +22,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 2,
+   "execution_count": 29,
    "id": "2c66e5da",
    "metadata": {},
    "outputs": [],
@@ -32,7 +32,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 3,
+   "execution_count": 30,
    "id": "01370375",
    "metadata": {},
    "outputs": [],
@@ -42,7 +42,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 4,
+   "execution_count": 31,
    "id": "bfb6142c",
    "metadata": {},
    "outputs": [],
@@ -52,7 +52,32 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 5,
+   "execution_count": 32,
+   "id": "91bc875d-829b-4c3d-8e6f-fc2dda30a3bd",
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "[-0.003186025367556387,\n",
+       " 0.011071979803637493,\n",
+       " -0.004020420763285827,\n",
+       " -0.011658221276953042,\n",
+       " -0.0010534035786864363]"
+      ]
+     },
+     "execution_count": 32,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "query_result[:5]"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 33,
    "id": "0356c3b7",
    "metadata": {},
    "outputs": [],
@@ -60,6 +85,31 @@
     "doc_result = embeddings.embed_documents([text])"
    ]
   },
+  {
+   "cell_type": "code",
+   "execution_count": 34,
+   "id": "a4b0d49e-0c73-44b6-aed5-5b426564e085",
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "[-0.003186025367556387,\n",
+       " 0.011071979803637493,\n",
+       " -0.004020420763285827,\n",
+       " -0.011658221276953042,\n",
+       " -0.0010534035786864363]"
+      ]
+     },
+     "execution_count": 34,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "doc_result[0][:5]"
+   ]
+  },
   {
    "cell_type": "markdown",
    "id": "bb61bbeb",
@@ -70,7 +120,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": null,
+   "execution_count": 1,
    "id": "c0b072cc",
    "metadata": {},
    "outputs": [],
@@ -80,17 +130,17 @@
   },
   {
    "cell_type": "code",
-   "execution_count": null,
+   "execution_count": 23,
    "id": "a56b70f5",
    "metadata": {},
    "outputs": [],
    "source": [
-    "embeddings = OpenAIEmbeddings()"
+    "embeddings = OpenAIEmbeddings(model=\"text-search-ada-doc-001\")"
    ]
   },
   {
    "cell_type": "code",
-   "execution_count": null,
+   "execution_count": 24,
    "id": "14aefb64",
    "metadata": {},
    "outputs": [],
@@ -100,7 +150,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": null,
+   "execution_count": 25,
    "id": "3c39ed33",
    "metadata": {},
    "outputs": [],
@@ -110,7 +160,32 @@
   },
   {
    "cell_type": "code",
-   "execution_count": null,
+   "execution_count": 26,
+   "id": "2ee7ce9f-d506-4810-8897-e44334412714",
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "[0.004452846988523035,\n",
+       " 0.034550655976098514,\n",
+       " -0.015029939040690051,\n",
+       " 0.03827273883655212,\n",
+       " 0.005785414075152477]"
+      ]
+     },
+     "execution_count": 26,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "query_result[:5]"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 27,
    "id": "e3221db6",
    "metadata": {},
    "outputs": [],
@@ -118,6 +193,31 @@
     "doc_result = embeddings.embed_documents([text])"
    ]
   },
+  {
+   "cell_type": "code",
+   "execution_count": 28,
+   "id": "a0865409-3a6d-468f-939f-abde17c7cac3",
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "[0.004452846988523035,\n",
+       " 0.034550655976098514,\n",
+       " -0.015029939040690051,\n",
+       " 0.03827273883655212,\n",
+       " 0.005785414075152477]"
+      ]
+     },
+     "execution_count": 28,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "doc_result[0][:5]"
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": null,
@@ -132,7 +232,7 @@
  ],
  "metadata": {
   "kernelspec": {
-   "display_name": "Python 3.11.1 64-bit",
+   "display_name": "Python 3 (ipykernel)",
    "language": "python",
    "name": "python3"
   },
@@ -146,7 +246,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.11.1"
+   "version": "3.9.1"
   },
   "vscode": {
    "interpreter": {