44 lines
2 KiB
JavaScript
44 lines
2 KiB
JavaScript
|
"use strict";
|
||
|
|
Object.defineProperty(exports, "__esModule", { value: true });
|
||
|
|
exports.getPlannerChatPrompt = exports.DEFAULT_STEP_EXECUTOR_HUMAN_CHAT_MESSAGE_TEMPLATE = exports.PLANNER_SYSTEM_PROMPT_MESSAGE_TEMPLATE = void 0;
|
||
|
|
const prompts_1 = require("@langchain/core/prompts");
|
||
|
|
exports.PLANNER_SYSTEM_PROMPT_MESSAGE_TEMPLATE = [
|
||
|
|
`Let's first understand the problem and devise a plan to solve the problem.`,
|
||
|
|
`Please output the plan starting with the header "Plan:"`,
|
||
|
|
`followed by a numbered list of steps.`,
|
||
|
|
`Please make the plan the minimum number of steps required`,
|
||
|
|
`to answer the query or complete the task accurately and precisely.`,
|
||
|
|
`You have a set of tools at your disposal to help you with this task:`,
|
||
|
|
"",
|
||
|
|
"{toolStrings}",
|
||
|
|
"",
|
||
|
|
`You must consider these tools when coming up with your plan.`,
|
||
|
|
`If the task is a question, the final step in the plan must be the following: "Given the above steps taken,`,
|
||
|
|
`please respond to the original query."`,
|
||
|
|
`At the end of your plan, say "<END_OF_PLAN>"`,
|
||
|
|
].join(" ");
|
||
|
|
exports.DEFAULT_STEP_EXECUTOR_HUMAN_CHAT_MESSAGE_TEMPLATE = `Previous steps: {previous_steps}
|
||
|
|
|
||
|
|
Current objective: {current_step}
|
||
|
|
|
||
|
|
{agent_scratchpad}
|
||
|
|
|
||
|
|
You may extract and combine relevant data from your previous steps when responding to me.`;
|
||
|
|
/**
|
||
|
|
* Add the tool descriptions to the planning system prompt in
|
||
|
|
* order to get a better suited plan that makes efficient use
|
||
|
|
* of the tools
|
||
|
|
* @param tools the tools available to the `planner`
|
||
|
|
* @returns
|
||
|
|
*/
|
||
|
|
const getPlannerChatPrompt = async (tools) => {
|
||
|
|
const toolStrings = tools
|
||
|
|
.map((tool) => `${tool.name}: ${tool.description}`)
|
||
|
|
.join("\n");
|
||
|
|
return /* #__PURE__ */ prompts_1.ChatPromptTemplate.fromMessages([
|
||
|
|
prompts_1.SystemMessagePromptTemplate.fromTemplate(exports.PLANNER_SYSTEM_PROMPT_MESSAGE_TEMPLATE),
|
||
|
|
prompts_1.HumanMessagePromptTemplate.fromTemplate(`{input}`),
|
||
|
|
]).partial({ toolStrings });
|
||
|
|
};
|
||
|
|
exports.getPlannerChatPrompt = getPlannerChatPrompt;
|