Yes, it should work fine.
I recently ported the Node.js RESTful API to Lambda and did not need to change the KMS code.
You just need to make sure that the role your Lambda function has has access rights to the key that you configure with AWS for use with encrypted / decrypted calls.