fix equations rendering, get gemini to work (#3)

* try fix max_retries from google call

* try fix llm with retry

* test set versions for packages

* revert testing changes

* update readme to match template

* updated index

* new $ equations prompt

* fix equations generated

Author: neagualexa

index.py

@@ -1,42 +1,38 @@
 import json
 try:
     from .src.module import chat_module
-    from .src.agents.utils.types import JsonType
 except ImportError:
     from src.module import chat_module
-    from src.agents.utils.types import JsonType
 
-def handler(event: JsonType, context):
+def handler(event, 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", None)
+    params = event.get("params", None)
 
     try:
         chatbot_response = chat_module(message, params)
@@ -49,7 +45,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,8 +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.
 ---
 """