handle result structure without body

[haider-zada]

Description

In this scenario, we are expecting a body key in the result object, but what if the result doesn't have a body key or if the result is a simple boolean? I actually, encountered a situation where a Lambda function was returning a bool, which caused the Lambda function to crash, leading to timeouts.

To address this, I added exception handling. If the result object contains a body key, it will function as expected. However, if the body key is absent, the Lambda function won't crash, preventing timeouts.

Motivation and Context

This will resolve the issue where, if the result object doesn't have a body key, or if the result is a string or boolean, it won't crash. The exception will be handled smoothly.

How Has This Been Tested?

I have added this change to the invoke.py and then ran the same lambda function, which resolved the issue and I was able to run the lambda functions.

Screenshots (if appropriate):

[guga-grigolia]

Hi @haider-zada I just came with this issue as well, when I am returning 204 response code, Flask automatically removes body from response and this line throws an exception. My approach would have been to check:
if hasattr(result, 'body') and isinstance(result['body'], bytes): instead of catching exception.

[haider-zada]

@guga-grigolia Thanks for your suggestion here, I have updated the PR

[haider-zada]

@dherault Could you please review this PR when you have some time.
Thank You!

[dherault]

Thanks for the PR.
Do you want it merged in v13 or v14 (serverless v4)?

[haider-zada]

@dherault I am using version 13.6.0 but it will definitely help many using version 14. But for me you can merge it in 13.6.0