fix: lookup default sg when using existing vpc

aws-samples · Jun 13, 2024 · 1418b5c · 1418b5c
1 parent 7a75d45
commit 1418b5c
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/lib/model-interfaces/idefics/index.ts b/lib/model-interfaces/idefics/index.ts
@@ -39,13 +39,13 @@ export class IdeficsInterface extends Construct {
     // in order to avoid using signed URLs and run out of input tokens
     // with the idefics model
     const defaultSecurityGroup = (props.config.vpc?.vpcId && props.config.vpc.vpcDefaultSecurityGroup) ?
-        props.config.vpc.vpcDefaultSecurityGroup : props.shared.vpc.vpcDefaultSecurityGroup
+        props.config.vpc.vpcDefaultSecurityGroup : props.shared.vpc.vpcDefaultSecurityGroup;
 
-    const vpcDefaultSecurityGroup = ec2.SecurityGroup.fromSecurityGroupId(
+    const vpcDefaultSecurityGroup = defaultSecurityGroup ? ec2.SecurityGroup.fromSecurityGroupId(
         this,
         'VPCDefaultSecurityGroup',
         defaultSecurityGroup
-    );
+    ) : ec2.SecurityGroup.fromLookupByName(this, 'VPCDefaultSecurityGroup', 'default', props.shared.vpc);
 
     const vpcEndpoint = props.shared.vpc.addInterfaceEndpoint(
       "PrivateApiEndpoint",