try fix summary

Author: neagualexa

index.py

@@ -1,10 +1,14 @@
 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, context):
+def handler(event: JsonType, context):
     """
     Lambda handler function
     """
@@ -45,7 +49,7 @@ def handler(event, context):
     # Create a response
     response = {
         "statusCode": 200,
-        "body": chatbot_response
+        "body": json.dumps(chatbot_response)
     }
 
     # Log the response for debugging purposes

src/module.py

@@ -6,11 +6,13 @@
 try:
     from .agents.utils.parse_json_context_to_prompt import parse_json_to_prompt
     from .agents.base_agent.base_agent import invoke_base_agent
+    from .agents.utils.types import JsonType
 except ImportError:
     from src.agents.utils.parse_json_context_to_prompt import parse_json_to_prompt
     from src.agents.base_agent.base_agent import invoke_base_agent
+    from src.agents.utils.types import JsonType
 
-def chat_module(message: Any, params: Params) -> Result:
+def chat_module(message: Any, params: Params) -> JsonType:
     """
     Function used by student to converse with a chatbot.
     ---