@langchain/classic 1.0.44 → 1.0.46

package/dist/experimental/autogpt/prompt.js.map

@@ -1 +1 @@
-{"version":3,"file":"prompt.js","names":[],"sources":["../../../src/experimental/autogpt/prompt.ts"],"sourcesContent":["import type { VectorStoreRetrieverInterface } from \"@langchain/core/vectorstores\";\nimport {\n  BaseChatPromptTemplate,\n  SerializedBasePromptTemplate,\n} from \"@langchain/core/prompts\";\nimport {\n  BaseMessage,\n  HumanMessage,\n  SystemMessage,\n} from \"@langchain/core/messages\";\nimport { PartialValues } from \"@langchain/core/utils/types\";\nimport { getPrompt } from \"./prompt_generator.js\";\nimport { ObjectTool } from \"./schema.js\";\n\n/**\n * Interface for the input parameters of the AutoGPTPrompt class.\n */\nexport interface AutoGPTPromptInput {\n  aiName: string;\n  aiRole: string;\n  tools: ObjectTool[];\n  tokenCounter: (text: string) => Promise<number>;\n  sendTokenLimit?: number;\n}\n\n/**\n * Class used to generate prompts for the AutoGPT model. It takes into\n * account the AI's name, role, tools, token counter, and send token\n * limit. The class also handles the formatting of messages and the\n * construction of the full prompt.\n */\nexport class AutoGPTPrompt\n  extends BaseChatPromptTemplate\n  implements AutoGPTPromptInput\n{\n  aiName: string;\n\n  aiRole: string;\n\n  tools: ObjectTool[];\n\n  tokenCounter: (text: string) => Promise<number>;\n\n  sendTokenLimit: number;\n\n  constructor(fields: AutoGPTPromptInput) {\n    super({ inputVariables: [\"goals\", \"memory\", \"messages\", \"user_input\"] });\n    this.aiName = fields.aiName;\n    this.aiRole = fields.aiRole;\n    this.tools = fields.tools;\n    this.tokenCounter = fields.tokenCounter;\n    this.sendTokenLimit = fields.sendTokenLimit || 4196;\n  }\n\n  _getPromptType() {\n    return \"autogpt\" as const;\n  }\n\n  /**\n   * Constructs the full prompt based on the provided goals.\n   * @param goals An array of goals.\n   * @returns The full prompt as a string.\n   */\n  constructFullPrompt(goals: string[]): string {\n    const promptStart = `Your decisions must always be made independently\n            without seeking user assistance. Play to your strengths\n            as an LLM and pursue simple strategies with no legal complications.\n            If you have completed all your tasks,\n            make sure to use the \"finish\" command.`;\n\n    let fullPrompt = `You are ${this.aiName}, ${this.aiRole}\\n${promptStart}\\n\\nGOALS:\\n\\n`;\n    goals.forEach((goal, index) => {\n      fullPrompt += `${index + 1}. ${goal}\\n`;\n    });\n\n    fullPrompt += `\\n\\n${getPrompt(this.tools)}`;\n    return fullPrompt;\n  }\n\n  /**\n   * Formats the messages based on the provided parameters.\n   * @param goals An array of goals.\n   * @param memory A VectorStoreRetriever instance.\n   * @param messages An array of previous messages.\n   * @param user_input The user's input.\n   * @returns An array of formatted messages.\n   */\n  async formatMessages({\n    goals,\n    memory,\n    messages: previousMessages,\n    user_input,\n  }: {\n    goals: string[];\n    memory: VectorStoreRetrieverInterface;\n    messages: BaseMessage[];\n    user_input: string;\n  }) {\n    const basePrompt = new SystemMessage(this.constructFullPrompt(goals));\n    const timePrompt = new SystemMessage(\n      `The current time and date is ${new Date().toLocaleString(\"en-US\")}`\n    );\n    if (\n      typeof basePrompt.content !== \"string\" ||\n      typeof timePrompt.content !== \"string\"\n    ) {\n      throw new Error(\"Non-string message content is not supported.\");\n    }\n    const usedTokens =\n      (await this.tokenCounter(basePrompt.content)) +\n      (await this.tokenCounter(timePrompt.content));\n    const relevantDocs = await memory.invoke(\n      JSON.stringify(previousMessages.slice(-10))\n    );\n    const relevantMemory = relevantDocs.map(\n      (d: { pageContent: string }) => d.pageContent\n    );\n    let relevantMemoryTokens = await relevantMemory.reduce(\n      async (acc: Promise<number>, doc: string) =>\n        (await acc) + (await this.tokenCounter(doc)),\n      Promise.resolve(0)\n    );\n\n    while (usedTokens + relevantMemoryTokens > 2500) {\n      relevantMemory.pop();\n      relevantMemoryTokens = await relevantMemory.reduce(\n        async (acc: Promise<number>, doc: string) =>\n          (await acc) + (await this.tokenCounter(doc)),\n        Promise.resolve(0)\n      );\n    }\n\n    const contentFormat = `This reminds you of these events from your past:\\n${relevantMemory.join(\n      \"\\n\"\n    )}\\n\\n`;\n    const memoryMessage = new SystemMessage(contentFormat);\n    if (typeof memoryMessage.content !== \"string\") {\n      throw new Error(\"Non-string message content is not supported.\");\n    }\n    const usedTokensWithMemory =\n      usedTokens + (await this.tokenCounter(memoryMessage.content));\n    const historicalMessages: BaseMessage[] = [];\n\n    for (const message of previousMessages.slice(-10).reverse()) {\n      if (typeof message.content !== \"string\") {\n        throw new Error(\"Non-string message content is not supported.\");\n      }\n      const messageTokens = await this.tokenCounter(message.content);\n      if (usedTokensWithMemory + messageTokens > this.sendTokenLimit - 1000) {\n        break;\n      }\n      historicalMessages.unshift(message);\n    }\n\n    const inputMessage = new HumanMessage(user_input);\n    const messages: BaseMessage[] = [\n      basePrompt,\n      timePrompt,\n      memoryMessage,\n      ...historicalMessages,\n      inputMessage,\n    ];\n    return messages;\n  }\n\n  /**\n   * This method is not implemented in the AutoGPTPrompt class and will\n   * throw an error if called.\n   * @param _values Partial values.\n   * @returns Throws an error.\n   */\n  async partial(_values: PartialValues): Promise<BaseChatPromptTemplate> {\n    throw new Error(\"Method not implemented.\");\n  }\n\n  serialize(): SerializedBasePromptTemplate {\n    throw new Error(\"Method not implemented.\");\n  }\n}\n"],"mappings":";;;;;;;;;;AA+BA,IAAa,gBAAb,cACU,uBAEV;CACE;CAEA;CAEA;CAEA;CAEA;CAEA,YAAY,QAA4B;EACtC,MAAM,EAAE,gBAAgB;GAAC;GAAS;GAAU;GAAY;EAAY,EAAE,CAAC;EACvE,KAAK,SAAS,OAAO;EACrB,KAAK,SAAS,OAAO;EACrB,KAAK,QAAQ,OAAO;EACpB,KAAK,eAAe,OAAO;EAC3B,KAAK,iBAAiB,OAAO,kBAAkB;CACjD;CAEA,iBAAiB;EACf,OAAO;CACT;;;;;;CAOA,oBAAoB,OAAyB;EAO3C,IAAI,aAAa,WAAW,KAAK,OAAO,IAAI,KAAK,OAAO;;;;;EACxD,MAAM,SAAS,MAAM,UAAU;GAC7B,cAAc,GAAG,QAAQ,EAAE,IAAI,KAAK;EACtC,CAAC;EAED,cAAc,OAAO,UAAU,KAAK,KAAK;EACzC,OAAO;CACT;;;;;;;;;CAUA,MAAM,eAAe,EACnB,OACA,QACA,UAAU,kBACV,cAMC;EACD,MAAM,aAAa,IAAI,cAAc,KAAK,oBAAoB,KAAK,CAAC;EACpE,MAAM,aAAa,IAAI,cACrB,iDAAgC,IAAI,KAAK,EAAA,CAAE,eAAe,OAAO,GACnE;EACA,IACE,OAAO,WAAW,YAAY,YAC9B,OAAO,WAAW,YAAY,UAE9B,MAAM,IAAI,MAAM,8CAA8C;EAEhE,MAAM,aACH,MAAM,KAAK,aAAa,WAAW,OAAO,IAC1C,MAAM,KAAK,aAAa,WAAW,OAAO;EAI7C,MAAM,kBAAiB,MAHI,OAAO,OAChC,KAAK,UAAU,iBAAiB,MAAM,GAAG,CAAC,CAC5C,EAAA,CACoC,KACjC,MAA+B,EAAE,WACpC;EACA,IAAI,uBAAuB,MAAM,eAAe,OAC9C,OAAO,KAAsB,QAC1B,MAAM,MAAQ,MAAM,KAAK,aAAa,GAAG,GAC5C,QAAQ,QAAQ,CAAC,CACnB;EAEA,OAAO,aAAa,uBAAuB,MAAM;GAC/C,eAAe,IAAI;GACnB,uBAAuB,MAAM,eAAe,OAC1C,OAAO,KAAsB,QAC1B,MAAM,MAAQ,MAAM,KAAK,aAAa,GAAG,GAC5C,QAAQ,QAAQ,CAAC,CACnB;EACF;EAKA,MAAM,gBAAgB,IAAI,cAAc,qDAHmC,eAAe,KACxF,IACF,EAAE,KACmD;EACrD,IAAI,OAAO,cAAc,YAAY,UACnC,MAAM,IAAI,MAAM,8CAA8C;EAEhE,MAAM,uBACJ,aAAc,MAAM,KAAK,aAAa,cAAc,OAAO;EAC7D,MAAM,qBAAoC,CAAC;EAE3C,KAAK,MAAM,WAAW,iBAAiB,MAAM,GAAG,CAAC,CAAC,QAAQ,GAAG;GAC3D,IAAI,OAAO,QAAQ,YAAY,UAC7B,MAAM,IAAI,MAAM,8CAA8C;GAGhE,IAAI,uBAAuB,MADC,KAAK,aAAa,QAAQ,OAAO,IAClB,KAAK,iBAAiB,KAC/D;GAEF,mBAAmB,QAAQ,OAAO;EACpC;EAEA,MAAM,eAAe,IAAI,aAAa,UAAU;EAQhD,OAAO;GANL;GACA;GACA;GACA,GAAG;GACH;EAEY;CAChB;;;;;;;CAQA,MAAM,QAAQ,SAAyD;EACrE,MAAM,IAAI,MAAM,yBAAyB;CAC3C;CAEA,YAA0C;EACxC,MAAM,IAAI,MAAM,yBAAyB;CAC3C;AACF"}
+{"version":3,"file":"prompt.js","names":[],"sources":["../../../src/experimental/autogpt/prompt.ts"],"sourcesContent":["import type { VectorStoreRetrieverInterface } from \"@langchain/core/vectorstores\";\nimport {\n  BaseChatPromptTemplate,\n  SerializedBasePromptTemplate,\n} from \"@langchain/core/prompts\";\nimport {\n  BaseMessage,\n  HumanMessage,\n  SystemMessage,\n} from \"@langchain/core/messages\";\nimport { PartialValues } from \"@langchain/core/utils/types\";\nimport { getPrompt } from \"./prompt_generator.js\";\nimport { ObjectTool } from \"./schema.js\";\n\n/**\n * Interface for the input parameters of the AutoGPTPrompt class.\n */\nexport interface AutoGPTPromptInput {\n  aiName: string;\n  aiRole: string;\n  tools: ObjectTool[];\n  tokenCounter: (text: string) => Promise<number>;\n  sendTokenLimit?: number;\n}\n\n/**\n * Class used to generate prompts for the AutoGPT model. It takes into\n * account the AI's name, role, tools, token counter, and send token\n * limit. The class also handles the formatting of messages and the\n * construction of the full prompt.\n */\nexport class AutoGPTPrompt\n  extends BaseChatPromptTemplate\n  implements AutoGPTPromptInput\n{\n  aiName: string;\n\n  aiRole: string;\n\n  tools: ObjectTool[];\n\n  tokenCounter: (text: string) => Promise<number>;\n\n  sendTokenLimit: number;\n\n  constructor(fields: AutoGPTPromptInput) {\n    super({ inputVariables: [\"goals\", \"memory\", \"messages\", \"user_input\"] });\n    this.aiName = fields.aiName;\n    this.aiRole = fields.aiRole;\n    this.tools = fields.tools;\n    this.tokenCounter = fields.tokenCounter;\n    this.sendTokenLimit = fields.sendTokenLimit || 4196;\n  }\n\n  _getPromptType() {\n    return \"autogpt\" as const;\n  }\n\n  /**\n   * Constructs the full prompt based on the provided goals.\n   * @param goals An array of goals.\n   * @returns The full prompt as a string.\n   */\n  constructFullPrompt(goals: string[]): string {\n    const promptStart = `Your decisions must always be made independently\n            without seeking user assistance. Play to your strengths\n            as an LLM and pursue simple strategies with no legal complications.\n            If you have completed all your tasks,\n            make sure to use the \"finish\" command.`;\n\n    let fullPrompt = `You are ${this.aiName}, ${this.aiRole}\\n${promptStart}\\n\\nGOALS:\\n\\n`;\n    goals.forEach((goal, index) => {\n      fullPrompt += `${index + 1}. ${goal}\\n`;\n    });\n\n    fullPrompt += `\\n\\n${getPrompt(this.tools)}`;\n    return fullPrompt;\n  }\n\n  /**\n   * Formats the messages based on the provided parameters.\n   * @param goals An array of goals.\n   * @param memory A VectorStoreRetriever instance.\n   * @param messages An array of previous messages.\n   * @param user_input The user's input.\n   * @returns An array of formatted messages.\n   */\n  async formatMessages({\n    goals,\n    memory,\n    messages: previousMessages,\n    user_input,\n  }: {\n    goals: string[];\n    memory: VectorStoreRetrieverInterface;\n    messages: BaseMessage[];\n    user_input: string;\n  }) {\n    const basePrompt = new SystemMessage(this.constructFullPrompt(goals));\n    const timePrompt = new SystemMessage(\n      `The current time and date is ${new Date().toLocaleString(\"en-US\")}`\n    );\n    if (\n      typeof basePrompt.content !== \"string\" ||\n      typeof timePrompt.content !== \"string\"\n    ) {\n      throw new Error(\"Non-string message content is not supported.\");\n    }\n    const usedTokens =\n      (await this.tokenCounter(basePrompt.content)) +\n      (await this.tokenCounter(timePrompt.content));\n    const relevantDocs = await memory.invoke(\n      JSON.stringify(previousMessages.slice(-10))\n    );\n    const relevantMemory = relevantDocs.map(\n      (d: { pageContent: string }) => d.pageContent\n    );\n    let relevantMemoryTokens = await relevantMemory.reduce(\n      async (acc: Promise<number>, doc: string) =>\n        (await acc) + (await this.tokenCounter(doc)),\n      Promise.resolve(0)\n    );\n\n    while (usedTokens + relevantMemoryTokens > 2500) {\n      relevantMemory.pop();\n      relevantMemoryTokens = await relevantMemory.reduce(\n        async (acc: Promise<number>, doc: string) =>\n          (await acc) + (await this.tokenCounter(doc)),\n        Promise.resolve(0)\n      );\n    }\n\n    const contentFormat = `This reminds you of these events from your past:\\n${relevantMemory.join(\n      \"\\n\"\n    )}\\n\\n`;\n    const memoryMessage = new SystemMessage(contentFormat);\n    if (typeof memoryMessage.content !== \"string\") {\n      throw new Error(\"Non-string message content is not supported.\");\n    }\n    const usedTokensWithMemory =\n      usedTokens + (await this.tokenCounter(memoryMessage.content));\n    const historicalMessages: BaseMessage[] = [];\n\n    for (const message of previousMessages.slice(-10).reverse()) {\n      if (typeof message.content !== \"string\") {\n        throw new Error(\"Non-string message content is not supported.\");\n      }\n      const messageTokens = await this.tokenCounter(message.content);\n      if (usedTokensWithMemory + messageTokens > this.sendTokenLimit - 1000) {\n        break;\n      }\n      historicalMessages.unshift(message);\n    }\n\n    const inputMessage = new HumanMessage(user_input);\n    const messages: BaseMessage[] = [\n      basePrompt,\n      timePrompt,\n      memoryMessage,\n      ...historicalMessages,\n      inputMessage,\n    ];\n    return messages;\n  }\n\n  /**\n   * This method is not implemented in the AutoGPTPrompt class and will\n   * throw an error if called.\n   * @param _values Partial values.\n   * @returns Throws an error.\n   */\n  async partial(_values: PartialValues): Promise<BaseChatPromptTemplate> {\n    throw new Error(\"Method not implemented.\");\n  }\n\n  serialize(): SerializedBasePromptTemplate {\n    throw new Error(\"Method not implemented.\");\n  }\n}\n"],"mappings":";;;;;;;;;;AA+BA,IAAa,gBAAb,cACU,uBAEV;CACE;CAEA;CAEA;CAEA;CAEA;CAEA,YAAY,QAA4B;EACtC,MAAM,EAAE,gBAAgB;GAAC;GAAS;GAAU;GAAY;EAAY,EAAE,CAAC;EACvE,KAAK,SAAS,OAAO;EACrB,KAAK,SAAS,OAAO;EACrB,KAAK,QAAQ,OAAO;EACpB,KAAK,eAAe,OAAO;EAC3B,KAAK,iBAAiB,OAAO,kBAAkB;CACjD;CAEA,iBAAiB;EACf,OAAO;CACT;;;;;;CAOA,oBAAoB,OAAyB;EAO3C,IAAI,aAAa,WAAW,KAAK,OAAO,IAAI,KAAK,OAAO;;;;;EACxD,MAAM,SAAS,MAAM,UAAU;GAC7B,cAAc,GAAG,QAAQ,EAAE,IAAI,KAAK;EACtC,CAAC;EAED,cAAc,OAAO,UAAU,KAAK,KAAK;EACzC,OAAO;CACT;;;;;;;;;CAUA,MAAM,eAAe,EACnB,OACA,QACA,UAAU,kBACV,cAMC;EACD,MAAM,aAAa,IAAI,cAAc,KAAK,oBAAoB,KAAK,CAAC;EACpE,MAAM,aAAa,IAAI,cACrB,iDAAgC,IAAI,KAAK,EAAA,CAAE,eAAe,OAAO,GACnE;EACA,IACE,OAAO,WAAW,YAAY,YAC9B,OAAO,WAAW,YAAY,UAE9B,MAAM,IAAI,MAAM,8CAA8C;EAEhE,MAAM,aACH,MAAM,KAAK,aAAa,WAAW,OAAO,IAC1C,MAAM,KAAK,aAAa,WAAW,OAAO;EAI7C,MAAM,kBAAiB,MAHI,OAAO,OAChC,KAAK,UAAU,iBAAiB,MAAM,GAAG,CAAC,CAC5C,EAAA,CACoC,KACjC,MAA+B,EAAE,WACpC;EACA,IAAI,uBAAuB,MAAM,eAAe,OAC9C,OAAO,KAAsB,QAC1B,MAAM,MAAQ,MAAM,KAAK,aAAa,GAAG,GAC5C,QAAQ,QAAQ,CAAC,CACnB;EAEA,OAAO,aAAa,uBAAuB,MAAM;GAC/C,eAAe,IAAI;GACnB,uBAAuB,MAAM,eAAe,OAC1C,OAAO,KAAsB,QAC1B,MAAM,MAAQ,MAAM,KAAK,aAAa,GAAG,GAC5C,QAAQ,QAAQ,CAAC,CACnB;EACF;EAEA,MAAM,gBAAgB,qDAAqD,eAAe,KACxF,IACF,EAAE;EACF,MAAM,gBAAgB,IAAI,cAAc,aAAa;EACrD,IAAI,OAAO,cAAc,YAAY,UACnC,MAAM,IAAI,MAAM,8CAA8C;EAEhE,MAAM,uBACJ,aAAc,MAAM,KAAK,aAAa,cAAc,OAAO;EAC7D,MAAM,qBAAoC,CAAC;EAE3C,KAAK,MAAM,WAAW,iBAAiB,MAAM,GAAG,CAAC,CAAC,QAAQ,GAAG;GAC3D,IAAI,OAAO,QAAQ,YAAY,UAC7B,MAAM,IAAI,MAAM,8CAA8C;GAGhE,IAAI,uBAAuB,MADC,KAAK,aAAa,QAAQ,OAAO,IAClB,KAAK,iBAAiB,KAC/D;GAEF,mBAAmB,QAAQ,OAAO;EACpC;EAEA,MAAM,eAAe,IAAI,aAAa,UAAU;EAQhD,OAAO;GANL;GACA;GACA;GACA,GAAG;GACH;EAEY;CAChB;;;;;;;CAQA,MAAM,QAAQ,SAAyD;EACrE,MAAM,IAAI,MAAM,yBAAyB;CAC3C;CAEA,YAA0C;EACxC,MAAM,IAAI,MAAM,yBAAyB;CAC3C;AACF"}

package/dist/experimental/autogpt/prompt_generator.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"prompt_generator.cjs","names":["FINISH_NAME"],"sources":["../../../src/experimental/autogpt/prompt_generator.ts"],"sourcesContent":["import { isInteropZodSchema } from \"@langchain/core/utils/types\";\nimport {\n  type JsonSchema7Type,\n  toJsonSchema,\n} from \"@langchain/core/utils/json_schema\";\nimport { ObjectTool, FINISH_NAME } from \"./schema.js\";\n\n/**\n * Class that generates prompts for generative agents. It maintains a list\n * of constraints, commands, resources, and performance evaluations.\n */\nexport class PromptGenerator {\n  constraints: string[];\n\n  commands: ObjectTool[];\n\n  resources: string[];\n\n  performance_evaluation: string[];\n\n  response_format: object;\n\n  constructor() {\n    this.constraints = [];\n    this.commands = [];\n    this.resources = [];\n    this.performance_evaluation = [];\n    this.response_format = {\n      thoughts: {\n        text: \"thought\",\n        reasoning: \"reasoning\",\n        plan: \"- short bulleted\\n- list that conveys\\n- long-term plan\",\n        criticism: \"constructive self-criticism\",\n        speak: \"thoughts summary to say to user\",\n      },\n      command: { name: \"command name\", args: { \"arg name\": \"value\" } },\n    };\n  }\n\n  /**\n   * Adds a constraint to the list of constraints.\n   * @param constraint The constraint to add.\n   * @returns void\n   */\n  add_constraint(constraint: string): void {\n    this.constraints.push(constraint);\n  }\n\n  /**\n   * Adds a tool to the list of commands.\n   * @param tool The tool to add.\n   * @returns void\n   */\n  add_tool(tool: ObjectTool): void {\n    this.commands.push(tool);\n  }\n\n  _generate_command_string(tool: ObjectTool): string {\n    let output = `\"${tool.name}\": ${tool.description}`;\n    const jsonSchema = (\n      isInteropZodSchema(tool.schema) ? toJsonSchema(tool.schema) : tool.schema\n    ) as { properties?: Record<string, JsonSchema7Type> } | undefined;\n    output += `, args json schema: ${JSON.stringify(jsonSchema?.properties)}`;\n    return output;\n  }\n\n  /**\n   * Adds a resource to the list of resources.\n   * @param resource The resource to add.\n   * @returns void\n   */\n  add_resource(resource: string): void {\n    this.resources.push(resource);\n  }\n\n  /**\n   * Adds a performance evaluation to the list of performance evaluations.\n   * @param evaluation The performance evaluation to add.\n   * @returns void\n   */\n  add_performance_evaluation(evaluation: string): void {\n    this.performance_evaluation.push(evaluation);\n  }\n\n  // oxlint-disable-next-line @typescript-eslint/no-explicit-any\n  _generate_numbered_list(items: any[], item_type = \"list\"): string {\n    if (item_type === \"command\") {\n      const command_strings = items.map(\n        (item, i) => `${i + 1}. ${this._generate_command_string(item)}`\n      );\n      const finish_description =\n        \"use this to signal that you have finished all your objectives\";\n      const finish_args =\n        '\"response\": \"final response to let people know you have finished your objectives\"';\n      const finish_string = `${\n        items.length + 1\n      }. ${FINISH_NAME}: ${finish_description}, args: ${finish_args}`;\n      return command_strings.concat([finish_string]).join(\"\\n\");\n    }\n\n    return items.map((item, i) => `${i + 1}. ${item}`).join(\"\\n\");\n  }\n\n  /**\n   * Generates a prompt string that includes the constraints, commands,\n   * resources, performance evaluations, and response format.\n   * @returns A string representing the prompt.\n   */\n  generate_prompt_string(): string {\n    const formatted_response_format = JSON.stringify(\n      this.response_format,\n      null,\n      4\n    );\n    const prompt_string =\n      `Constraints:\\n${this._generate_numbered_list(this.constraints)}\\n\\n` +\n      `Commands:\\n${this._generate_numbered_list(\n        this.commands,\n        \"command\"\n      )}\\n\\n` +\n      `Resources:\\n${this._generate_numbered_list(this.resources)}\\n\\n` +\n      `Performance Evaluation:\\n${this._generate_numbered_list(\n        this.performance_evaluation\n      )}\\n\\n` +\n      `You should only respond in JSON format as described below ` +\n      `\\nResponse Format: \\n${formatted_response_format} ` +\n      `\\nEnsure the response can be parsed by Python json.loads`;\n\n    return prompt_string;\n  }\n}\n\n/**\n * Function that generates a prompt string for a given list of tools.\n */\nexport function getPrompt(tools: ObjectTool[]): string {\n  const prompt_generator = new PromptGenerator();\n\n  prompt_generator.add_constraint(\n    \"~4000 word limit for short term memory. \" +\n      \"Your short term memory is short, \" +\n      \"so immediately save important information to files.\"\n  );\n  prompt_generator.add_constraint(\n    \"If you are unsure how you previously did something \" +\n      \"or want to recall past events, \" +\n      \"thinking about similar events will help you remember.\"\n  );\n  prompt_generator.add_constraint(\"No user assistance\");\n  prompt_generator.add_constraint(\n    'Exclusively use the commands listed in double quotes e.g. \"command name\"'\n  );\n\n  for (const tool of tools) {\n    prompt_generator.add_tool(tool);\n  }\n\n  prompt_generator.add_resource(\n    \"Internet access for searches and information gathering.\"\n  );\n  prompt_generator.add_resource(\"Long Term memory management.\");\n  prompt_generator.add_resource(\n    \"GPT-3.5 powered Agents for delegation of simple tasks.\"\n  );\n  prompt_generator.add_resource(\"File output.\");\n\n  prompt_generator.add_performance_evaluation(\n    \"Continuously review and analyze your actions \" +\n      \"to ensure you are performing to the best of your abilities.\"\n  );\n  prompt_generator.add_performance_evaluation(\n    \"Constructively self-criticize your big-picture behavior constantly.\"\n  );\n  prompt_generator.add_performance_evaluation(\n    \"Reflect on past decisions and strategies to refine your approach.\"\n  );\n  prompt_generator.add_performance_evaluation(\n    \"Every command has a cost, so be smart and efficient. \" +\n      \"Aim to complete tasks in the least number of steps.\"\n  );\n\n  const prompt_string = prompt_generator.generate_prompt_string();\n\n  return prompt_string;\n}\n"],"mappings":";;;;;;;;AAWA,IAAa,kBAAb,MAA6B;CAC3B;CAEA;CAEA;CAEA;CAEA;CAEA,cAAc;EACZ,KAAK,cAAc,CAAC;EACpB,KAAK,WAAW,CAAC;EACjB,KAAK,YAAY,CAAC;EAClB,KAAK,yBAAyB,CAAC;EAC/B,KAAK,kBAAkB;GACrB,UAAU;IACR,MAAM;IACN,WAAW;IACX,MAAM;IACN,WAAW;IACX,OAAO;GACT;GACA,SAAS;IAAE,MAAM;IAAgB,MAAM,EAAE,YAAY,QAAQ;GAAE;EACjE;CACF;;;;;;CAOA,eAAe,YAA0B;EACvC,KAAK,YAAY,KAAK,UAAU;CAClC;;;;;;CAOA,SAAS,MAAwB;EAC/B,KAAK,SAAS,KAAK,IAAI;CACzB;CAEA,yBAAyB,MAA0B;EACjD,IAAI,SAAS,IAAI,KAAK,KAAK,KAAK,KAAK;EACrC,MAAM,cAAA,GAAA,4BAAA,mBAAA,CACe,KAAK,MAAM,KAAA,GAAA,kCAAA,aAAA,CAAiB,KAAK,MAAM,IAAI,KAAK;EAErE,UAAU,uBAAuB,KAAK,UAAU,YAAY,UAAU;EACtE,OAAO;CACT;;;;;;CAOA,aAAa,UAAwB;EACnC,KAAK,UAAU,KAAK,QAAQ;CAC9B;;;;;;CAOA,2BAA2B,YAA0B;EACnD,KAAK,uBAAuB,KAAK,UAAU;CAC7C;CAGA,wBAAwB,OAAc,YAAY,QAAgB;EAChE,IAAI,cAAc,WAAW;GAC3B,MAAM,kBAAkB,MAAM,KAC3B,MAAM,MAAM,GAAG,IAAI,EAAE,IAAI,KAAK,yBAAyB,IAAI,GAC9D;GAKA,MAAM,gBAAgB,GACpB,MAAM,SAAS,EAChB,IAAIA,eAAAA,YAAY;GACjB,OAAO,gBAAgB,OAAO,CAAC,aAAa,CAAC,CAAC,CAAC,KAAK,IAAI;EAC1D;EAEA,OAAO,MAAM,KAAK,MAAM,MAAM,GAAG,IAAI,EAAE,IAAI,MAAM,CAAC,CAAC,KAAK,IAAI;CAC9D;;;;;;CAOA,yBAAiC;EAC/B,MAAM,4BAA4B,KAAK,UACrC,KAAK,iBACL,MACA,CACF;EAeA,OAAO,iBAbY,KAAK,wBAAwB,KAAK,WAAW,EAAE,iBAClD,KAAK,wBACjB,KAAK,UACL,SACF,EAAE,kBACa,KAAK,wBAAwB,KAAK,SAAS,EAAE,+BAChC,KAAK,wBAC/B,KAAK,sBACP,EAAE,qFAEsB,0BAA0B;CAItD;AACF;;;;AAKA,SAAgB,UAAU,OAA6B;CACrD,MAAM,mBAAmB,IAAI,gBAAgB;CAE7C,iBAAiB,eACf,8HAGF;CACA,iBAAiB,eACf,yIAGF;CACA,iBAAiB,eAAe,oBAAoB;CACpD,iBAAiB,eACf,4EACF;CAEA,KAAK,MAAM,QAAQ,OACjB,iBAAiB,SAAS,IAAI;CAGhC,iBAAiB,aACf,yDACF;CACA,iBAAiB,aAAa,8BAA8B;CAC5D,iBAAiB,aACf,wDACF;CACA,iBAAiB,aAAa,cAAc;CAE5C,iBAAiB,2BACf,0GAEF;CACA,iBAAiB,2BACf,qEACF;CACA,iBAAiB,2BACf,mEACF;CACA,iBAAiB,2BACf,0GAEF;CAIA,OAFsB,iBAAiB,uBAEpB;AACrB"}
+{"version":3,"file":"prompt_generator.cjs","names":["isInteropZodSchema","toJsonSchema","FINISH_NAME"],"sources":["../../../src/experimental/autogpt/prompt_generator.ts"],"sourcesContent":["import { isInteropZodSchema } from \"@langchain/core/utils/types\";\nimport {\n  type JsonSchema7Type,\n  toJsonSchema,\n} from \"@langchain/core/utils/json_schema\";\nimport { ObjectTool, FINISH_NAME } from \"./schema.js\";\n\n/**\n * Class that generates prompts for generative agents. It maintains a list\n * of constraints, commands, resources, and performance evaluations.\n */\nexport class PromptGenerator {\n  constraints: string[];\n\n  commands: ObjectTool[];\n\n  resources: string[];\n\n  performance_evaluation: string[];\n\n  response_format: object;\n\n  constructor() {\n    this.constraints = [];\n    this.commands = [];\n    this.resources = [];\n    this.performance_evaluation = [];\n    this.response_format = {\n      thoughts: {\n        text: \"thought\",\n        reasoning: \"reasoning\",\n        plan: \"- short bulleted\\n- list that conveys\\n- long-term plan\",\n        criticism: \"constructive self-criticism\",\n        speak: \"thoughts summary to say to user\",\n      },\n      command: { name: \"command name\", args: { \"arg name\": \"value\" } },\n    };\n  }\n\n  /**\n   * Adds a constraint to the list of constraints.\n   * @param constraint The constraint to add.\n   * @returns void\n   */\n  add_constraint(constraint: string): void {\n    this.constraints.push(constraint);\n  }\n\n  /**\n   * Adds a tool to the list of commands.\n   * @param tool The tool to add.\n   * @returns void\n   */\n  add_tool(tool: ObjectTool): void {\n    this.commands.push(tool);\n  }\n\n  _generate_command_string(tool: ObjectTool): string {\n    let output = `\"${tool.name}\": ${tool.description}`;\n    const jsonSchema = (\n      isInteropZodSchema(tool.schema) ? toJsonSchema(tool.schema) : tool.schema\n    ) as { properties?: Record<string, JsonSchema7Type> } | undefined;\n    output += `, args json schema: ${JSON.stringify(jsonSchema?.properties)}`;\n    return output;\n  }\n\n  /**\n   * Adds a resource to the list of resources.\n   * @param resource The resource to add.\n   * @returns void\n   */\n  add_resource(resource: string): void {\n    this.resources.push(resource);\n  }\n\n  /**\n   * Adds a performance evaluation to the list of performance evaluations.\n   * @param evaluation The performance evaluation to add.\n   * @returns void\n   */\n  add_performance_evaluation(evaluation: string): void {\n    this.performance_evaluation.push(evaluation);\n  }\n\n  // oxlint-disable-next-line @typescript-eslint/no-explicit-any\n  _generate_numbered_list(items: any[], item_type = \"list\"): string {\n    if (item_type === \"command\") {\n      const command_strings = items.map(\n        (item, i) => `${i + 1}. ${this._generate_command_string(item)}`\n      );\n      const finish_description =\n        \"use this to signal that you have finished all your objectives\";\n      const finish_args =\n        '\"response\": \"final response to let people know you have finished your objectives\"';\n      const finish_string = `${\n        items.length + 1\n      }. ${FINISH_NAME}: ${finish_description}, args: ${finish_args}`;\n      return command_strings.concat([finish_string]).join(\"\\n\");\n    }\n\n    return items.map((item, i) => `${i + 1}. ${item}`).join(\"\\n\");\n  }\n\n  /**\n   * Generates a prompt string that includes the constraints, commands,\n   * resources, performance evaluations, and response format.\n   * @returns A string representing the prompt.\n   */\n  generate_prompt_string(): string {\n    const formatted_response_format = JSON.stringify(\n      this.response_format,\n      null,\n      4\n    );\n    const prompt_string =\n      `Constraints:\\n${this._generate_numbered_list(this.constraints)}\\n\\n` +\n      `Commands:\\n${this._generate_numbered_list(\n        this.commands,\n        \"command\"\n      )}\\n\\n` +\n      `Resources:\\n${this._generate_numbered_list(this.resources)}\\n\\n` +\n      `Performance Evaluation:\\n${this._generate_numbered_list(\n        this.performance_evaluation\n      )}\\n\\n` +\n      `You should only respond in JSON format as described below ` +\n      `\\nResponse Format: \\n${formatted_response_format} ` +\n      `\\nEnsure the response can be parsed by Python json.loads`;\n\n    return prompt_string;\n  }\n}\n\n/**\n * Function that generates a prompt string for a given list of tools.\n */\nexport function getPrompt(tools: ObjectTool[]): string {\n  const prompt_generator = new PromptGenerator();\n\n  prompt_generator.add_constraint(\n    \"~4000 word limit for short term memory. \" +\n      \"Your short term memory is short, \" +\n      \"so immediately save important information to files.\"\n  );\n  prompt_generator.add_constraint(\n    \"If you are unsure how you previously did something \" +\n      \"or want to recall past events, \" +\n      \"thinking about similar events will help you remember.\"\n  );\n  prompt_generator.add_constraint(\"No user assistance\");\n  prompt_generator.add_constraint(\n    'Exclusively use the commands listed in double quotes e.g. \"command name\"'\n  );\n\n  for (const tool of tools) {\n    prompt_generator.add_tool(tool);\n  }\n\n  prompt_generator.add_resource(\n    \"Internet access for searches and information gathering.\"\n  );\n  prompt_generator.add_resource(\"Long Term memory management.\");\n  prompt_generator.add_resource(\n    \"GPT-3.5 powered Agents for delegation of simple tasks.\"\n  );\n  prompt_generator.add_resource(\"File output.\");\n\n  prompt_generator.add_performance_evaluation(\n    \"Continuously review and analyze your actions \" +\n      \"to ensure you are performing to the best of your abilities.\"\n  );\n  prompt_generator.add_performance_evaluation(\n    \"Constructively self-criticize your big-picture behavior constantly.\"\n  );\n  prompt_generator.add_performance_evaluation(\n    \"Reflect on past decisions and strategies to refine your approach.\"\n  );\n  prompt_generator.add_performance_evaluation(\n    \"Every command has a cost, so be smart and efficient. \" +\n      \"Aim to complete tasks in the least number of steps.\"\n  );\n\n  const prompt_string = prompt_generator.generate_prompt_string();\n\n  return prompt_string;\n}\n"],"mappings":";;;;;;;;AAWA,IAAa,kBAAb,MAA6B;CAC3B;CAEA;CAEA;CAEA;CAEA;CAEA,cAAc;EACZ,KAAK,cAAc,CAAC;EACpB,KAAK,WAAW,CAAC;EACjB,KAAK,YAAY,CAAC;EAClB,KAAK,yBAAyB,CAAC;EAC/B,KAAK,kBAAkB;GACrB,UAAU;IACR,MAAM;IACN,WAAW;IACX,MAAM;IACN,WAAW;IACX,OAAO;GACT;GACA,SAAS;IAAE,MAAM;IAAgB,MAAM,EAAE,YAAY,QAAQ;GAAE;EACjE;CACF;;;;;;CAOA,eAAe,YAA0B;EACvC,KAAK,YAAY,KAAK,UAAU;CAClC;;;;;;CAOA,SAAS,MAAwB;EAC/B,KAAK,SAAS,KAAK,IAAI;CACzB;CAEA,yBAAyB,MAA0B;EACjD,IAAI,SAAS,IAAI,KAAK,KAAK,KAAK,KAAK;EACrC,MAAM,cAAA,GACJA,4BAAAA,mBAAAA,CAAmB,KAAK,MAAM,KAAA,GAAIC,kCAAAA,aAAAA,CAAa,KAAK,MAAM,IAAI,KAAK;EAErE,UAAU,uBAAuB,KAAK,UAAU,YAAY,UAAU;EACtE,OAAO;CACT;;;;;;CAOA,aAAa,UAAwB;EACnC,KAAK,UAAU,KAAK,QAAQ;CAC9B;;;;;;CAOA,2BAA2B,YAA0B;EACnD,KAAK,uBAAuB,KAAK,UAAU;CAC7C;CAGA,wBAAwB,OAAc,YAAY,QAAgB;EAChE,IAAI,cAAc,WAAW;GAC3B,MAAM,kBAAkB,MAAM,KAC3B,MAAM,MAAM,GAAG,IAAI,EAAE,IAAI,KAAK,yBAAyB,IAAI,GAC9D;GAKA,MAAM,gBAAgB,GACpB,MAAM,SAAS,EAChB,IAAIC,eAAAA,YAAY;GACjB,OAAO,gBAAgB,OAAO,CAAC,aAAa,CAAC,CAAC,CAAC,KAAK,IAAI;EAC1D;EAEA,OAAO,MAAM,KAAK,MAAM,MAAM,GAAG,IAAI,EAAE,IAAI,MAAM,CAAC,CAAC,KAAK,IAAI;CAC9D;;;;;;CAOA,yBAAiC;EAC/B,MAAM,4BAA4B,KAAK,UACrC,KAAK,iBACL,MACA,CACF;EAeA,OAAO,iBAbY,KAAK,wBAAwB,KAAK,WAAW,EAAE,iBAClD,KAAK,wBACjB,KAAK,UACL,SACF,EAAE,kBACa,KAAK,wBAAwB,KAAK,SAAS,EAAE,+BAChC,KAAK,wBAC/B,KAAK,sBACP,EAAE,qFAEsB,0BAA0B;CAItD;AACF;;;;AAKA,SAAgB,UAAU,OAA6B;CACrD,MAAM,mBAAmB,IAAI,gBAAgB;CAE7C,iBAAiB,eACf,8HAGF;CACA,iBAAiB,eACf,yIAGF;CACA,iBAAiB,eAAe,oBAAoB;CACpD,iBAAiB,eACf,4EACF;CAEA,KAAK,MAAM,QAAQ,OACjB,iBAAiB,SAAS,IAAI;CAGhC,iBAAiB,aACf,yDACF;CACA,iBAAiB,aAAa,8BAA8B;CAC5D,iBAAiB,aACf,wDACF;CACA,iBAAiB,aAAa,cAAc;CAE5C,iBAAiB,2BACf,0GAEF;CACA,iBAAiB,2BACf,qEACF;CACA,iBAAiB,2BACf,mEACF;CACA,iBAAiB,2BACf,0GAEF;CAIA,OAFsB,iBAAiB,uBAEpB;AACrB"}

