fix expressions rendering

Author: neagualexa

index.py

@@ -9,34 +9,32 @@
 def handler(event: JsonType, context):
     """
     Lambda handler function
-    Args:
-        event (JsonType): The AWS Lambda event received by the gateway.
-        context (Any): The AWS Lambda context object.
-
     """
     # Log the input event for debugging purposes
-    print("Received event:", json.dumps(event, indent=2))
+    # print("Received event:", " ".join(json.dumps(event, indent=2).splitlines()))
 
-    if "body" not in event:
-        return {
-            "statusCode": 400,
-            "body": "Missing 'body' key in event. Please confirm the key in the json body."
-        }
-    body = json.loads(event["body"])
-        
-    if "message" not in body:
+    if "body" in event:
+        try:
+            event = json.loads(event["body"])
+        except json.JSONDecodeError:
+            return {
+                "statusCode": 400,
+                "body": "Invalid JSON format in the body or body not found. Please check the input."
+            }
+
+    if "message" not in event:
         return {
             "statusCode": 400,
             "body": "Missing 'message' key in event. Please confirm the key in the json body."
         }
-    if "params" not in body:
+    if "params" not in event:
         return {
             "statusCode": 400,
             "body": "Missing 'params' key in event. Please confirm the key in the json body. Make sure it contains the necessary conversation_id."
         }
 
-    message = body["message"]
-    params = body["params"]
+    message = event.get("message")
+    params = event.get("params")
 
     try:
         chatbot_response = chat_module(message, params)
@@ -49,7 +47,10 @@ def handler(event: JsonType, context):
     # Create a response
     response = {
         "statusCode": 200,
-        "body": json.dumps(chatbot_response)
+        "body": chatbot_response
     }
 
+    # Log the response for debugging purposes
+    print("Returning response:", " ".join(json.dumps(response, indent=2).splitlines()))
+
     return response

src/agents/utils/prompt_context_templates.py

@@ -73,7 +73,9 @@ def format_question_header(
 - Description: {content}
 {duration_text}
 
-> Note: Mathematical equations are in KaTeX format, preserve them the same. Use British English spellings.
+> Note: Mathematical equations are in KaTeX format, preserve them the same. Ensure mathematical equations are surrounded by one '$' for in-line equations and '$$' for block equations. 
+Example: '$E=mc^2$' or '$$E=mc^2$$'.
+Use British English spellings.
 
 ---
 """

src/agents/utils/testbench_agents.py

@@ -27,7 +27,7 @@
       STEP 2: Extract the parameters from the JSON
     """
     # NOTE: #### This is the testing message!! #####
-    message = "Hi" 
+    message = "Hi, how do I solve this problem?" 
     # NOTE: ########################################
 
     # replace "mock" in the message and conversation history with the actual message