handle multiple response messages

bento-platform · Jan 9, 2024 · 849812b · 849812b
1 parent b384e98
commit 849812b
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/bento_beacon/utils/beacon_response.py b/bento_beacon/utils/beacon_response.py
@@ -10,10 +10,14 @@ def init_response_data():
     g.response_data = {}
     g.response_info = {}
 
-
-# TODO: handle multiple messages
 def add_info_to_response(info):
-    g.response_info["message"] = info
+    add_message({"description": info, "level": "info"}) 
+
+
+def add_message(message_obj):
+    messages = g.response_info.get("messages", [])
+    messages.append(message_obj)
+    g.response_info["messages"] = messages
 
 
 def add_stats_to_response(ids):