Merge pull request #31 from seiflotfy/fix-keyerror

axiomhq · Jan 30, 2023 · 4171918 · 4171918
2 parents 4f43785 + e6c0e79
commit 4171918
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/handler.py b/handler.py
@@ -179,7 +179,8 @@ def lambda_handler(event: dict, context=None):
                 lambda_data = msg
 
         if lambda_data is not None:
-            ev.update({aws_fields["serviceName"]: lambda_data})
+            if "serviceName" in aws_fields:
+                ev.update({aws_fields["serviceName"]: lambda_data})
 
         events.append(ev)