Update lambda.py

Author: lmangani

lambda/lambda.py

@@ -1,16 +1,15 @@
-import os
 import json
 import chdb
-        
+
 def lambda_handler(event, context):
-    json_region = os.environ['AWS_REGION']
     query = event['query'] or "SELECT version()"
     format = event['default_format'] or "JSONCompact"
     res = chdb.query(query, format)
+    out = json.loads(res.data())
     return {
         "statusCode": 200,
         "headers": {
             "Content-Type": "application/json"
         },
-        "body": json.dumps(res.get_memview().tobytes())
+        "body": json.dumps(out)
     }