Tachi67 commited on
Commit
8cf8798
·
1 Parent(s): e56cc11

Update CtrlExMem_CoderFlow.py

Browse files
Files changed (1) hide show
  1. CtrlExMem_CoderFlow.py +6 -0
CtrlExMem_CoderFlow.py CHANGED
@@ -41,6 +41,12 @@ class CtrlExMem_CoderFlow(CtrlExMemFlow):
41
  output_payload["command_args"]["plan_file_location"] = fetched_state["memory_files"]["plan"]
42
  output_payload["command_args"]["plan"] = fetched_state["plan"]
43
  return output_payload
 
 
 
 
 
 
44
 
45
  else:
46
  return output_payload
 
41
  output_payload["command_args"]["plan_file_location"] = fetched_state["memory_files"]["plan"]
42
  output_payload["command_args"]["plan"] = fetched_state["plan"]
43
  return output_payload
44
+
45
+ elif command == "run_code":
46
+ keys_to_fetch_from_state = ["memory_files"]
47
+ fetched_state = self._fetch_state_attributes_by_keys(keys=keys_to_fetch_from_state)
48
+ output_payload["command_args"]["memory_files"] = fetched_state["memory_files"]
49
+ return output_payload
50
 
51
  else:
52
  return output_payload