fix(bedrock): fix agent version issue (#441)

awslabs · May 9, 2024 · 77aa63c · 77aa63c
1 parent 8c1aed5
commit 77aa63c
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/lambda/bedrock-custom-resources/custom_resources/bedrock_agent_alias.py b/lambda/bedrock-custom-resources/custom_resources/bedrock_agent_alias.py
@@ -166,7 +166,10 @@ def list_unused_agent_versions(agent_id: str, session: boto3.Session) -> List[in
 
 def get_version(routing_configuration: List[Dict]) -> str:
     if len(routing_configuration) == 1:
-        return routing_configuration[0]["agentVersion"]
+        try:
+            return routing_configuration[0]["agentVersion"]
+        except KeyError:
+            return None
     else:
         return None