|
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