Update CtrlExMem_CoderFlow.py
Browse files- 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
|