How to set context in AWS Lambda handler?

My code works fine with this code:

const {dialogflow} = require('actions-on-google');
const app = dialogflow();
// bunch of app.intent()
exports.handler = app;

Now I want to set context.callbackWaitsForEmptyEventLoop in AWS Lambda handler function in order to make my backend code work, and I did some search and tried this:

const {dialogflow} = require('actions-on-google');
const app = dialogflow();
// bunch of app.intent()
exports.handler = async function(event, context) {
    logger.info("got event: " + event)
    context.callbackWaitsForEmptyEventLoop = false;
    try {
        let body = JSON.parse(event.body);
        let headers = event.headers;
        let response = await app.handler(body, headers)
        logger.info(response)
        return response
    } catch (err) {
        logger.info("err: " + err)
    }
}; 

The code above provides a good response, but dialogflow received the following:

Webhook call failed. Error: 502 Bad Gateway

The documents only give the line exports.fulfillment = app which is not informative about how to set the content inside.

Can someone please help me with this? Any help is appreciated.

[Canain]

Try to see if this works:

app.middleware((conv, framework) => {
  if (framework.lambda) {
    framework.lambda.context.callbackWaitsForEmptyEventLoop = false
  }
})

// ... intent code

exports.handler = app

[Canain]

Hi @luomanke currently that code isn’t implemented yet in the library. You would have to use something like

exports.handler = async (event, context, callback) => {
  const body = event.body || event;
  const headers = event.headers || context.request.headers || [];
  const result = await app.handler(body, headers);
  if (result.status >= 200 && result.status < 300) {
    callback(null, result.body);
  } else {
    callback(JSON.stringify(result));
  }
}

The code in the documentation is for if you have AWS Lambda API Gateway/HTTP Proxy integration enabled.