|
37 | 37 | }, |
38 | 38 | { |
39 | 39 | "cell_type": "code", |
40 | | - "execution_count": 2, |
| 40 | + "execution_count": null, |
41 | 41 | "id": "8c304b93", |
42 | 42 | "metadata": {}, |
43 | 43 | "outputs": [], |
44 | 44 | "source": [ |
45 | 45 | "# install packages\n", |
46 | 46 | "\n", |
47 | | - "!python3 -m pip install -qU openai pandas wget elasticsearch\n", |
| 47 | + "! python3 -m pip install -qU openai pandas wget elasticsearch\n", |
48 | 48 | "\n", |
49 | 49 | "# import modules\n", |
50 | 50 | "\n", |
|
54 | 54 | "import zipfile\n", |
55 | 55 | "import pandas as pd\n", |
56 | 56 | "import json\n", |
57 | | - "import openai" |
| 57 | + "from openai import OpenAI" |
58 | 58 | ] |
59 | 59 | }, |
60 | 60 | { |
|
321 | 321 | }, |
322 | 322 | { |
323 | 323 | "cell_type": "code", |
324 | | - "execution_count": 13, |
| 324 | + "execution_count": null, |
325 | 325 | "id": "57385c69", |
326 | 326 | "metadata": {}, |
327 | 327 | "outputs": [], |
328 | 328 | "source": [ |
329 | | - "# Get OpenAI API key\n", |
330 | | - "OPENAI_API_KEY = getpass(\"Enter OpenAI API key\")\n", |
331 | | - "\n", |
332 | | - "# Set API key\n", |
333 | | - "openai.api_key = OPENAI_API_KEY\n", |
334 | | - "\n", |
335 | | - "# Define model\n", |
336 | | - "EMBEDDING_MODEL = \"text-embedding-3-small\"\n", |
| 329 | + "# Create OpenAI client\n", |
| 330 | + "openai_client = OpenAI()\n", |
337 | 331 | "\n", |
338 | 332 | "# Define question\n", |
339 | 333 | "question = 'Is the Atlantic the biggest ocean in the world?'\n", |
340 | 334 | "\n", |
341 | | - "# Create embedding\n", |
342 | | - "question_embedding = openai.Embedding.create(input=question, model=EMBEDDING_MODEL)\n" |
| 335 | + "question_embedding = openai_client.embeddings.create(\n", |
| 336 | + " input=question,\n", |
| 337 | + " model=\"text-embedding-3-small\"\n", |
| 338 | + ")" |
343 | 339 | ] |
344 | 340 | }, |
345 | 341 | { |
|
383 | 379 | }, |
384 | 380 | { |
385 | 381 | "cell_type": "code", |
386 | | - "execution_count": 14, |
| 382 | + "execution_count": null, |
387 | 383 | "id": "fc834fdd", |
388 | 384 | "metadata": {}, |
389 | 385 | "outputs": [ |
|
764 | 760 | " index = \"wikipedia_vector_index\",\n", |
765 | 761 | " knn={\n", |
766 | 762 | " \"field\": \"content_vector\",\n", |
767 | | - " \"query_vector\": question_embedding[\"data\"][0][\"embedding\"],\n", |
| 763 | + " \"query_vector\": question_embedding.data[0].embedding,\n", |
768 | 764 | " \"k\": 10,\n", |
769 | 765 | " \"num_candidates\": 100\n", |
770 | 766 | " }\n", |
|
0 commit comments