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",
              }
        """