That's what I'd start with also. Looks like you want to treat

context

as a dict but it's actually an object of a special AWS library type. Some kind of preconversion like

context = dict(context)

might help here but best would be to check the docs on how to best iterate over that context.