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",