NewPlanGenFlow.py

import json
from copy import deepcopy
from typing import Any, Dict
from flow_modules.Tachi67.PlanGeneratorFlowModule import PlanGeneratorAtomicFlow


class NewPlanGenFlow(PlanGeneratorAtomicFlow):
    def __init__(self, **kwargs):
        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",
              }
        """