package/dist/experimental/chains/violation_of_expectations/violation_of_expectations_chain.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"violation_of_expectations_chain.cjs","names":["BaseChain","JsonOutputFunctionsParser","StringOutputParser","HumanMessage","PREDICT_NEXT_USER_MESSAGE_FUNCTION","PREDICT_NEXT_USER_MESSAGE_PROMPT","PREDICTION_VIOLATIONS_FUNCTION","PREDICTION_VIOLATIONS_PROMPT","GENERATE_REVISED_PREDICTION_PROMPT","GENERATE_FACTS_PROMPT"],"sources":["../../../../src/experimental/chains/violation_of_expectations/violation_of_expectations_chain.ts"],"sourcesContent":["import type { BaseRetrieverInterface } from \"@langchain/core/retrievers\";\nimport { ChatOpenAI } from \"@langchain/openai\";\nimport {\n  BaseMessage,\n  HumanMessage,\n  isBaseMessage,\n} from \"@langchain/core/messages\";\nimport { ChainValues } from \"@langchain/core/utils/types\";\nimport { StringOutputParser } from \"@langchain/core/output_parsers\";\nimport { CallbackManagerForChainRun } from \"@langchain/core/callbacks/manager\";\nimport { JsonOutputFunctionsParser } from \"../../../output_parsers/openai_functions.js\";\nimport { BaseChain, ChainInputs } from \"../../../chains/base.js\";\nimport {\n  GetPredictionViolationsResponse,\n  MessageChunkResult,\n  PREDICTION_VIOLATIONS_FUNCTION,\n  PREDICT_NEXT_USER_MESSAGE_FUNCTION,\n  PredictNextUserMessageResponse,\n} from \"./types.js\";\nimport {\n  GENERATE_FACTS_PROMPT,\n  GENERATE_REVISED_PREDICTION_PROMPT,\n  PREDICTION_VIOLATIONS_PROMPT,\n  PREDICT_NEXT_USER_MESSAGE_PROMPT,\n} from \"./violation_of_expectations_prompt.js\";\n\n/**\n * Interface for the input parameters of the ViolationOfExpectationsChain class.\n */\nexport interface ViolationOfExpectationsChainInput extends ChainInputs {\n  /**\n   * The retriever to use for retrieving stored\n   * thoughts and insights.\n   */\n  retriever: BaseRetrieverInterface;\n  /**\n   * The LLM to use\n   */\n  llm: ChatOpenAI;\n}\n\n/**\n * Chain that generates key insights/facts of a user based on a\n * a chat conversation with an AI.\n */\nexport class ViolationOfExpectationsChain\n  extends BaseChain\n  implements ViolationOfExpectationsChainInput\n{\n  static lc_name() {\n    return \"ViolationOfExpectationsChain\";\n  }\n\n  _chainType(): string {\n    return \"violation_of_expectation_chain\";\n  }\n\n  chatHistoryKey = \"chat_history\";\n\n  thoughtsKey = \"thoughts\";\n\n  get inputKeys() {\n    return [this.chatHistoryKey];\n  }\n\n  get outputKeys() {\n    return [this.thoughtsKey];\n  }\n\n  retriever: BaseRetrieverInterface;\n\n  llm: ChatOpenAI;\n\n  jsonOutputParser: JsonOutputFunctionsParser;\n\n  stringOutputParser: StringOutputParser;\n\n  constructor(fields: ViolationOfExpectationsChainInput) {\n    super(fields);\n    this.retriever = fields.retriever;\n    this.llm = fields.llm;\n    this.jsonOutputParser = new JsonOutputFunctionsParser();\n    this.stringOutputParser = new StringOutputParser();\n  }\n\n  getChatHistoryString(chatHistory: BaseMessage[]): string {\n    return chatHistory\n      .map((chatMessage) => {\n        if (chatMessage._getType() === \"human\") {\n          return `Human: ${chatMessage.content}`;\n        } else if (chatMessage._getType() === \"ai\") {\n          return `AI: ${chatMessage.content}`;\n        } else {\n          return `${chatMessage.content}`;\n        }\n      })\n      .join(\"\\n\");\n  }\n\n  removeDuplicateStrings(strings: Array<string>): Array<string> {\n    return [...new Set(strings)];\n  }\n\n  /**\n   * This method breaks down the chat history into chunks of messages.\n   * Each chunk consists of a sequence of messages ending with an AI message and the subsequent user response, if any.\n   *\n   * @param {BaseMessage[]} chatHistory - The chat history to be chunked.\n   *\n   * @returns {MessageChunkResult[]} An array of message chunks. Each chunk includes a sequence of messages and the subsequent user response.\n   *\n   * @description\n   * The method iterates over the chat history and pushes each message into a temporary array.\n   * When it encounters an AI message, it checks for a subsequent user message.\n   * If a user message is found, it is considered as the user response to the AI message.\n   * If no user message is found after the AI message, the user response is undefined.\n   * The method then pushes the chunk (sequence of messages and user response) into the result array.\n   * This process continues until all messages in the chat history have been processed.\n   */\n  chunkMessagesByAIResponse(chatHistory: BaseMessage[]): MessageChunkResult[] {\n    const newArray: MessageChunkResult[] = [];\n    const tempArray: BaseMessage[] = [];\n\n    chatHistory.forEach((item, index) => {\n      tempArray.push(item);\n      if (item._getType() === \"ai\") {\n        let userResponse: BaseMessage | undefined = chatHistory[index + 1];\n        if (!userResponse || userResponse._getType() !== \"human\") {\n          userResponse = undefined;\n        }\n\n        newArray.push({\n          chunkedMessages: tempArray,\n          userResponse: userResponse\n            ? new HumanMessage(userResponse)\n            : undefined,\n        });\n      }\n    });\n\n    return newArray;\n  }\n\n  /**\n   * This method processes a chat history to generate insights about the user.\n   *\n   * @param {ChainValues} values - The input values for the chain. It should contain a key for chat history.\n   * @param {CallbackManagerForChainRun} [runManager] - Optional callback manager for the chain run.\n   *\n   * @returns {Promise<ChainValues>} A promise that resolves to a list of insights about the user.\n   *\n   * @throws {Error} If the chat history key is not found in the input values or if the chat history is not an array of BaseMessages.\n   *\n   * @description\n   * The method performs the following steps:\n   * 1. Checks if the chat history key is present in the input values and if the chat history is an array of BaseMessages.\n   * 2. Breaks the chat history into chunks of messages.\n   * 3. For each chunk, it generates an initial prediction for the user's next message.\n   * 4. For each prediction, it generates insights and prediction violations, and regenerates the prediction based on the violations.\n   * 5. For each set of messages, it generates a fact/insight about the user.\n   * The method returns a list of these insights.\n   */\n  async _call(\n    values: ChainValues,\n    runManager?: CallbackManagerForChainRun\n  ): Promise<ChainValues> {\n    if (!(this.chatHistoryKey in values)) {\n      throw new Error(`Chat history key ${this.chatHistoryKey} not found`);\n    }\n\n    const chatHistory: unknown[] = values[this.chatHistoryKey];\n\n    const isEveryMessageBaseMessage = chatHistory.every((message) =>\n      isBaseMessage(message)\n    );\n    if (!isEveryMessageBaseMessage) {\n      throw new Error(\"Chat history must be an array of BaseMessages\");\n    }\n\n    const messageChunks = this.chunkMessagesByAIResponse(\n      chatHistory as BaseMessage[]\n    );\n\n    // Generate the initial prediction for every user message.\n    const userPredictions = await Promise.all(\n      messageChunks.map(async (chatHistoryChunk) => ({\n        userPredictions: await this.predictNextUserMessage(\n          chatHistoryChunk.chunkedMessages\n        ),\n        userResponse: chatHistoryChunk.userResponse,\n        runManager,\n      }))\n    );\n\n    // Generate insights, and prediction violations for every user message.\n    // This call also regenerates the prediction based on the violations.\n    const predictionViolations = await Promise.all(\n      userPredictions.map((prediction) =>\n        this.getPredictionViolations({\n          userPredictions: prediction.userPredictions,\n          userResponse: prediction.userResponse,\n          runManager,\n        })\n      )\n    );\n\n    // Generate a fact/insight about the user for every set of messages.\n    const insights = await Promise.all(\n      predictionViolations.map((violation) =>\n        this.generateFacts({\n          userResponse: violation.userResponse,\n          predictions: {\n            revisedPrediction: violation.revisedPrediction,\n            explainedPredictionErrors: violation.explainedPredictionErrors,\n          },\n        })\n      )\n    );\n\n    return {\n      insights,\n    };\n  }\n\n  /**\n   * This method predicts the next user message based on the chat history.\n   *\n   * @param {BaseMessage[]} chatHistory - The chat history based on which the next user message is predicted.\n   * @param {CallbackManagerForChainRun} [runManager] - Optional callback manager for the chain run.\n   *\n   * @returns {Promise<PredictNextUserMessageResponse>} A promise that resolves to the predicted next user message, the user state, and any insights.\n   *\n   * @throws {Error} If the response from the language model does not contain the expected keys: 'userState', 'predictedUserMessage', and 'insights'.\n   */\n  private async predictNextUserMessage(\n    chatHistory: BaseMessage[],\n    runManager?: CallbackManagerForChainRun\n  ): Promise<PredictNextUserMessageResponse> {\n    const messageString = this.getChatHistoryString(chatHistory);\n\n    const llmWithFunctions = this.llm\n      .bindTools([PREDICT_NEXT_USER_MESSAGE_FUNCTION])\n      .withConfig({\n        function_call: { name: PREDICT_NEXT_USER_MESSAGE_FUNCTION.name },\n      });\n\n    const chain = PREDICT_NEXT_USER_MESSAGE_PROMPT.pipe(llmWithFunctions).pipe(\n      this.jsonOutputParser\n    );\n\n    const res = await chain.invoke(\n      {\n        chat_history: messageString,\n      },\n      runManager?.getChild(\"prediction\")\n    );\n\n    if (\n      !(\n        \"userState\" in res &&\n        \"predictedUserMessage\" in res &&\n        \"insights\" in res\n      )\n    ) {\n      throw new Error(`Invalid response from LLM: ${JSON.stringify(res)}`);\n    }\n\n    const predictionResponse = res as PredictNextUserMessageResponse;\n\n    // Query the retriever for relevant insights. Use the generates insights as a query.\n    const retrievedDocs = await this.retrieveRelevantInsights(\n      predictionResponse.insights\n    );\n    const relevantDocs = this.removeDuplicateStrings([\n      ...predictionResponse.insights,\n      ...retrievedDocs,\n    ]);\n\n    return {\n      ...predictionResponse,\n      insights: relevantDocs,\n    };\n  }\n\n  /**\n   * Retrieves relevant insights based on the provided insights.\n   *\n   * @param {Array<string>} insights - An array of insights to be used for retrieving relevant documents.\n   *\n   * @returns {Promise<Array<string>>} A promise that resolves to an array of relevant insights content.\n   */\n  private async retrieveRelevantInsights(\n    insights: Array<string>\n  ): Promise<Array<string>> {\n    // Only extract the first relevant doc from the retriever. We don't need more than one.\n    const relevantInsightsDocuments = await Promise.all(\n      insights.map(async (insight) => {\n        const relevantInsight = await this.retriever.invoke(insight);\n        return relevantInsight[0];\n      })\n    );\n\n    const relevantInsightsContent = relevantInsightsDocuments.map(\n      (document) => document.pageContent\n    );\n\n    return relevantInsightsContent;\n  }\n\n  /**\n   * This method generates prediction violations based on the predicted and actual user responses.\n   * It also generates a revised prediction based on the identified violations.\n   *\n   * @param {Object} params - The parameters for the method.\n   * @param {PredictNextUserMessageResponse} params.userPredictions - The predicted user message, user state, and insights.\n   * @param {BaseMessage} [params.userResponse] - The actual user response.\n   * @param {CallbackManagerForChainRun} [params.runManager] - Optional callback manager for the chain run.\n   *\n   * @returns {Promise<{ userResponse: BaseMessage | undefined; revisedPrediction: string; explainedPredictionErrors: Array<string>; }>} A promise that resolves to an object containing the actual user response, the revised prediction, and the explained prediction errors.\n   *\n   * @throws {Error} If the response from the language model does not contain the expected keys: 'violationExplanation', 'explainedPredictionErrors', and 'accuratePrediction'.\n   */\n  private async getPredictionViolations({\n    userPredictions,\n    userResponse,\n    runManager,\n  }: {\n    userPredictions: PredictNextUserMessageResponse;\n    userResponse?: BaseMessage;\n    runManager?: CallbackManagerForChainRun;\n  }): Promise<GetPredictionViolationsResponse> {\n    const llmWithFunctions = this.llm\n      .bindTools([PREDICTION_VIOLATIONS_FUNCTION])\n      .withConfig({\n        function_call: { name: PREDICTION_VIOLATIONS_FUNCTION.name },\n      });\n\n    const chain = PREDICTION_VIOLATIONS_PROMPT.pipe(llmWithFunctions).pipe(\n      this.jsonOutputParser\n    );\n\n    if (typeof userResponse?.content !== \"string\") {\n      throw new Error(\"This chain does not support non-string model output.\");\n    }\n    const res = (await chain.invoke(\n      {\n        predicted_output: userPredictions.predictedUserMessage,\n        actual_output: userResponse?.content ?? \"\",\n        user_insights: userPredictions.insights.join(\"\\n\"),\n      },\n      runManager?.getChild(\"prediction_violations\")\n    )) as Awaited<{\n      violationExplanation: string;\n      explainedPredictionErrors: Array<string>;\n      accuratePrediction: boolean;\n    }>;\n\n    // Generate a revised prediction based on violations.\n    const revisedPrediction = await this.generateRevisedPrediction({\n      originalPrediction: userPredictions.predictedUserMessage,\n      explainedPredictionErrors: res.explainedPredictionErrors,\n      userInsights: userPredictions.insights,\n      runManager,\n    });\n\n    return {\n      userResponse,\n      revisedPrediction,\n      explainedPredictionErrors: res.explainedPredictionErrors,\n    };\n  }\n\n  /**\n   * This method generates a revised prediction based on the original prediction, explained prediction errors, and user insights.\n   *\n   * @param {Object} params - The parameters for the method.\n   * @param {string} params.originalPrediction - The original prediction made by the model.\n   * @param {Array<string>} params.explainedPredictionErrors - An array of explained prediction errors.\n   * @param {Array<string>} params.userInsights - An array of insights about the user.\n   * @param {CallbackManagerForChainRun} [params.runManager] - Optional callback manager for the chain run.\n   *\n   * @returns {Promise<string>} A promise that resolves to a revised prediction.\n   */\n  private async generateRevisedPrediction({\n    originalPrediction,\n    explainedPredictionErrors,\n    userInsights,\n    runManager,\n  }: {\n    originalPrediction: string;\n    explainedPredictionErrors: Array<string>;\n    userInsights: Array<string>;\n    runManager?: CallbackManagerForChainRun;\n  }): Promise<string> {\n    const revisedPredictionChain = GENERATE_REVISED_PREDICTION_PROMPT.pipe(\n      this.llm\n    ).pipe(this.stringOutputParser);\n\n    const revisedPredictionRes = await revisedPredictionChain.invoke(\n      {\n        prediction: originalPrediction,\n        explained_prediction_errors: explainedPredictionErrors.join(\"\\n\"),\n        user_insights: userInsights.join(\"\\n\"),\n      },\n      runManager?.getChild(\"prediction_revision\")\n    );\n\n    return revisedPredictionRes;\n  }\n\n  /**\n   * This method generates facts or insights about the user based on the revised prediction, explained prediction errors, and the user's response.\n   *\n   * @param {Object} params - The parameters for the method.\n   * @param {BaseMessage} [params.userResponse] - The actual user response.\n   * @param {Object} params.predictions - The revised prediction and explained prediction errors.\n   * @param {string} params.predictions.revisedPrediction - The revised prediction made by the model.\n   * @param {Array<string>} params.predictions.explainedPredictionErrors - An array of explained prediction errors.\n   * @param {CallbackManagerForChainRun} [params.runManager] - Optional callback manager for the chain run.\n   *\n   * @returns {Promise<string>} A promise that resolves to a string containing the generated facts or insights about the user.\n   */\n  private async generateFacts({\n    userResponse,\n    predictions,\n    runManager,\n  }: {\n    userResponse?: BaseMessage;\n    /**\n     * Optional if the prediction was accurate.\n     */\n    predictions: {\n      revisedPrediction: string;\n      explainedPredictionErrors: Array<string>;\n    };\n    runManager?: CallbackManagerForChainRun;\n  }): Promise<string> {\n    const chain = GENERATE_FACTS_PROMPT.pipe(this.llm).pipe(\n      this.stringOutputParser\n    );\n\n    if (typeof userResponse?.content !== \"string\") {\n      throw new Error(\"This chain does not support non-string model output.\");\n    }\n    const res = await chain.invoke(\n      {\n        prediction_violations: predictions.explainedPredictionErrors.join(\"\\n\"),\n        prediction: predictions.revisedPrediction,\n        user_message: userResponse?.content ?? \"\",\n      },\n      runManager?.getChild(\"generate_facts\")\n    );\n\n    return res;\n  }\n\n  /**\n   * Static method that creates a ViolationOfExpectationsChain instance from a\n   * ChatOpenAI and retriever. It also accepts optional options\n   * to customize the chain.\n   *\n   * @param llm The ChatOpenAI instance.\n   * @param retriever The retriever used for similarity search.\n   * @param options Optional options to customize the chain.\n   *\n   * @returns A new instance of ViolationOfExpectationsChain.\n   */\n  static fromLLM(\n    llm: ChatOpenAI,\n    retriever: BaseRetrieverInterface,\n    options?: Partial<\n      Omit<ViolationOfExpectationsChainInput, \"llm\" | \"retriever\">\n    >\n  ): ViolationOfExpectationsChain {\n    return new this({\n      retriever,\n      llm,\n      ...options,\n    });\n  }\n}\n"],"mappings":";;;;;;;;;;;AA6CA,IAAa,+BAAb,cACUA,aAAAA,UAEV;CACE,OAAO,UAAU;EACf,OAAO;CACT;CAEA,aAAqB;EACnB,OAAO;CACT;CAEA,iBAAiB;CAEjB,cAAc;CAEd,IAAI,YAAY;EACd,OAAO,CAAC,KAAK,cAAc;CAC7B;CAEA,IAAI,aAAa;EACf,OAAO,CAAC,KAAK,WAAW;CAC1B;CAEA;CAEA;CAEA;CAEA;CAEA,YAAY,QAA2C;EACrD,MAAM,MAAM;EACZ,KAAK,YAAY,OAAO;EACxB,KAAK,MAAM,OAAO;EAClB,KAAK,mBAAmB,IAAIC,yBAAAA,0BAA0B;EACtD,KAAK,qBAAqB,IAAIC,+BAAAA,mBAAmB;CACnD;CAEA,qBAAqB,aAAoC;EACvD,OAAO,YACJ,KAAK,gBAAgB;GACpB,IAAI,YAAY,SAAS,MAAM,SAC7B,OAAO,UAAU,YAAY;QACxB,IAAI,YAAY,SAAS,MAAM,MACpC,OAAO,OAAO,YAAY;QAE1B,OAAO,GAAG,YAAY;EAE1B,CAAC,CAAC,CACD,KAAK,IAAI;CACd;CAEA,uBAAuB,SAAuC;EAC5D,OAAO,CAAC,GAAG,IAAI,IAAI,OAAO,CAAC;CAC7B;;;;;;;;;;;;;;;;;CAkBA,0BAA0B,aAAkD;EAC1E,MAAM,WAAiC,CAAC;EACxC,MAAM,YAA2B,CAAC;EAElC,YAAY,SAAS,MAAM,UAAU;GACnC,UAAU,KAAK,IAAI;GACnB,IAAI,KAAK,SAAS,MAAM,MAAM;IAC5B,IAAI,eAAwC,YAAY,QAAQ;IAChE,IAAI,CAAC,gBAAgB,aAAa,SAAS,MAAM,SAC/C,eAAe,KAAA;IAGjB,SAAS,KAAK;KACZ,iBAAiB;KACjB,cAAc,eACV,IAAIC,yBAAAA,aAAa,YAAY,IAC7B,KAAA;IACN,CAAC;GACH;EACF,CAAC;EAED,OAAO;CACT;;;;;;;;;;;;;;;;;;;;CAqBA,MAAM,MACJ,QACA,YACsB;EACtB,IAAI,EAAE,KAAK,kBAAkB,SAC3B,MAAM,IAAI,MAAM,oBAAoB,KAAK,eAAe,WAAW;EAGrE,MAAM,cAAyB,OAAO,KAAK;EAK3C,IAAI,CAH8B,YAAY,OAAO,aAAA,GAAA,yBAAA,cAAA,CACrC,OAAO,CAEM,GAC3B,MAAM,IAAI,MAAM,+CAA+C;EAGjE,MAAM,gBAAgB,KAAK,0BACzB,WACF;EAGA,MAAM,kBAAkB,MAAM,QAAQ,IACpC,cAAc,IAAI,OAAO,sBAAsB;GAC7C,iBAAiB,MAAM,KAAK,uBAC1B,iBAAiB,eACnB;GACA,cAAc,iBAAiB;GAC/B;EACF,EAAE,CACJ;EAIA,MAAM,uBAAuB,MAAM,QAAQ,IACzC,gBAAgB,KAAK,eACnB,KAAK,wBAAwB;GAC3B,iBAAiB,WAAW;GAC5B,cAAc,WAAW;GACzB;EACF,CAAC,CACH,CACF;EAeA,OAAO,EACL,UAAA,MAbqB,QAAQ,IAC7B,qBAAqB,KAAK,cACxB,KAAK,cAAc;GACjB,cAAc,UAAU;GACxB,aAAa;IACX,mBAAmB,UAAU;IAC7B,2BAA2B,UAAU;GACvC;EACF,CAAC,CACH,CACF,EAIA;CACF;;;;;;;;;;;CAYA,MAAc,uBACZ,aACA,YACyC;EACzC,MAAM,gBAAgB,KAAK,qBAAqB,WAAW;EAE3D,MAAM,mBAAmB,KAAK,IAC3B,UAAU,CAACC,cAAAA,kCAAkC,CAAC,CAAC,CAC/C,WAAW,EACV,eAAe,EAAE,MAAMA,cAAAA,mCAAmC,KAAK,EACjE,CAAC;EAMH,MAAM,MAAM,MAJEC,yCAAAA,iCAAiC,KAAK,gBAAgB,CAAC,CAAC,KACpE,KAAK,gBAGe,CAAC,CAAC,OACtB,EACE,cAAc,cAChB,GACA,YAAY,SAAS,YAAY,CACnC;EAEA,IACE,EACE,eAAe,OACf,0BAA0B,OAC1B,cAAc,MAGhB,MAAM,IAAI,MAAM,8BAA8B,KAAK,UAAU,GAAG,GAAG;EAGrE,MAAM,qBAAqB;EAG3B,MAAM,gBAAgB,MAAM,KAAK,yBAC/B,mBAAmB,QACrB;EACA,MAAM,eAAe,KAAK,uBAAuB,CAC/C,GAAG,mBAAmB,UACtB,GAAG,aACL,CAAC;EAED,OAAO;GACL,GAAG;GACH,UAAU;EACZ;CACF;;;;;;;;CASA,MAAc,yBACZ,UACwB;EAaxB,QAJgC,MAPQ,QAAQ,IAC9C,SAAS,IAAI,OAAO,YAAY;GAE9B,QAAO,MADuB,KAAK,UAAU,OAAO,OAAO,EAAA,CACpC;EACzB,CAAC,CACH,EAAA,CAE0D,KACvD,aAAa,SAAS,WAGI;CAC/B;;;;;;;;;;;;;;CAeA,MAAc,wBAAwB,EACpC,iBACA,cACA,cAK2C;EAC3C,MAAM,mBAAmB,KAAK,IAC3B,UAAU,CAACC,cAAAA,8BAA8B,CAAC,CAAC,CAC3C,WAAW,EACV,eAAe,EAAE,MAAMA,cAAAA,+BAA+B,KAAK,EAC7D,CAAC;EAEH,MAAM,QAAQC,yCAAAA,6BAA6B,KAAK,gBAAgB,CAAC,CAAC,KAChE,KAAK,gBACP;EAEA,IAAI,OAAO,cAAc,YAAY,UACnC,MAAM,IAAI,MAAM,sDAAsD;EAExE,MAAM,MAAO,MAAM,MAAM,OACvB;GACE,kBAAkB,gBAAgB;GAClC,eAAe,cAAc,WAAW;GACxC,eAAe,gBAAgB,SAAS,KAAK,IAAI;EACnD,GACA,YAAY,SAAS,uBAAuB,CAC9C;EAcA,OAAO;GACL;GACA,mBAAA,MAT8B,KAAK,0BAA0B;IAC7D,oBAAoB,gBAAgB;IACpC,2BAA2B,IAAI;IAC/B,cAAc,gBAAgB;IAC9B;GACF,CAAC;GAKC,2BAA2B,IAAI;EACjC;CACF;;;;;;;;;;;;CAaA,MAAc,0BAA0B,EACtC,oBACA,2BACA,cACA,cAMkB;EAclB,OAAO,MAbwBC,yCAAAA,mCAAmC,KAChE,KAAK,GACP,CAAC,CAAC,KAAK,KAAK,kBAE4C,CAAC,CAAC,OACxD;GACE,YAAY;GACZ,6BAA6B,0BAA0B,KAAK,IAAI;GAChE,eAAe,aAAa,KAAK,IAAI;EACvC,GACA,YAAY,SAAS,qBAAqB,CAC5C;CAGF;;;;;;;;;;;;;CAcA,MAAc,cAAc,EAC1B,cACA,aACA,cAWkB;EAClB,MAAM,QAAQC,yCAAAA,sBAAsB,KAAK,KAAK,GAAG,CAAC,CAAC,KACjD,KAAK,kBACP;EAEA,IAAI,OAAO,cAAc,YAAY,UACnC,MAAM,IAAI,MAAM,sDAAsD;EAWxE,OAAO,MATW,MAAM,OACtB;GACE,uBAAuB,YAAY,0BAA0B,KAAK,IAAI;GACtE,YAAY,YAAY;GACxB,cAAc,cAAc,WAAW;EACzC,GACA,YAAY,SAAS,gBAAgB,CACvC;CAGF;;;;;;;;;;;;CAaA,OAAO,QACL,KACA,WACA,SAG8B;EAC9B,OAAO,IAAI,KAAK;GACd;GACA;GACA,GAAG;EACL,CAAC;CACH;AACF"}
+{"version":3,"file":"violation_of_expectations_chain.cjs","names":["BaseChain","JsonOutputFunctionsParser","StringOutputParser","HumanMessage","isBaseMessage","PREDICT_NEXT_USER_MESSAGE_FUNCTION","PREDICT_NEXT_USER_MESSAGE_PROMPT","PREDICTION_VIOLATIONS_FUNCTION","PREDICTION_VIOLATIONS_PROMPT","GENERATE_REVISED_PREDICTION_PROMPT","GENERATE_FACTS_PROMPT"],"sources":["../../../../src/experimental/chains/violation_of_expectations/violation_of_expectations_chain.ts"],"sourcesContent":["import type { BaseRetrieverInterface } from \"@langchain/core/retrievers\";\nimport { ChatOpenAI } from \"@langchain/openai\";\nimport {\n  BaseMessage,\n  HumanMessage,\n  isBaseMessage,\n} from \"@langchain/core/messages\";\nimport { ChainValues } from \"@langchain/core/utils/types\";\nimport { StringOutputParser } from \"@langchain/core/output_parsers\";\nimport { CallbackManagerForChainRun } from \"@langchain/core/callbacks/manager\";\nimport { JsonOutputFunctionsParser } from \"../../../output_parsers/openai_functions.js\";\nimport { BaseChain, ChainInputs } from \"../../../chains/base.js\";\nimport {\n  GetPredictionViolationsResponse,\n  MessageChunkResult,\n  PREDICTION_VIOLATIONS_FUNCTION,\n  PREDICT_NEXT_USER_MESSAGE_FUNCTION,\n  PredictNextUserMessageResponse,\n} from \"./types.js\";\nimport {\n  GENERATE_FACTS_PROMPT,\n  GENERATE_REVISED_PREDICTION_PROMPT,\n  PREDICTION_VIOLATIONS_PROMPT,\n  PREDICT_NEXT_USER_MESSAGE_PROMPT,\n} from \"./violation_of_expectations_prompt.js\";\n\n/**\n * Interface for the input parameters of the ViolationOfExpectationsChain class.\n */\nexport interface ViolationOfExpectationsChainInput extends ChainInputs {\n  /**\n   * The retriever to use for retrieving stored\n   * thoughts and insights.\n   */\n  retriever: BaseRetrieverInterface;\n  /**\n   * The LLM to use\n   */\n  llm: ChatOpenAI;\n}\n\n/**\n * Chain that generates key insights/facts of a user based on a\n * a chat conversation with an AI.\n */\nexport class ViolationOfExpectationsChain\n  extends BaseChain\n  implements ViolationOfExpectationsChainInput\n{\n  static lc_name() {\n    return \"ViolationOfExpectationsChain\";\n  }\n\n  _chainType(): string {\n    return \"violation_of_expectation_chain\";\n  }\n\n  chatHistoryKey = \"chat_history\";\n\n  thoughtsKey = \"thoughts\";\n\n  get inputKeys() {\n    return [this.chatHistoryKey];\n  }\n\n  get outputKeys() {\n    return [this.thoughtsKey];\n  }\n\n  retriever: BaseRetrieverInterface;\n\n  llm: ChatOpenAI;\n\n  jsonOutputParser: JsonOutputFunctionsParser;\n\n  stringOutputParser: StringOutputParser;\n\n  constructor(fields: ViolationOfExpectationsChainInput) {\n    super(fields);\n    this.retriever = fields.retriever;\n    this.llm = fields.llm;\n    this.jsonOutputParser = new JsonOutputFunctionsParser();\n    this.stringOutputParser = new StringOutputParser();\n  }\n\n  getChatHistoryString(chatHistory: BaseMessage[]): string {\n    return chatHistory\n      .map((chatMessage) => {\n        if (chatMessage._getType() === \"human\") {\n          return `Human: ${chatMessage.content}`;\n        } else if (chatMessage._getType() === \"ai\") {\n          return `AI: ${chatMessage.content}`;\n        } else {\n          return `${chatMessage.content}`;\n        }\n      })\n      .join(\"\\n\");\n  }\n\n  removeDuplicateStrings(strings: Array<string>): Array<string> {\n    return [...new Set(strings)];\n  }\n\n  /**\n   * This method breaks down the chat history into chunks of messages.\n   * Each chunk consists of a sequence of messages ending with an AI message and the subsequent user response, if any.\n   *\n   * @param {BaseMessage[]} chatHistory - The chat history to be chunked.\n   *\n   * @returns {MessageChunkResult[]} An array of message chunks. Each chunk includes a sequence of messages and the subsequent user response.\n   *\n   * @description\n   * The method iterates over the chat history and pushes each message into a temporary array.\n   * When it encounters an AI message, it checks for a subsequent user message.\n   * If a user message is found, it is considered as the user response to the AI message.\n   * If no user message is found after the AI message, the user response is undefined.\n   * The method then pushes the chunk (sequence of messages and user response) into the result array.\n   * This process continues until all messages in the chat history have been processed.\n   */\n  chunkMessagesByAIResponse(chatHistory: BaseMessage[]): MessageChunkResult[] {\n    const newArray: MessageChunkResult[] = [];\n    const tempArray: BaseMessage[] = [];\n\n    chatHistory.forEach((item, index) => {\n      tempArray.push(item);\n      if (item._getType() === \"ai\") {\n        let userResponse: BaseMessage | undefined = chatHistory[index + 1];\n        if (!userResponse || userResponse._getType() !== \"human\") {\n          userResponse = undefined;\n        }\n\n        newArray.push({\n          chunkedMessages: tempArray,\n          userResponse: userResponse\n            ? new HumanMessage(userResponse)\n            : undefined,\n        });\n      }\n    });\n\n    return newArray;\n  }\n\n  /**\n   * This method processes a chat history to generate insights about the user.\n   *\n   * @param {ChainValues} values - The input values for the chain. It should contain a key for chat history.\n   * @param {CallbackManagerForChainRun} [runManager] - Optional callback manager for the chain run.\n   *\n   * @returns {Promise<ChainValues>} A promise that resolves to a list of insights about the user.\n   *\n   * @throws {Error} If the chat history key is not found in the input values or if the chat history is not an array of BaseMessages.\n   *\n   * @description\n   * The method performs the following steps:\n   * 1. Checks if the chat history key is present in the input values and if the chat history is an array of BaseMessages.\n   * 2. Breaks the chat history into chunks of messages.\n   * 3. For each chunk, it generates an initial prediction for the user's next message.\n   * 4. For each prediction, it generates insights and prediction violations, and regenerates the prediction based on the violations.\n   * 5. For each set of messages, it generates a fact/insight about the user.\n   * The method returns a list of these insights.\n   */\n  async _call(\n    values: ChainValues,\n    runManager?: CallbackManagerForChainRun\n  ): Promise<ChainValues> {\n    if (!(this.chatHistoryKey in values)) {\n      throw new Error(`Chat history key ${this.chatHistoryKey} not found`);\n    }\n\n    const chatHistory: unknown[] = values[this.chatHistoryKey];\n\n    const isEveryMessageBaseMessage = chatHistory.every((message) =>\n      isBaseMessage(message)\n    );\n    if (!isEveryMessageBaseMessage) {\n      throw new Error(\"Chat history must be an array of BaseMessages\");\n    }\n\n    const messageChunks = this.chunkMessagesByAIResponse(\n      chatHistory as BaseMessage[]\n    );\n\n    // Generate the initial prediction for every user message.\n    const userPredictions = await Promise.all(\n      messageChunks.map(async (chatHistoryChunk) => ({\n        userPredictions: await this.predictNextUserMessage(\n          chatHistoryChunk.chunkedMessages\n        ),\n        userResponse: chatHistoryChunk.userResponse,\n        runManager,\n      }))\n    );\n\n    // Generate insights, and prediction violations for every user message.\n    // This call also regenerates the prediction based on the violations.\n    const predictionViolations = await Promise.all(\n      userPredictions.map((prediction) =>\n        this.getPredictionViolations({\n          userPredictions: prediction.userPredictions,\n          userResponse: prediction.userResponse,\n          runManager,\n        })\n      )\n    );\n\n    // Generate a fact/insight about the user for every set of messages.\n    const insights = await Promise.all(\n      predictionViolations.map((violation) =>\n        this.generateFacts({\n          userResponse: violation.userResponse,\n          predictions: {\n            revisedPrediction: violation.revisedPrediction,\n            explainedPredictionErrors: violation.explainedPredictionErrors,\n          },\n        })\n      )\n    );\n\n    return {\n      insights,\n    };\n  }\n\n  /**\n   * This method predicts the next user message based on the chat history.\n   *\n   * @param {BaseMessage[]} chatHistory - The chat history based on which the next user message is predicted.\n   * @param {CallbackManagerForChainRun} [runManager] - Optional callback manager for the chain run.\n   *\n   * @returns {Promise<PredictNextUserMessageResponse>} A promise that resolves to the predicted next user message, the user state, and any insights.\n   *\n   * @throws {Error} If the response from the language model does not contain the expected keys: 'userState', 'predictedUserMessage', and 'insights'.\n   */\n  private async predictNextUserMessage(\n    chatHistory: BaseMessage[],\n    runManager?: CallbackManagerForChainRun\n  ): Promise<PredictNextUserMessageResponse> {\n    const messageString = this.getChatHistoryString(chatHistory);\n\n    const llmWithFunctions = this.llm\n      .bindTools([PREDICT_NEXT_USER_MESSAGE_FUNCTION])\n      .withConfig({\n        function_call: { name: PREDICT_NEXT_USER_MESSAGE_FUNCTION.name },\n      });\n\n    const chain = PREDICT_NEXT_USER_MESSAGE_PROMPT.pipe(llmWithFunctions).pipe(\n      this.jsonOutputParser\n    );\n\n    const res = await chain.invoke(\n      {\n        chat_history: messageString,\n      },\n      runManager?.getChild(\"prediction\")\n    );\n\n    if (\n      !(\n        \"userState\" in res &&\n        \"predictedUserMessage\" in res &&\n        \"insights\" in res\n      )\n    ) {\n      throw new Error(`Invalid response from LLM: ${JSON.stringify(res)}`);\n    }\n\n    const predictionResponse = res as PredictNextUserMessageResponse;\n\n    // Query the retriever for relevant insights. Use the generates insights as a query.\n    const retrievedDocs = await this.retrieveRelevantInsights(\n      predictionResponse.insights\n    );\n    const relevantDocs = this.removeDuplicateStrings([\n      ...predictionResponse.insights,\n      ...retrievedDocs,\n    ]);\n\n    return {\n      ...predictionResponse,\n      insights: relevantDocs,\n    };\n  }\n\n  /**\n   * Retrieves relevant insights based on the provided insights.\n   *\n   * @param {Array<string>} insights - An array of insights to be used for retrieving relevant documents.\n   *\n   * @returns {Promise<Array<string>>} A promise that resolves to an array of relevant insights content.\n   */\n  private async retrieveRelevantInsights(\n    insights: Array<string>\n  ): Promise<Array<string>> {\n    // Only extract the first relevant doc from the retriever. We don't need more than one.\n    const relevantInsightsDocuments = await Promise.all(\n      insights.map(async (insight) => {\n        const relevantInsight = await this.retriever.invoke(insight);\n        return relevantInsight[0];\n      })\n    );\n\n    const relevantInsightsContent = relevantInsightsDocuments.map(\n      (document) => document.pageContent\n    );\n\n    return relevantInsightsContent;\n  }\n\n  /**\n   * This method generates prediction violations based on the predicted and actual user responses.\n   * It also generates a revised prediction based on the identified violations.\n   *\n   * @param {Object} params - The parameters for the method.\n   * @param {PredictNextUserMessageResponse} params.userPredictions - The predicted user message, user state, and insights.\n   * @param {BaseMessage} [params.userResponse] - The actual user response.\n   * @param {CallbackManagerForChainRun} [params.runManager] - Optional callback manager for the chain run.\n   *\n   * @returns {Promise<{ userResponse: BaseMessage | undefined; revisedPrediction: string; explainedPredictionErrors: Array<string>; }>} A promise that resolves to an object containing the actual user response, the revised prediction, and the explained prediction errors.\n   *\n   * @throws {Error} If the response from the language model does not contain the expected keys: 'violationExplanation', 'explainedPredictionErrors', and 'accuratePrediction'.\n   */\n  private async getPredictionViolations({\n    userPredictions,\n    userResponse,\n    runManager,\n  }: {\n    userPredictions: PredictNextUserMessageResponse;\n    userResponse?: BaseMessage;\n    runManager?: CallbackManagerForChainRun;\n  }): Promise<GetPredictionViolationsResponse> {\n    const llmWithFunctions = this.llm\n      .bindTools([PREDICTION_VIOLATIONS_FUNCTION])\n      .withConfig({\n        function_call: { name: PREDICTION_VIOLATIONS_FUNCTION.name },\n      });\n\n    const chain = PREDICTION_VIOLATIONS_PROMPT.pipe(llmWithFunctions).pipe(\n      this.jsonOutputParser\n    );\n\n    if (typeof userResponse?.content !== \"string\") {\n      throw new Error(\"This chain does not support non-string model output.\");\n    }\n    const res = (await chain.invoke(\n      {\n        predicted_output: userPredictions.predictedUserMessage,\n        actual_output: userResponse?.content ?? \"\",\n        user_insights: userPredictions.insights.join(\"\\n\"),\n      },\n      runManager?.getChild(\"prediction_violations\")\n    )) as Awaited<{\n      violationExplanation: string;\n      explainedPredictionErrors: Array<string>;\n      accuratePrediction: boolean;\n    }>;\n\n    // Generate a revised prediction based on violations.\n    const revisedPrediction = await this.generateRevisedPrediction({\n      originalPrediction: userPredictions.predictedUserMessage,\n      explainedPredictionErrors: res.explainedPredictionErrors,\n      userInsights: userPredictions.insights,\n      runManager,\n    });\n\n    return {\n      userResponse,\n      revisedPrediction,\n      explainedPredictionErrors: res.explainedPredictionErrors,\n    };\n  }\n\n  /**\n   * This method generates a revised prediction based on the original prediction, explained prediction errors, and user insights.\n   *\n   * @param {Object} params - The parameters for the method.\n   * @param {string} params.originalPrediction - The original prediction made by the model.\n   * @param {Array<string>} params.explainedPredictionErrors - An array of explained prediction errors.\n   * @param {Array<string>} params.userInsights - An array of insights about the user.\n   * @param {CallbackManagerForChainRun} [params.runManager] - Optional callback manager for the chain run.\n   *\n   * @returns {Promise<string>} A promise that resolves to a revised prediction.\n   */\n  private async generateRevisedPrediction({\n    originalPrediction,\n    explainedPredictionErrors,\n    userInsights,\n    runManager,\n  }: {\n    originalPrediction: string;\n    explainedPredictionErrors: Array<string>;\n    userInsights: Array<string>;\n    runManager?: CallbackManagerForChainRun;\n  }): Promise<string> {\n    const revisedPredictionChain = GENERATE_REVISED_PREDICTION_PROMPT.pipe(\n      this.llm\n    ).pipe(this.stringOutputParser);\n\n    const revisedPredictionRes = await revisedPredictionChain.invoke(\n      {\n        prediction: originalPrediction,\n        explained_prediction_errors: explainedPredictionErrors.join(\"\\n\"),\n        user_insights: userInsights.join(\"\\n\"),\n      },\n      runManager?.getChild(\"prediction_revision\")\n    );\n\n    return revisedPredictionRes;\n  }\n\n  /**\n   * This method generates facts or insights about the user based on the revised prediction, explained prediction errors, and the user's response.\n   *\n   * @param {Object} params - The parameters for the method.\n   * @param {BaseMessage} [params.userResponse] - The actual user response.\n   * @param {Object} params.predictions - The revised prediction and explained prediction errors.\n   * @param {string} params.predictions.revisedPrediction - The revised prediction made by the model.\n   * @param {Array<string>} params.predictions.explainedPredictionErrors - An array of explained prediction errors.\n   * @param {CallbackManagerForChainRun} [params.runManager] - Optional callback manager for the chain run.\n   *\n   * @returns {Promise<string>} A promise that resolves to a string containing the generated facts or insights about the user.\n   */\n  private async generateFacts({\n    userResponse,\n    predictions,\n    runManager,\n  }: {\n    userResponse?: BaseMessage;\n    /**\n     * Optional if the prediction was accurate.\n     */\n    predictions: {\n      revisedPrediction: string;\n      explainedPredictionErrors: Array<string>;\n    };\n    runManager?: CallbackManagerForChainRun;\n  }): Promise<string> {\n    const chain = GENERATE_FACTS_PROMPT.pipe(this.llm).pipe(\n      this.stringOutputParser\n    );\n\n    if (typeof userResponse?.content !== \"string\") {\n      throw new Error(\"This chain does not support non-string model output.\");\n    }\n    const res = await chain.invoke(\n      {\n        prediction_violations: predictions.explainedPredictionErrors.join(\"\\n\"),\n        prediction: predictions.revisedPrediction,\n        user_message: userResponse?.content ?? \"\",\n      },\n      runManager?.getChild(\"generate_facts\")\n    );\n\n    return res;\n  }\n\n  /**\n   * Static method that creates a ViolationOfExpectationsChain instance from a\n   * ChatOpenAI and retriever. It also accepts optional options\n   * to customize the chain.\n   *\n   * @param llm The ChatOpenAI instance.\n   * @param retriever The retriever used for similarity search.\n   * @param options Optional options to customize the chain.\n   *\n   * @returns A new instance of ViolationOfExpectationsChain.\n   */\n  static fromLLM(\n    llm: ChatOpenAI,\n    retriever: BaseRetrieverInterface,\n    options?: Partial<\n      Omit<ViolationOfExpectationsChainInput, \"llm\" | \"retriever\">\n    >\n  ): ViolationOfExpectationsChain {\n    return new this({\n      retriever,\n      llm,\n      ...options,\n    });\n  }\n}\n"],"mappings":";;;;;;;;;;;AA6CA,IAAa,+BAAb,cACUA,aAAAA,UAEV;CACE,OAAO,UAAU;EACf,OAAO;CACT;CAEA,aAAqB;EACnB,OAAO;CACT;CAEA,iBAAiB;CAEjB,cAAc;CAEd,IAAI,YAAY;EACd,OAAO,CAAC,KAAK,cAAc;CAC7B;CAEA,IAAI,aAAa;EACf,OAAO,CAAC,KAAK,WAAW;CAC1B;CAEA;CAEA;CAEA;CAEA;CAEA,YAAY,QAA2C;EACrD,MAAM,MAAM;EACZ,KAAK,YAAY,OAAO;EACxB,KAAK,MAAM,OAAO;EAClB,KAAK,mBAAmB,IAAIC,yBAAAA,0BAA0B;EACtD,KAAK,qBAAqB,IAAIC,+BAAAA,mBAAmB;CACnD;CAEA,qBAAqB,aAAoC;EACvD,OAAO,YACJ,KAAK,gBAAgB;GACpB,IAAI,YAAY,SAAS,MAAM,SAC7B,OAAO,UAAU,YAAY;QACxB,IAAI,YAAY,SAAS,MAAM,MACpC,OAAO,OAAO,YAAY;QAE1B,OAAO,GAAG,YAAY;EAE1B,CAAC,CAAC,CACD,KAAK,IAAI;CACd;CAEA,uBAAuB,SAAuC;EAC5D,OAAO,CAAC,GAAG,IAAI,IAAI,OAAO,CAAC;CAC7B;;;;;;;;;;;;;;;;;CAkBA,0BAA0B,aAAkD;EAC1E,MAAM,WAAiC,CAAC;EACxC,MAAM,YAA2B,CAAC;EAElC,YAAY,SAAS,MAAM,UAAU;GACnC,UAAU,KAAK,IAAI;GACnB,IAAI,KAAK,SAAS,MAAM,MAAM;IAC5B,IAAI,eAAwC,YAAY,QAAQ;IAChE,IAAI,CAAC,gBAAgB,aAAa,SAAS,MAAM,SAC/C,eAAe,KAAA;IAGjB,SAAS,KAAK;KACZ,iBAAiB;KACjB,cAAc,eACV,IAAIC,yBAAAA,aAAa,YAAY,IAC7B,KAAA;IACN,CAAC;GACH;EACF,CAAC;EAED,OAAO;CACT;;;;;;;;;;;;;;;;;;;;CAqBA,MAAM,MACJ,QACA,YACsB;EACtB,IAAI,EAAE,KAAK,kBAAkB,SAC3B,MAAM,IAAI,MAAM,oBAAoB,KAAK,eAAe,WAAW;EAGrE,MAAM,cAAyB,OAAO,KAAK;EAK3C,IAAI,CAH8B,YAAY,OAAO,aAAA,GACnDC,yBAAAA,cAAAA,CAAc,OAAO,CAEM,GAC3B,MAAM,IAAI,MAAM,+CAA+C;EAGjE,MAAM,gBAAgB,KAAK,0BACzB,WACF;EAGA,MAAM,kBAAkB,MAAM,QAAQ,IACpC,cAAc,IAAI,OAAO,sBAAsB;GAC7C,iBAAiB,MAAM,KAAK,uBAC1B,iBAAiB,eACnB;GACA,cAAc,iBAAiB;GAC/B;EACF,EAAE,CACJ;EAIA,MAAM,uBAAuB,MAAM,QAAQ,IACzC,gBAAgB,KAAK,eACnB,KAAK,wBAAwB;GAC3B,iBAAiB,WAAW;GAC5B,cAAc,WAAW;GACzB;EACF,CAAC,CACH,CACF;EAeA,OAAO,EACL,UAAA,MAbqB,QAAQ,IAC7B,qBAAqB,KAAK,cACxB,KAAK,cAAc;GACjB,cAAc,UAAU;GACxB,aAAa;IACX,mBAAmB,UAAU;IAC7B,2BAA2B,UAAU;GACvC;EACF,CAAC,CACH,CACF,EAIA;CACF;;;;;;;;;;;CAYA,MAAc,uBACZ,aACA,YACyC;EACzC,MAAM,gBAAgB,KAAK,qBAAqB,WAAW;EAE3D,MAAM,mBAAmB,KAAK,IAC3B,UAAU,CAACC,cAAAA,kCAAkC,CAAC,CAAC,CAC/C,WAAW,EACV,eAAe,EAAE,MAAMA,cAAAA,mCAAmC,KAAK,EACjE,CAAC;EAMH,MAAM,MAAM,MAJEC,yCAAAA,iCAAiC,KAAK,gBAAgB,CAAC,CAAC,KACpE,KAAK,gBAGe,CAAC,CAAC,OACtB,EACE,cAAc,cAChB,GACA,YAAY,SAAS,YAAY,CACnC;EAEA,IACE,EACE,eAAe,OACf,0BAA0B,OAC1B,cAAc,MAGhB,MAAM,IAAI,MAAM,8BAA8B,KAAK,UAAU,GAAG,GAAG;EAGrE,MAAM,qBAAqB;EAG3B,MAAM,gBAAgB,MAAM,KAAK,yBAC/B,mBAAmB,QACrB;EACA,MAAM,eAAe,KAAK,uBAAuB,CAC/C,GAAG,mBAAmB,UACtB,GAAG,aACL,CAAC;EAED,OAAO;GACL,GAAG;GACH,UAAU;EACZ;CACF;;;;;;;;CASA,MAAc,yBACZ,UACwB;EAaxB,QAJgC,MAPQ,QAAQ,IAC9C,SAAS,IAAI,OAAO,YAAY;GAE9B,QAAO,MADuB,KAAK,UAAU,OAAO,OAAO,EAAA,CACpC;EACzB,CAAC,CACH,EAAA,CAE0D,KACvD,aAAa,SAAS,WAGI;CAC/B;;;;;;;;;;;;;;CAeA,MAAc,wBAAwB,EACpC,iBACA,cACA,cAK2C;EAC3C,MAAM,mBAAmB,KAAK,IAC3B,UAAU,CAACC,cAAAA,8BAA8B,CAAC,CAAC,CAC3C,WAAW,EACV,eAAe,EAAE,MAAMA,cAAAA,+BAA+B,KAAK,EAC7D,CAAC;EAEH,MAAM,QAAQC,yCAAAA,6BAA6B,KAAK,gBAAgB,CAAC,CAAC,KAChE,KAAK,gBACP;EAEA,IAAI,OAAO,cAAc,YAAY,UACnC,MAAM,IAAI,MAAM,sDAAsD;EAExE,MAAM,MAAO,MAAM,MAAM,OACvB;GACE,kBAAkB,gBAAgB;GAClC,eAAe,cAAc,WAAW;GACxC,eAAe,gBAAgB,SAAS,KAAK,IAAI;EACnD,GACA,YAAY,SAAS,uBAAuB,CAC9C;EAcA,OAAO;GACL;GACA,mBAAA,MAT8B,KAAK,0BAA0B;IAC7D,oBAAoB,gBAAgB;IACpC,2BAA2B,IAAI;IAC/B,cAAc,gBAAgB;IAC9B;GACF,CAAC;GAKC,2BAA2B,IAAI;EACjC;CACF;;;;;;;;;;;;CAaA,MAAc,0BAA0B,EACtC,oBACA,2BACA,cACA,cAMkB;EAclB,OAAO,MAbwBC,yCAAAA,mCAAmC,KAChE,KAAK,GACP,CAAC,CAAC,KAAK,KAAK,kBAE4C,CAAC,CAAC,OACxD;GACE,YAAY;GACZ,6BAA6B,0BAA0B,KAAK,IAAI;GAChE,eAAe,aAAa,KAAK,IAAI;EACvC,GACA,YAAY,SAAS,qBAAqB,CAC5C;CAGF;;;;;;;;;;;;;CAcA,MAAc,cAAc,EAC1B,cACA,aACA,cAWkB;EAClB,MAAM,QAAQC,yCAAAA,sBAAsB,KAAK,KAAK,GAAG,CAAC,CAAC,KACjD,KAAK,kBACP;EAEA,IAAI,OAAO,cAAc,YAAY,UACnC,MAAM,IAAI,MAAM,sDAAsD;EAWxE,OAAO,MATW,MAAM,OACtB;GACE,uBAAuB,YAAY,0BAA0B,KAAK,IAAI;GACtE,YAAY,YAAY;GACxB,cAAc,cAAc,WAAW;EACzC,GACA,YAAY,SAAS,gBAAgB,CACvC;CAGF;;;;;;;;;;;;CAaA,OAAO,QACL,KACA,WACA,SAG8B;EAC9B,OAAO,IAAI,KAAK;GACd;GACA;GACA,GAAG;EACL,CAAC;CACH;AACF"}

