From 02ef1b6c3287195f65264fe7c9273057f5fd0f43 Mon Sep 17 00:00:00 2001 From: Jun Yamog Date: Thu, 24 Aug 2023 15:04:24 +1200 Subject: [PATCH] Minor fixes: add missing dependency, use the env var for deployment --- demo-python/code/azure-search-vector-python-sample.ipynb | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/demo-python/code/azure-search-vector-python-sample.ipynb b/demo-python/code/azure-search-vector-python-sample.ipynb index 4a785df..e2b9e96 100644 --- a/demo-python/code/azure-search-vector-python-sample.ipynb +++ b/demo-python/code/azure-search-vector-python-sample.ipynb @@ -19,7 +19,9 @@ "source": [ "! pip install azure-search-documents --pre\n", "! pip install openai\n", - "! pip install python-dotenv" + "! pip install python-dotenv\n", + "! pip install tenacity\n", + "! pip install openai[datalib]" ] }, { @@ -71,6 +73,7 @@ "openai.api_key = os.getenv(\"AZURE_OPENAI_API_KEY\") \n", "openai.api_base = os.getenv(\"AZURE_OPENAI_ENDPOINT\") \n", "openai.api_version = os.getenv(\"AZURE_OPENAI_API_VERSION\") \n", + "embedding_deployment_model = os.getenv(\"AZURE_OPENAI_EMBEDDING_DEPLOYED_MODEL\")\n", "credential = AzureKeyCredential(key)" ] }, @@ -99,7 +102,7 @@ "# Function to generate embeddings for title and content fields, also used for query embeddings\n", "def generate_embeddings(text):\n", " response = openai.Embedding.create(\n", - " input=text, engine=\"text-embedding-ada-002\")\n", + " input=text, engine=embedding_deployment_model)\n", " embeddings = response['data'][0]['embedding']\n", " return embeddings\n", "\n",