Merge pull request #417 from aws-samples/lvn

feat: add exception handler
aws-samples · Oct 28, 2024 · dab8084 · dab8084
2 parents 186f193 + b00de06
commit dab8084
Show file tree

Hide file tree

Showing 3 changed files with 213 additions and 152 deletions.
diff --git a/source/lambda/online/common_logic/common_utils/response_utils.py b/source/lambda/online/common_logic/common_utils/response_utils.py
@@ -71,7 +71,7 @@ def stream_response(event_body:dict, response:dict):
     ws_connection_id = event_body["ws_connection_id"]
     custom_message_id = event_body["custom_message_id"]
     answer = response["answer"]
-    figure = response.get("ddb_additional_kwargs").get("figure")
+    figure = response.get("ddb_additional_kwargs", {}).get("figure")
     if isinstance(answer, str):
         answer = iter([answer])