package/dist/experimental/openai_assistant/index.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"index.cjs","names":["Runnable","OpenAIClient","StructuredTool","sleep"],"sources":["../../../src/experimental/openai_assistant/index.ts"],"sourcesContent":["import {\n  type ClientOptions,\n  type OpenAIChatModelId,\n  OpenAIClient,\n} from \"@langchain/openai\";\nimport { StructuredTool, StructuredToolInterface } from \"@langchain/core/tools\";\nimport { Runnable, RunnableConfig } from \"@langchain/core/runnables\";\nimport { isInteropZodSchema } from \"@langchain/core/utils/types\";\nimport { toJsonSchema } from \"@langchain/core/utils/json_schema\";\nimport { ToolDefinition } from \"@langchain/core/language_models/base\";\nimport { sleep } from \"../../util/time.js\";\nimport type {\n  OpenAIAssistantFinish,\n  OpenAIAssistantAction,\n  OpenAIToolType,\n} from \"./schema.js\";\n\n// oxlint-disable-next-line @typescript-eslint/no-explicit-any\ntype ThreadMessage = any;\n// oxlint-disable-next-line @typescript-eslint/no-explicit-any\ntype RequiredActionFunctionToolCall = any;\n\ntype ExtractRunOutput<AsAgent extends boolean | undefined> =\n  AsAgent extends true\n    ? OpenAIAssistantFinish | OpenAIAssistantAction[]\n    : ThreadMessage[] | RequiredActionFunctionToolCall[];\n\nexport type OpenAIAssistantRunnableInput<\n  AsAgent extends boolean | undefined = undefined,\n> = {\n  client?: OpenAIClient;\n  clientOptions?: ClientOptions;\n  assistantId: string;\n  pollIntervalMs?: number;\n  asAgent?: AsAgent;\n};\n\nexport class OpenAIAssistantRunnable<\n  AsAgent extends boolean | undefined,\n  // oxlint-disable-next-line @typescript-eslint/no-explicit-any\n  RunInput extends Record<string, any> = Record<string, any>,\n> extends Runnable<RunInput, ExtractRunOutput<AsAgent>> {\n  lc_namespace = [\"langchain\", \"experimental\", \"openai_assistant\"];\n\n  private client: OpenAIClient;\n\n  assistantId: string;\n\n  pollIntervalMs = 1000;\n\n  asAgent?: AsAgent;\n\n  constructor(fields: OpenAIAssistantRunnableInput<AsAgent>) {\n    super(fields);\n    this.client = fields.client ?? new OpenAIClient(fields?.clientOptions);\n    this.assistantId = fields.assistantId;\n    this.asAgent = fields.asAgent ?? this.asAgent;\n  }\n\n  static async createAssistant<AsAgent extends boolean>({\n    model,\n    name,\n    instructions,\n    tools,\n    client,\n    clientOptions,\n    asAgent,\n    pollIntervalMs,\n    fileIds,\n  }: Omit<OpenAIAssistantRunnableInput<AsAgent>, \"assistantId\"> & {\n    model: OpenAIChatModelId;\n    name?: string;\n    instructions?: string;\n    tools?: OpenAIToolType | Array<StructuredTool>;\n    fileIds?: string[];\n  }) {\n    const formattedTools =\n      tools?.map((tool) => {\n        // oxlint-disable-next-line no-instanceof/no-instanceof\n        if (tool instanceof StructuredTool) {\n          return formatToOpenAIAssistantTool(tool);\n        }\n        return tool;\n      }) ?? [];\n    const oaiClient = client ?? new OpenAIClient(clientOptions);\n    const assistant = await oaiClient.beta.assistants.create({\n      name,\n      instructions,\n      tools: formattedTools,\n      model,\n      file_ids: fileIds,\n      // oxlint-disable-next-line @typescript-eslint/no-explicit-any\n    } as any);\n\n    return new this({\n      client: oaiClient,\n      assistantId: assistant.id,\n      asAgent,\n      pollIntervalMs,\n    });\n  }\n\n  async invoke(\n    input: RunInput,\n    _options?: RunnableConfig\n  ): Promise<ExtractRunOutput<AsAgent>> {\n    // oxlint-disable-next-line @typescript-eslint/no-explicit-any\n    let run: any;\n    if (this.asAgent && input.steps && input.steps.length > 0) {\n      const parsedStepsInput = await this._parseStepsInput(input);\n      run = await this.client.beta.threads.runs.submitToolOutputs(\n        parsedStepsInput.runId,\n        {\n          thread_id: parsedStepsInput.threadId,\n          tool_outputs: parsedStepsInput.toolOutputs,\n        }\n      );\n    } else if (!(\"threadId\" in input)) {\n      const thread = {\n        messages: [\n          {\n            role: \"user\",\n            content: input.content,\n            attachments: input.attachments,\n            metadata: input.messagesMetadata,\n          },\n        ],\n        metadata: input.threadMetadata,\n      };\n      run = await this._createThreadAndRun({\n        ...input,\n        thread,\n      });\n    } else if (!(\"runId\" in input)) {\n      await this.client.beta.threads.messages.create(input.threadId, {\n        content: input.content,\n        role: \"user\",\n        attachments: input.attachments,\n        metadata: input.messagesMetadata,\n        // oxlint-disable-next-line @typescript-eslint/no-explicit-any\n      } as any);\n      run = await this._createRun(input);\n    } else {\n      // Submitting tool outputs to an existing run, outside the AgentExecutor\n      // framework.\n      run = await this.client.beta.threads.runs.submitToolOutputs(input.runId, {\n        thread_id: input.threadId,\n        tool_outputs: input.toolOutputs,\n      });\n    }\n\n    return this._getResponse(run.id, run.thread_id);\n  }\n\n  /**\n   * Delete an assistant.\n   *\n   * @link {https://platform.openai.com/docs/api-reference/assistants/deleteAssistant}\n   * @returns {Promise<AssistantDeleted>}\n   */\n  public async deleteAssistant() {\n    return await this.client.beta.assistants.delete(this.assistantId);\n  }\n\n  /**\n   * Retrieves an assistant.\n   *\n   * @link {https://platform.openai.com/docs/api-reference/assistants/getAssistant}\n   * @returns {Promise<OpenAIClient.Beta.Assistants.Assistant>}\n   */\n  public async getAssistant() {\n    return await this.client.beta.assistants.retrieve(this.assistantId);\n  }\n\n  /**\n   * Modifies an assistant.\n   *\n   * @link {https://platform.openai.com/docs/api-reference/assistants/modifyAssistant}\n   * @returns {Promise<OpenAIClient.Beta.Assistants.Assistant>}\n   */\n  public async modifyAssistant<AsAgent extends boolean>({\n    model,\n    name,\n    instructions,\n    fileIds,\n  }: Omit<OpenAIAssistantRunnableInput<AsAgent>, \"assistantId\" | \"tools\"> & {\n    model?: OpenAIChatModelId;\n    name?: string;\n    instructions?: string;\n    fileIds?: string[];\n  }) {\n    return await this.client.beta.assistants.update(this.assistantId, {\n      name,\n      instructions,\n      model,\n      file_ids: fileIds,\n      // oxlint-disable-next-line @typescript-eslint/no-explicit-any\n    } as any);\n  }\n\n  private async _parseStepsInput(input: RunInput): Promise<RunInput> {\n    const {\n      action: { runId, threadId },\n    } = input.steps[input.steps.length - 1];\n    const run = await this._waitForRun(runId, threadId);\n    const toolCalls = run.required_action?.submit_tool_outputs.tool_calls;\n    if (!toolCalls) {\n      return input;\n    }\n    // oxlint-disable-next-line @typescript-eslint/no-explicit-any\n    const toolOutputs = toolCalls.flatMap((toolCall: any) => {\n      const matchedAction = (\n        input.steps as {\n          action: OpenAIAssistantAction;\n          observation: string;\n        }[]\n      ).find((step) => step.action.toolCallId === toolCall.id);\n\n      return matchedAction\n        ? [\n            {\n              output: matchedAction.observation,\n              tool_call_id: matchedAction.action.toolCallId,\n            },\n          ]\n        : [];\n    });\n    return { toolOutputs, runId, threadId } as unknown as RunInput;\n  }\n\n  private async _createRun({\n    instructions,\n    model,\n    tools,\n    metadata,\n    threadId,\n  }: RunInput) {\n    const run = this.client.beta.threads.runs.create(threadId, {\n      assistant_id: this.assistantId,\n      instructions,\n      model,\n      tools,\n      metadata,\n    });\n    return run;\n  }\n\n  private async _createThreadAndRun(input: RunInput) {\n    const params: Record<string, unknown> = [\n      \"instructions\",\n      \"model\",\n      \"tools\",\n      \"run_metadata\",\n    ]\n      .filter((key) => key in input)\n      .reduce(\n        (obj, key) => {\n          const newObj = obj;\n          newObj[key] = input[key];\n          return newObj;\n        },\n        {} as Record<string, unknown>\n      );\n    const run = this.client.beta.threads.createAndRun({\n      ...params,\n      thread: input.thread,\n      assistant_id: this.assistantId,\n    });\n    return run;\n  }\n\n  private async _waitForRun(runId: string, threadId: string) {\n    let inProgress = true;\n    // oxlint-disable-next-line @typescript-eslint/no-explicit-any\n    let run = {} as any;\n    while (inProgress) {\n      run = await this.client.beta.threads.runs.retrieve(runId, {\n        thread_id: threadId,\n      });\n      inProgress = [\"in_progress\", \"queued\"].includes(run.status);\n      if (inProgress) {\n        await sleep(this.pollIntervalMs);\n      }\n    }\n    return run;\n  }\n\n  private async _getResponse(\n    runId: string,\n    threadId: string\n  ): Promise<ExtractRunOutput<AsAgent>>;\n\n  private async _getResponse(\n    runId: string,\n    threadId: string\n  ): Promise<\n    | OpenAIAssistantFinish\n    | OpenAIAssistantAction[]\n    | ThreadMessage[]\n    | RequiredActionFunctionToolCall[]\n  > {\n    const run = await this._waitForRun(runId, threadId);\n    if (run.status === \"completed\") {\n      const messages = await this.client.beta.threads.messages.list(threadId, {\n        order: \"desc\",\n      });\n      const newMessages = messages.data.filter((msg) => msg.run_id === runId);\n      if (!this.asAgent) {\n        return newMessages;\n      }\n      const answer = newMessages.flatMap((msg) => msg.content);\n      if (answer.every((item) => item.type === \"text\")) {\n        const answerString = answer\n          .map((item) => item.type === \"text\" && item.text.value)\n          .join(\"\\n\");\n        return {\n          returnValues: {\n            output: answerString,\n            runId,\n            threadId,\n          },\n          log: \"\",\n          runId,\n          threadId,\n        };\n      }\n    } else if (run.status === \"requires_action\") {\n      if (!this.asAgent) {\n        return run.required_action?.submit_tool_outputs.tool_calls ?? [];\n      }\n      const actions: OpenAIAssistantAction[] = [];\n      run.required_action?.submit_tool_outputs.tool_calls.forEach(\n        // oxlint-disable-next-line @typescript-eslint/no-explicit-any\n        (item: any) => {\n          const functionCall = item.function;\n          const args = JSON.parse(functionCall.arguments);\n          actions.push({\n            tool: functionCall.name,\n            toolInput: args,\n            toolCallId: item.id,\n            log: \"\",\n            runId,\n            threadId,\n          });\n        }\n      );\n      return actions;\n    }\n    const runInfo = JSON.stringify(run, null, 2);\n    throw new Error(\n      `Unexpected run status ${run.status}.\\nFull run info:\\n\\n${runInfo}`\n    );\n  }\n}\n\nexport function formatToOpenAIAssistantTool(\n  tool: StructuredToolInterface\n): ToolDefinition {\n  return {\n    type: \"function\",\n    function: {\n      name: tool.name,\n      description: tool.description,\n      parameters: isInteropZodSchema(tool.schema)\n        ? toJsonSchema(tool.schema)\n        : tool.schema,\n    },\n  };\n}\n"],"mappings":";;;;;;;;;;;;;AAqCA,IAAa,0BAAb,cAIUA,0BAAAA,SAA8C;CACtD,eAAe;EAAC;EAAa;EAAgB;CAAkB;CAE/D;CAEA;CAEA,iBAAiB;CAEjB;CAEA,YAAY,QAA+C;EACzD,MAAM,MAAM;EACZ,KAAK,SAAS,OAAO,UAAU,IAAIC,kBAAAA,aAAa,QAAQ,aAAa;EACrE,KAAK,cAAc,OAAO;EAC1B,KAAK,UAAU,OAAO,WAAW,KAAK;CACxC;CAEA,aAAa,gBAAyC,EACpD,OACA,MACA,cACA,OACA,QACA,eACA,SACA,gBACA,WAOC;EACD,MAAM,iBACJ,OAAO,KAAK,SAAS;GAEnB,IAAI,gBAAgBC,sBAAAA,gBAClB,OAAO,4BAA4B,IAAI;GAEzC,OAAO;EACT,CAAC,KAAK,CAAC;EACT,MAAM,YAAY,UAAU,IAAID,kBAAAA,aAAa,aAAa;EAC1D,MAAM,YAAY,MAAM,UAAU,KAAK,WAAW,OAAO;GACvD;GACA;GACA,OAAO;GACP;GACA,UAAU;EAEZ,CAAQ;EAER,OAAO,IAAI,KAAK;GACd,QAAQ;GACR,aAAa,UAAU;GACvB;GACA;EACF,CAAC;CACH;CAEA,MAAM,OACJ,OACA,UACoC;EAEpC,IAAI;EACJ,IAAI,KAAK,WAAW,MAAM,SAAS,MAAM,MAAM,SAAS,GAAG;GACzD,MAAM,mBAAmB,MAAM,KAAK,iBAAiB,KAAK;GAC1D,MAAM,MAAM,KAAK,OAAO,KAAK,QAAQ,KAAK,kBACxC,iBAAiB,OACjB;IACE,WAAW,iBAAiB;IAC5B,cAAc,iBAAiB;GACjC,CACF;EACF,OAAO,IAAI,EAAE,cAAc,QAAQ;GACjC,MAAM,SAAS;IACb,UAAU,CACR;KACE,MAAM;KACN,SAAS,MAAM;KACf,aAAa,MAAM;KACnB,UAAU,MAAM;IAClB,CACF;IACA,UAAU,MAAM;GAClB;GACA,MAAM,MAAM,KAAK,oBAAoB;IACnC,GAAG;IACH;GACF,CAAC;EACH,OAAO,IAAI,EAAE,WAAW,QAAQ;GAC9B,MAAM,KAAK,OAAO,KAAK,QAAQ,SAAS,OAAO,MAAM,UAAU;IAC7D,SAAS,MAAM;IACf,MAAM;IACN,aAAa,MAAM;IACnB,UAAU,MAAM;GAElB,CAAQ;GACR,MAAM,MAAM,KAAK,WAAW,KAAK;EACnC,OAGE,MAAM,MAAM,KAAK,OAAO,KAAK,QAAQ,KAAK,kBAAkB,MAAM,OAAO;GACvE,WAAW,MAAM;GACjB,cAAc,MAAM;EACtB,CAAC;EAGH,OAAO,KAAK,aAAa,IAAI,IAAI,IAAI,SAAS;CAChD;;;;;;;CAQA,MAAa,kBAAkB;EAC7B,OAAO,MAAM,KAAK,OAAO,KAAK,WAAW,OAAO,KAAK,WAAW;CAClE;;;;;;;CAQA,MAAa,eAAe;EAC1B,OAAO,MAAM,KAAK,OAAO,KAAK,WAAW,SAAS,KAAK,WAAW;CACpE;;;;;;;CAQA,MAAa,gBAAyC,EACpD,OACA,MACA,cACA,WAMC;EACD,OAAO,MAAM,KAAK,OAAO,KAAK,WAAW,OAAO,KAAK,aAAa;GAChE;GACA;GACA;GACA,UAAU;EAEZ,CAAQ;CACV;CAEA,MAAc,iBAAiB,OAAoC;EACjE,MAAM,EACJ,QAAQ,EAAE,OAAO,eACf,MAAM,MAAM,MAAM,MAAM,SAAS;EAErC,MAAM,aAAY,MADA,KAAK,YAAY,OAAO,QAAQ,EAAA,CAC5B,iBAAiB,oBAAoB;EAC3D,IAAI,CAAC,WACH,OAAO;EAoBT,OAAO;GAAE,aAjBW,UAAU,SAAS,aAAkB;IACvD,MAAM,gBACJ,MAAM,MAIN,MAAM,SAAS,KAAK,OAAO,eAAe,SAAS,EAAE;IAEvD,OAAO,gBACH,CACE;KACE,QAAQ,cAAc;KACtB,cAAc,cAAc,OAAO;IACrC,CACF,IACA,CAAC;GACP,CACmB;GAAG;GAAO;EAAS;CACxC;CAEA,MAAc,WAAW,EACvB,cACA,OACA,OACA,UACA,YACW;EAQX,OAPY,KAAK,OAAO,KAAK,QAAQ,KAAK,OAAO,UAAU;GACzD,cAAc,KAAK;GACnB;GACA;GACA;GACA;EACF,CACS;CACX;CAEA,MAAc,oBAAoB,OAAiB;EACjD,MAAM,SAAkC;GACtC;GACA;GACA;GACA;EACF,CAAC,CACE,QAAQ,QAAQ,OAAO,KAAK,CAAC,CAC7B,QACE,KAAK,QAAQ;GACZ,MAAM,SAAS;GACf,OAAO,OAAO,MAAM;GACpB,OAAO;EACT,GACA,CAAC,CACH;EAMF,OALY,KAAK,OAAO,KAAK,QAAQ,aAAa;GAChD,GAAG;GACH,QAAQ,MAAM;GACd,cAAc,KAAK;EACrB,CACS;CACX;CAEA,MAAc,YAAY,OAAe,UAAkB;EACzD,IAAI,aAAa;EAEjB,IAAI,MAAM,CAAC;EACX,OAAO,YAAY;GACjB,MAAM,MAAM,KAAK,OAAO,KAAK,QAAQ,KAAK,SAAS,OAAO,EACxD,WAAW,SACb,CAAC;GACD,aAAa,CAAC,eAAe,QAAQ,CAAC,CAAC,SAAS,IAAI,MAAM;GAC1D,IAAI,YACF,MAAME,kBAAAA,MAAM,KAAK,cAAc;EAEnC;EACA,OAAO;CACT;CAOA,MAAc,aACZ,OACA,UAMA;EACA,MAAM,MAAM,MAAM,KAAK,YAAY,OAAO,QAAQ;EAClD,IAAI,IAAI,WAAW,aAAa;GAI9B,MAAM,eAAc,MAHG,KAAK,OAAO,KAAK,QAAQ,SAAS,KAAK,UAAU,EACtE,OAAO,OACT,CAAC,EAAA,CAC4B,KAAK,QAAQ,QAAQ,IAAI,WAAW,KAAK;GACtE,IAAI,CAAC,KAAK,SACR,OAAO;GAET,MAAM,SAAS,YAAY,SAAS,QAAQ,IAAI,OAAO;GACvD,IAAI,OAAO,OAAO,SAAS,KAAK,SAAS,MAAM,GAI7C,OAAO;IACL,cAAc;KACZ,QALiB,OAClB,KAAK,SAAS,KAAK,SAAS,UAAU,KAAK,KAAK,KAAK,CAAC,CACtD,KAAK,IAGe;KACnB;KACA;IACF;IACA,KAAK;IACL;IACA;GACF;EAEJ,OAAO,IAAI,IAAI,WAAW,mBAAmB;GAC3C,IAAI,CAAC,KAAK,SACR,OAAO,IAAI,iBAAiB,oBAAoB,cAAc,CAAC;GAEjE,MAAM,UAAmC,CAAC;GAC1C,IAAI,iBAAiB,oBAAoB,WAAW,SAEjD,SAAc;IACb,MAAM,eAAe,KAAK;IAC1B,MAAM,OAAO,KAAK,MAAM,aAAa,SAAS;IAC9C,QAAQ,KAAK;KACX,MAAM,aAAa;KACnB,WAAW;KACX,YAAY,KAAK;KACjB,KAAK;KACL;KACA;IACF,CAAC;GACH,CACF;GACA,OAAO;EACT;EACA,MAAM,UAAU,KAAK,UAAU,KAAK,MAAM,CAAC;EAC3C,MAAM,IAAI,MACR,yBAAyB,IAAI,OAAO,uBAAuB,SAC7D;CACF;AACF;AAEA,SAAgB,4BACd,MACgB;CAChB,OAAO;EACL,MAAM;EACN,UAAU;GACR,MAAM,KAAK;GACX,aAAa,KAAK;GAClB,aAAA,GAAA,4BAAA,mBAAA,CAA+B,KAAK,MAAM,KAAA,GAAA,kCAAA,aAAA,CACzB,KAAK,MAAM,IACxB,KAAK;EACX;CACF;AACF"}
+{"version":3,"file":"index.cjs","names":["Runnable","OpenAIClient","StructuredTool","sleep","isInteropZodSchema","toJsonSchema"],"sources":["../../../src/experimental/openai_assistant/index.ts"],"sourcesContent":["import {\n  type ClientOptions,\n  type OpenAIChatModelId,\n  OpenAIClient,\n} from \"@langchain/openai\";\nimport { StructuredTool, StructuredToolInterface } from \"@langchain/core/tools\";\nimport { Runnable, RunnableConfig } from \"@langchain/core/runnables\";\nimport { isInteropZodSchema } from \"@langchain/core/utils/types\";\nimport { toJsonSchema } from \"@langchain/core/utils/json_schema\";\nimport { ToolDefinition } from \"@langchain/core/language_models/base\";\nimport { sleep } from \"../../util/time.js\";\nimport type {\n  OpenAIAssistantFinish,\n  OpenAIAssistantAction,\n  OpenAIToolType,\n} from \"./schema.js\";\n\n// oxlint-disable-next-line @typescript-eslint/no-explicit-any\ntype ThreadMessage = any;\n// oxlint-disable-next-line @typescript-eslint/no-explicit-any\ntype RequiredActionFunctionToolCall = any;\n\ntype ExtractRunOutput<AsAgent extends boolean | undefined> =\n  AsAgent extends true\n    ? OpenAIAssistantFinish | OpenAIAssistantAction[]\n    : ThreadMessage[] | RequiredActionFunctionToolCall[];\n\nexport type OpenAIAssistantRunnableInput<\n  AsAgent extends boolean | undefined = undefined,\n> = {\n  client?: OpenAIClient;\n  clientOptions?: ClientOptions;\n  assistantId: string;\n  pollIntervalMs?: number;\n  asAgent?: AsAgent;\n};\n\nexport class OpenAIAssistantRunnable<\n  AsAgent extends boolean | undefined,\n  // oxlint-disable-next-line @typescript-eslint/no-explicit-any\n  RunInput extends Record<string, any> = Record<string, any>,\n> extends Runnable<RunInput, ExtractRunOutput<AsAgent>> {\n  lc_namespace = [\"langchain\", \"experimental\", \"openai_assistant\"];\n\n  private client: OpenAIClient;\n\n  assistantId: string;\n\n  pollIntervalMs = 1000;\n\n  asAgent?: AsAgent;\n\n  constructor(fields: OpenAIAssistantRunnableInput<AsAgent>) {\n    super(fields);\n    this.client = fields.client ?? new OpenAIClient(fields?.clientOptions);\n    this.assistantId = fields.assistantId;\n    this.asAgent = fields.asAgent ?? this.asAgent;\n  }\n\n  static async createAssistant<AsAgent extends boolean>({\n    model,\n    name,\n    instructions,\n    tools,\n    client,\n    clientOptions,\n    asAgent,\n    pollIntervalMs,\n    fileIds,\n  }: Omit<OpenAIAssistantRunnableInput<AsAgent>, \"assistantId\"> & {\n    model: OpenAIChatModelId;\n    name?: string;\n    instructions?: string;\n    tools?: OpenAIToolType | Array<StructuredTool>;\n    fileIds?: string[];\n  }) {\n    const formattedTools =\n      tools?.map((tool) => {\n        // oxlint-disable-next-line no-instanceof/no-instanceof\n        if (tool instanceof StructuredTool) {\n          return formatToOpenAIAssistantTool(tool);\n        }\n        return tool;\n      }) ?? [];\n    const oaiClient = client ?? new OpenAIClient(clientOptions);\n    const assistant = await oaiClient.beta.assistants.create({\n      name,\n      instructions,\n      tools: formattedTools,\n      model,\n      file_ids: fileIds,\n      // oxlint-disable-next-line @typescript-eslint/no-explicit-any\n    } as any);\n\n    return new this({\n      client: oaiClient,\n      assistantId: assistant.id,\n      asAgent,\n      pollIntervalMs,\n    });\n  }\n\n  async invoke(\n    input: RunInput,\n    _options?: RunnableConfig\n  ): Promise<ExtractRunOutput<AsAgent>> {\n    // oxlint-disable-next-line @typescript-eslint/no-explicit-any\n    let run: any;\n    if (this.asAgent && input.steps && input.steps.length > 0) {\n      const parsedStepsInput = await this._parseStepsInput(input);\n      run = await this.client.beta.threads.runs.submitToolOutputs(\n        parsedStepsInput.runId,\n        {\n          thread_id: parsedStepsInput.threadId,\n          tool_outputs: parsedStepsInput.toolOutputs,\n        }\n      );\n    } else if (!(\"threadId\" in input)) {\n      const thread = {\n        messages: [\n          {\n            role: \"user\",\n            content: input.content,\n            attachments: input.attachments,\n            metadata: input.messagesMetadata,\n          },\n        ],\n        metadata: input.threadMetadata,\n      };\n      run = await this._createThreadAndRun({\n        ...input,\n        thread,\n      });\n    } else if (!(\"runId\" in input)) {\n      await this.client.beta.threads.messages.create(input.threadId, {\n        content: input.content,\n        role: \"user\",\n        attachments: input.attachments,\n        metadata: input.messagesMetadata,\n        // oxlint-disable-next-line @typescript-eslint/no-explicit-any\n      } as any);\n      run = await this._createRun(input);\n    } else {\n      // Submitting tool outputs to an existing run, outside the AgentExecutor\n      // framework.\n      run = await this.client.beta.threads.runs.submitToolOutputs(input.runId, {\n        thread_id: input.threadId,\n        tool_outputs: input.toolOutputs,\n      });\n    }\n\n    return this._getResponse(run.id, run.thread_id);\n  }\n\n  /**\n   * Delete an assistant.\n   *\n   * @link {https://platform.openai.com/docs/api-reference/assistants/deleteAssistant}\n   * @returns {Promise<AssistantDeleted>}\n   */\n  public async deleteAssistant() {\n    return await this.client.beta.assistants.delete(this.assistantId);\n  }\n\n  /**\n   * Retrieves an assistant.\n   *\n   * @link {https://platform.openai.com/docs/api-reference/assistants/getAssistant}\n   * @returns {Promise<OpenAIClient.Beta.Assistants.Assistant>}\n   */\n  public async getAssistant() {\n    return await this.client.beta.assistants.retrieve(this.assistantId);\n  }\n\n  /**\n   * Modifies an assistant.\n   *\n   * @link {https://platform.openai.com/docs/api-reference/assistants/modifyAssistant}\n   * @returns {Promise<OpenAIClient.Beta.Assistants.Assistant>}\n   */\n  public async modifyAssistant<AsAgent extends boolean>({\n    model,\n    name,\n    instructions,\n    fileIds,\n  }: Omit<OpenAIAssistantRunnableInput<AsAgent>, \"assistantId\" | \"tools\"> & {\n    model?: OpenAIChatModelId;\n    name?: string;\n    instructions?: string;\n    fileIds?: string[];\n  }) {\n    return await this.client.beta.assistants.update(this.assistantId, {\n      name,\n      instructions,\n      model,\n      file_ids: fileIds,\n      // oxlint-disable-next-line @typescript-eslint/no-explicit-any\n    } as any);\n  }\n\n  private async _parseStepsInput(input: RunInput): Promise<RunInput> {\n    const {\n      action: { runId, threadId },\n    } = input.steps[input.steps.length - 1];\n    const run = await this._waitForRun(runId, threadId);\n    const toolCalls = run.required_action?.submit_tool_outputs.tool_calls;\n    if (!toolCalls) {\n      return input;\n    }\n    // oxlint-disable-next-line @typescript-eslint/no-explicit-any\n    const toolOutputs = toolCalls.flatMap((toolCall: any) => {\n      const matchedAction = (\n        input.steps as {\n          action: OpenAIAssistantAction;\n          observation: string;\n        }[]\n      ).find((step) => step.action.toolCallId === toolCall.id);\n\n      return matchedAction\n        ? [\n            {\n              output: matchedAction.observation,\n              tool_call_id: matchedAction.action.toolCallId,\n            },\n          ]\n        : [];\n    });\n    return { toolOutputs, runId, threadId } as unknown as RunInput;\n  }\n\n  private async _createRun({\n    instructions,\n    model,\n    tools,\n    metadata,\n    threadId,\n  }: RunInput) {\n    const run = this.client.beta.threads.runs.create(threadId, {\n      assistant_id: this.assistantId,\n      instructions,\n      model,\n      tools,\n      metadata,\n    });\n    return run;\n  }\n\n  private async _createThreadAndRun(input: RunInput) {\n    const params: Record<string, unknown> = [\n      \"instructions\",\n      \"model\",\n      \"tools\",\n      \"run_metadata\",\n    ]\n      .filter((key) => key in input)\n      .reduce(\n        (obj, key) => {\n          const newObj = obj;\n          newObj[key] = input[key];\n          return newObj;\n        },\n        {} as Record<string, unknown>\n      );\n    const run = this.client.beta.threads.createAndRun({\n      ...params,\n      thread: input.thread,\n      assistant_id: this.assistantId,\n    });\n    return run;\n  }\n\n  private async _waitForRun(runId: string, threadId: string) {\n    let inProgress = true;\n    // oxlint-disable-next-line @typescript-eslint/no-explicit-any\n    let run = {} as any;\n    while (inProgress) {\n      run = await this.client.beta.threads.runs.retrieve(runId, {\n        thread_id: threadId,\n      });\n      inProgress = [\"in_progress\", \"queued\"].includes(run.status);\n      if (inProgress) {\n        await sleep(this.pollIntervalMs);\n      }\n    }\n    return run;\n  }\n\n  private async _getResponse(\n    runId: string,\n    threadId: string\n  ): Promise<ExtractRunOutput<AsAgent>>;\n\n  private async _getResponse(\n    runId: string,\n    threadId: string\n  ): Promise<\n    | OpenAIAssistantFinish\n    | OpenAIAssistantAction[]\n    | ThreadMessage[]\n    | RequiredActionFunctionToolCall[]\n  > {\n    const run = await this._waitForRun(runId, threadId);\n    if (run.status === \"completed\") {\n      const messages = await this.client.beta.threads.messages.list(threadId, {\n        order: \"desc\",\n      });\n      const newMessages = messages.data.filter((msg) => msg.run_id === runId);\n      if (!this.asAgent) {\n        return newMessages;\n      }\n      const answer = newMessages.flatMap((msg) => msg.content);\n      if (answer.every((item) => item.type === \"text\")) {\n        const answerString = answer\n          .map((item) => item.type === \"text\" && item.text.value)\n          .join(\"\\n\");\n        return {\n          returnValues: {\n            output: answerString,\n            runId,\n            threadId,\n          },\n          log: \"\",\n          runId,\n          threadId,\n        };\n      }\n    } else if (run.status === \"requires_action\") {\n      if (!this.asAgent) {\n        return run.required_action?.submit_tool_outputs.tool_calls ?? [];\n      }\n      const actions: OpenAIAssistantAction[] = [];\n      run.required_action?.submit_tool_outputs.tool_calls.forEach(\n        // oxlint-disable-next-line @typescript-eslint/no-explicit-any\n        (item: any) => {\n          const functionCall = item.function;\n          const args = JSON.parse(functionCall.arguments);\n          actions.push({\n            tool: functionCall.name,\n            toolInput: args,\n            toolCallId: item.id,\n            log: \"\",\n            runId,\n            threadId,\n          });\n        }\n      );\n      return actions;\n    }\n    const runInfo = JSON.stringify(run, null, 2);\n    throw new Error(\n      `Unexpected run status ${run.status}.\\nFull run info:\\n\\n${runInfo}`\n    );\n  }\n}\n\nexport function formatToOpenAIAssistantTool(\n  tool: StructuredToolInterface\n): ToolDefinition {\n  return {\n    type: \"function\",\n    function: {\n      name: tool.name,\n      description: tool.description,\n      parameters: isInteropZodSchema(tool.schema)\n        ? toJsonSchema(tool.schema)\n        : tool.schema,\n    },\n  };\n}\n"],"mappings":";;;;;;;;;;;;;AAqCA,IAAa,0BAAb,cAIUA,0BAAAA,SAA8C;CACtD,eAAe;EAAC;EAAa;EAAgB;CAAkB;CAE/D;CAEA;CAEA,iBAAiB;CAEjB;CAEA,YAAY,QAA+C;EACzD,MAAM,MAAM;EACZ,KAAK,SAAS,OAAO,UAAU,IAAIC,kBAAAA,aAAa,QAAQ,aAAa;EACrE,KAAK,cAAc,OAAO;EAC1B,KAAK,UAAU,OAAO,WAAW,KAAK;CACxC;CAEA,aAAa,gBAAyC,EACpD,OACA,MACA,cACA,OACA,QACA,eACA,SACA,gBACA,WAOC;EACD,MAAM,iBACJ,OAAO,KAAK,SAAS;GAEnB,IAAI,gBAAgBC,sBAAAA,gBAClB,OAAO,4BAA4B,IAAI;GAEzC,OAAO;EACT,CAAC,KAAK,CAAC;EACT,MAAM,YAAY,UAAU,IAAID,kBAAAA,aAAa,aAAa;EAC1D,MAAM,YAAY,MAAM,UAAU,KAAK,WAAW,OAAO;GACvD;GACA;GACA,OAAO;GACP;GACA,UAAU;EAEZ,CAAQ;EAER,OAAO,IAAI,KAAK;GACd,QAAQ;GACR,aAAa,UAAU;GACvB;GACA;EACF,CAAC;CACH;CAEA,MAAM,OACJ,OACA,UACoC;EAEpC,IAAI;EACJ,IAAI,KAAK,WAAW,MAAM,SAAS,MAAM,MAAM,SAAS,GAAG;GACzD,MAAM,mBAAmB,MAAM,KAAK,iBAAiB,KAAK;GAC1D,MAAM,MAAM,KAAK,OAAO,KAAK,QAAQ,KAAK,kBACxC,iBAAiB,OACjB;IACE,WAAW,iBAAiB;IAC5B,cAAc,iBAAiB;GACjC,CACF;EACF,OAAO,IAAI,EAAE,cAAc,QAAQ;GACjC,MAAM,SAAS;IACb,UAAU,CACR;KACE,MAAM;KACN,SAAS,MAAM;KACf,aAAa,MAAM;KACnB,UAAU,MAAM;IAClB,CACF;IACA,UAAU,MAAM;GAClB;GACA,MAAM,MAAM,KAAK,oBAAoB;IACnC,GAAG;IACH;GACF,CAAC;EACH,OAAO,IAAI,EAAE,WAAW,QAAQ;GAC9B,MAAM,KAAK,OAAO,KAAK,QAAQ,SAAS,OAAO,MAAM,UAAU;IAC7D,SAAS,MAAM;IACf,MAAM;IACN,aAAa,MAAM;IACnB,UAAU,MAAM;GAElB,CAAQ;GACR,MAAM,MAAM,KAAK,WAAW,KAAK;EACnC,OAGE,MAAM,MAAM,KAAK,OAAO,KAAK,QAAQ,KAAK,kBAAkB,MAAM,OAAO;GACvE,WAAW,MAAM;GACjB,cAAc,MAAM;EACtB,CAAC;EAGH,OAAO,KAAK,aAAa,IAAI,IAAI,IAAI,SAAS;CAChD;;;;;;;CAQA,MAAa,kBAAkB;EAC7B,OAAO,MAAM,KAAK,OAAO,KAAK,WAAW,OAAO,KAAK,WAAW;CAClE;;;;;;;CAQA,MAAa,eAAe;EAC1B,OAAO,MAAM,KAAK,OAAO,KAAK,WAAW,SAAS,KAAK,WAAW;CACpE;;;;;;;CAQA,MAAa,gBAAyC,EACpD,OACA,MACA,cACA,WAMC;EACD,OAAO,MAAM,KAAK,OAAO,KAAK,WAAW,OAAO,KAAK,aAAa;GAChE;GACA;GACA;GACA,UAAU;EAEZ,CAAQ;CACV;CAEA,MAAc,iBAAiB,OAAoC;EACjE,MAAM,EACJ,QAAQ,EAAE,OAAO,eACf,MAAM,MAAM,MAAM,MAAM,SAAS;EAErC,MAAM,aAAY,MADA,KAAK,YAAY,OAAO,QAAQ,EAAA,CAC5B,iBAAiB,oBAAoB;EAC3D,IAAI,CAAC,WACH,OAAO;EAoBT,OAAO;GAAE,aAjBW,UAAU,SAAS,aAAkB;IACvD,MAAM,gBACJ,MAAM,MAIN,MAAM,SAAS,KAAK,OAAO,eAAe,SAAS,EAAE;IAEvD,OAAO,gBACH,CACE;KACE,QAAQ,cAAc;KACtB,cAAc,cAAc,OAAO;IACrC,CACF,IACA,CAAC;GACP,CACmB;GAAG;GAAO;EAAS;CACxC;CAEA,MAAc,WAAW,EACvB,cACA,OACA,OACA,UACA,YACW;EAQX,OAPY,KAAK,OAAO,KAAK,QAAQ,KAAK,OAAO,UAAU;GACzD,cAAc,KAAK;GACnB;GACA;GACA;GACA;EACF,CACS;CACX;CAEA,MAAc,oBAAoB,OAAiB;EACjD,MAAM,SAAkC;GACtC;GACA;GACA;GACA;EACF,CAAC,CACE,QAAQ,QAAQ,OAAO,KAAK,CAAC,CAC7B,QACE,KAAK,QAAQ;GACZ,MAAM,SAAS;GACf,OAAO,OAAO,MAAM;GACpB,OAAO;EACT,GACA,CAAC,CACH;EAMF,OALY,KAAK,OAAO,KAAK,QAAQ,aAAa;GAChD,GAAG;GACH,QAAQ,MAAM;GACd,cAAc,KAAK;EACrB,CACS;CACX;CAEA,MAAc,YAAY,OAAe,UAAkB;EACzD,IAAI,aAAa;EAEjB,IAAI,MAAM,CAAC;EACX,OAAO,YAAY;GACjB,MAAM,MAAM,KAAK,OAAO,KAAK,QAAQ,KAAK,SAAS,OAAO,EACxD,WAAW,SACb,CAAC;GACD,aAAa,CAAC,eAAe,QAAQ,CAAC,CAAC,SAAS,IAAI,MAAM;GAC1D,IAAI,YACF,MAAME,kBAAAA,MAAM,KAAK,cAAc;EAEnC;EACA,OAAO;CACT;CAOA,MAAc,aACZ,OACA,UAMA;EACA,MAAM,MAAM,MAAM,KAAK,YAAY,OAAO,QAAQ;EAClD,IAAI,IAAI,WAAW,aAAa;GAI9B,MAAM,eAAc,MAHG,KAAK,OAAO,KAAK,QAAQ,SAAS,KAAK,UAAU,EACtE,OAAO,OACT,CAAC,EAAA,CAC4B,KAAK,QAAQ,QAAQ,IAAI,WAAW,KAAK;GACtE,IAAI,CAAC,KAAK,SACR,OAAO;GAET,MAAM,SAAS,YAAY,SAAS,QAAQ,IAAI,OAAO;GACvD,IAAI,OAAO,OAAO,SAAS,KAAK,SAAS,MAAM,GAI7C,OAAO;IACL,cAAc;KACZ,QALiB,OAClB,KAAK,SAAS,KAAK,SAAS,UAAU,KAAK,KAAK,KAAK,CAAC,CACtD,KAAK,IAGe;KACnB;KACA;IACF;IACA,KAAK;IACL;IACA;GACF;EAEJ,OAAO,IAAI,IAAI,WAAW,mBAAmB;GAC3C,IAAI,CAAC,KAAK,SACR,OAAO,IAAI,iBAAiB,oBAAoB,cAAc,CAAC;GAEjE,MAAM,UAAmC,CAAC;GAC1C,IAAI,iBAAiB,oBAAoB,WAAW,SAEjD,SAAc;IACb,MAAM,eAAe,KAAK;IAC1B,MAAM,OAAO,KAAK,MAAM,aAAa,SAAS;IAC9C,QAAQ,KAAK;KACX,MAAM,aAAa;KACnB,WAAW;KACX,YAAY,KAAK;KACjB,KAAK;KACL;KACA;IACF,CAAC;GACH,CACF;GACA,OAAO;EACT;EACA,MAAM,UAAU,KAAK,UAAU,KAAK,MAAM,CAAC;EAC3C,MAAM,IAAI,MACR,yBAAyB,IAAI,OAAO,uBAAuB,SAC7D;CACF;AACF;AAEA,SAAgB,4BACd,MACgB;CAChB,OAAO;EACL,MAAM;EACN,UAAU;GACR,MAAM,KAAK;GACX,aAAa,KAAK;GAClB,aAAA,GAAYC,4BAAAA,mBAAAA,CAAmB,KAAK,MAAM,KAAA,GACtCC,kCAAAA,aAAAA,CAAa,KAAK,MAAM,IACxB,KAAK;EACX;CACF;AACF"}

