chore(agent): update adddependency to higher lvl cdk method

krokoko · Jan 24, 2025 · 95b44ef · 95b44ef
1 parent 4e1da7c
commit 95b44ef
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/cdk-lib/bedrock/agents/agent.ts b/src/cdk-lib/bedrock/agents/agent.ts
@@ -378,8 +378,6 @@ export class Agent extends AgentBase {
           },
         }),
       });
-      // add the appropriate permissions to use the FM
-      this.foundationModel.grantInvoke(this.role);
     }
     // ------------------------------------------------------
     // Set Lazy Props initial values
@@ -441,7 +439,9 @@ export class Agent extends AgentBase {
     // Add explicit dependency between the agent resource and the agent's role default policy
     // See https://github.com/awslabs/generative-ai-cdk-constructs/issues/899
     if (!props.existingRole) {
-      this.__resource.node.addDependency(this.role.node.findChild('DefaultPolicy'));
+      // add the appropriate permissions to use the FM
+      const grant = this.foundationModel.grantInvoke(this.role);
+      grant.applyBefore(this.__resource);
     }
 
     this.testAlias = AgentAlias.fromAttributes(this, 'DefaultAlias', {