Merge branch 'deploy/prototype' into 4313-deployment-override

chat/src/handlers/chat.py

@@ -1,21 +1,26 @@
 import traceback
-from chat.src.event_config import EventConfig
-from src.helpers.response import prepare_response
+from handlers.event_config import EventConfig
+from helpers.response import prepare_response
 
 
-def handler(event={}, _context={}):
-    config = EventConfig(event)
-    socket = config.setup_websocket()
-    if config.is_logged_in and socket is not None:
+def handler(event, _context):
+    try:
+        config = EventConfig(event)
+        config.setup_websocket()
+
+        if not config.is_logged_in:
+            config.socket.send({"type": "error", "message": "Unauthorized"})
+            return {"statusCode": 401, "body": "Unauthorized"}
+
         if config.debug_mode:
-            socket.send(config.debug_message())
+            config.socket.send(config.debug_message())
 
         config.setup_llm_request()
         final_response = prepare_response(config)
-        socket.send(final_response)
+        config.socket.send(final_response)
         return {"statusCode": 200}
-    elif socket is not None:
-        socket.send({"type": "error", "message": "Unauthorized"})
-        return {"statusCode": 401, "body": "Unauthorized"}
-    else:
-        return {"statusCode": 500, "body": "Internal Server Error"}
+    except Exception as err:
+        error_message = traceback.format_exc()
+        config.socket.send(error_message)
+        print(event)
+        raise err

chat/template.yaml

@@ -192,7 +192,7 @@ Resources:
         - x86_64
       Layers:
         - !Ref ChatDependencies
-      MemorySize: 128  
+      MemorySize: 1024
       Handler: handlers/chat.handler
       Timeout: 300
       Environment: