Update Planner_JarvisFlow.py
Browse files- Planner_JarvisFlow.py +10 -0
Planner_JarvisFlow.py
CHANGED
@@ -4,6 +4,16 @@ from flow_modules.Tachi67.PlanWriterFlowModule import PlanWriterFlow
|
|
4 |
from aiflows.base_flows import CircularFlow
|
5 |
|
6 |
class Planner_JarvisFlow(PlanWriterFlow):
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
7 |
# TODO: generalize this flow to avoid code duplication
|
8 |
@CircularFlow.output_msg_payload_processor
|
9 |
def detect_finish_or_continue(self, output_payload: Dict[str, Any], src_flow) -> Dict[str, Any]:
|
|
|
4 |
from aiflows.base_flows import CircularFlow
|
5 |
|
6 |
class Planner_JarvisFlow(PlanWriterFlow):
|
7 |
+
"""This flow inherits from PlanWriterFlow (https://huggingface.co/Tachi67/PlanWriterFlowModule), and is used to generate a plan for Jarvis.
|
8 |
+
*Input Interface*:
|
9 |
+
- `goal` (str): the goal of the planner, the goal comes from the user's query when calling Jarvis.
|
10 |
+
- `memory_files` (dict): a dictionary of memory files, the keys are the names of the memory files, the values are the locations of the memory files.
|
11 |
+
|
12 |
+
*Output Interfaces*:
|
13 |
+
- `plan` (str): the generated plan, the plan string will be written to the plan file and returned to the flow state of the Jarvis flow.
|
14 |
+
- `summary` (str): the summary of the planner.
|
15 |
+
- `status` (str): the status of the planner, can be "finished" or "unfinished".
|
16 |
+
"""
|
17 |
# TODO: generalize this flow to avoid code duplication
|
18 |
@CircularFlow.output_msg_payload_processor
|
19 |
def detect_finish_or_continue(self, output_payload: Dict[str, Any], src_flow) -> Dict[str, Any]:
|