diff --git a/examples/Get_embeddings.ipynb b/examples/Get_embeddings.ipynb index e1b1732..aff14cf 100644 --- a/examples/Get_embeddings.ipynb +++ b/examples/Get_embeddings.ipynb @@ -28,8 +28,11 @@ "source": [ "import openai\n", "\n", - "embedding = openai.Embedding.create(input=\"Sample document text goes here\", engine=\"text-similarity-davinci-001\")['data'][0]['embedding']\n", - "len(embedding)" + "embedding = openai.Embedding.create(\n", + " input=\"Sample document text goes here\",\n", + " engine=\"text-similarity-davinci-001\"\n", + ")[\"data\"][0][\"embedding\"]\n", + "len(embedding)\n" ] }, { @@ -49,21 +52,23 @@ "import openai\n", "from tenacity import retry, wait_random_exponential, stop_after_attempt\n", "\n", + "\n", "@retry(wait=wait_random_exponential(min=1, max=20), stop=stop_after_attempt(6))\n", - "def get_embedding(text: str, engine=\"text-similarity-davinci-001\") -> List[float]:\n", + "def get_embedding(text: str, engine=\"text-similarity-davinci-001\") -> list[float]:\n", "\n", " # replace newlines, which can negatively affect performance.\n", " text = text.replace(\"\\n\", \" \")\n", "\n", " return openai.Embedding.create(input=[text], engine=engine)[\"data\"][0][\"embedding\"]\n", "\n", + "\n", "embedding = get_embedding(\"Sample query text goes here\", engine=\"text-search-ada-query-001\")\n", - "print(len(embedding))" + "print(len(embedding))\n" ] }, { "cell_type": "code", - "execution_count": 53, + "execution_count": 3, "metadata": {}, "outputs": [ { @@ -76,16 +81,14 @@ ], "source": [ "embedding = get_embedding(\"Sample document text goes here\", engine=\"text-search-ada-doc-001\")\n", - "print(len(embedding))" + "print(len(embedding))\n" ] } ], "metadata": { - "interpreter": { - "hash": "be4b5d5b73a21c599de40d6deb1129796d12dc1cc33a738f7bac13269cfcafe8" - }, "kernelspec": { - "display_name": "Python 3.7.3 64-bit ('base': conda)", + "display_name": "Python 3.9.9 ('openai')", + "language": "python", "name": "python3" }, "language_info": { @@ -98,9 +101,14 @@ "name": "python", "nbconvert_exporter": "python", "pygments_lexer": "ipython3", - "version": "3.7.3" + "version": "3.9.9" }, - "orig_nbformat": 4 + "orig_nbformat": 4, + "vscode": { + "interpreter": { + "hash": "365536dcbde60510dc9073d6b991cd35db2d9bac356a11f5b64279a5e6708b97" + } + } }, "nbformat": 4, "nbformat_minor": 2