Skip to content

Commit 1ac9077

Browse files
committed
fix the tutorial in AIConnectorHelper when fetching domain_url
Signed-off-by: Xun Zhang <xunzh@amazon.com>
1 parent f6a25f5 commit 1ac9077

File tree

1 file changed

+14
-1
lines changed

1 file changed

+14
-1
lines changed

docs/tutorials/aws/AIConnectorHelper.ipynb

Lines changed: 14 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -37,9 +37,22 @@
3737
" domain_info = response['DomainStatus']\n",
3838
"\n",
3939
" # Extract the domain URL and ARN\n",
40-
" domain_url = domain_info['Endpoint']\n",
4140
" domain_arn = domain_info['ARN']\n",
4241
"\n",
42+
" # Check if domain has VPC endpoints\n",
43+
" if 'Endpoints' in domain_info:\n",
44+
" # VPC domain case\n",
45+
" if 'vpc' in domain_info['Endpoints']:\n",
46+
" domain_url = domain_info['Endpoints']['vpc']\n",
47+
" else:\n",
48+
" domain_url = next(iter(domain_info['Endpoints'].values()))\n",
49+
" # Non-VPC domain case\n",
50+
" elif 'Endpoint' in domain_info:\n",
51+
" domain_url = domain_info['Endpoint']\n",
52+
" else:\n",
53+
" print(f\"No endpoint found for domain '{domain_name}'\")\n",
54+
" return None, None\n",
55+
"\n",
4356
" return f'https://{domain_url}', domain_arn\n",
4457
"\n",
4558
" except opensearch_client.exceptions.ResourceNotFoundException:\n",

0 commit comments

Comments
 (0)