package/dist/experimental/plan_and_execute/agent_executor.cjs

@@ -53,10 +53,11 @@ var PlanAndExecuteAgentExecutor = class PlanAndExecuteAgentExecutor extends requ
 	* @returns A new LLMPlanner instance.
 	*/
 	static async getDefaultPlanner({ llm, tools }) {
-		return new require_base$1.LLMPlanner(new require_llm_chain.LLMChain({
+		const plannerLlmChain = new require_llm_chain.LLMChain({
 			llm,
 			prompt: await require_prompt.getPlannerChatPrompt(tools)
-		}), new require_outputParser.PlanOutputParser());
+		});
+		return new require_base$1.LLMPlanner(plannerLlmChain, new require_outputParser.PlanOutputParser());
 	}
 	/**
 	* Static method that returns a default step executor for the agent. It

package/dist/experimental/plan_and_execute/agent_executor.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"agent_executor.cjs","names":["BaseChain","ListStepContainer","LLMPlanner","LLMChain","getPlannerChatPrompt","PlanOutputParser","DEFAULT_STEP_EXECUTOR_HUMAN_CHAT_MESSAGE_TEMPLATE","StructuredChatAgent","ChainStepExecutor","AgentExecutor","ChatAgent"],"sources":["../../../src/experimental/plan_and_execute/agent_executor.ts"],"sourcesContent":["import type { BaseLanguageModelInterface } from \"@langchain/core/language_models/base\";\nimport { ChainValues } from \"@langchain/core/utils/types\";\nimport { Tool, DynamicStructuredTool } from \"@langchain/core/tools\";\nimport { CallbackManagerForChainRun } from \"@langchain/core/callbacks/manager\";\nimport { BaseChain, ChainInputs } from \"../../chains/base.js\";\nimport {\n  BasePlanner,\n  BaseStepContainer,\n  BaseStepExecutor,\n  ListStepContainer,\n  LLMPlanner,\n  ChainStepExecutor,\n} from \"./base.js\";\nimport { AgentExecutor } from \"../../agents/executor.js\";\nimport {\n  DEFAULT_STEP_EXECUTOR_HUMAN_CHAT_MESSAGE_TEMPLATE,\n  getPlannerChatPrompt,\n} from \"./prompt.js\";\nimport { LLMChain } from \"../../chains/llm_chain.js\";\nimport { PlanOutputParser } from \"./outputParser.js\";\nimport { ChatAgent } from \"../../agents/chat/index.js\";\nimport { StructuredChatAgent } from \"../../agents/index.js\";\nimport { SerializedLLMChain } from \"../../chains/serde.js\";\n\n/**\n * A utility function to distiguish a dynamicstructuredtool over other tools.\n * @param tool the tool to test\n * @returns bool\n */\nexport function isDynamicStructuredTool(\n  tool: Tool | DynamicStructuredTool\n): tool is DynamicStructuredTool {\n  // We check for the existence of the static lc_name method in the object's constructor\n  return (\n    // oxlint-disable-next-line @typescript-eslint/no-explicit-any\n    typeof (tool.constructor as any).lc_name === \"function\" &&\n    // oxlint-disable-next-line @typescript-eslint/no-explicit-any\n    (tool.constructor as any).lc_name() === \"DynamicStructuredTool\"\n  );\n}\n\n/**\n * Interface for the input to the PlanAndExecuteAgentExecutor class. It\n * extends ChainInputs and includes additional properties for the planner,\n * step executor, step container, and input and output keys.\n */\nexport interface PlanAndExecuteAgentExecutorInput extends ChainInputs {\n  planner: BasePlanner;\n  stepExecutor: BaseStepExecutor;\n  stepContainer?: BaseStepContainer;\n  inputKey?: string;\n  outputKey?: string;\n}\n\n/**\n * Class representing a plan-and-execute agent executor. This agent\n * decides on the full sequence of actions upfront, then executes them all\n * without updating the plan. This is suitable for complex or long-running\n * tasks that require maintaining long-term objectives and focus.\n */\nexport class PlanAndExecuteAgentExecutor extends BaseChain {\n  static lc_name() {\n    return \"PlanAndExecuteAgentExecutor\";\n  }\n\n  private planner: BasePlanner;\n\n  private stepExecutor: BaseStepExecutor;\n\n  private stepContainer: BaseStepContainer = new ListStepContainer();\n\n  private inputKey = \"input\";\n\n  private outputKey = \"output\";\n\n  constructor(input: PlanAndExecuteAgentExecutorInput) {\n    super(input);\n    this.planner = input.planner;\n    this.stepExecutor = input.stepExecutor;\n    this.stepContainer = input.stepContainer ?? this.stepContainer;\n    this.inputKey = input.inputKey ?? this.inputKey;\n    this.outputKey = input.outputKey ?? this.outputKey;\n  }\n\n  get inputKeys() {\n    return [this.inputKey];\n  }\n\n  get outputKeys() {\n    return [this.outputKey];\n  }\n\n  /**\n   * Static method that returns a default planner for the agent. It creates\n   * a new LLMChain with a given LLM and a fixed prompt, and uses it to\n   * create a new LLMPlanner with a PlanOutputParser.\n   * @param llm The Large Language Model (LLM) used to generate responses.\n   * @returns A new LLMPlanner instance.\n   */\n\n  static async getDefaultPlanner({\n    llm,\n    tools,\n  }: {\n    llm: BaseLanguageModelInterface;\n    tools: Tool[] | DynamicStructuredTool[];\n  }) {\n    const plannerLlmChain = new LLMChain({\n      llm,\n      prompt: await getPlannerChatPrompt(tools),\n    });\n    return new LLMPlanner(plannerLlmChain, new PlanOutputParser());\n  }\n\n  /**\n   * Static method that returns a default step executor for the agent. It\n   * creates a new ChatAgent from a given LLM and a set of tools, and uses\n   * it to create a new ChainStepExecutor.\n   * @param llm The Large Language Model (LLM) used to generate responses.\n   * @param tools The set of tools used by the agent.\n   * @param humanMessageTemplate The template for human messages. If not provided, a default template is used.\n   * @returns A new ChainStepExecutor instance.\n   */\n  static getDefaultStepExecutor({\n    llm,\n    tools,\n    humanMessageTemplate = DEFAULT_STEP_EXECUTOR_HUMAN_CHAT_MESSAGE_TEMPLATE,\n  }: {\n    llm: BaseLanguageModelInterface;\n    tools: Tool[] | DynamicStructuredTool[];\n    humanMessageTemplate?: string;\n  }) {\n    let agent;\n\n    if (tools.length > 0 && isDynamicStructuredTool(tools[0])) {\n      agent = StructuredChatAgent.fromLLMAndTools(llm, tools, {\n        humanMessageTemplate,\n        inputVariables: [\"previous_steps\", \"current_step\", \"agent_scratchpad\"],\n      });\n      return new ChainStepExecutor(\n        AgentExecutor.fromAgentAndTools({\n          agent,\n          tools,\n        })\n      );\n    }\n\n    agent = ChatAgent.fromLLMAndTools(llm, tools as Tool[], {\n      humanMessageTemplate,\n    });\n    return new ChainStepExecutor(\n      AgentExecutor.fromAgentAndTools({\n        agent,\n        tools,\n      })\n    );\n  }\n\n  /**\n   * Static method that creates a new PlanAndExecuteAgentExecutor from a\n   * given LLM, a set of tools, and optionally a human message template. It\n   * uses the getDefaultPlanner and getDefaultStepExecutor methods to create\n   * the planner and step executor for the new agent executor.\n   * @param llm The Large Language Model (LLM) used to generate responses.\n   * @param tools The set of tools used by the agent.\n   * @param humanMessageTemplate The template for human messages. If not provided, a default template is used.\n   * @returns A new PlanAndExecuteAgentExecutor instance.\n   */\n  static async fromLLMAndTools({\n    llm,\n    tools,\n    humanMessageTemplate,\n  }: {\n    llm: BaseLanguageModelInterface;\n    tools: Tool[] | DynamicStructuredTool[];\n    humanMessageTemplate?: string;\n  } & Omit<PlanAndExecuteAgentExecutorInput, \"planner\" | \"stepExecutor\">) {\n    const executor = new PlanAndExecuteAgentExecutor({\n      planner: await PlanAndExecuteAgentExecutor.getDefaultPlanner({\n        llm,\n        tools,\n      }),\n      stepExecutor: PlanAndExecuteAgentExecutor.getDefaultStepExecutor({\n        llm,\n        tools,\n        humanMessageTemplate,\n      }),\n    });\n    return executor;\n  }\n\n  /** @ignore */\n  async _call(\n    inputs: ChainValues,\n    runManager?: CallbackManagerForChainRun\n  ): Promise<ChainValues> {\n    const plan = await this.planner.plan(inputs.input, runManager?.getChild());\n    if (!plan.steps?.length) {\n      throw new Error(\n        \"Could not create and parse a plan to answer your question - please try again.\"\n      );\n    }\n    plan.steps[plan.steps.length - 1].text +=\n      ` The original question was: ${inputs.input}.`;\n    for (const step of plan.steps) {\n      const newInputs = {\n        ...inputs,\n        previous_steps: JSON.stringify(this.stepContainer.getSteps()),\n        current_step: step.text,\n      };\n      const response = await this.stepExecutor.step(\n        newInputs,\n        runManager?.getChild()\n      );\n      this.stepContainer.addStep(step, response);\n    }\n    return { [this.outputKey]: this.stepContainer.getFinalResponse() };\n  }\n\n  _chainType() {\n    return \"agent_executor\" as const;\n  }\n\n  serialize(): SerializedLLMChain {\n    throw new Error(\"Cannot serialize an AgentExecutor\");\n  }\n}\n"],"mappings":";;;;;;;;;;;;;;;AA6BA,SAAgB,wBACd,MAC+B;CAE/B,OAEE,OAAQ,KAAK,YAAoB,YAAY,cAE5C,KAAK,YAAoB,QAAQ,MAAM;AAE5C;;;;;;;AAqBA,IAAa,8BAAb,MAAa,oCAAoCA,aAAAA,UAAU;CACzD,OAAO,UAAU;EACf,OAAO;CACT;CAEA;CAEA;CAEA,gBAA2C,IAAIC,eAAAA,kBAAkB;CAEjE,WAAmB;CAEnB,YAAoB;CAEpB,YAAY,OAAyC;EACnD,MAAM,KAAK;EACX,KAAK,UAAU,MAAM;EACrB,KAAK,eAAe,MAAM;EAC1B,KAAK,gBAAgB,MAAM,iBAAiB,KAAK;EACjD,KAAK,WAAW,MAAM,YAAY,KAAK;EACvC,KAAK,YAAY,MAAM,aAAa,KAAK;CAC3C;CAEA,IAAI,YAAY;EACd,OAAO,CAAC,KAAK,QAAQ;CACvB;CAEA,IAAI,aAAa;EACf,OAAO,CAAC,KAAK,SAAS;CACxB;;;;;;;;CAUA,aAAa,kBAAkB,EAC7B,KACA,SAIC;EAKD,OAAO,IAAIC,eAAAA,WAAW,IAJMC,kBAAAA,SAAS;GACnC;GACA,QAAQ,MAAMC,eAAAA,qBAAqB,KAAK;EAC1C,CACoC,GAAG,IAAIC,qBAAAA,iBAAiB,CAAC;CAC/D;;;;;;;;;;CAWA,OAAO,uBAAuB,EAC5B,KACA,OACA,uBAAuBC,eAAAA,qDAKtB;EACD,IAAI;EAEJ,IAAI,MAAM,SAAS,KAAK,wBAAwB,MAAM,EAAE,GAAG;GACzD,QAAQC,gBAAAA,oBAAoB,gBAAgB,KAAK,OAAO;IACtD;IACA,gBAAgB;KAAC;KAAkB;KAAgB;IAAkB;GACvE,CAAC;GACD,OAAO,IAAIC,eAAAA,kBACTC,iBAAAA,cAAc,kBAAkB;IAC9B;IACA;GACF,CAAC,CACH;EACF;EAEA,QAAQC,cAAAA,UAAU,gBAAgB,KAAK,OAAiB,EACtD,qBACF,CAAC;EACD,OAAO,IAAIF,eAAAA,kBACTC,iBAAAA,cAAc,kBAAkB;GAC9B;GACA;EACF,CAAC,CACH;CACF;;;;;;;;;;;CAYA,aAAa,gBAAgB,EAC3B,KACA,OACA,wBAKsE;EAYtE,OAAO,IAXc,4BAA4B;GAC/C,SAAS,MAAM,4BAA4B,kBAAkB;IAC3D;IACA;GACF,CAAC;GACD,cAAc,4BAA4B,uBAAuB;IAC/D;IACA;IACA;GACF,CAAC;EACH,CACc;CAChB;;CAGA,MAAM,MACJ,QACA,YACsB;EACtB,MAAM,OAAO,MAAM,KAAK,QAAQ,KAAK,OAAO,OAAO,YAAY,SAAS,CAAC;EACzE,IAAI,CAAC,KAAK,OAAO,QACf,MAAM,IAAI,MACR,+EACF;EAEF,KAAK,MAAM,KAAK,MAAM,SAAS,EAAE,CAAC,QAChC,+BAA+B,OAAO,MAAM;EAC9C,KAAK,MAAM,QAAQ,KAAK,OAAO;GAC7B,MAAM,YAAY;IAChB,GAAG;IACH,gBAAgB,KAAK,UAAU,KAAK,cAAc,SAAS,CAAC;IAC5D,cAAc,KAAK;GACrB;GACA,MAAM,WAAW,MAAM,KAAK,aAAa,KACvC,WACA,YAAY,SAAS,CACvB;GACA,KAAK,cAAc,QAAQ,MAAM,QAAQ;EAC3C;EACA,OAAO,GAAG,KAAK,YAAY,KAAK,cAAc,iBAAiB,EAAE;CACnE;CAEA,aAAa;EACX,OAAO;CACT;CAEA,YAAgC;EAC9B,MAAM,IAAI,MAAM,mCAAmC;CACrD;AACF"}
+{"version":3,"file":"agent_executor.cjs","names":["BaseChain","ListStepContainer","LLMChain","getPlannerChatPrompt","LLMPlanner","PlanOutputParser","DEFAULT_STEP_EXECUTOR_HUMAN_CHAT_MESSAGE_TEMPLATE","StructuredChatAgent","ChainStepExecutor","AgentExecutor","ChatAgent"],"sources":["../../../src/experimental/plan_and_execute/agent_executor.ts"],"sourcesContent":["import type { BaseLanguageModelInterface } from \"@langchain/core/language_models/base\";\nimport { ChainValues } from \"@langchain/core/utils/types\";\nimport { Tool, DynamicStructuredTool } from \"@langchain/core/tools\";\nimport { CallbackManagerForChainRun } from \"@langchain/core/callbacks/manager\";\nimport { BaseChain, ChainInputs } from \"../../chains/base.js\";\nimport {\n  BasePlanner,\n  BaseStepContainer,\n  BaseStepExecutor,\n  ListStepContainer,\n  LLMPlanner,\n  ChainStepExecutor,\n} from \"./base.js\";\nimport { AgentExecutor } from \"../../agents/executor.js\";\nimport {\n  DEFAULT_STEP_EXECUTOR_HUMAN_CHAT_MESSAGE_TEMPLATE,\n  getPlannerChatPrompt,\n} from \"./prompt.js\";\nimport { LLMChain } from \"../../chains/llm_chain.js\";\nimport { PlanOutputParser } from \"./outputParser.js\";\nimport { ChatAgent } from \"../../agents/chat/index.js\";\nimport { StructuredChatAgent } from \"../../agents/index.js\";\nimport { SerializedLLMChain } from \"../../chains/serde.js\";\n\n/**\n * A utility function to distiguish a dynamicstructuredtool over other tools.\n * @param tool the tool to test\n * @returns bool\n */\nexport function isDynamicStructuredTool(\n  tool: Tool | DynamicStructuredTool\n): tool is DynamicStructuredTool {\n  // We check for the existence of the static lc_name method in the object's constructor\n  return (\n    // oxlint-disable-next-line @typescript-eslint/no-explicit-any\n    typeof (tool.constructor as any).lc_name === \"function\" &&\n    // oxlint-disable-next-line @typescript-eslint/no-explicit-any\n    (tool.constructor as any).lc_name() === \"DynamicStructuredTool\"\n  );\n}\n\n/**\n * Interface for the input to the PlanAndExecuteAgentExecutor class. It\n * extends ChainInputs and includes additional properties for the planner,\n * step executor, step container, and input and output keys.\n */\nexport interface PlanAndExecuteAgentExecutorInput extends ChainInputs {\n  planner: BasePlanner;\n  stepExecutor: BaseStepExecutor;\n  stepContainer?: BaseStepContainer;\n  inputKey?: string;\n  outputKey?: string;\n}\n\n/**\n * Class representing a plan-and-execute agent executor. This agent\n * decides on the full sequence of actions upfront, then executes them all\n * without updating the plan. This is suitable for complex or long-running\n * tasks that require maintaining long-term objectives and focus.\n */\nexport class PlanAndExecuteAgentExecutor extends BaseChain {\n  static lc_name() {\n    return \"PlanAndExecuteAgentExecutor\";\n  }\n\n  private planner: BasePlanner;\n\n  private stepExecutor: BaseStepExecutor;\n\n  private stepContainer: BaseStepContainer = new ListStepContainer();\n\n  private inputKey = \"input\";\n\n  private outputKey = \"output\";\n\n  constructor(input: PlanAndExecuteAgentExecutorInput) {\n    super(input);\n    this.planner = input.planner;\n    this.stepExecutor = input.stepExecutor;\n    this.stepContainer = input.stepContainer ?? this.stepContainer;\n    this.inputKey = input.inputKey ?? this.inputKey;\n    this.outputKey = input.outputKey ?? this.outputKey;\n  }\n\n  get inputKeys() {\n    return [this.inputKey];\n  }\n\n  get outputKeys() {\n    return [this.outputKey];\n  }\n\n  /**\n   * Static method that returns a default planner for the agent. It creates\n   * a new LLMChain with a given LLM and a fixed prompt, and uses it to\n   * create a new LLMPlanner with a PlanOutputParser.\n   * @param llm The Large Language Model (LLM) used to generate responses.\n   * @returns A new LLMPlanner instance.\n   */\n\n  static async getDefaultPlanner({\n    llm,\n    tools,\n  }: {\n    llm: BaseLanguageModelInterface;\n    tools: Tool[] | DynamicStructuredTool[];\n  }) {\n    const plannerLlmChain = new LLMChain({\n      llm,\n      prompt: await getPlannerChatPrompt(tools),\n    });\n    return new LLMPlanner(plannerLlmChain, new PlanOutputParser());\n  }\n\n  /**\n   * Static method that returns a default step executor for the agent. It\n   * creates a new ChatAgent from a given LLM and a set of tools, and uses\n   * it to create a new ChainStepExecutor.\n   * @param llm The Large Language Model (LLM) used to generate responses.\n   * @param tools The set of tools used by the agent.\n   * @param humanMessageTemplate The template for human messages. If not provided, a default template is used.\n   * @returns A new ChainStepExecutor instance.\n   */\n  static getDefaultStepExecutor({\n    llm,\n    tools,\n    humanMessageTemplate = DEFAULT_STEP_EXECUTOR_HUMAN_CHAT_MESSAGE_TEMPLATE,\n  }: {\n    llm: BaseLanguageModelInterface;\n    tools: Tool[] | DynamicStructuredTool[];\n    humanMessageTemplate?: string;\n  }) {\n    let agent;\n\n    if (tools.length > 0 && isDynamicStructuredTool(tools[0])) {\n      agent = StructuredChatAgent.fromLLMAndTools(llm, tools, {\n        humanMessageTemplate,\n        inputVariables: [\"previous_steps\", \"current_step\", \"agent_scratchpad\"],\n      });\n      return new ChainStepExecutor(\n        AgentExecutor.fromAgentAndTools({\n          agent,\n          tools,\n        })\n      );\n    }\n\n    agent = ChatAgent.fromLLMAndTools(llm, tools as Tool[], {\n      humanMessageTemplate,\n    });\n    return new ChainStepExecutor(\n      AgentExecutor.fromAgentAndTools({\n        agent,\n        tools,\n      })\n    );\n  }\n\n  /**\n   * Static method that creates a new PlanAndExecuteAgentExecutor from a\n   * given LLM, a set of tools, and optionally a human message template. It\n   * uses the getDefaultPlanner and getDefaultStepExecutor methods to create\n   * the planner and step executor for the new agent executor.\n   * @param llm The Large Language Model (LLM) used to generate responses.\n   * @param tools The set of tools used by the agent.\n   * @param humanMessageTemplate The template for human messages. If not provided, a default template is used.\n   * @returns A new PlanAndExecuteAgentExecutor instance.\n   */\n  static async fromLLMAndTools({\n    llm,\n    tools,\n    humanMessageTemplate,\n  }: {\n    llm: BaseLanguageModelInterface;\n    tools: Tool[] | DynamicStructuredTool[];\n    humanMessageTemplate?: string;\n  } & Omit<PlanAndExecuteAgentExecutorInput, \"planner\" | \"stepExecutor\">) {\n    const executor = new PlanAndExecuteAgentExecutor({\n      planner: await PlanAndExecuteAgentExecutor.getDefaultPlanner({\n        llm,\n        tools,\n      }),\n      stepExecutor: PlanAndExecuteAgentExecutor.getDefaultStepExecutor({\n        llm,\n        tools,\n        humanMessageTemplate,\n      }),\n    });\n    return executor;\n  }\n\n  /** @ignore */\n  async _call(\n    inputs: ChainValues,\n    runManager?: CallbackManagerForChainRun\n  ): Promise<ChainValues> {\n    const plan = await this.planner.plan(inputs.input, runManager?.getChild());\n    if (!plan.steps?.length) {\n      throw new Error(\n        \"Could not create and parse a plan to answer your question - please try again.\"\n      );\n    }\n    plan.steps[plan.steps.length - 1].text +=\n      ` The original question was: ${inputs.input}.`;\n    for (const step of plan.steps) {\n      const newInputs = {\n        ...inputs,\n        previous_steps: JSON.stringify(this.stepContainer.getSteps()),\n        current_step: step.text,\n      };\n      const response = await this.stepExecutor.step(\n        newInputs,\n        runManager?.getChild()\n      );\n      this.stepContainer.addStep(step, response);\n    }\n    return { [this.outputKey]: this.stepContainer.getFinalResponse() };\n  }\n\n  _chainType() {\n    return \"agent_executor\" as const;\n  }\n\n  serialize(): SerializedLLMChain {\n    throw new Error(\"Cannot serialize an AgentExecutor\");\n  }\n}\n"],"mappings":";;;;;;;;;;;;;;;AA6BA,SAAgB,wBACd,MAC+B;CAE/B,OAEE,OAAQ,KAAK,YAAoB,YAAY,cAE5C,KAAK,YAAoB,QAAQ,MAAM;AAE5C;;;;;;;AAqBA,IAAa,8BAAb,MAAa,oCAAoCA,aAAAA,UAAU;CACzD,OAAO,UAAU;EACf,OAAO;CACT;CAEA;CAEA;CAEA,gBAA2C,IAAIC,eAAAA,kBAAkB;CAEjE,WAAmB;CAEnB,YAAoB;CAEpB,YAAY,OAAyC;EACnD,MAAM,KAAK;EACX,KAAK,UAAU,MAAM;EACrB,KAAK,eAAe,MAAM;EAC1B,KAAK,gBAAgB,MAAM,iBAAiB,KAAK;EACjD,KAAK,WAAW,MAAM,YAAY,KAAK;EACvC,KAAK,YAAY,MAAM,aAAa,KAAK;CAC3C;CAEA,IAAI,YAAY;EACd,OAAO,CAAC,KAAK,QAAQ;CACvB;CAEA,IAAI,aAAa;EACf,OAAO,CAAC,KAAK,SAAS;CACxB;;;;;;;;CAUA,aAAa,kBAAkB,EAC7B,KACA,SAIC;EACD,MAAM,kBAAkB,IAAIC,kBAAAA,SAAS;GACnC;GACA,QAAQ,MAAMC,eAAAA,qBAAqB,KAAK;EAC1C,CAAC;EACD,OAAO,IAAIC,eAAAA,WAAW,iBAAiB,IAAIC,qBAAAA,iBAAiB,CAAC;CAC/D;;;;;;;;;;CAWA,OAAO,uBAAuB,EAC5B,KACA,OACA,uBAAuBC,eAAAA,qDAKtB;EACD,IAAI;EAEJ,IAAI,MAAM,SAAS,KAAK,wBAAwB,MAAM,EAAE,GAAG;GACzD,QAAQC,gBAAAA,oBAAoB,gBAAgB,KAAK,OAAO;IACtD;IACA,gBAAgB;KAAC;KAAkB;KAAgB;IAAkB;GACvE,CAAC;GACD,OAAO,IAAIC,eAAAA,kBACTC,iBAAAA,cAAc,kBAAkB;IAC9B;IACA;GACF,CAAC,CACH;EACF;EAEA,QAAQC,cAAAA,UAAU,gBAAgB,KAAK,OAAiB,EACtD,qBACF,CAAC;EACD,OAAO,IAAIF,eAAAA,kBACTC,iBAAAA,cAAc,kBAAkB;GAC9B;GACA;EACF,CAAC,CACH;CACF;;;;;;;;;;;CAYA,aAAa,gBAAgB,EAC3B,KACA,OACA,wBAKsE;EAYtE,OAAO,IAXc,4BAA4B;GAC/C,SAAS,MAAM,4BAA4B,kBAAkB;IAC3D;IACA;GACF,CAAC;GACD,cAAc,4BAA4B,uBAAuB;IAC/D;IACA;IACA;GACF,CAAC;EACH,CACc;CAChB;;CAGA,MAAM,MACJ,QACA,YACsB;EACtB,MAAM,OAAO,MAAM,KAAK,QAAQ,KAAK,OAAO,OAAO,YAAY,SAAS,CAAC;EACzE,IAAI,CAAC,KAAK,OAAO,QACf,MAAM,IAAI,MACR,+EACF;EAEF,KAAK,MAAM,KAAK,MAAM,SAAS,EAAE,CAAC,QAChC,+BAA+B,OAAO,MAAM;EAC9C,KAAK,MAAM,QAAQ,KAAK,OAAO;GAC7B,MAAM,YAAY;IAChB,GAAG;IACH,gBAAgB,KAAK,UAAU,KAAK,cAAc,SAAS,CAAC;IAC5D,cAAc,KAAK;GACrB;GACA,MAAM,WAAW,MAAM,KAAK,aAAa,KACvC,WACA,YAAY,SAAS,CACvB;GACA,KAAK,cAAc,QAAQ,MAAM,QAAQ;EAC3C;EACA,OAAO,GAAG,KAAK,YAAY,KAAK,cAAc,iBAAiB,EAAE;CACnE;CAEA,aAAa;EACX,OAAO;CACT;CAEA,YAAgC;EAC9B,MAAM,IAAI,MAAM,mCAAmC;CACrD;AACF"}

package/dist/experimental/plan_and_execute/agent_executor.js

@@ -53,10 +53,11 @@ var PlanAndExecuteAgentExecutor = class PlanAndExecuteAgentExecutor extends Base
 	* @returns A new LLMPlanner instance.
 	*/
 	static async getDefaultPlanner({ llm, tools }) {
-		return new LLMPlanner(new LLMChain({
+		const plannerLlmChain = new LLMChain({
 			llm,
 			prompt: await getPlannerChatPrompt(tools)
-		}), new PlanOutputParser());
+		});
+		return new LLMPlanner(plannerLlmChain, new PlanOutputParser());
 	}
 	/**
 	* Static method that returns a default step executor for the agent. It

package/dist/experimental/plan_and_execute/agent_executor.js.map

@@ -1 +1 @@
-{"version":3,"file":"agent_executor.js","names":[],"sources":["../../../src/experimental/plan_and_execute/agent_executor.ts"],"sourcesContent":["import type { BaseLanguageModelInterface } from \"@langchain/core/language_models/base\";\nimport { ChainValues } from \"@langchain/core/utils/types\";\nimport { Tool, DynamicStructuredTool } from \"@langchain/core/tools\";\nimport { CallbackManagerForChainRun } from \"@langchain/core/callbacks/manager\";\nimport { BaseChain, ChainInputs } from \"../../chains/base.js\";\nimport {\n  BasePlanner,\n  BaseStepContainer,\n  BaseStepExecutor,\n  ListStepContainer,\n  LLMPlanner,\n  ChainStepExecutor,\n} from \"./base.js\";\nimport { AgentExecutor } from \"../../agents/executor.js\";\nimport {\n  DEFAULT_STEP_EXECUTOR_HUMAN_CHAT_MESSAGE_TEMPLATE,\n  getPlannerChatPrompt,\n} from \"./prompt.js\";\nimport { LLMChain } from \"../../chains/llm_chain.js\";\nimport { PlanOutputParser } from \"./outputParser.js\";\nimport { ChatAgent } from \"../../agents/chat/index.js\";\nimport { StructuredChatAgent } from \"../../agents/index.js\";\nimport { SerializedLLMChain } from \"../../chains/serde.js\";\n\n/**\n * A utility function to distiguish a dynamicstructuredtool over other tools.\n * @param tool the tool to test\n * @returns bool\n */\nexport function isDynamicStructuredTool(\n  tool: Tool | DynamicStructuredTool\n): tool is DynamicStructuredTool {\n  // We check for the existence of the static lc_name method in the object's constructor\n  return (\n    // oxlint-disable-next-line @typescript-eslint/no-explicit-any\n    typeof (tool.constructor as any).lc_name === \"function\" &&\n    // oxlint-disable-next-line @typescript-eslint/no-explicit-any\n    (tool.constructor as any).lc_name() === \"DynamicStructuredTool\"\n  );\n}\n\n/**\n * Interface for the input to the PlanAndExecuteAgentExecutor class. It\n * extends ChainInputs and includes additional properties for the planner,\n * step executor, step container, and input and output keys.\n */\nexport interface PlanAndExecuteAgentExecutorInput extends ChainInputs {\n  planner: BasePlanner;\n  stepExecutor: BaseStepExecutor;\n  stepContainer?: BaseStepContainer;\n  inputKey?: string;\n  outputKey?: string;\n}\n\n/**\n * Class representing a plan-and-execute agent executor. This agent\n * decides on the full sequence of actions upfront, then executes them all\n * without updating the plan. This is suitable for complex or long-running\n * tasks that require maintaining long-term objectives and focus.\n */\nexport class PlanAndExecuteAgentExecutor extends BaseChain {\n  static lc_name() {\n    return \"PlanAndExecuteAgentExecutor\";\n  }\n\n  private planner: BasePlanner;\n\n  private stepExecutor: BaseStepExecutor;\n\n  private stepContainer: BaseStepContainer = new ListStepContainer();\n\n  private inputKey = \"input\";\n\n  private outputKey = \"output\";\n\n  constructor(input: PlanAndExecuteAgentExecutorInput) {\n    super(input);\n    this.planner = input.planner;\n    this.stepExecutor = input.stepExecutor;\n    this.stepContainer = input.stepContainer ?? this.stepContainer;\n    this.inputKey = input.inputKey ?? this.inputKey;\n    this.outputKey = input.outputKey ?? this.outputKey;\n  }\n\n  get inputKeys() {\n    return [this.inputKey];\n  }\n\n  get outputKeys() {\n    return [this.outputKey];\n  }\n\n  /**\n   * Static method that returns a default planner for the agent. It creates\n   * a new LLMChain with a given LLM and a fixed prompt, and uses it to\n   * create a new LLMPlanner with a PlanOutputParser.\n   * @param llm The Large Language Model (LLM) used to generate responses.\n   * @returns A new LLMPlanner instance.\n   */\n\n  static async getDefaultPlanner({\n    llm,\n    tools,\n  }: {\n    llm: BaseLanguageModelInterface;\n    tools: Tool[] | DynamicStructuredTool[];\n  }) {\n    const plannerLlmChain = new LLMChain({\n      llm,\n      prompt: await getPlannerChatPrompt(tools),\n    });\n    return new LLMPlanner(plannerLlmChain, new PlanOutputParser());\n  }\n\n  /**\n   * Static method that returns a default step executor for the agent. It\n   * creates a new ChatAgent from a given LLM and a set of tools, and uses\n   * it to create a new ChainStepExecutor.\n   * @param llm The Large Language Model (LLM) used to generate responses.\n   * @param tools The set of tools used by the agent.\n   * @param humanMessageTemplate The template for human messages. If not provided, a default template is used.\n   * @returns A new ChainStepExecutor instance.\n   */\n  static getDefaultStepExecutor({\n    llm,\n    tools,\n    humanMessageTemplate = DEFAULT_STEP_EXECUTOR_HUMAN_CHAT_MESSAGE_TEMPLATE,\n  }: {\n    llm: BaseLanguageModelInterface;\n    tools: Tool[] | DynamicStructuredTool[];\n    humanMessageTemplate?: string;\n  }) {\n    let agent;\n\n    if (tools.length > 0 && isDynamicStructuredTool(tools[0])) {\n      agent = StructuredChatAgent.fromLLMAndTools(llm, tools, {\n        humanMessageTemplate,\n        inputVariables: [\"previous_steps\", \"current_step\", \"agent_scratchpad\"],\n      });\n      return new ChainStepExecutor(\n        AgentExecutor.fromAgentAndTools({\n          agent,\n          tools,\n        })\n      );\n    }\n\n    agent = ChatAgent.fromLLMAndTools(llm, tools as Tool[], {\n      humanMessageTemplate,\n    });\n    return new ChainStepExecutor(\n      AgentExecutor.fromAgentAndTools({\n        agent,\n        tools,\n      })\n    );\n  }\n\n  /**\n   * Static method that creates a new PlanAndExecuteAgentExecutor from a\n   * given LLM, a set of tools, and optionally a human message template. It\n   * uses the getDefaultPlanner and getDefaultStepExecutor methods to create\n   * the planner and step executor for the new agent executor.\n   * @param llm The Large Language Model (LLM) used to generate responses.\n   * @param tools The set of tools used by the agent.\n   * @param humanMessageTemplate The template for human messages. If not provided, a default template is used.\n   * @returns A new PlanAndExecuteAgentExecutor instance.\n   */\n  static async fromLLMAndTools({\n    llm,\n    tools,\n    humanMessageTemplate,\n  }: {\n    llm: BaseLanguageModelInterface;\n    tools: Tool[] | DynamicStructuredTool[];\n    humanMessageTemplate?: string;\n  } & Omit<PlanAndExecuteAgentExecutorInput, \"planner\" | \"stepExecutor\">) {\n    const executor = new PlanAndExecuteAgentExecutor({\n      planner: await PlanAndExecuteAgentExecutor.getDefaultPlanner({\n        llm,\n        tools,\n      }),\n      stepExecutor: PlanAndExecuteAgentExecutor.getDefaultStepExecutor({\n        llm,\n        tools,\n        humanMessageTemplate,\n      }),\n    });\n    return executor;\n  }\n\n  /** @ignore */\n  async _call(\n    inputs: ChainValues,\n    runManager?: CallbackManagerForChainRun\n  ): Promise<ChainValues> {\n    const plan = await this.planner.plan(inputs.input, runManager?.getChild());\n    if (!plan.steps?.length) {\n      throw new Error(\n        \"Could not create and parse a plan to answer your question - please try again.\"\n      );\n    }\n    plan.steps[plan.steps.length - 1].text +=\n      ` The original question was: ${inputs.input}.`;\n    for (const step of plan.steps) {\n      const newInputs = {\n        ...inputs,\n        previous_steps: JSON.stringify(this.stepContainer.getSteps()),\n        current_step: step.text,\n      };\n      const response = await this.stepExecutor.step(\n        newInputs,\n        runManager?.getChild()\n      );\n      this.stepContainer.addStep(step, response);\n    }\n    return { [this.outputKey]: this.stepContainer.getFinalResponse() };\n  }\n\n  _chainType() {\n    return \"agent_executor\" as const;\n  }\n\n  serialize(): SerializedLLMChain {\n    throw new Error(\"Cannot serialize an AgentExecutor\");\n  }\n}\n"],"mappings":";;;;;;;;;;;;;;;AA6BA,SAAgB,wBACd,MAC+B;CAE/B,OAEE,OAAQ,KAAK,YAAoB,YAAY,cAE5C,KAAK,YAAoB,QAAQ,MAAM;AAE5C;;;;;;;AAqBA,IAAa,8BAAb,MAAa,oCAAoC,UAAU;CACzD,OAAO,UAAU;EACf,OAAO;CACT;CAEA;CAEA;CAEA,gBAA2C,IAAI,kBAAkB;CAEjE,WAAmB;CAEnB,YAAoB;CAEpB,YAAY,OAAyC;EACnD,MAAM,KAAK;EACX,KAAK,UAAU,MAAM;EACrB,KAAK,eAAe,MAAM;EAC1B,KAAK,gBAAgB,MAAM,iBAAiB,KAAK;EACjD,KAAK,WAAW,MAAM,YAAY,KAAK;EACvC,KAAK,YAAY,MAAM,aAAa,KAAK;CAC3C;CAEA,IAAI,YAAY;EACd,OAAO,CAAC,KAAK,QAAQ;CACvB;CAEA,IAAI,aAAa;EACf,OAAO,CAAC,KAAK,SAAS;CACxB;;;;;;;;CAUA,aAAa,kBAAkB,EAC7B,KACA,SAIC;EAKD,OAAO,IAAI,WAAW,IAJM,SAAS;GACnC;GACA,QAAQ,MAAM,qBAAqB,KAAK;EAC1C,CACoC,GAAG,IAAI,iBAAiB,CAAC;CAC/D;;;;;;;;;;CAWA,OAAO,uBAAuB,EAC5B,KACA,OACA,uBAAuB,qDAKtB;EACD,IAAI;EAEJ,IAAI,MAAM,SAAS,KAAK,wBAAwB,MAAM,EAAE,GAAG;GACzD,QAAQ,oBAAoB,gBAAgB,KAAK,OAAO;IACtD;IACA,gBAAgB;KAAC;KAAkB;KAAgB;IAAkB;GACvE,CAAC;GACD,OAAO,IAAI,kBACT,cAAc,kBAAkB;IAC9B;IACA;GACF,CAAC,CACH;EACF;EAEA,QAAQ,UAAU,gBAAgB,KAAK,OAAiB,EACtD,qBACF,CAAC;EACD,OAAO,IAAI,kBACT,cAAc,kBAAkB;GAC9B;GACA;EACF,CAAC,CACH;CACF;;;;;;;;;;;CAYA,aAAa,gBAAgB,EAC3B,KACA,OACA,wBAKsE;EAYtE,OAAO,IAXc,4BAA4B;GAC/C,SAAS,MAAM,4BAA4B,kBAAkB;IAC3D;IACA;GACF,CAAC;GACD,cAAc,4BAA4B,uBAAuB;IAC/D;IACA;IACA;GACF,CAAC;EACH,CACc;CAChB;;CAGA,MAAM,MACJ,QACA,YACsB;EACtB,MAAM,OAAO,MAAM,KAAK,QAAQ,KAAK,OAAO,OAAO,YAAY,SAAS,CAAC;EACzE,IAAI,CAAC,KAAK,OAAO,QACf,MAAM,IAAI,MACR,+EACF;EAEF,KAAK,MAAM,KAAK,MAAM,SAAS,EAAE,CAAC,QAChC,+BAA+B,OAAO,MAAM;EAC9C,KAAK,MAAM,QAAQ,KAAK,OAAO;GAC7B,MAAM,YAAY;IAChB,GAAG;IACH,gBAAgB,KAAK,UAAU,KAAK,cAAc,SAAS,CAAC;IAC5D,cAAc,KAAK;GACrB;GACA,MAAM,WAAW,MAAM,KAAK,aAAa,KACvC,WACA,YAAY,SAAS,CACvB;GACA,KAAK,cAAc,QAAQ,MAAM,QAAQ;EAC3C;EACA,OAAO,GAAG,KAAK,YAAY,KAAK,cAAc,iBAAiB,EAAE;CACnE;CAEA,aAAa;EACX,OAAO;CACT;CAEA,YAAgC;EAC9B,MAAM,IAAI,MAAM,mCAAmC;CACrD;AACF"}
+{"version":3,"file":"agent_executor.js","names":[],"sources":["../../../src/experimental/plan_and_execute/agent_executor.ts"],"sourcesContent":["import type { BaseLanguageModelInterface } from \"@langchain/core/language_models/base\";\nimport { ChainValues } from \"@langchain/core/utils/types\";\nimport { Tool, DynamicStructuredTool } from \"@langchain/core/tools\";\nimport { CallbackManagerForChainRun } from \"@langchain/core/callbacks/manager\";\nimport { BaseChain, ChainInputs } from \"../../chains/base.js\";\nimport {\n  BasePlanner,\n  BaseStepContainer,\n  BaseStepExecutor,\n  ListStepContainer,\n  LLMPlanner,\n  ChainStepExecutor,\n} from \"./base.js\";\nimport { AgentExecutor } from \"../../agents/executor.js\";\nimport {\n  DEFAULT_STEP_EXECUTOR_HUMAN_CHAT_MESSAGE_TEMPLATE,\n  getPlannerChatPrompt,\n} from \"./prompt.js\";\nimport { LLMChain } from \"../../chains/llm_chain.js\";\nimport { PlanOutputParser } from \"./outputParser.js\";\nimport { ChatAgent } from \"../../agents/chat/index.js\";\nimport { StructuredChatAgent } from \"../../agents/index.js\";\nimport { SerializedLLMChain } from \"../../chains/serde.js\";\n\n/**\n * A utility function to distiguish a dynamicstructuredtool over other tools.\n * @param tool the tool to test\n * @returns bool\n */\nexport function isDynamicStructuredTool(\n  tool: Tool | DynamicStructuredTool\n): tool is DynamicStructuredTool {\n  // We check for the existence of the static lc_name method in the object's constructor\n  return (\n    // oxlint-disable-next-line @typescript-eslint/no-explicit-any\n    typeof (tool.constructor as any).lc_name === \"function\" &&\n    // oxlint-disable-next-line @typescript-eslint/no-explicit-any\n    (tool.constructor as any).lc_name() === \"DynamicStructuredTool\"\n  );\n}\n\n/**\n * Interface for the input to the PlanAndExecuteAgentExecutor class. It\n * extends ChainInputs and includes additional properties for the planner,\n * step executor, step container, and input and output keys.\n */\nexport interface PlanAndExecuteAgentExecutorInput extends ChainInputs {\n  planner: BasePlanner;\n  stepExecutor: BaseStepExecutor;\n  stepContainer?: BaseStepContainer;\n  inputKey?: string;\n  outputKey?: string;\n}\n\n/**\n * Class representing a plan-and-execute agent executor. This agent\n * decides on the full sequence of actions upfront, then executes them all\n * without updating the plan. This is suitable for complex or long-running\n * tasks that require maintaining long-term objectives and focus.\n */\nexport class PlanAndExecuteAgentExecutor extends BaseChain {\n  static lc_name() {\n    return \"PlanAndExecuteAgentExecutor\";\n  }\n\n  private planner: BasePlanner;\n\n  private stepExecutor: BaseStepExecutor;\n\n  private stepContainer: BaseStepContainer = new ListStepContainer();\n\n  private inputKey = \"input\";\n\n  private outputKey = \"output\";\n\n  constructor(input: PlanAndExecuteAgentExecutorInput) {\n    super(input);\n    this.planner = input.planner;\n    this.stepExecutor = input.stepExecutor;\n    this.stepContainer = input.stepContainer ?? this.stepContainer;\n    this.inputKey = input.inputKey ?? this.inputKey;\n    this.outputKey = input.outputKey ?? this.outputKey;\n  }\n\n  get inputKeys() {\n    return [this.inputKey];\n  }\n\n  get outputKeys() {\n    return [this.outputKey];\n  }\n\n  /**\n   * Static method that returns a default planner for the agent. It creates\n   * a new LLMChain with a given LLM and a fixed prompt, and uses it to\n   * create a new LLMPlanner with a PlanOutputParser.\n   * @param llm The Large Language Model (LLM) used to generate responses.\n   * @returns A new LLMPlanner instance.\n   */\n\n  static async getDefaultPlanner({\n    llm,\n    tools,\n  }: {\n    llm: BaseLanguageModelInterface;\n    tools: Tool[] | DynamicStructuredTool[];\n  }) {\n    const plannerLlmChain = new LLMChain({\n      llm,\n      prompt: await getPlannerChatPrompt(tools),\n    });\n    return new LLMPlanner(plannerLlmChain, new PlanOutputParser());\n  }\n\n  /**\n   * Static method that returns a default step executor for the agent. It\n   * creates a new ChatAgent from a given LLM and a set of tools, and uses\n   * it to create a new ChainStepExecutor.\n   * @param llm The Large Language Model (LLM) used to generate responses.\n   * @param tools The set of tools used by the agent.\n   * @param humanMessageTemplate The template for human messages. If not provided, a default template is used.\n   * @returns A new ChainStepExecutor instance.\n   */\n  static getDefaultStepExecutor({\n    llm,\n    tools,\n    humanMessageTemplate = DEFAULT_STEP_EXECUTOR_HUMAN_CHAT_MESSAGE_TEMPLATE,\n  }: {\n    llm: BaseLanguageModelInterface;\n    tools: Tool[] | DynamicStructuredTool[];\n    humanMessageTemplate?: string;\n  }) {\n    let agent;\n\n    if (tools.length > 0 && isDynamicStructuredTool(tools[0])) {\n      agent = StructuredChatAgent.fromLLMAndTools(llm, tools, {\n        humanMessageTemplate,\n        inputVariables: [\"previous_steps\", \"current_step\", \"agent_scratchpad\"],\n      });\n      return new ChainStepExecutor(\n        AgentExecutor.fromAgentAndTools({\n          agent,\n          tools,\n        })\n      );\n    }\n\n    agent = ChatAgent.fromLLMAndTools(llm, tools as Tool[], {\n      humanMessageTemplate,\n    });\n    return new ChainStepExecutor(\n      AgentExecutor.fromAgentAndTools({\n        agent,\n        tools,\n      })\n    );\n  }\n\n  /**\n   * Static method that creates a new PlanAndExecuteAgentExecutor from a\n   * given LLM, a set of tools, and optionally a human message template. It\n   * uses the getDefaultPlanner and getDefaultStepExecutor methods to create\n   * the planner and step executor for the new agent executor.\n   * @param llm The Large Language Model (LLM) used to generate responses.\n   * @param tools The set of tools used by the agent.\n   * @param humanMessageTemplate The template for human messages. If not provided, a default template is used.\n   * @returns A new PlanAndExecuteAgentExecutor instance.\n   */\n  static async fromLLMAndTools({\n    llm,\n    tools,\n    humanMessageTemplate,\n  }: {\n    llm: BaseLanguageModelInterface;\n    tools: Tool[] | DynamicStructuredTool[];\n    humanMessageTemplate?: string;\n  } & Omit<PlanAndExecuteAgentExecutorInput, \"planner\" | \"stepExecutor\">) {\n    const executor = new PlanAndExecuteAgentExecutor({\n      planner: await PlanAndExecuteAgentExecutor.getDefaultPlanner({\n        llm,\n        tools,\n      }),\n      stepExecutor: PlanAndExecuteAgentExecutor.getDefaultStepExecutor({\n        llm,\n        tools,\n        humanMessageTemplate,\n      }),\n    });\n    return executor;\n  }\n\n  /** @ignore */\n  async _call(\n    inputs: ChainValues,\n    runManager?: CallbackManagerForChainRun\n  ): Promise<ChainValues> {\n    const plan = await this.planner.plan(inputs.input, runManager?.getChild());\n    if (!plan.steps?.length) {\n      throw new Error(\n        \"Could not create and parse a plan to answer your question - please try again.\"\n      );\n    }\n    plan.steps[plan.steps.length - 1].text +=\n      ` The original question was: ${inputs.input}.`;\n    for (const step of plan.steps) {\n      const newInputs = {\n        ...inputs,\n        previous_steps: JSON.stringify(this.stepContainer.getSteps()),\n        current_step: step.text,\n      };\n      const response = await this.stepExecutor.step(\n        newInputs,\n        runManager?.getChild()\n      );\n      this.stepContainer.addStep(step, response);\n    }\n    return { [this.outputKey]: this.stepContainer.getFinalResponse() };\n  }\n\n  _chainType() {\n    return \"agent_executor\" as const;\n  }\n\n  serialize(): SerializedLLMChain {\n    throw new Error(\"Cannot serialize an AgentExecutor\");\n  }\n}\n"],"mappings":";;;;;;;;;;;;;;;AA6BA,SAAgB,wBACd,MAC+B;CAE/B,OAEE,OAAQ,KAAK,YAAoB,YAAY,cAE5C,KAAK,YAAoB,QAAQ,MAAM;AAE5C;;;;;;;AAqBA,IAAa,8BAAb,MAAa,oCAAoC,UAAU;CACzD,OAAO,UAAU;EACf,OAAO;CACT;CAEA;CAEA;CAEA,gBAA2C,IAAI,kBAAkB;CAEjE,WAAmB;CAEnB,YAAoB;CAEpB,YAAY,OAAyC;EACnD,MAAM,KAAK;EACX,KAAK,UAAU,MAAM;EACrB,KAAK,eAAe,MAAM;EAC1B,KAAK,gBAAgB,MAAM,iBAAiB,KAAK;EACjD,KAAK,WAAW,MAAM,YAAY,KAAK;EACvC,KAAK,YAAY,MAAM,aAAa,KAAK;CAC3C;CAEA,IAAI,YAAY;EACd,OAAO,CAAC,KAAK,QAAQ;CACvB;CAEA,IAAI,aAAa;EACf,OAAO,CAAC,KAAK,SAAS;CACxB;;;;;;;;CAUA,aAAa,kBAAkB,EAC7B,KACA,SAIC;EACD,MAAM,kBAAkB,IAAI,SAAS;GACnC;GACA,QAAQ,MAAM,qBAAqB,KAAK;EAC1C,CAAC;EACD,OAAO,IAAI,WAAW,iBAAiB,IAAI,iBAAiB,CAAC;CAC/D;;;;;;;;;;CAWA,OAAO,uBAAuB,EAC5B,KACA,OACA,uBAAuB,qDAKtB;EACD,IAAI;EAEJ,IAAI,MAAM,SAAS,KAAK,wBAAwB,MAAM,EAAE,GAAG;GACzD,QAAQ,oBAAoB,gBAAgB,KAAK,OAAO;IACtD;IACA,gBAAgB;KAAC;KAAkB;KAAgB;IAAkB;GACvE,CAAC;GACD,OAAO,IAAI,kBACT,cAAc,kBAAkB;IAC9B;IACA;GACF,CAAC,CACH;EACF;EAEA,QAAQ,UAAU,gBAAgB,KAAK,OAAiB,EACtD,qBACF,CAAC;EACD,OAAO,IAAI,kBACT,cAAc,kBAAkB;GAC9B;GACA;EACF,CAAC,CACH;CACF;;;;;;;;;;;CAYA,aAAa,gBAAgB,EAC3B,KACA,OACA,wBAKsE;EAYtE,OAAO,IAXc,4BAA4B;GAC/C,SAAS,MAAM,4BAA4B,kBAAkB;IAC3D;IACA;GACF,CAAC;GACD,cAAc,4BAA4B,uBAAuB;IAC/D;IACA;IACA;GACF,CAAC;EACH,CACc;CAChB;;CAGA,MAAM,MACJ,QACA,YACsB;EACtB,MAAM,OAAO,MAAM,KAAK,QAAQ,KAAK,OAAO,OAAO,YAAY,SAAS,CAAC;EACzE,IAAI,CAAC,KAAK,OAAO,QACf,MAAM,IAAI,MACR,+EACF;EAEF,KAAK,MAAM,KAAK,MAAM,SAAS,EAAE,CAAC,QAChC,+BAA+B,OAAO,MAAM;EAC9C,KAAK,MAAM,QAAQ,KAAK,OAAO;GAC7B,MAAM,YAAY;IAChB,GAAG;IACH,gBAAgB,KAAK,UAAU,KAAK,cAAc,SAAS,CAAC;IAC5D,cAAc,KAAK;GACrB;GACA,MAAM,WAAW,MAAM,KAAK,aAAa,KACvC,WACA,YAAY,SAAS,CACvB;GACA,KAAK,cAAc,QAAQ,MAAM,QAAQ;EAC3C;EACA,OAAO,GAAG,KAAK,YAAY,KAAK,cAAc,iBAAiB,EAAE;CACnE;CAEA,aAAa;EACX,OAAO;CACT;CAEA,YAAgC;EAC9B,MAAM,IAAI,MAAM,mCAAmC;CACrD;AACF"}

package/dist/hub/base.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"base.cjs","names":[],"sources":["../../src/hub/base.ts"],"sourcesContent":["import type { BaseLanguageModel } from \"@langchain/core/language_models/base\";\nimport type { Runnable } from \"@langchain/core/runnables\";\n\nimport type { Client, ClientConfig } from \"langsmith\";\nimport type { PromptCommit } from \"langsmith/schemas\";\n\n/**\n * Push a prompt to the hub.\n * If the specified repo doesn't already exist, it will be created.\n * @param repoFullName The full name of the repo.\n * @param runnable The prompt to push.\n * @param options\n * @returns The URL of the newly pushed prompt in the hub.\n */\nexport async function basePush(\n  repoFullName: string,\n  runnable: Runnable,\n  options?: {\n    apiKey?: string;\n    apiUrl?: string;\n    parentCommitHash?: string;\n    /** @deprecated Use isPublic instead. */\n    newRepoIsPublic?: boolean;\n    isPublic?: boolean;\n    /** @deprecated Use description instead. */\n    newRepoDescription?: string;\n    description?: string;\n    readme?: string;\n    tags?: string[];\n  }\n): Promise<string> {\n  const Client = await loadLangSmith();\n  const client = new Client(options);\n  const payloadOptions = {\n    object: runnable,\n    parentCommitHash: options?.parentCommitHash,\n    isPublic: options?.isPublic ?? options?.newRepoIsPublic,\n    description: options?.description ?? options?.newRepoDescription,\n    readme: options?.readme,\n    tags: options?.tags,\n  };\n  return client.pushPrompt(repoFullName, payloadOptions);\n}\n\nexport async function basePull(\n  ownerRepoCommit: string,\n  options?: { apiKey?: string; apiUrl?: string; includeModel?: boolean }\n): Promise<PromptCommit> {\n  const Client = await loadLangSmith();\n  const client = new Client(options);\n\n  const promptObject = await client.pullPromptCommit(ownerRepoCommit, {\n    includeModel: options?.includeModel,\n  });\n\n  if (promptObject.manifest.kwargs?.metadata === undefined) {\n    promptObject.manifest.kwargs = {\n      ...promptObject.manifest.kwargs,\n      metadata: {},\n    };\n  }\n\n  promptObject.manifest.kwargs.metadata = {\n    ...promptObject.manifest.kwargs.metadata,\n    lc_hub_owner: promptObject.owner,\n    lc_hub_repo: promptObject.repo,\n    lc_hub_commit_hash: promptObject.commit_hash,\n  };\n\n  // Some nested mustache prompts have improperly parsed variables that include a dot.\n  if (promptObject.manifest.kwargs.template_format === \"mustache\") {\n    const stripDotNotation = (varName: string) => varName.split(\".\")[0];\n\n    const { input_variables } = promptObject.manifest.kwargs;\n    if (Array.isArray(input_variables)) {\n      promptObject.manifest.kwargs.input_variables =\n        input_variables.map(stripDotNotation);\n    }\n\n    const { messages } = promptObject.manifest.kwargs;\n    if (Array.isArray(messages)) {\n      // oxlint-disable-next-line @typescript-eslint/no-explicit-any\n      promptObject.manifest.kwargs.messages = messages.map((message: any) => {\n        const nestedVars = message?.kwargs?.prompt?.kwargs?.input_variables;\n        if (Array.isArray(nestedVars)) {\n          message.kwargs.prompt.kwargs.input_variables =\n            nestedVars.map(stripDotNotation);\n        }\n        return message;\n      });\n    }\n  }\n  return promptObject;\n}\n\nexport function generateModelImportMap(\n  // oxlint-disable-next-line @typescript-eslint/no-explicit-any\n  modelClass?: new (...args: any[]) => BaseLanguageModel\n) {\n  // oxlint-disable-next-line @typescript-eslint/no-explicit-any\n  const modelImportMap: Record<string, any> = {};\n  if (modelClass !== undefined) {\n    // oxlint-disable-next-line @typescript-eslint/no-explicit-any\n    const modelLcName = (modelClass as any)?.lc_name();\n    let importMapKey;\n    if (modelLcName === \"ChatOpenAI\") {\n      importMapKey = \"chat_models__openai\";\n    } else if (modelLcName === \"ChatAnthropic\") {\n      importMapKey = \"chat_models__anthropic\";\n    } else if (modelLcName === \"ChatAzureOpenAI\") {\n      importMapKey = \"chat_models__openai\";\n    } else if (modelLcName === \"ChatVertexAI\") {\n      importMapKey = \"chat_models__vertexai\";\n    } else if (modelLcName === \"ChatGoogleGenerativeAI\") {\n      importMapKey = \"chat_models__google_genai\";\n    } else if (modelLcName === \"ChatBedrockConverse\") {\n      importMapKey = \"chat_models__chat_bedrock_converse\";\n    } else if (modelLcName === \"ChatMistral\") {\n      importMapKey = \"chat_models__mistralai\";\n    } else if (modelLcName === \"ChatFireworks\") {\n      importMapKey = \"chat_models__fireworks\";\n    } else if (modelLcName === \"ChatGroq\") {\n      importMapKey = \"chat_models__groq\";\n    } else {\n      throw new Error(\"Received unsupported model class when pulling prompt.\");\n    }\n    modelImportMap[importMapKey] = {\n      ...modelImportMap[importMapKey],\n      [modelLcName]: modelClass,\n    };\n  }\n  return modelImportMap;\n}\n\nexport function generateOptionalImportMap(\n  // oxlint-disable-next-line @typescript-eslint/no-explicit-any\n  modelClass?: new (...args: any[]) => BaseLanguageModel\n) {\n  // oxlint-disable-next-line @typescript-eslint/no-explicit-any\n  const optionalImportMap: Record<string, any> = {};\n  if (modelClass !== undefined) {\n    // oxlint-disable-next-line @typescript-eslint/no-explicit-any\n    const modelLcName = (modelClass as any)?.lc_name();\n    let optionalImportMapKey;\n    if (modelLcName === \"ChatGoogleGenerativeAI\") {\n      optionalImportMapKey = \"langchain_google_genai/chat_models\";\n    } else if (modelLcName === \"ChatBedrockConverse\") {\n      optionalImportMapKey = \"langchain_aws/chat_models\";\n    } else if (modelLcName === \"ChatGroq\") {\n      optionalImportMapKey = \"langchain_groq/chat_models\";\n    }\n    if (optionalImportMapKey !== undefined) {\n      optionalImportMap[optionalImportMapKey] = {\n        [modelLcName]: modelClass,\n      };\n    }\n  }\n  return optionalImportMap;\n}\n\nexport function bindOutputSchema<T extends Runnable>(loadedSequence: T) {\n  if (\n    \"first\" in loadedSequence &&\n    loadedSequence.first !== null &&\n    typeof loadedSequence.first === \"object\" &&\n    \"schema\" in loadedSequence.first &&\n    \"last\" in loadedSequence &&\n    loadedSequence.last !== null &&\n    typeof loadedSequence.last === \"object\"\n  ) {\n    if (\n      \"bound\" in loadedSequence.last &&\n      loadedSequence.last.bound !== null &&\n      typeof loadedSequence.last.bound === \"object\" &&\n      \"withStructuredOutput\" in loadedSequence.last.bound &&\n      typeof loadedSequence.last.bound.withStructuredOutput === \"function\"\n    ) {\n      loadedSequence.last.bound =\n        loadedSequence.last.bound.withStructuredOutput(\n          loadedSequence.first.schema\n        );\n    } else if (\n      \"withStructuredOutput\" in loadedSequence.last &&\n      typeof loadedSequence.last.withStructuredOutput === \"function\"\n    ) {\n      loadedSequence.last = loadedSequence.last.withStructuredOutput(\n        loadedSequence.first.schema\n      );\n    }\n  }\n  return loadedSequence;\n}\n\n/**\n * Dynamically load the LangSmith client.\n * @returns The LangSmith client.\n */\nasync function loadLangSmith(): Promise<new (config?: ClientConfig) => Client> {\n  try {\n    const { Client } = await import(\"langsmith\");\n    return Client;\n  } catch (error) {\n    // oxlint-disable-next-line no-instanceof/no-instanceof\n    const errorMessage = error instanceof Error ? error.message : String(error);\n    throw new Error(\n      `Error loading \"langsmith\" package, install it via \\`npm install langsmith\\` before you use this function.\\nError: ${errorMessage}`\n    );\n  }\n}\n"],"mappings":";;;;;;;;;AAcA,eAAsB,SACpB,cACA,UACA,SAaiB;CAEjB,MAAM,SAAS,KAAI,OADE,cAAc,IACT,OAAO;CACjC,MAAM,iBAAiB;EACrB,QAAQ;EACR,kBAAkB,SAAS;EAC3B,UAAU,SAAS,YAAY,SAAS;EACxC,aAAa,SAAS,eAAe,SAAS;EAC9C,QAAQ,SAAS;EACjB,MAAM,SAAS;CACjB;CACA,OAAO,OAAO,WAAW,cAAc,cAAc;AACvD;AAEA,eAAsB,SACpB,iBACA,SACuB;CAIvB,MAAM,eAAe,MAAM,KAFR,OADE,cAAc,IACT,OAEM,CAAC,CAAC,iBAAiB,iBAAiB,EAClE,cAAc,SAAS,aACzB,CAAC;CAED,IAAI,aAAa,SAAS,QAAQ,aAAa,KAAA,GAC7C,aAAa,SAAS,SAAS;EAC7B,GAAG,aAAa,SAAS;EACzB,UAAU,CAAC;CACb;CAGF,aAAa,SAAS,OAAO,WAAW;EACtC,GAAG,aAAa,SAAS,OAAO;EAChC,cAAc,aAAa;EAC3B,aAAa,aAAa;EAC1B,oBAAoB,aAAa;CACnC;CAGA,IAAI,aAAa,SAAS,OAAO,oBAAoB,YAAY;EAC/D,MAAM,oBAAoB,YAAoB,QAAQ,MAAM,GAAG,CAAC,CAAC;EAEjE,MAAM,EAAE,oBAAoB,aAAa,SAAS;EAClD,IAAI,MAAM,QAAQ,eAAe,GAC/B,aAAa,SAAS,OAAO,kBAC3B,gBAAgB,IAAI,gBAAgB;EAGxC,MAAM,EAAE,aAAa,aAAa,SAAS;EAC3C,IAAI,MAAM,QAAQ,QAAQ,GAExB,aAAa,SAAS,OAAO,WAAW,SAAS,KAAK,YAAiB;GACrE,MAAM,aAAa,SAAS,QAAQ,QAAQ,QAAQ;GACpD,IAAI,MAAM,QAAQ,UAAU,GAC1B,QAAQ,OAAO,OAAO,OAAO,kBAC3B,WAAW,IAAI,gBAAgB;GAEnC,OAAO;EACT,CAAC;CAEL;CACA,OAAO;AACT;AAEA,SAAgB,uBAEd,YACA;CAEA,MAAM,iBAAsC,CAAC;CAC7C,IAAI,eAAe,KAAA,GAAW;EAE5B,MAAM,cAAe,YAAoB,QAAQ;EACjD,IAAI;EACJ,IAAI,gBAAgB,cAClB,eAAe;OACV,IAAI,gBAAgB,iBACzB,eAAe;OACV,IAAI,gBAAgB,mBACzB,eAAe;OACV,IAAI,gBAAgB,gBACzB,eAAe;OACV,IAAI,gBAAgB,0BACzB,eAAe;OACV,IAAI,gBAAgB,uBACzB,eAAe;OACV,IAAI,gBAAgB,eACzB,eAAe;OACV,IAAI,gBAAgB,iBACzB,eAAe;OACV,IAAI,gBAAgB,YACzB,eAAe;OAEf,MAAM,IAAI,MAAM,uDAAuD;EAEzE,eAAe,gBAAgB;GAC7B,GAAG,eAAe;IACjB,cAAc;EACjB;CACF;CACA,OAAO;AACT;AAEA,SAAgB,0BAEd,YACA;CAEA,MAAM,oBAAyC,CAAC;CAChD,IAAI,eAAe,KAAA,GAAW;EAE5B,MAAM,cAAe,YAAoB,QAAQ;EACjD,IAAI;EACJ,IAAI,gBAAgB,0BAClB,uBAAuB;OAClB,IAAI,gBAAgB,uBACzB,uBAAuB;OAClB,IAAI,gBAAgB,YACzB,uBAAuB;EAEzB,IAAI,yBAAyB,KAAA,GAC3B,kBAAkB,wBAAwB,GACvC,cAAc,WACjB;CAEJ;CACA,OAAO;AACT;AAEA,SAAgB,iBAAqC,gBAAmB;CACtE,IACE,WAAW,kBACX,eAAe,UAAU,QACzB,OAAO,eAAe,UAAU,YAChC,YAAY,eAAe,SAC3B,UAAU,kBACV,eAAe,SAAS,QACxB,OAAO,eAAe,SAAS,UAG7B;MAAA,WAAW,eAAe,QAC1B,eAAe,KAAK,UAAU,QAC9B,OAAO,eAAe,KAAK,UAAU,YACrC,0BAA0B,eAAe,KAAK,SAC9C,OAAO,eAAe,KAAK,MAAM,yBAAyB,YAE1D,eAAe,KAAK,QAClB,eAAe,KAAK,MAAM,qBACxB,eAAe,MAAM,MACvB;OACG,IACL,0BAA0B,eAAe,QACzC,OAAO,eAAe,KAAK,yBAAyB,YAEpD,eAAe,OAAO,eAAe,KAAK,qBACxC,eAAe,MAAM,MACvB;CAAA;CAGJ,OAAO;AACT;;;;;AAMA,eAAe,gBAAgE;CAC7E,IAAI;EACF,MAAM,EAAE,WAAW,MAAM,OAAO;EAChC,OAAO;CACT,SAAS,OAAO;EAEd,MAAM,eAAe,iBAAiB,QAAQ,MAAM,UAAU,OAAO,KAAK;EAC1E,MAAM,IAAI,MACR,qHAAqH,cACvH;CACF;AACF"}
+{"version":3,"file":"base.cjs","names":[],"sources":["../../src/hub/base.ts"],"sourcesContent":["import type { BaseLanguageModel } from \"@langchain/core/language_models/base\";\nimport type { Runnable } from \"@langchain/core/runnables\";\n\nimport type { Client, ClientConfig } from \"langsmith\";\nimport type { PromptCommit } from \"langsmith/schemas\";\n\n/**\n * Push a prompt to the hub.\n * If the specified repo doesn't already exist, it will be created.\n * @param repoFullName The full name of the repo.\n * @param runnable The prompt to push.\n * @param options\n * @returns The URL of the newly pushed prompt in the hub.\n */\nexport async function basePush(\n  repoFullName: string,\n  runnable: Runnable,\n  options?: {\n    apiKey?: string;\n    apiUrl?: string;\n    parentCommitHash?: string;\n    /** @deprecated Use isPublic instead. */\n    newRepoIsPublic?: boolean;\n    isPublic?: boolean;\n    /** @deprecated Use description instead. */\n    newRepoDescription?: string;\n    description?: string;\n    readme?: string;\n    tags?: string[];\n  }\n): Promise<string> {\n  const Client = await loadLangSmith();\n  const client = new Client(options);\n  const payloadOptions = {\n    object: runnable,\n    parentCommitHash: options?.parentCommitHash,\n    isPublic: options?.isPublic ?? options?.newRepoIsPublic,\n    description: options?.description ?? options?.newRepoDescription,\n    readme: options?.readme,\n    tags: options?.tags,\n  };\n  return client.pushPrompt(repoFullName, payloadOptions);\n}\n\nexport async function basePull(\n  ownerRepoCommit: string,\n  options?: { apiKey?: string; apiUrl?: string; includeModel?: boolean }\n): Promise<PromptCommit> {\n  const Client = await loadLangSmith();\n  const client = new Client(options);\n\n  const promptObject = await client.pullPromptCommit(ownerRepoCommit, {\n    includeModel: options?.includeModel,\n  });\n\n  if (promptObject.manifest.kwargs?.metadata === undefined) {\n    promptObject.manifest.kwargs = {\n      ...promptObject.manifest.kwargs,\n      metadata: {},\n    };\n  }\n\n  promptObject.manifest.kwargs.metadata = {\n    ...promptObject.manifest.kwargs.metadata,\n    lc_hub_owner: promptObject.owner,\n    lc_hub_repo: promptObject.repo,\n    lc_hub_commit_hash: promptObject.commit_hash,\n  };\n\n  // Some nested mustache prompts have improperly parsed variables that include a dot.\n  if (promptObject.manifest.kwargs.template_format === \"mustache\") {\n    const stripDotNotation = (varName: string) => varName.split(\".\")[0];\n\n    const { input_variables } = promptObject.manifest.kwargs;\n    if (Array.isArray(input_variables)) {\n      promptObject.manifest.kwargs.input_variables =\n        input_variables.map(stripDotNotation);\n    }\n\n    const { messages } = promptObject.manifest.kwargs;\n    if (Array.isArray(messages)) {\n      // oxlint-disable-next-line @typescript-eslint/no-explicit-any\n      promptObject.manifest.kwargs.messages = messages.map((message: any) => {\n        const nestedVars = message?.kwargs?.prompt?.kwargs?.input_variables;\n        if (Array.isArray(nestedVars)) {\n          message.kwargs.prompt.kwargs.input_variables =\n            nestedVars.map(stripDotNotation);\n        }\n        return message;\n      });\n    }\n  }\n  return promptObject;\n}\n\nexport function generateModelImportMap(\n  // oxlint-disable-next-line @typescript-eslint/no-explicit-any\n  modelClass?: new (...args: any[]) => BaseLanguageModel\n) {\n  // oxlint-disable-next-line @typescript-eslint/no-explicit-any\n  const modelImportMap: Record<string, any> = {};\n  if (modelClass !== undefined) {\n    // oxlint-disable-next-line @typescript-eslint/no-explicit-any\n    const modelLcName = (modelClass as any)?.lc_name();\n    let importMapKey;\n    if (modelLcName === \"ChatOpenAI\") {\n      importMapKey = \"chat_models__openai\";\n    } else if (modelLcName === \"ChatAnthropic\") {\n      importMapKey = \"chat_models__anthropic\";\n    } else if (modelLcName === \"ChatAzureOpenAI\") {\n      importMapKey = \"chat_models__openai\";\n    } else if (modelLcName === \"ChatVertexAI\") {\n      importMapKey = \"chat_models__vertexai\";\n    } else if (modelLcName === \"ChatGoogleGenerativeAI\") {\n      importMapKey = \"chat_models__google_genai\";\n    } else if (modelLcName === \"ChatBedrockConverse\") {\n      importMapKey = \"chat_models__chat_bedrock_converse\";\n    } else if (modelLcName === \"ChatMistral\") {\n      importMapKey = \"chat_models__mistralai\";\n    } else if (modelLcName === \"ChatFireworks\") {\n      importMapKey = \"chat_models__fireworks\";\n    } else if (modelLcName === \"ChatGroq\") {\n      importMapKey = \"chat_models__groq\";\n    } else {\n      throw new Error(\"Received unsupported model class when pulling prompt.\");\n    }\n    modelImportMap[importMapKey] = {\n      ...modelImportMap[importMapKey],\n      [modelLcName]: modelClass,\n    };\n  }\n  return modelImportMap;\n}\n\nexport function generateOptionalImportMap(\n  // oxlint-disable-next-line @typescript-eslint/no-explicit-any\n  modelClass?: new (...args: any[]) => BaseLanguageModel\n) {\n  // oxlint-disable-next-line @typescript-eslint/no-explicit-any\n  const optionalImportMap: Record<string, any> = {};\n  if (modelClass !== undefined) {\n    // oxlint-disable-next-line @typescript-eslint/no-explicit-any\n    const modelLcName = (modelClass as any)?.lc_name();\n    let optionalImportMapKey;\n    if (modelLcName === \"ChatGoogleGenerativeAI\") {\n      optionalImportMapKey = \"langchain_google_genai/chat_models\";\n    } else if (modelLcName === \"ChatBedrockConverse\") {\n      optionalImportMapKey = \"langchain_aws/chat_models\";\n    } else if (modelLcName === \"ChatGroq\") {\n      optionalImportMapKey = \"langchain_groq/chat_models\";\n    }\n    if (optionalImportMapKey !== undefined) {\n      optionalImportMap[optionalImportMapKey] = {\n        [modelLcName]: modelClass,\n      };\n    }\n  }\n  return optionalImportMap;\n}\n\nexport function bindOutputSchema<T extends Runnable>(loadedSequence: T) {\n  if (\n    \"first\" in loadedSequence &&\n    loadedSequence.first !== null &&\n    typeof loadedSequence.first === \"object\" &&\n    \"schema\" in loadedSequence.first &&\n    \"last\" in loadedSequence &&\n    loadedSequence.last !== null &&\n    typeof loadedSequence.last === \"object\"\n  ) {\n    if (\n      \"bound\" in loadedSequence.last &&\n      loadedSequence.last.bound !== null &&\n      typeof loadedSequence.last.bound === \"object\" &&\n      \"withStructuredOutput\" in loadedSequence.last.bound &&\n      typeof loadedSequence.last.bound.withStructuredOutput === \"function\"\n    ) {\n      loadedSequence.last.bound =\n        loadedSequence.last.bound.withStructuredOutput(\n          loadedSequence.first.schema\n        );\n    } else if (\n      \"withStructuredOutput\" in loadedSequence.last &&\n      typeof loadedSequence.last.withStructuredOutput === \"function\"\n    ) {\n      loadedSequence.last = loadedSequence.last.withStructuredOutput(\n        loadedSequence.first.schema\n      );\n    }\n  }\n  return loadedSequence;\n}\n\n/**\n * Dynamically load the LangSmith client.\n * @returns The LangSmith client.\n */\nasync function loadLangSmith(): Promise<new (config?: ClientConfig) => Client> {\n  try {\n    const { Client } = await import(\"langsmith\");\n    return Client;\n  } catch (error) {\n    // oxlint-disable-next-line no-instanceof/no-instanceof\n    const errorMessage = error instanceof Error ? error.message : String(error);\n    throw new Error(\n      `Error loading \"langsmith\" package, install it via \\`npm install langsmith\\` before you use this function.\\nError: ${errorMessage}`\n    );\n  }\n}\n"],"mappings":";;;;;;;;;AAcA,eAAsB,SACpB,cACA,UACA,SAaiB;CAEjB,MAAM,SAAS,KAAI,OADE,cAAc,IACT,OAAO;CACjC,MAAM,iBAAiB;EACrB,QAAQ;EACR,kBAAkB,SAAS;EAC3B,UAAU,SAAS,YAAY,SAAS;EACxC,aAAa,SAAS,eAAe,SAAS;EAC9C,QAAQ,SAAS;EACjB,MAAM,SAAS;CACjB;CACA,OAAO,OAAO,WAAW,cAAc,cAAc;AACvD;AAEA,eAAsB,SACpB,iBACA,SACuB;CAIvB,MAAM,eAAe,MAAM,KAFR,OADE,cAAc,IACT,OAEM,CAAC,CAAC,iBAAiB,iBAAiB,EAClE,cAAc,SAAS,aACzB,CAAC;CAED,IAAI,aAAa,SAAS,QAAQ,aAAa,KAAA,GAC7C,aAAa,SAAS,SAAS;EAC7B,GAAG,aAAa,SAAS;EACzB,UAAU,CAAC;CACb;CAGF,aAAa,SAAS,OAAO,WAAW;EACtC,GAAG,aAAa,SAAS,OAAO;EAChC,cAAc,aAAa;EAC3B,aAAa,aAAa;EAC1B,oBAAoB,aAAa;CACnC;CAGA,IAAI,aAAa,SAAS,OAAO,oBAAoB,YAAY;EAC/D,MAAM,oBAAoB,YAAoB,QAAQ,MAAM,GAAG,CAAC,CAAC;EAEjE,MAAM,EAAE,oBAAoB,aAAa,SAAS;EAClD,IAAI,MAAM,QAAQ,eAAe,GAC/B,aAAa,SAAS,OAAO,kBAC3B,gBAAgB,IAAI,gBAAgB;EAGxC,MAAM,EAAE,aAAa,aAAa,SAAS;EAC3C,IAAI,MAAM,QAAQ,QAAQ,GAExB,aAAa,SAAS,OAAO,WAAW,SAAS,KAAK,YAAiB;GACrE,MAAM,aAAa,SAAS,QAAQ,QAAQ,QAAQ;GACpD,IAAI,MAAM,QAAQ,UAAU,GAC1B,QAAQ,OAAO,OAAO,OAAO,kBAC3B,WAAW,IAAI,gBAAgB;GAEnC,OAAO;EACT,CAAC;CAEL;CACA,OAAO;AACT;AAEA,SAAgB,uBAEd,YACA;CAEA,MAAM,iBAAsC,CAAC;CAC7C,IAAI,eAAe,KAAA,GAAW;EAE5B,MAAM,cAAe,YAAoB,QAAQ;EACjD,IAAI;EACJ,IAAI,gBAAgB,cAClB,eAAe;OACV,IAAI,gBAAgB,iBACzB,eAAe;OACV,IAAI,gBAAgB,mBACzB,eAAe;OACV,IAAI,gBAAgB,gBACzB,eAAe;OACV,IAAI,gBAAgB,0BACzB,eAAe;OACV,IAAI,gBAAgB,uBACzB,eAAe;OACV,IAAI,gBAAgB,eACzB,eAAe;OACV,IAAI,gBAAgB,iBACzB,eAAe;OACV,IAAI,gBAAgB,YACzB,eAAe;OAEf,MAAM,IAAI,MAAM,uDAAuD;EAEzE,eAAe,gBAAgB;GAC7B,GAAG,eAAe;IACjB,cAAc;EACjB;CACF;CACA,OAAO;AACT;AAEA,SAAgB,0BAEd,YACA;CAEA,MAAM,oBAAyC,CAAC;CAChD,IAAI,eAAe,KAAA,GAAW;EAE5B,MAAM,cAAe,YAAoB,QAAQ;EACjD,IAAI;EACJ,IAAI,gBAAgB,0BAClB,uBAAuB;OAClB,IAAI,gBAAgB,uBACzB,uBAAuB;OAClB,IAAI,gBAAgB,YACzB,uBAAuB;EAEzB,IAAI,yBAAyB,KAAA,GAC3B,kBAAkB,wBAAwB,GACvC,cAAc,WACjB;CAEJ;CACA,OAAO;AACT;AAEA,SAAgB,iBAAqC,gBAAmB;CACtE,IACE,WAAW,kBACX,eAAe,UAAU,QACzB,OAAO,eAAe,UAAU,YAChC,YAAY,eAAe,SAC3B,UAAU,kBACV,eAAe,SAAS,QACxB,OAAO,eAAe,SAAS,UAC/B;EACA,IACE,WAAW,eAAe,QAC1B,eAAe,KAAK,UAAU,QAC9B,OAAO,eAAe,KAAK,UAAU,YACrC,0BAA0B,eAAe,KAAK,SAC9C,OAAO,eAAe,KAAK,MAAM,yBAAyB,YAE1D,eAAe,KAAK,QAClB,eAAe,KAAK,MAAM,qBACxB,eAAe,MAAM,MACvB;OACG,IACL,0BAA0B,eAAe,QACzC,OAAO,eAAe,KAAK,yBAAyB,YAEpD,eAAe,OAAO,eAAe,KAAK,qBACxC,eAAe,MAAM,MACvB;CAEJ;CACA,OAAO;AACT;;;;;AAMA,eAAe,gBAAgE;CAC7E,IAAI;EACF,MAAM,EAAE,WAAW,MAAM,OAAO;EAChC,OAAO;CACT,SAAS,OAAO;EAEd,MAAM,eAAe,iBAAiB,QAAQ,MAAM,UAAU,OAAO,KAAK;EAC1E,MAAM,IAAI,MACR,qHAAqH,cACvH;CACF;AACF"}

package/dist/hub/base.js.map

@@ -1 +1 @@
-{"version":3,"file":"base.js","names":[],"sources":["../../src/hub/base.ts"],"sourcesContent":["import type { BaseLanguageModel } from \"@langchain/core/language_models/base\";\nimport type { Runnable } from \"@langchain/core/runnables\";\n\nimport type { Client, ClientConfig } from \"langsmith\";\nimport type { PromptCommit } from \"langsmith/schemas\";\n\n/**\n * Push a prompt to the hub.\n * If the specified repo doesn't already exist, it will be created.\n * @param repoFullName The full name of the repo.\n * @param runnable The prompt to push.\n * @param options\n * @returns The URL of the newly pushed prompt in the hub.\n */\nexport async function basePush(\n  repoFullName: string,\n  runnable: Runnable,\n  options?: {\n    apiKey?: string;\n    apiUrl?: string;\n    parentCommitHash?: string;\n    /** @deprecated Use isPublic instead. */\n    newRepoIsPublic?: boolean;\n    isPublic?: boolean;\n    /** @deprecated Use description instead. */\n    newRepoDescription?: string;\n    description?: string;\n    readme?: string;\n    tags?: string[];\n  }\n): Promise<string> {\n  const Client = await loadLangSmith();\n  const client = new Client(options);\n  const payloadOptions = {\n    object: runnable,\n    parentCommitHash: options?.parentCommitHash,\n    isPublic: options?.isPublic ?? options?.newRepoIsPublic,\n    description: options?.description ?? options?.newRepoDescription,\n    readme: options?.readme,\n    tags: options?.tags,\n  };\n  return client.pushPrompt(repoFullName, payloadOptions);\n}\n\nexport async function basePull(\n  ownerRepoCommit: string,\n  options?: { apiKey?: string; apiUrl?: string; includeModel?: boolean }\n): Promise<PromptCommit> {\n  const Client = await loadLangSmith();\n  const client = new Client(options);\n\n  const promptObject = await client.pullPromptCommit(ownerRepoCommit, {\n    includeModel: options?.includeModel,\n  });\n\n  if (promptObject.manifest.kwargs?.metadata === undefined) {\n    promptObject.manifest.kwargs = {\n      ...promptObject.manifest.kwargs,\n      metadata: {},\n    };\n  }\n\n  promptObject.manifest.kwargs.metadata = {\n    ...promptObject.manifest.kwargs.metadata,\n    lc_hub_owner: promptObject.owner,\n    lc_hub_repo: promptObject.repo,\n    lc_hub_commit_hash: promptObject.commit_hash,\n  };\n\n  // Some nested mustache prompts have improperly parsed variables that include a dot.\n  if (promptObject.manifest.kwargs.template_format === \"mustache\") {\n    const stripDotNotation = (varName: string) => varName.split(\".\")[0];\n\n    const { input_variables } = promptObject.manifest.kwargs;\n    if (Array.isArray(input_variables)) {\n      promptObject.manifest.kwargs.input_variables =\n        input_variables.map(stripDotNotation);\n    }\n\n    const { messages } = promptObject.manifest.kwargs;\n    if (Array.isArray(messages)) {\n      // oxlint-disable-next-line @typescript-eslint/no-explicit-any\n      promptObject.manifest.kwargs.messages = messages.map((message: any) => {\n        const nestedVars = message?.kwargs?.prompt?.kwargs?.input_variables;\n        if (Array.isArray(nestedVars)) {\n          message.kwargs.prompt.kwargs.input_variables =\n            nestedVars.map(stripDotNotation);\n        }\n        return message;\n      });\n    }\n  }\n  return promptObject;\n}\n\nexport function generateModelImportMap(\n  // oxlint-disable-next-line @typescript-eslint/no-explicit-any\n  modelClass?: new (...args: any[]) => BaseLanguageModel\n) {\n  // oxlint-disable-next-line @typescript-eslint/no-explicit-any\n  const modelImportMap: Record<string, any> = {};\n  if (modelClass !== undefined) {\n    // oxlint-disable-next-line @typescript-eslint/no-explicit-any\n    const modelLcName = (modelClass as any)?.lc_name();\n    let importMapKey;\n    if (modelLcName === \"ChatOpenAI\") {\n      importMapKey = \"chat_models__openai\";\n    } else if (modelLcName === \"ChatAnthropic\") {\n      importMapKey = \"chat_models__anthropic\";\n    } else if (modelLcName === \"ChatAzureOpenAI\") {\n      importMapKey = \"chat_models__openai\";\n    } else if (modelLcName === \"ChatVertexAI\") {\n      importMapKey = \"chat_models__vertexai\";\n    } else if (modelLcName === \"ChatGoogleGenerativeAI\") {\n      importMapKey = \"chat_models__google_genai\";\n    } else if (modelLcName === \"ChatBedrockConverse\") {\n      importMapKey = \"chat_models__chat_bedrock_converse\";\n    } else if (modelLcName === \"ChatMistral\") {\n      importMapKey = \"chat_models__mistralai\";\n    } else if (modelLcName === \"ChatFireworks\") {\n      importMapKey = \"chat_models__fireworks\";\n    } else if (modelLcName === \"ChatGroq\") {\n      importMapKey = \"chat_models__groq\";\n    } else {\n      throw new Error(\"Received unsupported model class when pulling prompt.\");\n    }\n    modelImportMap[importMapKey] = {\n      ...modelImportMap[importMapKey],\n      [modelLcName]: modelClass,\n    };\n  }\n  return modelImportMap;\n}\n\nexport function generateOptionalImportMap(\n  // oxlint-disable-next-line @typescript-eslint/no-explicit-any\n  modelClass?: new (...args: any[]) => BaseLanguageModel\n) {\n  // oxlint-disable-next-line @typescript-eslint/no-explicit-any\n  const optionalImportMap: Record<string, any> = {};\n  if (modelClass !== undefined) {\n    // oxlint-disable-next-line @typescript-eslint/no-explicit-any\n    const modelLcName = (modelClass as any)?.lc_name();\n    let optionalImportMapKey;\n    if (modelLcName === \"ChatGoogleGenerativeAI\") {\n      optionalImportMapKey = \"langchain_google_genai/chat_models\";\n    } else if (modelLcName === \"ChatBedrockConverse\") {\n      optionalImportMapKey = \"langchain_aws/chat_models\";\n    } else if (modelLcName === \"ChatGroq\") {\n      optionalImportMapKey = \"langchain_groq/chat_models\";\n    }\n    if (optionalImportMapKey !== undefined) {\n      optionalImportMap[optionalImportMapKey] = {\n        [modelLcName]: modelClass,\n      };\n    }\n  }\n  return optionalImportMap;\n}\n\nexport function bindOutputSchema<T extends Runnable>(loadedSequence: T) {\n  if (\n    \"first\" in loadedSequence &&\n    loadedSequence.first !== null &&\n    typeof loadedSequence.first === \"object\" &&\n    \"schema\" in loadedSequence.first &&\n    \"last\" in loadedSequence &&\n    loadedSequence.last !== null &&\n    typeof loadedSequence.last === \"object\"\n  ) {\n    if (\n      \"bound\" in loadedSequence.last &&\n      loadedSequence.last.bound !== null &&\n      typeof loadedSequence.last.bound === \"object\" &&\n      \"withStructuredOutput\" in loadedSequence.last.bound &&\n      typeof loadedSequence.last.bound.withStructuredOutput === \"function\"\n    ) {\n      loadedSequence.last.bound =\n        loadedSequence.last.bound.withStructuredOutput(\n          loadedSequence.first.schema\n        );\n    } else if (\n      \"withStructuredOutput\" in loadedSequence.last &&\n      typeof loadedSequence.last.withStructuredOutput === \"function\"\n    ) {\n      loadedSequence.last = loadedSequence.last.withStructuredOutput(\n        loadedSequence.first.schema\n      );\n    }\n  }\n  return loadedSequence;\n}\n\n/**\n * Dynamically load the LangSmith client.\n * @returns The LangSmith client.\n */\nasync function loadLangSmith(): Promise<new (config?: ClientConfig) => Client> {\n  try {\n    const { Client } = await import(\"langsmith\");\n    return Client;\n  } catch (error) {\n    // oxlint-disable-next-line no-instanceof/no-instanceof\n    const errorMessage = error instanceof Error ? error.message : String(error);\n    throw new Error(\n      `Error loading \"langsmith\" package, install it via \\`npm install langsmith\\` before you use this function.\\nError: ${errorMessage}`\n    );\n  }\n}\n"],"mappings":";;;;;;;;;AAcA,eAAsB,SACpB,cACA,UACA,SAaiB;CAEjB,MAAM,SAAS,KAAI,OADE,cAAc,IACT,OAAO;CACjC,MAAM,iBAAiB;EACrB,QAAQ;EACR,kBAAkB,SAAS;EAC3B,UAAU,SAAS,YAAY,SAAS;EACxC,aAAa,SAAS,eAAe,SAAS;EAC9C,QAAQ,SAAS;EACjB,MAAM,SAAS;CACjB;CACA,OAAO,OAAO,WAAW,cAAc,cAAc;AACvD;AAEA,eAAsB,SACpB,iBACA,SACuB;CAIvB,MAAM,eAAe,MAAM,KAFR,OADE,cAAc,IACT,OAEM,CAAC,CAAC,iBAAiB,iBAAiB,EAClE,cAAc,SAAS,aACzB,CAAC;CAED,IAAI,aAAa,SAAS,QAAQ,aAAa,KAAA,GAC7C,aAAa,SAAS,SAAS;EAC7B,GAAG,aAAa,SAAS;EACzB,UAAU,CAAC;CACb;CAGF,aAAa,SAAS,OAAO,WAAW;EACtC,GAAG,aAAa,SAAS,OAAO;EAChC,cAAc,aAAa;EAC3B,aAAa,aAAa;EAC1B,oBAAoB,aAAa;CACnC;CAGA,IAAI,aAAa,SAAS,OAAO,oBAAoB,YAAY;EAC/D,MAAM,oBAAoB,YAAoB,QAAQ,MAAM,GAAG,CAAC,CAAC;EAEjE,MAAM,EAAE,oBAAoB,aAAa,SAAS;EAClD,IAAI,MAAM,QAAQ,eAAe,GAC/B,aAAa,SAAS,OAAO,kBAC3B,gBAAgB,IAAI,gBAAgB;EAGxC,MAAM,EAAE,aAAa,aAAa,SAAS;EAC3C,IAAI,MAAM,QAAQ,QAAQ,GAExB,aAAa,SAAS,OAAO,WAAW,SAAS,KAAK,YAAiB;GACrE,MAAM,aAAa,SAAS,QAAQ,QAAQ,QAAQ;GACpD,IAAI,MAAM,QAAQ,UAAU,GAC1B,QAAQ,OAAO,OAAO,OAAO,kBAC3B,WAAW,IAAI,gBAAgB;GAEnC,OAAO;EACT,CAAC;CAEL;CACA,OAAO;AACT;AAEA,SAAgB,uBAEd,YACA;CAEA,MAAM,iBAAsC,CAAC;CAC7C,IAAI,eAAe,KAAA,GAAW;EAE5B,MAAM,cAAe,YAAoB,QAAQ;EACjD,IAAI;EACJ,IAAI,gBAAgB,cAClB,eAAe;OACV,IAAI,gBAAgB,iBACzB,eAAe;OACV,IAAI,gBAAgB,mBACzB,eAAe;OACV,IAAI,gBAAgB,gBACzB,eAAe;OACV,IAAI,gBAAgB,0BACzB,eAAe;OACV,IAAI,gBAAgB,uBACzB,eAAe;OACV,IAAI,gBAAgB,eACzB,eAAe;OACV,IAAI,gBAAgB,iBACzB,eAAe;OACV,IAAI,gBAAgB,YACzB,eAAe;OAEf,MAAM,IAAI,MAAM,uDAAuD;EAEzE,eAAe,gBAAgB;GAC7B,GAAG,eAAe;IACjB,cAAc;EACjB;CACF;CACA,OAAO;AACT;AAEA,SAAgB,0BAEd,YACA;CAEA,MAAM,oBAAyC,CAAC;CAChD,IAAI,eAAe,KAAA,GAAW;EAE5B,MAAM,cAAe,YAAoB,QAAQ;EACjD,IAAI;EACJ,IAAI,gBAAgB,0BAClB,uBAAuB;OAClB,IAAI,gBAAgB,uBACzB,uBAAuB;OAClB,IAAI,gBAAgB,YACzB,uBAAuB;EAEzB,IAAI,yBAAyB,KAAA,GAC3B,kBAAkB,wBAAwB,GACvC,cAAc,WACjB;CAEJ;CACA,OAAO;AACT;AAEA,SAAgB,iBAAqC,gBAAmB;CACtE,IACE,WAAW,kBACX,eAAe,UAAU,QACzB,OAAO,eAAe,UAAU,YAChC,YAAY,eAAe,SAC3B,UAAU,kBACV,eAAe,SAAS,QACxB,OAAO,eAAe,SAAS,UAG7B;MAAA,WAAW,eAAe,QAC1B,eAAe,KAAK,UAAU,QAC9B,OAAO,eAAe,KAAK,UAAU,YACrC,0BAA0B,eAAe,KAAK,SAC9C,OAAO,eAAe,KAAK,MAAM,yBAAyB,YAE1D,eAAe,KAAK,QAClB,eAAe,KAAK,MAAM,qBACxB,eAAe,MAAM,MACvB;OACG,IACL,0BAA0B,eAAe,QACzC,OAAO,eAAe,KAAK,yBAAyB,YAEpD,eAAe,OAAO,eAAe,KAAK,qBACxC,eAAe,MAAM,MACvB;CAAA;CAGJ,OAAO;AACT;;;;;AAMA,eAAe,gBAAgE;CAC7E,IAAI;EACF,MAAM,EAAE,WAAW,MAAM,OAAO;EAChC,OAAO;CACT,SAAS,OAAO;EAEd,MAAM,eAAe,iBAAiB,QAAQ,MAAM,UAAU,OAAO,KAAK;EAC1E,MAAM,IAAI,MACR,qHAAqH,cACvH;CACF;AACF"}
+{"version":3,"file":"base.js","names":[],"sources":["../../src/hub/base.ts"],"sourcesContent":["import type { BaseLanguageModel } from \"@langchain/core/language_models/base\";\nimport type { Runnable } from \"@langchain/core/runnables\";\n\nimport type { Client, ClientConfig } from \"langsmith\";\nimport type { PromptCommit } from \"langsmith/schemas\";\n\n/**\n * Push a prompt to the hub.\n * If the specified repo doesn't already exist, it will be created.\n * @param repoFullName The full name of the repo.\n * @param runnable The prompt to push.\n * @param options\n * @returns The URL of the newly pushed prompt in the hub.\n */\nexport async function basePush(\n  repoFullName: string,\n  runnable: Runnable,\n  options?: {\n    apiKey?: string;\n    apiUrl?: string;\n    parentCommitHash?: string;\n    /** @deprecated Use isPublic instead. */\n    newRepoIsPublic?: boolean;\n    isPublic?: boolean;\n    /** @deprecated Use description instead. */\n    newRepoDescription?: string;\n    description?: string;\n    readme?: string;\n    tags?: string[];\n  }\n): Promise<string> {\n  const Client = await loadLangSmith();\n  const client = new Client(options);\n  const payloadOptions = {\n    object: runnable,\n    parentCommitHash: options?.parentCommitHash,\n    isPublic: options?.isPublic ?? options?.newRepoIsPublic,\n    description: options?.description ?? options?.newRepoDescription,\n    readme: options?.readme,\n    tags: options?.tags,\n  };\n  return client.pushPrompt(repoFullName, payloadOptions);\n}\n\nexport async function basePull(\n  ownerRepoCommit: string,\n  options?: { apiKey?: string; apiUrl?: string; includeModel?: boolean }\n): Promise<PromptCommit> {\n  const Client = await loadLangSmith();\n  const client = new Client(options);\n\n  const promptObject = await client.pullPromptCommit(ownerRepoCommit, {\n    includeModel: options?.includeModel,\n  });\n\n  if (promptObject.manifest.kwargs?.metadata === undefined) {\n    promptObject.manifest.kwargs = {\n      ...promptObject.manifest.kwargs,\n      metadata: {},\n    };\n  }\n\n  promptObject.manifest.kwargs.metadata = {\n    ...promptObject.manifest.kwargs.metadata,\n    lc_hub_owner: promptObject.owner,\n    lc_hub_repo: promptObject.repo,\n    lc_hub_commit_hash: promptObject.commit_hash,\n  };\n\n  // Some nested mustache prompts have improperly parsed variables that include a dot.\n  if (promptObject.manifest.kwargs.template_format === \"mustache\") {\n    const stripDotNotation = (varName: string) => varName.split(\".\")[0];\n\n    const { input_variables } = promptObject.manifest.kwargs;\n    if (Array.isArray(input_variables)) {\n      promptObject.manifest.kwargs.input_variables =\n        input_variables.map(stripDotNotation);\n    }\n\n    const { messages } = promptObject.manifest.kwargs;\n    if (Array.isArray(messages)) {\n      // oxlint-disable-next-line @typescript-eslint/no-explicit-any\n      promptObject.manifest.kwargs.messages = messages.map((message: any) => {\n        const nestedVars = message?.kwargs?.prompt?.kwargs?.input_variables;\n        if (Array.isArray(nestedVars)) {\n          message.kwargs.prompt.kwargs.input_variables =\n            nestedVars.map(stripDotNotation);\n        }\n        return message;\n      });\n    }\n  }\n  return promptObject;\n}\n\nexport function generateModelImportMap(\n  // oxlint-disable-next-line @typescript-eslint/no-explicit-any\n  modelClass?: new (...args: any[]) => BaseLanguageModel\n) {\n  // oxlint-disable-next-line @typescript-eslint/no-explicit-any\n  const modelImportMap: Record<string, any> = {};\n  if (modelClass !== undefined) {\n    // oxlint-disable-next-line @typescript-eslint/no-explicit-any\n    const modelLcName = (modelClass as any)?.lc_name();\n    let importMapKey;\n    if (modelLcName === \"ChatOpenAI\") {\n      importMapKey = \"chat_models__openai\";\n    } else if (modelLcName === \"ChatAnthropic\") {\n      importMapKey = \"chat_models__anthropic\";\n    } else if (modelLcName === \"ChatAzureOpenAI\") {\n      importMapKey = \"chat_models__openai\";\n    } else if (modelLcName === \"ChatVertexAI\") {\n      importMapKey = \"chat_models__vertexai\";\n    } else if (modelLcName === \"ChatGoogleGenerativeAI\") {\n      importMapKey = \"chat_models__google_genai\";\n    } else if (modelLcName === \"ChatBedrockConverse\") {\n      importMapKey = \"chat_models__chat_bedrock_converse\";\n    } else if (modelLcName === \"ChatMistral\") {\n      importMapKey = \"chat_models__mistralai\";\n    } else if (modelLcName === \"ChatFireworks\") {\n      importMapKey = \"chat_models__fireworks\";\n    } else if (modelLcName === \"ChatGroq\") {\n      importMapKey = \"chat_models__groq\";\n    } else {\n      throw new Error(\"Received unsupported model class when pulling prompt.\");\n    }\n    modelImportMap[importMapKey] = {\n      ...modelImportMap[importMapKey],\n      [modelLcName]: modelClass,\n    };\n  }\n  return modelImportMap;\n}\n\nexport function generateOptionalImportMap(\n  // oxlint-disable-next-line @typescript-eslint/no-explicit-any\n  modelClass?: new (...args: any[]) => BaseLanguageModel\n) {\n  // oxlint-disable-next-line @typescript-eslint/no-explicit-any\n  const optionalImportMap: Record<string, any> = {};\n  if (modelClass !== undefined) {\n    // oxlint-disable-next-line @typescript-eslint/no-explicit-any\n    const modelLcName = (modelClass as any)?.lc_name();\n    let optionalImportMapKey;\n    if (modelLcName === \"ChatGoogleGenerativeAI\") {\n      optionalImportMapKey = \"langchain_google_genai/chat_models\";\n    } else if (modelLcName === \"ChatBedrockConverse\") {\n      optionalImportMapKey = \"langchain_aws/chat_models\";\n    } else if (modelLcName === \"ChatGroq\") {\n      optionalImportMapKey = \"langchain_groq/chat_models\";\n    }\n    if (optionalImportMapKey !== undefined) {\n      optionalImportMap[optionalImportMapKey] = {\n        [modelLcName]: modelClass,\n      };\n    }\n  }\n  return optionalImportMap;\n}\n\nexport function bindOutputSchema<T extends Runnable>(loadedSequence: T) {\n  if (\n    \"first\" in loadedSequence &&\n    loadedSequence.first !== null &&\n    typeof loadedSequence.first === \"object\" &&\n    \"schema\" in loadedSequence.first &&\n    \"last\" in loadedSequence &&\n    loadedSequence.last !== null &&\n    typeof loadedSequence.last === \"object\"\n  ) {\n    if (\n      \"bound\" in loadedSequence.last &&\n      loadedSequence.last.bound !== null &&\n      typeof loadedSequence.last.bound === \"object\" &&\n      \"withStructuredOutput\" in loadedSequence.last.bound &&\n      typeof loadedSequence.last.bound.withStructuredOutput === \"function\"\n    ) {\n      loadedSequence.last.bound =\n        loadedSequence.last.bound.withStructuredOutput(\n          loadedSequence.first.schema\n        );\n    } else if (\n      \"withStructuredOutput\" in loadedSequence.last &&\n      typeof loadedSequence.last.withStructuredOutput === \"function\"\n    ) {\n      loadedSequence.last = loadedSequence.last.withStructuredOutput(\n        loadedSequence.first.schema\n      );\n    }\n  }\n  return loadedSequence;\n}\n\n/**\n * Dynamically load the LangSmith client.\n * @returns The LangSmith client.\n */\nasync function loadLangSmith(): Promise<new (config?: ClientConfig) => Client> {\n  try {\n    const { Client } = await import(\"langsmith\");\n    return Client;\n  } catch (error) {\n    // oxlint-disable-next-line no-instanceof/no-instanceof\n    const errorMessage = error instanceof Error ? error.message : String(error);\n    throw new Error(\n      `Error loading \"langsmith\" package, install it via \\`npm install langsmith\\` before you use this function.\\nError: ${errorMessage}`\n    );\n  }\n}\n"],"mappings":";;;;;;;;;AAcA,eAAsB,SACpB,cACA,UACA,SAaiB;CAEjB,MAAM,SAAS,KAAI,OADE,cAAc,IACT,OAAO;CACjC,MAAM,iBAAiB;EACrB,QAAQ;EACR,kBAAkB,SAAS;EAC3B,UAAU,SAAS,YAAY,SAAS;EACxC,aAAa,SAAS,eAAe,SAAS;EAC9C,QAAQ,SAAS;EACjB,MAAM,SAAS;CACjB;CACA,OAAO,OAAO,WAAW,cAAc,cAAc;AACvD;AAEA,eAAsB,SACpB,iBACA,SACuB;CAIvB,MAAM,eAAe,MAAM,KAFR,OADE,cAAc,IACT,OAEM,CAAC,CAAC,iBAAiB,iBAAiB,EAClE,cAAc,SAAS,aACzB,CAAC;CAED,IAAI,aAAa,SAAS,QAAQ,aAAa,KAAA,GAC7C,aAAa,SAAS,SAAS;EAC7B,GAAG,aAAa,SAAS;EACzB,UAAU,CAAC;CACb;CAGF,aAAa,SAAS,OAAO,WAAW;EACtC,GAAG,aAAa,SAAS,OAAO;EAChC,cAAc,aAAa;EAC3B,aAAa,aAAa;EAC1B,oBAAoB,aAAa;CACnC;CAGA,IAAI,aAAa,SAAS,OAAO,oBAAoB,YAAY;EAC/D,MAAM,oBAAoB,YAAoB,QAAQ,MAAM,GAAG,CAAC,CAAC;EAEjE,MAAM,EAAE,oBAAoB,aAAa,SAAS;EAClD,IAAI,MAAM,QAAQ,eAAe,GAC/B,aAAa,SAAS,OAAO,kBAC3B,gBAAgB,IAAI,gBAAgB;EAGxC,MAAM,EAAE,aAAa,aAAa,SAAS;EAC3C,IAAI,MAAM,QAAQ,QAAQ,GAExB,aAAa,SAAS,OAAO,WAAW,SAAS,KAAK,YAAiB;GACrE,MAAM,aAAa,SAAS,QAAQ,QAAQ,QAAQ;GACpD,IAAI,MAAM,QAAQ,UAAU,GAC1B,QAAQ,OAAO,OAAO,OAAO,kBAC3B,WAAW,IAAI,gBAAgB;GAEnC,OAAO;EACT,CAAC;CAEL;CACA,OAAO;AACT;AAEA,SAAgB,uBAEd,YACA;CAEA,MAAM,iBAAsC,CAAC;CAC7C,IAAI,eAAe,KAAA,GAAW;EAE5B,MAAM,cAAe,YAAoB,QAAQ;EACjD,IAAI;EACJ,IAAI,gBAAgB,cAClB,eAAe;OACV,IAAI,gBAAgB,iBACzB,eAAe;OACV,IAAI,gBAAgB,mBACzB,eAAe;OACV,IAAI,gBAAgB,gBACzB,eAAe;OACV,IAAI,gBAAgB,0BACzB,eAAe;OACV,IAAI,gBAAgB,uBACzB,eAAe;OACV,IAAI,gBAAgB,eACzB,eAAe;OACV,IAAI,gBAAgB,iBACzB,eAAe;OACV,IAAI,gBAAgB,YACzB,eAAe;OAEf,MAAM,IAAI,MAAM,uDAAuD;EAEzE,eAAe,gBAAgB;GAC7B,GAAG,eAAe;IACjB,cAAc;EACjB;CACF;CACA,OAAO;AACT;AAEA,SAAgB,0BAEd,YACA;CAEA,MAAM,oBAAyC,CAAC;CAChD,IAAI,eAAe,KAAA,GAAW;EAE5B,MAAM,cAAe,YAAoB,QAAQ;EACjD,IAAI;EACJ,IAAI,gBAAgB,0BAClB,uBAAuB;OAClB,IAAI,gBAAgB,uBACzB,uBAAuB;OAClB,IAAI,gBAAgB,YACzB,uBAAuB;EAEzB,IAAI,yBAAyB,KAAA,GAC3B,kBAAkB,wBAAwB,GACvC,cAAc,WACjB;CAEJ;CACA,OAAO;AACT;AAEA,SAAgB,iBAAqC,gBAAmB;CACtE,IACE,WAAW,kBACX,eAAe,UAAU,QACzB,OAAO,eAAe,UAAU,YAChC,YAAY,eAAe,SAC3B,UAAU,kBACV,eAAe,SAAS,QACxB,OAAO,eAAe,SAAS,UAC/B;EACA,IACE,WAAW,eAAe,QAC1B,eAAe,KAAK,UAAU,QAC9B,OAAO,eAAe,KAAK,UAAU,YACrC,0BAA0B,eAAe,KAAK,SAC9C,OAAO,eAAe,KAAK,MAAM,yBAAyB,YAE1D,eAAe,KAAK,QAClB,eAAe,KAAK,MAAM,qBACxB,eAAe,MAAM,MACvB;OACG,IACL,0BAA0B,eAAe,QACzC,OAAO,eAAe,KAAK,yBAAyB,YAEpD,eAAe,OAAO,eAAe,KAAK,qBACxC,eAAe,MAAM,MACvB;CAEJ;CACA,OAAO;AACT;;;;;AAMA,eAAe,gBAAgE;CAC7E,IAAI;EACF,MAAM,EAAE,WAAW,MAAM,OAAO;EAChC,OAAO;CACT,SAAS,OAAO;EAEd,MAAM,eAAe,iBAAiB,QAAQ,MAAM,UAAU,OAAO,KAAK;EAC1E,MAAM,IAAI,MACR,qHAAqH,cACvH;CACF;AACF"}

package/dist/hub/index.cjs

@@ -28,7 +28,8 @@ const require_base = require("./base.cjs");
 async function pull(ownerRepoCommit, options) {
 	const promptObject = await require_base.basePull(ownerRepoCommit, options);
 	try {
-		return require_base.bindOutputSchema(await require_load_index.load(JSON.stringify(promptObject.manifest), options?.secrets, require_base.generateOptionalImportMap(options?.modelClass), require_base.generateModelImportMap(options?.modelClass), options?.secretsFromEnv));
+		const loadedPrompt = await require_load_index.load(JSON.stringify(promptObject.manifest), options?.secrets, require_base.generateOptionalImportMap(options?.modelClass), require_base.generateModelImportMap(options?.modelClass), options?.secretsFromEnv);
+		return require_base.bindOutputSchema(loadedPrompt);
 	} catch (e) {
 		if (options?.includeModel) throw new Error([
 			e.message,

package/dist/hub/index.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"index.cjs","names":["basePull","bindOutputSchema","load","generateOptionalImportMap","generateModelImportMap"],"sources":["../../src/hub/index.ts"],"sourcesContent":["import { Runnable } from \"@langchain/core/runnables\";\nimport type { BaseLanguageModel } from \"@langchain/core/language_models/base\";\nimport { load } from \"../load/index.js\";\nimport {\n  basePush,\n  basePull,\n  generateModelImportMap,\n  generateOptionalImportMap,\n  bindOutputSchema,\n} from \"./base.js\";\n\nexport { basePush as push };\n\n/**\n * Pull a prompt from the hub.\n *\n * @param ownerRepoCommit The name of the repo containing the prompt, as well as an optional commit hash separated by a slash.\n * @param options.apiKey LangSmith API key to use when pulling the prompt\n * @param options.apiUrl LangSmith API URL to use when pulling the prompt\n * @param options.includeModel Whether to also instantiate and attach a model instance to the prompt,\n *   if the prompt has associated model metadata. If set to true, invoking the resulting pulled prompt will\n *   also invoke the instantiated model. For non-OpenAI models, you must also set \"modelClass\" to the\n *   correct class of the model.\n * @param options.modelClass If includeModel is true, the class of the model to instantiate. Required\n *   for non-OpenAI models. If you are running in Node or another environment that supports dynamic imports,\n *   you may instead import this function from \"langchain/hub/node\" and pass \"includeModel: true\" instead\n *   of specifying this parameter.\n * @param options.secrets A map of secrets to use when loading, e.g.\n *   {'OPENAI_API_KEY': 'sk-...'}`.\n *   If a secret is not found in the map, it will be loaded from the\n *   environment if `secrets_from_env` is `True`. Should only be needed when\n *   `includeModel` is `true`.\n * @param options.secretsFromEnv Whether to load secrets from environment variables.\n *   Use with caution and only with trusted prompts.\n * @returns\n */\nexport async function pull<T extends Runnable>(\n  ownerRepoCommit: string,\n  options?: {\n    apiKey?: string;\n    apiUrl?: string;\n    includeModel?: boolean;\n    // oxlint-disable-next-line @typescript-eslint/no-explicit-any\n    modelClass?: new (...args: any[]) => BaseLanguageModel;\n    secrets?: Record<string, string>;\n    secretsFromEnv?: boolean;\n  }\n) {\n  const promptObject = await basePull(ownerRepoCommit, options);\n  try {\n    const loadedPrompt = await load<T>(\n      JSON.stringify(promptObject.manifest),\n      options?.secrets,\n      generateOptionalImportMap(options?.modelClass),\n      generateModelImportMap(options?.modelClass),\n      options?.secretsFromEnv\n    );\n    return bindOutputSchema(loadedPrompt);\n    // oxlint-disable-next-line @typescript-eslint/no-explicit-any\n  } catch (e: any) {\n    if (options?.includeModel) {\n      throw new Error(\n        [\n          e.message,\n          \"\",\n          `To load prompts with an associated non-OpenAI model, you must use the \"langchain/hub/node\" entrypoint, or pass a \"modelClass\" parameter like this:`,\n          \"\",\n          \"```\",\n          `import { pull } from \"langchain/hub\";`,\n          `import { ChatAnthropic } from \"@langchain/anthropic\";`,\n          \"\",\n          `const prompt = await pull(\"my-prompt\", {`,\n          `  includeModel: true,`,\n          `  modelClass: ChatAnthropic,`,\n          `});`,\n          \"```\",\n        ].join(\"\\n\")\n      );\n    } else {\n      throw e;\n    }\n  }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;AAoCA,eAAsB,KACpB,iBACA,SASA;CACA,MAAM,eAAe,MAAMA,aAAAA,SAAS,iBAAiB,OAAO;CAC5D,IAAI;EAQF,OAAOC,aAAAA,iBAAiB,MAPGC,mBAAAA,KACzB,KAAK,UAAU,aAAa,QAAQ,GACpC,SAAS,SACTC,aAAAA,0BAA0B,SAAS,UAAU,GAC7CC,aAAAA,uBAAuB,SAAS,UAAU,GAC1C,SAAS,cACX,CACoC;CAEtC,SAAS,GAAQ;EACf,IAAI,SAAS,cACX,MAAM,IAAI,MACR;GACE,EAAE;GACF;GACA;GACA;GACA;GACA;GACA;GACA;GACA;GACA;GACA;GACA;GACA;EACF,CAAC,CAAC,KAAK,IAAI,CACb;OAEA,MAAM;CAEV;AACF"}
+{"version":3,"file":"index.cjs","names":["basePull","load","generateOptionalImportMap","generateModelImportMap","bindOutputSchema"],"sources":["../../src/hub/index.ts"],"sourcesContent":["import { Runnable } from \"@langchain/core/runnables\";\nimport type { BaseLanguageModel } from \"@langchain/core/language_models/base\";\nimport { load } from \"../load/index.js\";\nimport {\n  basePush,\n  basePull,\n  generateModelImportMap,\n  generateOptionalImportMap,\n  bindOutputSchema,\n} from \"./base.js\";\n\nexport { basePush as push };\n\n/**\n * Pull a prompt from the hub.\n *\n * @param ownerRepoCommit The name of the repo containing the prompt, as well as an optional commit hash separated by a slash.\n * @param options.apiKey LangSmith API key to use when pulling the prompt\n * @param options.apiUrl LangSmith API URL to use when pulling the prompt\n * @param options.includeModel Whether to also instantiate and attach a model instance to the prompt,\n *   if the prompt has associated model metadata. If set to true, invoking the resulting pulled prompt will\n *   also invoke the instantiated model. For non-OpenAI models, you must also set \"modelClass\" to the\n *   correct class of the model.\n * @param options.modelClass If includeModel is true, the class of the model to instantiate. Required\n *   for non-OpenAI models. If you are running in Node or another environment that supports dynamic imports,\n *   you may instead import this function from \"langchain/hub/node\" and pass \"includeModel: true\" instead\n *   of specifying this parameter.\n * @param options.secrets A map of secrets to use when loading, e.g.\n *   {'OPENAI_API_KEY': 'sk-...'}`.\n *   If a secret is not found in the map, it will be loaded from the\n *   environment if `secrets_from_env` is `True`. Should only be needed when\n *   `includeModel` is `true`.\n * @param options.secretsFromEnv Whether to load secrets from environment variables.\n *   Use with caution and only with trusted prompts.\n * @returns\n */\nexport async function pull<T extends Runnable>(\n  ownerRepoCommit: string,\n  options?: {\n    apiKey?: string;\n    apiUrl?: string;\n    includeModel?: boolean;\n    // oxlint-disable-next-line @typescript-eslint/no-explicit-any\n    modelClass?: new (...args: any[]) => BaseLanguageModel;\n    secrets?: Record<string, string>;\n    secretsFromEnv?: boolean;\n  }\n) {\n  const promptObject = await basePull(ownerRepoCommit, options);\n  try {\n    const loadedPrompt = await load<T>(\n      JSON.stringify(promptObject.manifest),\n      options?.secrets,\n      generateOptionalImportMap(options?.modelClass),\n      generateModelImportMap(options?.modelClass),\n      options?.secretsFromEnv\n    );\n    return bindOutputSchema(loadedPrompt);\n    // oxlint-disable-next-line @typescript-eslint/no-explicit-any\n  } catch (e: any) {\n    if (options?.includeModel) {\n      throw new Error(\n        [\n          e.message,\n          \"\",\n          `To load prompts with an associated non-OpenAI model, you must use the \"langchain/hub/node\" entrypoint, or pass a \"modelClass\" parameter like this:`,\n          \"\",\n          \"```\",\n          `import { pull } from \"langchain/hub\";`,\n          `import { ChatAnthropic } from \"@langchain/anthropic\";`,\n          \"\",\n          `const prompt = await pull(\"my-prompt\", {`,\n          `  includeModel: true,`,\n          `  modelClass: ChatAnthropic,`,\n          `});`,\n          \"```\",\n        ].join(\"\\n\")\n      );\n    } else {\n      throw e;\n    }\n  }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;AAoCA,eAAsB,KACpB,iBACA,SASA;CACA,MAAM,eAAe,MAAMA,aAAAA,SAAS,iBAAiB,OAAO;CAC5D,IAAI;EACF,MAAM,eAAe,MAAMC,mBAAAA,KACzB,KAAK,UAAU,aAAa,QAAQ,GACpC,SAAS,SACTC,aAAAA,0BAA0B,SAAS,UAAU,GAC7CC,aAAAA,uBAAuB,SAAS,UAAU,GAC1C,SAAS,cACX;EACA,OAAOC,aAAAA,iBAAiB,YAAY;CAEtC,SAAS,GAAQ;EACf,IAAI,SAAS,cACX,MAAM,IAAI,MACR;GACE,EAAE;GACF;GACA;GACA;GACA;GACA;GACA;GACA;GACA;GACA;GACA;GACA;GACA;EACF,CAAC,CAAC,KAAK,IAAI,CACb;OAEA,MAAM;CAEV;AACF"}

package/dist/hub/index.js

@@ -27,7 +27,8 @@ import { basePull, basePush, bindOutputSchema, generateModelImportMap, generateO
 async function pull(ownerRepoCommit, options) {
 	const promptObject = await basePull(ownerRepoCommit, options);
 	try {
-		return bindOutputSchema(await load(JSON.stringify(promptObject.manifest), options?.secrets, generateOptionalImportMap(options?.modelClass), generateModelImportMap(options?.modelClass), options?.secretsFromEnv));
+		const loadedPrompt = await load(JSON.stringify(promptObject.manifest), options?.secrets, generateOptionalImportMap(options?.modelClass), generateModelImportMap(options?.modelClass), options?.secretsFromEnv);
+		return bindOutputSchema(loadedPrompt);
 	} catch (e) {
 		if (options?.includeModel) throw new Error([
 			e.message,

package/dist/hub/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","names":[],"sources":["../../src/hub/index.ts"],"sourcesContent":["import { Runnable } from \"@langchain/core/runnables\";\nimport type { BaseLanguageModel } from \"@langchain/core/language_models/base\";\nimport { load } from \"../load/index.js\";\nimport {\n  basePush,\n  basePull,\n  generateModelImportMap,\n  generateOptionalImportMap,\n  bindOutputSchema,\n} from \"./base.js\";\n\nexport { basePush as push };\n\n/**\n * Pull a prompt from the hub.\n *\n * @param ownerRepoCommit The name of the repo containing the prompt, as well as an optional commit hash separated by a slash.\n * @param options.apiKey LangSmith API key to use when pulling the prompt\n * @param options.apiUrl LangSmith API URL to use when pulling the prompt\n * @param options.includeModel Whether to also instantiate and attach a model instance to the prompt,\n *   if the prompt has associated model metadata. If set to true, invoking the resulting pulled prompt will\n *   also invoke the instantiated model. For non-OpenAI models, you must also set \"modelClass\" to the\n *   correct class of the model.\n * @param options.modelClass If includeModel is true, the class of the model to instantiate. Required\n *   for non-OpenAI models. If you are running in Node or another environment that supports dynamic imports,\n *   you may instead import this function from \"langchain/hub/node\" and pass \"includeModel: true\" instead\n *   of specifying this parameter.\n * @param options.secrets A map of secrets to use when loading, e.g.\n *   {'OPENAI_API_KEY': 'sk-...'}`.\n *   If a secret is not found in the map, it will be loaded from the\n *   environment if `secrets_from_env` is `True`. Should only be needed when\n *   `includeModel` is `true`.\n * @param options.secretsFromEnv Whether to load secrets from environment variables.\n *   Use with caution and only with trusted prompts.\n * @returns\n */\nexport async function pull<T extends Runnable>(\n  ownerRepoCommit: string,\n  options?: {\n    apiKey?: string;\n    apiUrl?: string;\n    includeModel?: boolean;\n    // oxlint-disable-next-line @typescript-eslint/no-explicit-any\n    modelClass?: new (...args: any[]) => BaseLanguageModel;\n    secrets?: Record<string, string>;\n    secretsFromEnv?: boolean;\n  }\n) {\n  const promptObject = await basePull(ownerRepoCommit, options);\n  try {\n    const loadedPrompt = await load<T>(\n      JSON.stringify(promptObject.manifest),\n      options?.secrets,\n      generateOptionalImportMap(options?.modelClass),\n      generateModelImportMap(options?.modelClass),\n      options?.secretsFromEnv\n    );\n    return bindOutputSchema(loadedPrompt);\n    // oxlint-disable-next-line @typescript-eslint/no-explicit-any\n  } catch (e: any) {\n    if (options?.includeModel) {\n      throw new Error(\n        [\n          e.message,\n          \"\",\n          `To load prompts with an associated non-OpenAI model, you must use the \"langchain/hub/node\" entrypoint, or pass a \"modelClass\" parameter like this:`,\n          \"\",\n          \"```\",\n          `import { pull } from \"langchain/hub\";`,\n          `import { ChatAnthropic } from \"@langchain/anthropic\";`,\n          \"\",\n          `const prompt = await pull(\"my-prompt\", {`,\n          `  includeModel: true,`,\n          `  modelClass: ChatAnthropic,`,\n          `});`,\n          \"```\",\n        ].join(\"\\n\")\n      );\n    } else {\n      throw e;\n    }\n  }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;AAoCA,eAAsB,KACpB,iBACA,SASA;CACA,MAAM,eAAe,MAAM,SAAS,iBAAiB,OAAO;CAC5D,IAAI;EAQF,OAAO,iBAAiB,MAPG,KACzB,KAAK,UAAU,aAAa,QAAQ,GACpC,SAAS,SACT,0BAA0B,SAAS,UAAU,GAC7C,uBAAuB,SAAS,UAAU,GAC1C,SAAS,cACX,CACoC;CAEtC,SAAS,GAAQ;EACf,IAAI,SAAS,cACX,MAAM,IAAI,MACR;GACE,EAAE;GACF;GACA;GACA;GACA;GACA;GACA;GACA;GACA;GACA;GACA;GACA;GACA;EACF,CAAC,CAAC,KAAK,IAAI,CACb;OAEA,MAAM;CAEV;AACF"}
+{"version":3,"file":"index.js","names":[],"sources":["../../src/hub/index.ts"],"sourcesContent":["import { Runnable } from \"@langchain/core/runnables\";\nimport type { BaseLanguageModel } from \"@langchain/core/language_models/base\";\nimport { load } from \"../load/index.js\";\nimport {\n  basePush,\n  basePull,\n  generateModelImportMap,\n  generateOptionalImportMap,\n  bindOutputSchema,\n} from \"./base.js\";\n\nexport { basePush as push };\n\n/**\n * Pull a prompt from the hub.\n *\n * @param ownerRepoCommit The name of the repo containing the prompt, as well as an optional commit hash separated by a slash.\n * @param options.apiKey LangSmith API key to use when pulling the prompt\n * @param options.apiUrl LangSmith API URL to use when pulling the prompt\n * @param options.includeModel Whether to also instantiate and attach a model instance to the prompt,\n *   if the prompt has associated model metadata. If set to true, invoking the resulting pulled prompt will\n *   also invoke the instantiated model. For non-OpenAI models, you must also set \"modelClass\" to the\n *   correct class of the model.\n * @param options.modelClass If includeModel is true, the class of the model to instantiate. Required\n *   for non-OpenAI models. If you are running in Node or another environment that supports dynamic imports,\n *   you may instead import this function from \"langchain/hub/node\" and pass \"includeModel: true\" instead\n *   of specifying this parameter.\n * @param options.secrets A map of secrets to use when loading, e.g.\n *   {'OPENAI_API_KEY': 'sk-...'}`.\n *   If a secret is not found in the map, it will be loaded from the\n *   environment if `secrets_from_env` is `True`. Should only be needed when\n *   `includeModel` is `true`.\n * @param options.secretsFromEnv Whether to load secrets from environment variables.\n *   Use with caution and only with trusted prompts.\n * @returns\n */\nexport async function pull<T extends Runnable>(\n  ownerRepoCommit: string,\n  options?: {\n    apiKey?: string;\n    apiUrl?: string;\n    includeModel?: boolean;\n    // oxlint-disable-next-line @typescript-eslint/no-explicit-any\n    modelClass?: new (...args: any[]) => BaseLanguageModel;\n    secrets?: Record<string, string>;\n    secretsFromEnv?: boolean;\n  }\n) {\n  const promptObject = await basePull(ownerRepoCommit, options);\n  try {\n    const loadedPrompt = await load<T>(\n      JSON.stringify(promptObject.manifest),\n      options?.secrets,\n      generateOptionalImportMap(options?.modelClass),\n      generateModelImportMap(options?.modelClass),\n      options?.secretsFromEnv\n    );\n    return bindOutputSchema(loadedPrompt);\n    // oxlint-disable-next-line @typescript-eslint/no-explicit-any\n  } catch (e: any) {\n    if (options?.includeModel) {\n      throw new Error(\n        [\n          e.message,\n          \"\",\n          `To load prompts with an associated non-OpenAI model, you must use the \"langchain/hub/node\" entrypoint, or pass a \"modelClass\" parameter like this:`,\n          \"\",\n          \"```\",\n          `import { pull } from \"langchain/hub\";`,\n          `import { ChatAnthropic } from \"@langchain/anthropic\";`,\n          \"\",\n          `const prompt = await pull(\"my-prompt\", {`,\n          `  includeModel: true,`,\n          `  modelClass: ChatAnthropic,`,\n          `});`,\n          \"```\",\n        ].join(\"\\n\")\n      );\n    } else {\n      throw e;\n    }\n  }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;AAoCA,eAAsB,KACpB,iBACA,SASA;CACA,MAAM,eAAe,MAAM,SAAS,iBAAiB,OAAO;CAC5D,IAAI;EACF,MAAM,eAAe,MAAM,KACzB,KAAK,UAAU,aAAa,QAAQ,GACpC,SAAS,SACT,0BAA0B,SAAS,UAAU,GAC7C,uBAAuB,SAAS,UAAU,GAC1C,SAAS,cACX;EACA,OAAO,iBAAiB,YAAY;CAEtC,SAAS,GAAQ;EACf,IAAI,SAAS,cACX,MAAM,IAAI,MACR;GACE,EAAE;GACF;GACA;GACA;GACA;GACA;GACA;GACA;GACA;GACA;GACA;GACA;GACA;EACF,CAAC,CAAC,KAAK,IAAI,CACb;OAEA,MAAM;CAEV;AACF"}