Update ReplanningFlow.py
Browse files- ReplanningFlow.py +14 -0
ReplanningFlow.py
CHANGED
|
@@ -4,6 +4,20 @@ from aiflows.base_flows import CircularFlow
|
|
| 4 |
from flow_modules.Tachi67.PlanWriterFlowModule import PlanWriterFlow
|
| 5 |
|
| 6 |
class ReplanningFlow(PlanWriterFlow):
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 7 |
def _on_reach_max_round(self):
|
| 8 |
self._state_update_dict({
|
| 9 |
"plan": "The maximum amount of rounds was reached before the model generated the plan.",
|
|
|
|
| 4 |
from flow_modules.Tachi67.PlanWriterFlowModule import PlanWriterFlow
|
| 5 |
|
| 6 |
class ReplanningFlow(PlanWriterFlow):
|
| 7 |
+
"""This flow inherits from PlanWriterFlow.
|
| 8 |
+
By changing prompts and injecting proper information to the controller and the PlanGenerator, we are able to achieve the replanning.
|
| 9 |
+
|
| 10 |
+
*Input Interface*:
|
| 11 |
+
- `goal` (str): information on the old plan (e.g. what is wrong)
|
| 12 |
+
- `plan` (str): the old plan
|
| 13 |
+
- `plan_file_location` (str): the location of the old plan file
|
| 14 |
+
|
| 15 |
+
*Output Interface*:
|
| 16 |
+
- `plan` (str): the new plan
|
| 17 |
+
- `status`: "finished" or "unfinished"
|
| 18 |
+
- `summary` (str): summary of the flow, will be written to the log file of the caller flow.
|
| 19 |
+
- `result` (str): result of the flow, will be passed to the controller of the caller flow.
|
| 20 |
+
"""
|
| 21 |
def _on_reach_max_round(self):
|
| 22 |
self._state_update_dict({
|
| 23 |
"plan": "The maximum amount of rounds was reached before the model generated the plan.",
|