Added code to create bedrock vpc endpoint as the models list timesout

aws-samples · Jun 19, 2024 · 75f5704 · 75f5704
1 parent 1418b5c
commit 75f5704
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/lib/shared/index.ts b/lib/shared/index.ts
@@ -149,10 +149,16 @@ export class Shared extends Construct {
           if (props.config.bedrock?.region !== cdk.Stack.of(this).region) {
             throw new Error(`Bedrock is only supported in the same region as the stack when using private website (Bedrock region: ${props.config.bedrock?.region}, Stack region: ${cdk.Stack.of(this).region}).`);
           }
+
           vpc.addInterfaceEndpoint("BedrockEndpoint", {
-            service: new ec2.InterfaceVpcEndpointService('com.amazonaws.'+cdk.Aws.REGION+'.bedrock-runtime', 443),
-            privateDnsEnabled: true
+            service: ec2.InterfaceVpcEndpointAwsService.BEDROCK,
           });
+
+          vpc.addInterfaceEndpoint("BedrockRuntimeEndpoint", {
+            service: ec2.InterfaceVpcEndpointAwsService.BEDROCK_RUNTIME,
+          });
+
+
         }
 
         // Create VPC Endpoint for Kendra