Class InitialTaskCreationChain

bmtools/agent/BabyagiTools.py:64–77

Chain to generates tasks.

        return cls(prompt=prompt, llm=llm, verbose=verbose)
    
class InitialTaskCreationChain(LLMChain):
    """Chain to generates tasks."""

    @classmethod
    def from_llm(cls, llm: BaseLLM, verbose: bool = True) -> LLMChain:
        """Get the response parser."""
        task_creation_template = (
            "You are a planner who is an expert at coming up with a todo list for a given objective. For a simple objective, do not generate a complex todo list. Generate the first (only one) task needed to do for this objective: {objective}"
        )
        prompt = PromptTemplate(
            template=task_creation_template,
            input_variables=["objective"],
        )
        return cls(prompt=prompt, llm=llm, verbose=verbose)
    
class TaskPrioritizationChain(LLMChain):
    """Chain to prioritize tasks."""