Update CtrlExMem_CoderFlow.py

CtrlExMem_CoderFlow.py

@@ -41,6 +41,12 @@ class CtrlExMem_CoderFlow(CtrlExMemFlow):
             output_payload["command_args"]["plan_file_location"] = fetched_state["memory_files"]["plan"]
             output_payload["command_args"]["plan"] = fetched_state["plan"]
             return output_payload
+        
+        elif command == "run_code":
+            keys_to_fetch_from_state = ["memory_files"]
+            fetched_state = self._fetch_state_attributes_by_keys(keys=keys_to_fetch_from_state)
+            output_payload["command_args"]["memory_files"] = fetched_state["memory_files"]
+            return output_payload
 
         else:
             return output_payload