fix for deepseek notebook

Signed-off-by: Xun Zhang <xunzh@amazon.com>

Author: Zhangxunmt

docs/tutorials/aws/AIConnectorHelper.ipynb

@@ -46,7 +46,7 @@
     "                    domain_url = domain_info['Endpoints']['vpc']\n",
     "                else:\n",
     "                    domain_url = next(iter(domain_info['Endpoints'].values()))\n",
-    "                # Non-VPC domain case\n",
+    "            # Non-VPC domain case\n",
     "            elif 'Endpoint' in domain_info:\n",
     "                domain_url = domain_info['Endpoint']\n",
     "            else:\n",

docs/tutorials/aws/DeepSeek_demo_notebook_for_RAG.ipynb

@@ -65,8 +65,22 @@
     "        try:\n",
     "            response = self.opensearch_client.describe_elasticsearch_domain(DomainName=domain_name)\n",
     "            domain_info = response['DomainStatus']\n",
-    "            domain_url = domain_info['Endpoint']\n",
     "            domain_arn = domain_info['ARN']\n",
+    "\n",
+    "            # Check if domain has VPC endpoints\n",
+    "            if 'Endpoints' in domain_info:\n",
+    "                # VPC domain case\n",
+    "                if 'vpc' in domain_info['Endpoints']:\n",
+    "                    domain_url = domain_info['Endpoints']['vpc']\n",
+    "                else:\n",
+    "                    domain_url = next(iter(domain_info['Endpoints'].values()))\n",
+    "            # Non-VPC domain case\n",
+    "            elif 'Endpoint' in domain_info:\n",
+    "                domain_url = domain_info['Endpoint']\n",
+    "            else:\n",
+    "                print(f\"No endpoint found for domain '{domain_name}'\")\n",
+    "                return None, None\n",
+    "\n",
     "            return f'https://{domain_url}', domain_arn\n",
     "        except self.opensearch_client.exceptions.ResourceNotFoundException:\n",
     "            print(f\"Domain '{domain_name}' not found.\")\n",