Update MemoryWritingAtomicFlow.py

MemoryWritingAtomicFlow.py

@@ -9,18 +9,21 @@ class MemoryWritingAtomicFlow(AtomicFlow):
         assert any(item in input_data for item in self.supported_memories), "no memories to write"
 
     def _call(self, input_data: Dict[str, Any]):
-        if "summary" in input_data:
-            assert "logs" in input_data["memory_files"], "there is summary to write, but no logs file in memory_files"
-            logs_file_location = input_data["memory_files"]["logs"]
-            summary_to_write = input_data["summary"]
-            with open(logs_file_location, 'a') as file:
-                file.write(summary_to_write + "\n")
+        try:
+            if "summary" in input_data:
+                assert "logs" in input_data["memory_files"], "there is summary to write, but no logs file in memory_files"
+                logs_file_location = input_data["memory_files"]["logs"]
+                summary_to_write = input_data["summary"]
+                with open(logs_file_location, 'a') as file:
+                    file.write(summary_to_write + "\n")
+            return {"MemWrite_output": "Memory Write was successful"}
+        except Exception as e:
+            return {"MemWrite_output": str(e)}
+            
 
     def run(
             self,
             input_data: Dict[str, Any]
     ):
         self._check_input(input_data)
-        self._call(input_data)
-        response = {}
-        response["MemWrite_output"] = "Memory Write was successful"
+        return self._call(input_data)