import { PromptTemplate } from "@langchain/core/prompts";
import { LLMChain } from "../../chains/llm_chain.js";
/** Chain to generate tasks. */
export class TaskCreationChain extends LLMChain {
    static lc_name() {
        return "TaskCreationChain";
    }
    /**
     * Creates a new TaskCreationChain instance. It takes an object of type
     * LLMChainInput as input, omitting the 'prompt' field. It uses the
     * PromptTemplate class to create a new prompt based on the task creation
     * template and the input variables. The new TaskCreationChain instance is
     * then created with this prompt and the remaining fields from the input
     * object.
     * @param fields An object of type LLMChainInput, omitting the 'prompt' field.
     * @returns A new instance of TaskCreationChain.
     */
    static fromLLM(fields) {
        const taskCreationTemplate = `You are an task creation AI that uses the result of an execution agent` +
            ` to create new tasks with the following objective: {objective},` +
            ` The last completed task has the result: {result}.` +
            ` This result was based on this task description: {task_description}.` +
            ` These are incomplete tasks: {incomplete_tasks}.` +
            ` Based on the result, create new tasks to be completed` +
            ` by the AI system that do not overlap with incomplete tasks.` +
            ` Return the tasks as an array.`;
        const prompt = new PromptTemplate({
            template: taskCreationTemplate,
            inputVariables: [
                "result",
                "task_description",
                "incomplete_tasks",
                "objective",
            ],
        });
        return new TaskCreationChain({ prompt, ...fields });
    }
}