38 lines
1.7 KiB
JavaScript
38 lines
1.7 KiB
JavaScript
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 });
|
|
}
|
|
}
|