diff --git a/docs/src/notebooks/code_qa_bot.ipynb b/docs/src/notebooks/code_qa_bot.ipynb index de787aca..5ff18a0c 100644 --- a/docs/src/notebooks/code_qa_bot.ipynb +++ b/docs/src/notebooks/code_qa_bot.ipynb @@ -44,15 +44,14 @@ "metadata": {}, "outputs": [], "source": [ - "import openai\n", + "from openai import OpenAI\n", "import os\n", "\n", "# Configuring the environment variable OPENAI_API_KEY\n", "if \"OPENAI_API_KEY\" not in os.environ:\n", - " # OR set the key here as a variable\n", - " openai.api_key = \"sk-...\"\n", - " \n", - "assert len(openai.Model.list()[\"data\"]) > 0" + " os.environ[\"OPENAI_API_KEY\"] = \"sk-...\"\n", + "client = OpenAI()\n", + "assert len(client.models.list().data) > 0" ] }, { diff --git a/docs/src/notebooks/youtube_transcript_search.ipynb b/docs/src/notebooks/youtube_transcript_search.ipynb index b39a9c8e..e3cac08a 100644 --- a/docs/src/notebooks/youtube_transcript_search.ipynb +++ b/docs/src/notebooks/youtube_transcript_search.ipynb @@ -27,11 +27,11 @@ "output_type": "stream", "text": [ "\n", - "\u001B[1m[\u001B[0m\u001B[34;49mnotice\u001B[0m\u001B[1;39;49m]\u001B[0m\u001B[39;49m A new release of pip is available: \u001B[0m\u001B[31;49m23.0\u001B[0m\u001B[39;49m -> \u001B[0m\u001B[32;49m23.1.1\u001B[0m\n", - "\u001B[1m[\u001B[0m\u001B[34;49mnotice\u001B[0m\u001B[1;39;49m]\u001B[0m\u001B[39;49m To update, run: \u001B[0m\u001B[32;49mpip install --upgrade pip\u001B[0m\n", + "\u001b[1m[\u001b[0m\u001b[34;49mnotice\u001b[0m\u001b[1;39;49m]\u001b[0m\u001b[39;49m A new release of pip is available: \u001b[0m\u001b[31;49m23.0\u001b[0m\u001b[39;49m -> \u001b[0m\u001b[32;49m23.1.1\u001b[0m\n", + "\u001b[1m[\u001b[0m\u001b[34;49mnotice\u001b[0m\u001b[1;39;49m]\u001b[0m\u001b[39;49m To update, run: \u001b[0m\u001b[32;49mpip install --upgrade pip\u001b[0m\n", "\n", - "\u001B[1m[\u001B[0m\u001B[34;49mnotice\u001B[0m\u001B[1;39;49m]\u001B[0m\u001B[39;49m A new release of pip is available: \u001B[0m\u001B[31;49m23.0\u001B[0m\u001B[39;49m -> \u001B[0m\u001B[32;49m23.1.1\u001B[0m\n", - "\u001B[1m[\u001B[0m\u001B[34;49mnotice\u001B[0m\u001B[1;39;49m]\u001B[0m\u001B[39;49m To update, run: \u001B[0m\u001B[32;49mpip install --upgrade pip\u001B[0m\n" + "\u001b[1m[\u001b[0m\u001b[34;49mnotice\u001b[0m\u001b[1;39;49m]\u001b[0m\u001b[39;49m A new release of pip is available: \u001b[0m\u001b[31;49m23.0\u001b[0m\u001b[39;49m -> \u001b[0m\u001b[32;49m23.1.1\u001b[0m\n", + "\u001b[1m[\u001b[0m\u001b[34;49mnotice\u001b[0m\u001b[1;39;49m]\u001b[0m\u001b[39;49m To update, run: \u001b[0m\u001b[32;49mpip install --upgrade pip\u001b[0m\n" ] } ], @@ -206,15 +206,16 @@ "metadata": {}, "outputs": [], "source": [ - "import openai\n", + "from openai import OpenAI\n", "import os\n", "\n", "# Configuring the environment variable OPENAI_API_KEY\n", "if \"OPENAI_API_KEY\" not in os.environ:\n", " # OR set the key here as a variable\n", - " openai.api_key = \"sk-...\"\n", + " os.environ[\"OPENAI_API_KEY\"] = \"sk-...\"\n", " \n", - "assert len(openai.Model.list()[\"data\"]) > 0" + "client = OpenAI()\n", + "assert len(client.models.list().data) > 0" ] }, { @@ -234,8 +235,8 @@ "outputs": [], "source": [ "def embed_func(c): \n", - " rs = openai.Embedding.create(input=c, engine=\"text-embedding-ada-002\")\n", - " return [record[\"embedding\"] for record in rs[\"data\"]]" + " rs = client.embeddings.create(input=c, model=\"text-embedding-ada-002\")\n", + " return [rs.data[0].embedding]" ] }, { @@ -536,9 +537,8 @@ ], "source": [ "def complete(prompt):\n", - " # query text-davinci-003\n", - " res = openai.Completion.create(\n", - " engine='text-davinci-003',\n", + " res = client.completions.create(\n", + " model='text-davinci-003',\n", " prompt=prompt,\n", " temperature=0,\n", " max_tokens=400,\n", @@ -547,7 +547,7 @@ " presence_penalty=0,\n", " stop=None\n", " )\n", - " return res['choices'][0]['text'].strip()\n", + " return res.choices[0].text\n", "\n", "# check that it works\n", "query = \"who was the 12th person on the moon and when did they land?\"\n",