Update endpoint request handling and validation

app.py

@@ -5,6 +5,7 @@ or
 """
 import ast
 from json import JSONDecodeError
+from logging import getLogger
 import mathactive.microlessons.num_one as num_one_quiz
 import os
 import sentry_sdk
@@ -26,8 +27,10 @@ from pydantic import BaseModel
 from dotenv import load_dotenv
 load_dotenv()
 
+log = getLogger(__name__)
+
 sentry_sdk.init(
-    dsn=os.environ.get('SENTRY_DNS'),
+    dsn=os.environ.get('SENTRY_DSN'),
 
     # Set traces_sample_rate to 1.0 to capture 100%
     # of transactions for performance monitoring.
@@ -148,7 +151,7 @@ def intent_classification_ep(content: Text = None):
     content = {"message": ml_response}
     return JSONResponse(content=content)
 
-
+import json
 @app.post("/nlu")
 async def evaluate_user_message_with_nlu_api(request: Request):
     """ Calls nlu evaluation and returns the nlu_response
@@ -161,11 +164,20 @@ async def evaluate_user_message_with_nlu_api(request: Request):
       {'type':'integer', 'data': '8', 'confidence': 0}
       {'type':'sentiment', 'data': 'negative', 'confidence': 0.99}
     """
+    log.info(f'Received request: {request}')
+    log.info(f'Request header: {request.headers}')
+    log.info(f'Request body: {request.body()}')
+
     try:
         data_dict = await request.json()
     except JSONDecodeError:
+        log.error(f'Request.json failed: {dir(request)}')
         data_dict = {}
-    message_data = data_dict.get('message_data', {})
+    message_data = data_dict.get('message_data')
+    
+    if not message_data:
+        log.error(f'Data_dict: {data_dict}')
+        message_data = data_dict.get('message', {})
     nlu_response = evaluate_message_with_nlu(message_data)
     return JSONResponse(content=nlu_response)
 

scripts/make_request.py

@@ -81,7 +81,7 @@ run_full_nlu_endpoint_payload_test(b'{"message_data": {"author_id": "57787919091
 run_full_nlu_endpoint_payload_test(b'{"message_data": {"author_id": "@event.message._vnd.v1.chat.owner", "author_type": "@event.message._vnd.v1.author.type", "contact_uuid": "@event.message._vnd.v1.chat.contact_uuid", "message_body": "@event.message.text.body", "message_direction": "@event.message._vnd.v1.direction", "message_id": "@event.message.id", "message_inserted_at": "@event.message._vnd.v1.chat.inserted_at", "message_updated_at": "@event.message._vnd.v1.chat.updated_at"}}')
 
 # Case: Wrong payload object
-run_full_nlu_endpoint_payload_test(b'{"message_data": {"_vnd": {"v1": {"author": {"id": 54327547257, "name": "Jin", "type": "OWNER"}, "card_uuid": None, "chat": {"assigned_to": None, "contact_uuid": "f7889-f78dfgb798-f786ah89g7-f78f9a", "inserted_at": "2023-03-28T13:21:47.581221Z", "owner": "+43789789146", "permalink": "", "state": "OPEN", "state_reason": "Re-opened by inbound message.", "unread_count": 97, "updated_at": "2023-04-07T21:05:27.389948Z", "uuid": "dfg9a78-d76a786dghas-78d9fga789g-a78d69a9"}, "direction": "inbound", "faq_uuid": None, "in_reply_to": None, "inserted_at": "2023-04-07T21:05:27.368580Z", "labels": [], "last_status": None, "last_status_timestamp": None, "on_fallback_channel": False, "rendered_content": None, "uuid": "hf78s7s89b-789fb68d9fg-789fb789dfb-f79sfb789"}}, "from": 5475248689, "id": "SBDE4zgAAy7887sfdT35SHFS", "text": {"body": 1000}, "timestamp": 1680901527, "type": "text"}, "type": "message"}')
+run_full_nlu_endpoint_payload_test(b'{"message_data": {"_vnd": {"v1": {"author": {"id": 54327547257, "name": "Jin", "type": "OWNER"}, "card_uuid": None, "chat": {"assigned_to": None, "contact_uuid": "f7889-f78dfgb798-f786ah89g7-f78f9a", "inserted_at": "2023-03-28T13:21:47.581221Z", "owner": "+43789789146", "permalink": "", "state": "OPEN", "state_reason": "Re-opened by inbound message.", "unread_count": 97, "updated_at": "2023-04-07T21:05:27.389948Z", "uuid": "dfg9a78-d76a786dghas-78d9fga789g-a78d69a9"}, "direction": "inbound", "faq_uuid": None, "in_reply_to": None, "inserted_at": "2023-04-07T21:05:27.368580Z", "labels": [], "last_status": None, "last_status_timestamp": None, "on_fallback_channel": False, "rendered_content": None, "uuid": "hf78s7s89b-789fb68d9fg-789fb789dfb-f79sfb789"}}, "from": 5475248689, "id": "SBDE4zgAAy7887sfdT35SHFS", "text": {"body": 1000}, "timestamp": 1680901527, "type": "text"}, "type": "message"}')"
 
 
 # run_simulated_request('intent-classification', 'exit')