File size: 1,290 Bytes
8f80245 2dccc9b 8f80245 2dccc9b 8f80245 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 |
from flow_modules.aiflows.PlanGeneratorFlowModule import PlanGeneratorAtomicFlow
class NewPlanGenFlow(PlanGeneratorAtomicFlow):
"""
This flow module is responsible for generating a new plan based on the previous plan and the given requirements.
This flow is inherited from PlanGeneratorAtomicFlow. (https://huggingface.co/aiflows/PlanGeneratorFlowModule)
*Input Interface Non Initialized*:
- `goal`
- `old_plan`
*Input Interface Initialized*:
- `goal`
- `old_plan`
*Output Interface*:
- `new_plan`
*Configuration Parameters*:
- `system_message_prompt_template`: The template of the system message.
- `init_human_message_prompt_template`: The template of the init user message.
- `human_message_prompt_template`: The template of the user message.
"""
def __init__(self, **kwargs):
"""
Initialization of the flow module.
:param kwargs: The configuration parameters of the flow module.
"""
super().__init__(**kwargs)
self.hint_for_model = """
Make sure your response is in the following format:
Response Format:
{
"new_plan": "A new plan, refined from previous plan, with given requirements",
}
""" |