From 12f5e2ba26fef08ca4172a554847fc23438e2d98 Mon Sep 17 00:00:00 2001 From: Robert Tinn Date: Fri, 13 Jun 2025 14:34:15 +0100 Subject: [PATCH 1/2] Update elastic search cookbook to use responses API --- .../elasticsearch-semantic-search.ipynb | 28 ++++++++----------- registry.yaml | 4 +-- 2 files changed, 14 insertions(+), 18 deletions(-) diff --git a/examples/vector_databases/elasticsearch/elasticsearch-semantic-search.ipynb b/examples/vector_databases/elasticsearch/elasticsearch-semantic-search.ipynb index dacf58b54d..b5efb2a1d3 100644 --- a/examples/vector_databases/elasticsearch/elasticsearch-semantic-search.ipynb +++ b/examples/vector_databases/elasticsearch/elasticsearch-semantic-search.ipynb @@ -37,14 +37,14 @@ }, { "cell_type": "code", - "execution_count": 2, + "execution_count": null, "id": "8c304b93", "metadata": {}, "outputs": [], "source": [ "# install packages\n", "\n", - "!python3 -m pip install -qU openai pandas wget elasticsearch\n", + "! python3 -m pip install -qU openai pandas wget elasticsearch\n", "\n", "# import modules\n", "\n", @@ -54,7 +54,7 @@ "import zipfile\n", "import pandas as pd\n", "import json\n", - "import openai" + "from openai import OpenAI" ] }, { @@ -321,25 +321,21 @@ }, { "cell_type": "code", - "execution_count": 13, + "execution_count": null, "id": "57385c69", "metadata": {}, "outputs": [], "source": [ - "# Get OpenAI API key\n", - "OPENAI_API_KEY = getpass(\"Enter OpenAI API key\")\n", - "\n", - "# Set API key\n", - "openai.api_key = OPENAI_API_KEY\n", - "\n", - "# Define model\n", - "EMBEDDING_MODEL = \"text-embedding-3-small\"\n", + "# Create OpenAI client\n", + "openai_client = OpenAI()\n", "\n", "# Define question\n", "question = 'Is the Atlantic the biggest ocean in the world?'\n", "\n", - "# Create embedding\n", - "question_embedding = openai.Embedding.create(input=question, model=EMBEDDING_MODEL)\n" + "question_embedding = openai_client.embeddings.create(\n", + " input=question,\n", + " model=\"text-embedding-3-small\"\n", + ")" ] }, { @@ -383,7 +379,7 @@ }, { "cell_type": "code", - "execution_count": 14, + "execution_count": null, "id": "fc834fdd", "metadata": {}, "outputs": [ @@ -764,7 +760,7 @@ " index = \"wikipedia_vector_index\",\n", " knn={\n", " \"field\": \"content_vector\",\n", - " \"query_vector\": question_embedding[\"data\"][0][\"embedding\"],\n", + " \"query_vector\": question_embedding.data[0].embedding,\n", " \"k\": 10,\n", " \"num_candidates\": 100\n", " }\n", diff --git a/registry.yaml b/registry.yaml index c821bd077c..6cfb8048a0 100644 --- a/registry.yaml +++ b/registry.yaml @@ -861,12 +861,12 @@ - title: Semantic search using Elasticsearch and OpenAI path: examples/vector_databases/elasticsearch/elasticsearch-semantic-search.ipynb - date: 2023-08-29 + date: 2025-06-13 authors: - leemthompo tags: - embeddings - - completions + - responses - title: Using Hologres as a vector database for OpenAI embeddings path: >- From 6e700865d4f6efb1b43029adad8dc61994f8ce8d Mon Sep 17 00:00:00 2001 From: Robert Tinn Date: Mon, 16 Jun 2025 15:53:59 +0100 Subject: [PATCH 2/2] Change registry to only have embeddings tag --- registry.yaml | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/registry.yaml b/registry.yaml index 6cfb8048a0..6d8b6ae325 100644 --- a/registry.yaml +++ b/registry.yaml @@ -861,12 +861,11 @@ - title: Semantic search using Elasticsearch and OpenAI path: examples/vector_databases/elasticsearch/elasticsearch-semantic-search.ipynb - date: 2025-06-13 + date: 2023-08-29 authors: - leemthompo tags: - embeddings - - responses - title: Using Hologres as a vector database for OpenAI embeddings path: >-