30 lines
1.4 KiB
JavaScript
30 lines
1.4 KiB
JavaScript
import { PromptTemplate } from "@langchain/core/prompts";
|
|
import { LLMChain } from "../../chains/llm_chain.js";
|
|
/** Chain to prioritize tasks. */
|
|
export class TaskPrioritizationChain extends LLMChain {
|
|
static lc_name() {
|
|
return "TaskPrioritizationChain";
|
|
}
|
|
/**
|
|
* Static method to create a new TaskPrioritizationChain from a
|
|
* BaseLanguageModel. It generates a prompt using the PromptTemplate class
|
|
* and the task prioritization template, and returns a new instance of
|
|
* TaskPrioritizationChain.
|
|
* @param fields Object with fields used to initialize the chain, excluding the prompt.
|
|
* @returns A new instance of TaskPrioritizationChain.
|
|
*/
|
|
static fromLLM(fields) {
|
|
const taskPrioritizationTemplate = `You are a task prioritization AI tasked with cleaning the formatting of ` +
|
|
`and reprioritizing the following tasks: {task_names}.` +
|
|
` Consider the ultimate objective of your team: {objective}.` +
|
|
` Do not remove any tasks. Return the result as a numbered list, like:` +
|
|
` #. First task` +
|
|
` #. Second task` +
|
|
` Start the task list with number {next_task_id}.`;
|
|
const prompt = new PromptTemplate({
|
|
template: taskPrioritizationTemplate,
|
|
inputVariables: ["task_names", "next_task_id", "objective"],
|
|
});
|
|
return new TaskPrioritizationChain({ prompt, ...fields });
|
|
}
|
|
}
|