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": {