Skip to content

Commit d7a1068

Browse files
robtinnshikhar-cyber
authored andcommitted
Update elastic search cookbook to use latest API (#1899)
1 parent 74a1bda commit d7a1068

File tree

2 files changed

+12
-17
lines changed

2 files changed

+12
-17
lines changed

examples/vector_databases/elasticsearch/elasticsearch-semantic-search.ipynb

Lines changed: 12 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -37,14 +37,14 @@
3737
},
3838
{
3939
"cell_type": "code",
40-
"execution_count": 2,
40+
"execution_count": null,
4141
"id": "8c304b93",
4242
"metadata": {},
4343
"outputs": [],
4444
"source": [
4545
"# install packages\n",
4646
"\n",
47-
"!python3 -m pip install -qU openai pandas wget elasticsearch\n",
47+
"! python3 -m pip install -qU openai pandas wget elasticsearch\n",
4848
"\n",
4949
"# import modules\n",
5050
"\n",
@@ -54,7 +54,7 @@
5454
"import zipfile\n",
5555
"import pandas as pd\n",
5656
"import json\n",
57-
"import openai"
57+
"from openai import OpenAI"
5858
]
5959
},
6060
{
@@ -321,25 +321,21 @@
321321
},
322322
{
323323
"cell_type": "code",
324-
"execution_count": 13,
324+
"execution_count": null,
325325
"id": "57385c69",
326326
"metadata": {},
327327
"outputs": [],
328328
"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",
337331
"\n",
338332
"# Define question\n",
339333
"question = 'Is the Atlantic the biggest ocean in the world?'\n",
340334
"\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+
")"
343339
]
344340
},
345341
{
@@ -383,7 +379,7 @@
383379
},
384380
{
385381
"cell_type": "code",
386-
"execution_count": 14,
382+
"execution_count": null,
387383
"id": "fc834fdd",
388384
"metadata": {},
389385
"outputs": [
@@ -764,7 +760,7 @@
764760
" index = \"wikipedia_vector_index\",\n",
765761
" knn={\n",
766762
" \"field\": \"content_vector\",\n",
767-
" \"query_vector\": question_embedding[\"data\"][0][\"embedding\"],\n",
763+
" \"query_vector\": question_embedding.data[0].embedding,\n",
768764
" \"k\": 10,\n",
769765
" \"num_candidates\": 100\n",
770766
" }\n",

registry.yaml

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -880,7 +880,6 @@
880880
- leemthompo
881881
tags:
882882
- embeddings
883-
- completions
884883

885884
- title: Using Hologres as a vector database for OpenAI embeddings
886885
path: >-

0 commit comments

Comments
 (0)