langchain 1.0.0-alpha.4 → 1.0.0-alpha.6

package/dist/agents/utils.js

@@ -1,6 +1,6 @@
 import { MultipleToolsBoundError } from "./errors.js";
 import { PROMPT_RUNNABLE_NAME } from "./constants.js";
-import { AIMessage, SystemMessage, isAIMessage, isAIMessageChunk, isBaseMessage, isBaseMessageChunk } from "@langchain/core/messages";
+import { AIMessage, AIMessageChunk, SystemMessage } from "@langchain/core/messages";
 import { Runnable, RunnableBinding, RunnableLambda, RunnableSequence } from "@langchain/core/runnables";
 
 //#region src/agents/utils.ts
@@ -21,11 +21,11 @@ const CONTENT_PATTERN = /<content>(.*?)<\/content>/s;
 * @internal
 */
 function _addInlineAgentName(message) {
-	const isAI = isBaseMessage(message) && (isAIMessage(message) || isBaseMessageChunk(message) && isAIMessageChunk(message));
-	if (!isAI || !message.name) return message;
+	if (!AIMessage.isInstance(message) || AIMessageChunk.isInstance(message)) return message;
+	if (!message.name) return message;
 	const { name } = message;
 	if (typeof message.content === "string") return new AIMessage({
-		...Object.keys(message.lc_kwargs ?? {}).length > 0 ? message.lc_kwargs : message,
+		...message.lc_kwargs,
 		content: `<name>${name}</name><content>${message.content}</content>`,
 		name: void 0
 	});
@@ -68,7 +68,7 @@ function _addInlineAgentName(message) {
 * @internal
 */
 function _removeInlineAgentName(message) {
-	if (!isAIMessage(message) || !message.content) return message;
+	if (!AIMessage.isInstance(message) || !message.content) return message;
 	let updatedContent = [];
 	let updatedName;
 	if (Array.isArray(message.content)) updatedContent = message.content.filter((block) => {
@@ -130,7 +130,7 @@ function getPromptRunnable(prompt) {
 		promptRunnable = RunnableLambda.from((state) => {
 			return [systemMessage, ...state.messages ?? []];
 		}).withConfig({ runName: PROMPT_RUNNABLE_NAME });
-	} else if (isBaseMessage(prompt) && prompt._getType() === "system") promptRunnable = RunnableLambda.from((state) => [prompt, ...state.messages]).withConfig({ runName: PROMPT_RUNNABLE_NAME });
+	} else if (SystemMessage.isInstance(prompt)) promptRunnable = RunnableLambda.from((state) => [prompt, ...state.messages]).withConfig({ runName: PROMPT_RUNNABLE_NAME });
 	else if (typeof prompt === "function") promptRunnable = RunnableLambda.from(prompt).withConfig({ runName: PROMPT_RUNNABLE_NAME });
 	else if (Runnable.isRunnable(prompt)) promptRunnable = prompt;
 	else throw new Error(`Got unexpected type for 'prompt': ${typeof prompt}`);
@@ -226,7 +226,7 @@ function validateLLMHasNoBoundTools(llm) {
 */
 function hasToolCalls(messages) {
 	const lastMessage = messages.at(-1);
-	return Boolean(lastMessage instanceof AIMessage && lastMessage.tool_calls && lastMessage.tool_calls.length > 0);
+	return Boolean(AIMessage.isInstance(lastMessage) && lastMessage.tool_calls && lastMessage.tool_calls.length > 0);
 }
 const CHAT_MODELS_THAT_SUPPORT_JSON_SCHEMA_OUTPUT = ["ChatOpenAI", "FakeToolCallingModel"];
 /**

package/dist/agents/utils.js.map

@@ -1 +1 @@
-{"version":3,"file":"utils.js","names":["message: T","updatedContent: MessageContent","updatedName: string | undefined","tool: ClientTool | ServerTool","model: LanguageModelLike","model: unknown","llm: LanguageModelLike","prompt?: Prompt","promptRunnable: Runnable","state: typeof MessagesAnnotation.State","toolClasses: (ClientTool | ServerTool)[]","options: Partial<BaseChatModelCallOptions>","model","nextSteps: unknown[]","step: RunnableLike","messages: BaseMessage[]"],"sources":["../../src/agents/utils.ts"],"sourcesContent":["/* eslint-disable no-instanceof/no-instanceof */\nimport {\n  AIMessage,\n  BaseMessage,\n  BaseMessageLike,\n  MessageContent,\n  SystemMessage,\n  AIMessageChunk,\n  isAIMessage,\n  isAIMessageChunk,\n  isBaseMessage,\n  isBaseMessageChunk,\n} from \"@langchain/core/messages\";\nimport { MessagesAnnotation } from \"@langchain/langgraph\";\nimport {\n  BaseChatModel,\n  type BindToolsInput,\n  type BaseChatModelCallOptions,\n} from \"@langchain/core/language_models/chat_models\";\nimport {\n  LanguageModelLike,\n  BaseLanguageModelInput,\n} from \"@langchain/core/language_models/base\";\nimport {\n  Runnable,\n  RunnableLike,\n  RunnableConfig,\n  RunnableLambda,\n  RunnableSequence,\n  RunnableBinding,\n} from \"@langchain/core/runnables\";\n\nimport { MultipleToolsBoundError } from \"./errors.js\";\nimport { PROMPT_RUNNABLE_NAME } from \"./constants.js\";\nimport {\n  ServerTool,\n  ClientTool,\n  ConfigurableModelInterface,\n  Prompt,\n} from \"./types.js\";\n\nconst NAME_PATTERN = /<name>(.*?)<\\/name>/s;\nconst CONTENT_PATTERN = /<content>(.*?)<\\/content>/s;\n\nexport type AgentNameMode = \"inline\";\n\n/**\n * Attach formatted agent names to the messages passed to and from a language model.\n *\n * This is useful for making a message history with multiple agents more coherent.\n *\n * NOTE: agent name is consumed from the message.name field.\n * If you're using an agent built with createAgent, name is automatically set.\n * If you're building a custom agent, make sure to set the name on the AI message returned by the LLM.\n *\n * @param message - Message to add agent name formatting to\n * @returns Message with agent name formatting\n *\n * @internal\n */\nexport function _addInlineAgentName<T extends BaseMessageLike>(\n  message: T\n): T | AIMessage {\n  const isAI =\n    isBaseMessage(message) &&\n    (isAIMessage(message) ||\n      (isBaseMessageChunk(message) && isAIMessageChunk(message)));\n\n  if (!isAI || !message.name) {\n    return message;\n  }\n\n  const { name } = message;\n\n  if (typeof message.content === \"string\") {\n    return new AIMessage({\n      ...(Object.keys(message.lc_kwargs ?? {}).length > 0\n        ? message.lc_kwargs\n        : message),\n      content: `<name>${name}</name><content>${message.content}</content>`,\n      name: undefined,\n    });\n  }\n\n  const updatedContent = [];\n  let textBlockCount = 0;\n\n  for (const contentBlock of message.content) {\n    if (typeof contentBlock === \"string\") {\n      textBlockCount += 1;\n      updatedContent.push(\n        `<name>${name}</name><content>${contentBlock}</content>`\n      );\n    } else if (\n      typeof contentBlock === \"object\" &&\n      \"type\" in contentBlock &&\n      contentBlock.type === \"text\"\n    ) {\n      textBlockCount += 1;\n      updatedContent.push({\n        ...contentBlock,\n        text: `<name>${name}</name><content>${contentBlock.text}</content>`,\n      });\n    } else {\n      updatedContent.push(contentBlock);\n    }\n  }\n\n  if (!textBlockCount) {\n    updatedContent.unshift({\n      type: \"text\",\n      text: `<name>${name}</name><content></content>`,\n    });\n  }\n  return new AIMessage({\n    ...message.lc_kwargs,\n    content: updatedContent as MessageContent,\n    name: undefined,\n  });\n}\n\n/**\n * Remove explicit name and content XML tags from the AI message content.\n *\n * Examples:\n *\n * @example\n * ```typescript\n * removeInlineAgentName(new AIMessage({ content: \"<name>assistant</name><content>Hello</content>\", name: \"assistant\" }))\n * // AIMessage with content: \"Hello\"\n *\n * removeInlineAgentName(new AIMessage({ content: [{type: \"text\", text: \"<name>assistant</name><content>Hello</content>\"}], name: \"assistant\" }))\n * // AIMessage with content: [{type: \"text\", text: \"Hello\"}]\n * ```\n *\n * @internal\n */\nexport function _removeInlineAgentName<T extends BaseMessage>(message: T): T {\n  if (!isAIMessage(message) || !message.content) {\n    return message;\n  }\n\n  let updatedContent: MessageContent = [];\n  let updatedName: string | undefined;\n\n  if (Array.isArray(message.content)) {\n    updatedContent = message.content\n      .filter((block) => {\n        if (block.type === \"text\" && typeof block.text === \"string\") {\n          const nameMatch = block.text.match(NAME_PATTERN);\n          const contentMatch = block.text.match(CONTENT_PATTERN);\n          // don't include empty content blocks that were added because there was no text block to modify\n          if (nameMatch && (!contentMatch || contentMatch[1] === \"\")) {\n            // capture name from text block\n            // eslint-disable-next-line prefer-destructuring\n            updatedName = nameMatch[1];\n            return false;\n          }\n          return true;\n        }\n        return true;\n      })\n      .map((block) => {\n        if (block.type === \"text\" && typeof block.text === \"string\") {\n          const nameMatch = block.text.match(NAME_PATTERN);\n          const contentMatch = block.text.match(CONTENT_PATTERN);\n\n          if (!nameMatch || !contentMatch) {\n            return block;\n          }\n\n          // capture name from text block\n          // eslint-disable-next-line prefer-destructuring\n          updatedName = nameMatch[1];\n\n          return {\n            ...block,\n            text: contentMatch[1],\n          };\n        }\n        return block;\n      });\n  } else {\n    const content = message.content as string;\n    const nameMatch = content.match(NAME_PATTERN);\n    const contentMatch = content.match(CONTENT_PATTERN);\n\n    if (!nameMatch || !contentMatch) {\n      return message;\n    }\n\n    // eslint-disable-next-line prefer-destructuring\n    updatedName = nameMatch[1];\n    // eslint-disable-next-line prefer-destructuring\n    updatedContent = contentMatch[1];\n  }\n\n  return new AIMessage({\n    ...(Object.keys(message.lc_kwargs ?? {}).length > 0\n      ? message.lc_kwargs\n      : message),\n    content: updatedContent,\n    name: updatedName,\n  }) as T;\n}\n\nexport function isClientTool(\n  tool: ClientTool | ServerTool\n): tool is ClientTool {\n  return Runnable.isRunnable(tool);\n}\n\nexport function isBaseChatModel(\n  model: LanguageModelLike\n): model is BaseChatModel {\n  return (\n    \"invoke\" in model &&\n    typeof model.invoke === \"function\" &&\n    \"_modelType\" in model\n  );\n}\n\n// eslint-disable-next-line @typescript-eslint/no-explicit-any\nexport function isConfigurableModel(\n  model: unknown\n): model is ConfigurableModelInterface {\n  return (\n    typeof model === \"object\" &&\n    model != null &&\n    \"_queuedMethodOperations\" in model &&\n    \"_model\" in model &&\n    typeof model._model === \"function\"\n  );\n}\n\nfunction _isChatModelWithBindTools(\n  llm: LanguageModelLike\n): llm is BaseChatModel & Required<Pick<BaseChatModel, \"bindTools\">> {\n  if (!isBaseChatModel(llm)) return false;\n  return \"bindTools\" in llm && typeof llm.bindTools === \"function\";\n}\n\nexport function getPromptRunnable(prompt?: Prompt): Runnable {\n  let promptRunnable: Runnable;\n\n  if (prompt == null) {\n    promptRunnable = RunnableLambda.from(\n      (state: typeof MessagesAnnotation.State) => state.messages\n    ).withConfig({ runName: PROMPT_RUNNABLE_NAME });\n  } else if (typeof prompt === \"string\") {\n    const systemMessage = new SystemMessage(prompt);\n    promptRunnable = RunnableLambda.from(\n      (state: typeof MessagesAnnotation.State) => {\n        return [systemMessage, ...(state.messages ?? [])];\n      }\n    ).withConfig({ runName: PROMPT_RUNNABLE_NAME });\n  } else if (isBaseMessage(prompt) && prompt._getType() === \"system\") {\n    promptRunnable = RunnableLambda.from(\n      (state: typeof MessagesAnnotation.State) => [prompt, ...state.messages]\n    ).withConfig({ runName: PROMPT_RUNNABLE_NAME });\n  } else if (typeof prompt === \"function\") {\n    promptRunnable = RunnableLambda.from(prompt).withConfig({\n      runName: PROMPT_RUNNABLE_NAME,\n    });\n  } else if (Runnable.isRunnable(prompt)) {\n    promptRunnable = prompt;\n  } else {\n    throw new Error(`Got unexpected type for 'prompt': ${typeof prompt}`);\n  }\n\n  return promptRunnable;\n}\n\nexport async function shouldBindTools(\n  llm: LanguageModelLike,\n  tools: (ClientTool | ServerTool)[]\n): Promise<boolean> {\n  // If model is a RunnableSequence, find a RunnableBinding or BaseChatModel in its steps\n  let model = llm;\n  if (RunnableSequence.isRunnableSequence(model)) {\n    model =\n      model.steps.find(\n        (step) =>\n          RunnableBinding.isRunnableBinding(step) ||\n          isBaseChatModel(step) ||\n          isConfigurableModel(step)\n      ) || model;\n  }\n\n  if (isConfigurableModel(model)) {\n    model = await model._model();\n  }\n\n  // If not a RunnableBinding, we should bind tools\n  if (!RunnableBinding.isRunnableBinding(model)) {\n    return true;\n  }\n\n  let boundTools = (() => {\n    // check if model.kwargs contain the tools key\n    if (\n      model.kwargs != null &&\n      typeof model.kwargs === \"object\" &&\n      \"tools\" in model.kwargs &&\n      Array.isArray(model.kwargs.tools)\n    ) {\n      return (model.kwargs.tools ?? null) as BindToolsInput[] | null;\n    }\n\n    // Some models can bind the tools via `withConfig()` instead of `bind()`\n    if (\n      model.config != null &&\n      typeof model.config === \"object\" &&\n      \"tools\" in model.config &&\n      Array.isArray(model.config.tools)\n    ) {\n      return (model.config.tools ?? null) as BindToolsInput[] | null;\n    }\n\n    return null;\n  })();\n\n  // google-style\n  if (\n    boundTools != null &&\n    boundTools.length === 1 &&\n    \"functionDeclarations\" in boundTools[0]\n  ) {\n    boundTools = boundTools[0].functionDeclarations;\n  }\n\n  // If no tools in kwargs, we should bind tools\n  if (boundTools == null) return true;\n\n  // Check if tools count matches\n  if (tools.length !== boundTools.length) {\n    throw new Error(\n      \"Number of tools in the model.bindTools() and tools passed to createAgent must match\"\n    );\n  }\n\n  const toolNames = new Set<string>(\n    tools.flatMap((tool) => (isClientTool(tool) ? tool.name : []))\n  );\n\n  const boundToolNames = new Set<string>();\n\n  for (const boundTool of boundTools) {\n    let boundToolName: string | undefined;\n\n    // OpenAI-style tool\n    if (\"type\" in boundTool && boundTool.type === \"function\") {\n      boundToolName = boundTool.function.name;\n    }\n    // Anthropic or Google-style tool\n    else if (\"name\" in boundTool) {\n      boundToolName = boundTool.name;\n    }\n    // Bedrock-style tool\n    else if (\"toolSpec\" in boundTool && \"name\" in boundTool.toolSpec) {\n      boundToolName = boundTool.toolSpec.name;\n    }\n    // unknown tool type so we'll ignore it\n    else {\n      continue;\n    }\n\n    if (boundToolName) {\n      boundToolNames.add(boundToolName);\n    }\n  }\n\n  const missingTools = [...toolNames].filter((x) => !boundToolNames.has(x));\n  if (missingTools.length > 0) {\n    throw new Error(\n      `Missing tools '${missingTools}' in the model.bindTools().` +\n        `Tools in the model.bindTools() must match the tools passed to createAgent.`\n    );\n  }\n\n  return false;\n}\n\nconst _simpleBindTools = (\n  llm: LanguageModelLike,\n  toolClasses: (ClientTool | ServerTool)[],\n  options: Partial<BaseChatModelCallOptions> = {}\n) => {\n  if (_isChatModelWithBindTools(llm)) {\n    return llm.bindTools(toolClasses, options);\n  }\n\n  if (\n    RunnableBinding.isRunnableBinding(llm) &&\n    _isChatModelWithBindTools(llm.bound)\n  ) {\n    const newBound = llm.bound.bindTools(toolClasses, options);\n\n    if (RunnableBinding.isRunnableBinding(newBound)) {\n      return new RunnableBinding({\n        bound: newBound.bound,\n        config: { ...llm.config, ...newBound.config },\n        kwargs: { ...llm.kwargs, ...newBound.kwargs },\n        configFactories: newBound.configFactories ?? llm.configFactories,\n      });\n    }\n\n    return new RunnableBinding({\n      bound: newBound,\n      config: llm.config,\n      kwargs: llm.kwargs,\n      configFactories: llm.configFactories,\n    });\n  }\n\n  return null;\n};\n\nexport async function bindTools(\n  llm: LanguageModelLike,\n  toolClasses: (ClientTool | ServerTool)[],\n  options: Partial<BaseChatModelCallOptions> = {}\n): Promise<\n  | RunnableSequence<any, any>\n  | RunnableBinding<any, any, RunnableConfig<Record<string, any>>>\n  | Runnable<BaseLanguageModelInput, AIMessageChunk, BaseChatModelCallOptions>\n> {\n  const model = _simpleBindTools(llm, toolClasses, options);\n  if (model) return model;\n\n  if (isConfigurableModel(llm)) {\n    const model = _simpleBindTools(await llm._model(), toolClasses, options);\n    if (model) return model;\n  }\n\n  if (RunnableSequence.isRunnableSequence(llm)) {\n    const modelStep = llm.steps.findIndex(\n      (step) =>\n        RunnableBinding.isRunnableBinding(step) ||\n        isBaseChatModel(step) ||\n        isConfigurableModel(step)\n    );\n\n    if (modelStep >= 0) {\n      const model = _simpleBindTools(\n        llm.steps[modelStep],\n        toolClasses,\n        options\n      );\n      if (model) {\n        const nextSteps: unknown[] = llm.steps.slice();\n        nextSteps.splice(modelStep, 1, model);\n\n        return RunnableSequence.from(\n          nextSteps as [RunnableLike, ...RunnableLike[], RunnableLike]\n        );\n      }\n    }\n  }\n\n  throw new Error(`llm ${llm} must define bindTools method.`);\n}\n\n/**\n * Check if the LLM already has bound tools and throw if it does.\n *\n * @param llm - The LLM to check.\n * @returns void\n */\nexport function validateLLMHasNoBoundTools(llm: LanguageModelLike): void {\n  /**\n   * If llm is a function, we can't validate until runtime, so skip\n   */\n  if (typeof llm === \"function\") {\n    return;\n  }\n\n  let model = llm;\n\n  /**\n   * If model is a RunnableSequence, find a RunnableBinding in its steps\n   */\n  if (RunnableSequence.isRunnableSequence(model)) {\n    model =\n      model.steps.find((step: RunnableLike) =>\n        RunnableBinding.isRunnableBinding(step)\n      ) || model;\n  }\n\n  /**\n   * If model is configurable, get the underlying model\n   */\n  if (isConfigurableModel(model)) {\n    /**\n     * Can't validate async model retrieval in constructor\n     */\n    return;\n  }\n\n  /**\n   * Check if model is a RunnableBinding with bound tools\n   */\n  if (RunnableBinding.isRunnableBinding(model)) {\n    const hasToolsInKwargs =\n      model.kwargs != null &&\n      typeof model.kwargs === \"object\" &&\n      \"tools\" in model.kwargs &&\n      Array.isArray(model.kwargs.tools) &&\n      model.kwargs.tools.length > 0;\n\n    const hasToolsInConfig =\n      model.config != null &&\n      typeof model.config === \"object\" &&\n      \"tools\" in model.config &&\n      Array.isArray(model.config.tools) &&\n      model.config.tools.length > 0;\n\n    if (hasToolsInKwargs || hasToolsInConfig) {\n      throw new MultipleToolsBoundError();\n    }\n  }\n\n  /**\n   * Also check if model has tools property directly (e.g., FakeToolCallingModel)\n   */\n  if (\n    \"tools\" in model &&\n    model.tools !== undefined &&\n    Array.isArray(model.tools) &&\n    model.tools.length > 0\n  ) {\n    throw new MultipleToolsBoundError();\n  }\n}\n\n/**\n * Check if the last message in the messages array has tool calls.\n *\n * @param messages - The messages to check.\n * @returns True if the last message has tool calls, false otherwise.\n */\nexport function hasToolCalls(messages: BaseMessage[]): boolean {\n  const lastMessage = messages.at(-1);\n  return Boolean(\n    lastMessage instanceof AIMessage &&\n      lastMessage.tool_calls &&\n      lastMessage.tool_calls.length > 0\n  );\n}\n\n/**\n * Check if the model name supports structured output\n * @param modelName - The name of the model\n * @returns True if the model supports structured output, false otherwise\n */\nexport function hasSupportForStructuredOutput(modelName?: string): boolean {\n  return (\n    modelName?.startsWith(\"gpt-4\") || modelName?.startsWith(\"gpt-5\") || false\n  );\n}\n\nconst CHAT_MODELS_THAT_SUPPORT_JSON_SCHEMA_OUTPUT = [\n  \"ChatOpenAI\",\n  \"FakeToolCallingModel\",\n];\n\n/**\n * Identifies the models that support JSON schema output\n * @param model - The model to check\n * @returns True if the model supports JSON schema output, false otherwise\n */\nexport function hasSupportForJsonSchemaOutput(\n  model: LanguageModelLike\n): boolean {\n  if (!isBaseChatModel(model)) {\n    return false;\n  }\n\n  const chatModelClass = model.getName();\n  if (\n    CHAT_MODELS_THAT_SUPPORT_JSON_SCHEMA_OUTPUT.includes(chatModelClass) &&\n    ((chatModelClass === \"ChatOpenAI\" &&\n      /**\n       * OpenAI models\n       */\n      \"model\" in model &&\n      typeof model.model === \"string\" &&\n      model.model.startsWith(\"gpt-4\")) ||\n      /**\n       * for testing purposes only\n       */\n      (chatModelClass === \"FakeToolCallingModel\" &&\n        \"structuredResponse\" in model))\n  ) {\n    return true;\n  }\n\n  return false;\n}\n"],"mappings":";;;;;;AAyCA,MAAM,eAAe;AACrB,MAAM,kBAAkB;;;;;;;;;;;;;;;AAkBxB,SAAgB,oBACdA,SACe;CACf,MAAM,OACJ,cAAc,QAAQ,KACrB,YAAY,QAAQ,IAClB,mBAAmB,QAAQ,IAAI,iBAAiB,QAAQ;AAE7D,KAAI,CAAC,QAAQ,CAAC,QAAQ,KACpB,QAAO;CAGT,MAAM,EAAE,MAAM,GAAG;AAEjB,KAAI,OAAO,QAAQ,YAAY,SAC7B,QAAO,IAAI,UAAU;EACnB,GAAI,OAAO,KAAK,QAAQ,aAAa,CAAE,EAAC,CAAC,SAAS,IAC9C,QAAQ,YACR;EACJ,SAAS,CAAC,MAAM,EAAE,KAAK,gBAAgB,EAAE,QAAQ,QAAQ,UAAU,CAAC;EACpE,MAAM;CACP;CAGH,MAAM,iBAAiB,CAAE;CACzB,IAAI,iBAAiB;AAErB,MAAK,MAAM,gBAAgB,QAAQ,QACjC,KAAI,OAAO,iBAAiB,UAAU;EACpC,kBAAkB;EAClB,eAAe,KACb,CAAC,MAAM,EAAE,KAAK,gBAAgB,EAAE,aAAa,UAAU,CAAC,CACzD;CACF,WACC,OAAO,iBAAiB,YACxB,UAAU,gBACV,aAAa,SAAS,QACtB;EACA,kBAAkB;EAClB,eAAe,KAAK;GAClB,GAAG;GACH,MAAM,CAAC,MAAM,EAAE,KAAK,gBAAgB,EAAE,aAAa,KAAK,UAAU,CAAC;EACpE,EAAC;CACH,OACC,eAAe,KAAK,aAAa;AAIrC,KAAI,CAAC,gBACH,eAAe,QAAQ;EACrB,MAAM;EACN,MAAM,CAAC,MAAM,EAAE,KAAK,0BAA0B,CAAC;CAChD,EAAC;AAEJ,QAAO,IAAI,UAAU;EACnB,GAAG,QAAQ;EACX,SAAS;EACT,MAAM;CACP;AACF;;;;;;;;;;;;;;;;;AAkBD,SAAgB,uBAA8CA,SAAe;AAC3E,KAAI,CAAC,YAAY,QAAQ,IAAI,CAAC,QAAQ,QACpC,QAAO;CAGT,IAAIC,iBAAiC,CAAE;CACvC,IAAIC;AAEJ,KAAI,MAAM,QAAQ,QAAQ,QAAQ,EAChC,iBAAiB,QAAQ,QACtB,OAAO,CAAC,UAAU;AACjB,MAAI,MAAM,SAAS,UAAU,OAAO,MAAM,SAAS,UAAU;GAC3D,MAAM,YAAY,MAAM,KAAK,MAAM,aAAa;GAChD,MAAM,eAAe,MAAM,KAAK,MAAM,gBAAgB;AAEtD,OAAI,cAAc,CAAC,gBAAgB,aAAa,OAAO,KAAK;IAG1D,cAAc,UAAU;AACxB,WAAO;GACR;AACD,UAAO;EACR;AACD,SAAO;CACR,EAAC,CACD,IAAI,CAAC,UAAU;AACd,MAAI,MAAM,SAAS,UAAU,OAAO,MAAM,SAAS,UAAU;GAC3D,MAAM,YAAY,MAAM,KAAK,MAAM,aAAa;GAChD,MAAM,eAAe,MAAM,KAAK,MAAM,gBAAgB;AAEtD,OAAI,CAAC,aAAa,CAAC,aACjB,QAAO;GAKT,cAAc,UAAU;AAExB,UAAO;IACL,GAAG;IACH,MAAM,aAAa;GACpB;EACF;AACD,SAAO;CACR,EAAC;MACC;EACL,MAAM,UAAU,QAAQ;EACxB,MAAM,YAAY,QAAQ,MAAM,aAAa;EAC7C,MAAM,eAAe,QAAQ,MAAM,gBAAgB;AAEnD,MAAI,CAAC,aAAa,CAAC,aACjB,QAAO;EAIT,cAAc,UAAU;EAExB,iBAAiB,aAAa;CAC/B;AAED,QAAO,IAAI,UAAU;EACnB,GAAI,OAAO,KAAK,QAAQ,aAAa,CAAE,EAAC,CAAC,SAAS,IAC9C,QAAQ,YACR;EACJ,SAAS;EACT,MAAM;CACP;AACF;AAED,SAAgB,aACdC,MACoB;AACpB,QAAO,SAAS,WAAW,KAAK;AACjC;AAED,SAAgB,gBACdC,OACwB;AACxB,QACE,YAAY,SACZ,OAAO,MAAM,WAAW,cACxB,gBAAgB;AAEnB;AAGD,SAAgB,oBACdC,OACqC;AACrC,QACE,OAAO,UAAU,YACjB,SAAS,QACT,6BAA6B,SAC7B,YAAY,SACZ,OAAO,MAAM,WAAW;AAE3B;AAED,SAAS,0BACPC,KACmE;AACnE,KAAI,CAAC,gBAAgB,IAAI,CAAE,QAAO;AAClC,QAAO,eAAe,OAAO,OAAO,IAAI,cAAc;AACvD;AAED,SAAgB,kBAAkBC,QAA2B;CAC3D,IAAIC;AAEJ,KAAI,UAAU,MACZ,iBAAiB,eAAe,KAC9B,CAACC,UAA2C,MAAM,SACnD,CAAC,WAAW,EAAE,SAAS,qBAAsB,EAAC;UACtC,OAAO,WAAW,UAAU;EACrC,MAAM,gBAAgB,IAAI,cAAc;EACxC,iBAAiB,eAAe,KAC9B,CAACA,UAA2C;AAC1C,UAAO,CAAC,eAAe,GAAI,MAAM,YAAY,CAAE,CAAE;EAClD,EACF,CAAC,WAAW,EAAE,SAAS,qBAAsB,EAAC;CAChD,WAAU,cAAc,OAAO,IAAI,OAAO,UAAU,KAAK,UACxD,iBAAiB,eAAe,KAC9B,CAACA,UAA2C,CAAC,QAAQ,GAAG,MAAM,QAAS,EACxE,CAAC,WAAW,EAAE,SAAS,qBAAsB,EAAC;UACtC,OAAO,WAAW,YAC3B,iBAAiB,eAAe,KAAK,OAAO,CAAC,WAAW,EACtD,SAAS,qBACV,EAAC;UACO,SAAS,WAAW,OAAO,EACpC,iBAAiB;KAEjB,OAAM,IAAI,MAAM,CAAC,kCAAkC,EAAE,OAAO,QAAQ;AAGtE,QAAO;AACR;AAgHD,MAAM,mBAAmB,CACvBH,KACAI,aACAC,UAA6C,CAAE,MAC5C;AACH,KAAI,0BAA0B,IAAI,CAChC,QAAO,IAAI,UAAU,aAAa,QAAQ;AAG5C,KACE,gBAAgB,kBAAkB,IAAI,IACtC,0BAA0B,IAAI,MAAM,EACpC;EACA,MAAM,WAAW,IAAI,MAAM,UAAU,aAAa,QAAQ;AAE1D,MAAI,gBAAgB,kBAAkB,SAAS,CAC7C,QAAO,IAAI,gBAAgB;GACzB,OAAO,SAAS;GAChB,QAAQ;IAAE,GAAG,IAAI;IAAQ,GAAG,SAAS;GAAQ;GAC7C,QAAQ;IAAE,GAAG,IAAI;IAAQ,GAAG,SAAS;GAAQ;GAC7C,iBAAiB,SAAS,mBAAmB,IAAI;EAClD;AAGH,SAAO,IAAI,gBAAgB;GACzB,OAAO;GACP,QAAQ,IAAI;GACZ,QAAQ,IAAI;GACZ,iBAAiB,IAAI;EACtB;CACF;AAED,QAAO;AACR;AAED,eAAsB,UACpBL,KACAI,aACAC,UAA6C,CAAE,GAK/C;CACA,MAAM,QAAQ,iBAAiB,KAAK,aAAa,QAAQ;AACzD,KAAI,MAAO,QAAO;AAElB,KAAI,oBAAoB,IAAI,EAAE;EAC5B,MAAMC,UAAQ,iBAAiB,MAAM,IAAI,QAAQ,EAAE,aAAa,QAAQ;AACxE,MAAIA,QAAO,QAAOA;CACnB;AAED,KAAI,iBAAiB,mBAAmB,IAAI,EAAE;EAC5C,MAAM,YAAY,IAAI,MAAM,UAC1B,CAAC,SACC,gBAAgB,kBAAkB,KAAK,IACvC,gBAAgB,KAAK,IACrB,oBAAoB,KAAK,CAC5B;AAED,MAAI,aAAa,GAAG;GAClB,MAAMA,UAAQ,iBACZ,IAAI,MAAM,YACV,aACA,QACD;AACD,OAAIA,SAAO;IACT,MAAMC,YAAuB,IAAI,MAAM,OAAO;IAC9C,UAAU,OAAO,WAAW,GAAGD,QAAM;AAErC,WAAO,iBAAiB,KACtB,UACD;GACF;EACF;CACF;AAED,OAAM,IAAI,MAAM,CAAC,IAAI,EAAE,IAAI,8BAA8B,CAAC;AAC3D;;;;;;;AAQD,SAAgB,2BAA2BN,KAA8B;;;;AAIvE,KAAI,OAAO,QAAQ,WACjB;CAGF,IAAI,QAAQ;;;;AAKZ,KAAI,iBAAiB,mBAAmB,MAAM,EAC5C,QACE,MAAM,MAAM,KAAK,CAACQ,SAChB,gBAAgB,kBAAkB,KAAK,CACxC,IAAI;;;;AAMT,KAAI,oBAAoB,MAAM;;;;AAI5B;;;;AAMF,KAAI,gBAAgB,kBAAkB,MAAM,EAAE;EAC5C,MAAM,mBACJ,MAAM,UAAU,QAChB,OAAO,MAAM,WAAW,YACxB,WAAW,MAAM,UACjB,MAAM,QAAQ,MAAM,OAAO,MAAM,IACjC,MAAM,OAAO,MAAM,SAAS;EAE9B,MAAM,mBACJ,MAAM,UAAU,QAChB,OAAO,MAAM,WAAW,YACxB,WAAW,MAAM,UACjB,MAAM,QAAQ,MAAM,OAAO,MAAM,IACjC,MAAM,OAAO,MAAM,SAAS;AAE9B,MAAI,oBAAoB,iBACtB,OAAM,IAAI;CAEb;;;;AAKD,KACE,WAAW,SACX,MAAM,UAAU,UAChB,MAAM,QAAQ,MAAM,MAAM,IAC1B,MAAM,MAAM,SAAS,EAErB,OAAM,IAAI;AAEb;;;;;;;AAQD,SAAgB,aAAaC,UAAkC;CAC7D,MAAM,cAAc,SAAS,GAAG,GAAG;AACnC,QAAO,QACL,uBAAuB,aACrB,YAAY,cACZ,YAAY,WAAW,SAAS,EACnC;AACF;AAaD,MAAM,8CAA8C,CAClD,cACA,sBACD;;;;;;AAOD,SAAgB,8BACdX,OACS;AACT,KAAI,CAAC,gBAAgB,MAAM,CACzB,QAAO;CAGT,MAAM,iBAAiB,MAAM,SAAS;AACtC,KACE,4CAA4C,SAAS,eAAe,KAClE,mBAAmB,gBAInB,WAAW,SACX,OAAO,MAAM,UAAU,YACvB,MAAM,MAAM,WAAW,QAAQ,IAI9B,mBAAmB,0BAClB,wBAAwB,OAE5B,QAAO;AAGT,QAAO;AACR"}
+{"version":3,"file":"utils.js","names":["message: T","updatedContent: MessageContent","updatedName: string | undefined","tool: ClientTool | ServerTool","model: LanguageModelLike","model: unknown","llm: LanguageModelLike","prompt?: Prompt","promptRunnable: Runnable","state: typeof MessagesAnnotation.State","toolClasses: (ClientTool | ServerTool)[]","options: Partial<BaseChatModelCallOptions>","model","nextSteps: unknown[]","step: RunnableLike","messages: BaseMessage[]"],"sources":["../../src/agents/utils.ts"],"sourcesContent":["import {\n  AIMessage,\n  AIMessageChunk,\n  BaseMessage,\n  BaseMessageLike,\n  SystemMessage,\n  MessageContent,\n} from \"@langchain/core/messages\";\nimport { MessagesAnnotation } from \"@langchain/langgraph\";\nimport {\n  BaseChatModel,\n  type BindToolsInput,\n  type BaseChatModelCallOptions,\n} from \"@langchain/core/language_models/chat_models\";\nimport {\n  LanguageModelLike,\n  BaseLanguageModelInput,\n} from \"@langchain/core/language_models/base\";\nimport {\n  Runnable,\n  RunnableLike,\n  RunnableConfig,\n  RunnableLambda,\n  RunnableSequence,\n  RunnableBinding,\n} from \"@langchain/core/runnables\";\n\nimport { MultipleToolsBoundError } from \"./errors.js\";\nimport { PROMPT_RUNNABLE_NAME } from \"./constants.js\";\nimport {\n  ServerTool,\n  ClientTool,\n  ConfigurableModelInterface,\n  Prompt,\n} from \"./types.js\";\n\nconst NAME_PATTERN = /<name>(.*?)<\\/name>/s;\nconst CONTENT_PATTERN = /<content>(.*?)<\\/content>/s;\n\nexport type AgentNameMode = \"inline\";\n\n/**\n * Attach formatted agent names to the messages passed to and from a language model.\n *\n * This is useful for making a message history with multiple agents more coherent.\n *\n * NOTE: agent name is consumed from the message.name field.\n * If you're using an agent built with createAgent, name is automatically set.\n * If you're building a custom agent, make sure to set the name on the AI message returned by the LLM.\n *\n * @param message - Message to add agent name formatting to\n * @returns Message with agent name formatting\n *\n * @internal\n */\nexport function _addInlineAgentName<T extends BaseMessageLike>(\n  message: T\n): T | AIMessage {\n  if (!AIMessage.isInstance(message) || AIMessageChunk.isInstance(message)) {\n    return message;\n  }\n\n  if (!message.name) {\n    return message;\n  }\n\n  const { name } = message;\n\n  if (typeof message.content === \"string\") {\n    return new AIMessage({\n      ...message.lc_kwargs,\n      content: `<name>${name}</name><content>${message.content}</content>`,\n      name: undefined,\n    });\n  }\n\n  const updatedContent = [];\n  let textBlockCount = 0;\n\n  for (const contentBlock of message.content) {\n    if (typeof contentBlock === \"string\") {\n      textBlockCount += 1;\n      updatedContent.push(\n        `<name>${name}</name><content>${contentBlock}</content>`\n      );\n    } else if (\n      typeof contentBlock === \"object\" &&\n      \"type\" in contentBlock &&\n      contentBlock.type === \"text\"\n    ) {\n      textBlockCount += 1;\n      updatedContent.push({\n        ...contentBlock,\n        text: `<name>${name}</name><content>${contentBlock.text}</content>`,\n      });\n    } else {\n      updatedContent.push(contentBlock);\n    }\n  }\n\n  if (!textBlockCount) {\n    updatedContent.unshift({\n      type: \"text\",\n      text: `<name>${name}</name><content></content>`,\n    });\n  }\n  return new AIMessage({\n    ...message.lc_kwargs,\n    content: updatedContent as MessageContent,\n    name: undefined,\n  });\n}\n\n/**\n * Remove explicit name and content XML tags from the AI message content.\n *\n * Examples:\n *\n * @example\n * ```typescript\n * removeInlineAgentName(new AIMessage({ content: \"<name>assistant</name><content>Hello</content>\", name: \"assistant\" }))\n * // AIMessage with content: \"Hello\"\n *\n * removeInlineAgentName(new AIMessage({ content: [{type: \"text\", text: \"<name>assistant</name><content>Hello</content>\"}], name: \"assistant\" }))\n * // AIMessage with content: [{type: \"text\", text: \"Hello\"}]\n * ```\n *\n * @internal\n */\nexport function _removeInlineAgentName<T extends BaseMessage>(message: T): T {\n  if (!AIMessage.isInstance(message) || !message.content) {\n    return message;\n  }\n\n  let updatedContent: MessageContent = [];\n  let updatedName: string | undefined;\n\n  if (Array.isArray(message.content)) {\n    updatedContent = message.content\n      .filter((block) => {\n        if (block.type === \"text\" && typeof block.text === \"string\") {\n          const nameMatch = block.text.match(NAME_PATTERN);\n          const contentMatch = block.text.match(CONTENT_PATTERN);\n          // don't include empty content blocks that were added because there was no text block to modify\n          if (nameMatch && (!contentMatch || contentMatch[1] === \"\")) {\n            // capture name from text block\n            updatedName = nameMatch[1];\n            return false;\n          }\n          return true;\n        }\n        return true;\n      })\n      .map((block) => {\n        if (block.type === \"text\" && typeof block.text === \"string\") {\n          const nameMatch = block.text.match(NAME_PATTERN);\n          const contentMatch = block.text.match(CONTENT_PATTERN);\n\n          if (!nameMatch || !contentMatch) {\n            return block;\n          }\n\n          // capture name from text block\n          updatedName = nameMatch[1];\n\n          return {\n            ...block,\n            text: contentMatch[1],\n          };\n        }\n        return block;\n      });\n  } else {\n    const content = message.content as string;\n    const nameMatch = content.match(NAME_PATTERN);\n    const contentMatch = content.match(CONTENT_PATTERN);\n\n    if (!nameMatch || !contentMatch) {\n      return message;\n    }\n\n    updatedName = nameMatch[1];\n    updatedContent = contentMatch[1];\n  }\n\n  return new AIMessage({\n    ...(Object.keys(message.lc_kwargs ?? {}).length > 0\n      ? message.lc_kwargs\n      : message),\n    content: updatedContent,\n    name: updatedName,\n  }) as T;\n}\n\nexport function isClientTool(\n  tool: ClientTool | ServerTool\n): tool is ClientTool {\n  return Runnable.isRunnable(tool);\n}\n\nexport function isBaseChatModel(\n  model: LanguageModelLike\n): model is BaseChatModel {\n  return (\n    \"invoke\" in model &&\n    typeof model.invoke === \"function\" &&\n    \"_modelType\" in model\n  );\n}\n\nexport function isConfigurableModel(\n  model: unknown\n): model is ConfigurableModelInterface {\n  return (\n    typeof model === \"object\" &&\n    model != null &&\n    \"_queuedMethodOperations\" in model &&\n    \"_model\" in model &&\n    typeof model._model === \"function\"\n  );\n}\n\nfunction _isChatModelWithBindTools(\n  llm: LanguageModelLike\n): llm is BaseChatModel & Required<Pick<BaseChatModel, \"bindTools\">> {\n  if (!isBaseChatModel(llm)) return false;\n  return \"bindTools\" in llm && typeof llm.bindTools === \"function\";\n}\n\nexport function getPromptRunnable(prompt?: Prompt): Runnable {\n  let promptRunnable: Runnable;\n\n  if (prompt == null) {\n    promptRunnable = RunnableLambda.from(\n      (state: typeof MessagesAnnotation.State) => state.messages\n    ).withConfig({ runName: PROMPT_RUNNABLE_NAME });\n  } else if (typeof prompt === \"string\") {\n    const systemMessage = new SystemMessage(prompt);\n    promptRunnable = RunnableLambda.from(\n      (state: typeof MessagesAnnotation.State) => {\n        return [systemMessage, ...(state.messages ?? [])];\n      }\n    ).withConfig({ runName: PROMPT_RUNNABLE_NAME });\n  } else if (SystemMessage.isInstance(prompt)) {\n    promptRunnable = RunnableLambda.from(\n      (state: typeof MessagesAnnotation.State) => [prompt, ...state.messages]\n    ).withConfig({ runName: PROMPT_RUNNABLE_NAME });\n  } else if (typeof prompt === \"function\") {\n    promptRunnable = RunnableLambda.from(prompt).withConfig({\n      runName: PROMPT_RUNNABLE_NAME,\n    });\n  } else if (Runnable.isRunnable(prompt)) {\n    promptRunnable = prompt;\n  } else {\n    throw new Error(`Got unexpected type for 'prompt': ${typeof prompt}`);\n  }\n\n  return promptRunnable;\n}\n\nexport async function shouldBindTools(\n  llm: LanguageModelLike,\n  tools: (ClientTool | ServerTool)[]\n): Promise<boolean> {\n  // If model is a RunnableSequence, find a RunnableBinding or BaseChatModel in its steps\n  let model = llm;\n  if (RunnableSequence.isRunnableSequence(model)) {\n    model =\n      model.steps.find(\n        (step) =>\n          RunnableBinding.isRunnableBinding(step) ||\n          isBaseChatModel(step) ||\n          isConfigurableModel(step)\n      ) || model;\n  }\n\n  if (isConfigurableModel(model)) {\n    model = await model._model();\n  }\n\n  // If not a RunnableBinding, we should bind tools\n  if (!RunnableBinding.isRunnableBinding(model)) {\n    return true;\n  }\n\n  let boundTools = (() => {\n    // check if model.kwargs contain the tools key\n    if (\n      model.kwargs != null &&\n      typeof model.kwargs === \"object\" &&\n      \"tools\" in model.kwargs &&\n      Array.isArray(model.kwargs.tools)\n    ) {\n      return (model.kwargs.tools ?? null) as BindToolsInput[] | null;\n    }\n\n    // Some models can bind the tools via `withConfig()` instead of `bind()`\n    if (\n      model.config != null &&\n      typeof model.config === \"object\" &&\n      \"tools\" in model.config &&\n      Array.isArray(model.config.tools)\n    ) {\n      return (model.config.tools ?? null) as BindToolsInput[] | null;\n    }\n\n    return null;\n  })();\n\n  // google-style\n  if (\n    boundTools != null &&\n    boundTools.length === 1 &&\n    \"functionDeclarations\" in boundTools[0]\n  ) {\n    boundTools = boundTools[0].functionDeclarations;\n  }\n\n  // If no tools in kwargs, we should bind tools\n  if (boundTools == null) return true;\n\n  // Check if tools count matches\n  if (tools.length !== boundTools.length) {\n    throw new Error(\n      \"Number of tools in the model.bindTools() and tools passed to createAgent must match\"\n    );\n  }\n\n  const toolNames = new Set<string>(\n    tools.flatMap((tool) => (isClientTool(tool) ? tool.name : []))\n  );\n\n  const boundToolNames = new Set<string>();\n\n  for (const boundTool of boundTools) {\n    let boundToolName: string | undefined;\n\n    // OpenAI-style tool\n    if (\"type\" in boundTool && boundTool.type === \"function\") {\n      boundToolName = boundTool.function.name;\n    }\n    // Anthropic or Google-style tool\n    else if (\"name\" in boundTool) {\n      boundToolName = boundTool.name;\n    }\n    // Bedrock-style tool\n    else if (\"toolSpec\" in boundTool && \"name\" in boundTool.toolSpec) {\n      boundToolName = boundTool.toolSpec.name;\n    }\n    // unknown tool type so we'll ignore it\n    else {\n      continue;\n    }\n\n    if (boundToolName) {\n      boundToolNames.add(boundToolName);\n    }\n  }\n\n  const missingTools = [...toolNames].filter((x) => !boundToolNames.has(x));\n  if (missingTools.length > 0) {\n    throw new Error(\n      `Missing tools '${missingTools}' in the model.bindTools().` +\n        `Tools in the model.bindTools() must match the tools passed to createAgent.`\n    );\n  }\n\n  return false;\n}\n\nconst _simpleBindTools = (\n  llm: LanguageModelLike,\n  toolClasses: (ClientTool | ServerTool)[],\n  options: Partial<BaseChatModelCallOptions> = {}\n) => {\n  if (_isChatModelWithBindTools(llm)) {\n    return llm.bindTools(toolClasses, options);\n  }\n\n  if (\n    RunnableBinding.isRunnableBinding(llm) &&\n    _isChatModelWithBindTools(llm.bound)\n  ) {\n    const newBound = llm.bound.bindTools(toolClasses, options);\n\n    if (RunnableBinding.isRunnableBinding(newBound)) {\n      return new RunnableBinding({\n        bound: newBound.bound,\n        config: { ...llm.config, ...newBound.config },\n        kwargs: { ...llm.kwargs, ...newBound.kwargs },\n        configFactories: newBound.configFactories ?? llm.configFactories,\n      });\n    }\n\n    return new RunnableBinding({\n      bound: newBound,\n      config: llm.config,\n      kwargs: llm.kwargs,\n      configFactories: llm.configFactories,\n    });\n  }\n\n  return null;\n};\n\nexport async function bindTools(\n  llm: LanguageModelLike,\n  toolClasses: (ClientTool | ServerTool)[],\n  options: Partial<BaseChatModelCallOptions> = {}\n): Promise<\n  | RunnableSequence<unknown, unknown>\n  | RunnableBinding<unknown, unknown, RunnableConfig<Record<string, unknown>>>\n  | Runnable<BaseLanguageModelInput, AIMessageChunk, BaseChatModelCallOptions>\n> {\n  const model = _simpleBindTools(llm, toolClasses, options);\n  if (model) return model;\n\n  if (isConfigurableModel(llm)) {\n    const model = _simpleBindTools(await llm._model(), toolClasses, options);\n    if (model) return model;\n  }\n\n  if (RunnableSequence.isRunnableSequence(llm)) {\n    const modelStep = llm.steps.findIndex(\n      (step) =>\n        RunnableBinding.isRunnableBinding(step) ||\n        isBaseChatModel(step) ||\n        isConfigurableModel(step)\n    );\n\n    if (modelStep >= 0) {\n      const model = _simpleBindTools(\n        llm.steps[modelStep],\n        toolClasses,\n        options\n      );\n      if (model) {\n        const nextSteps: unknown[] = llm.steps.slice();\n        nextSteps.splice(modelStep, 1, model);\n\n        return RunnableSequence.from(\n          nextSteps as [RunnableLike, ...RunnableLike[], RunnableLike]\n        );\n      }\n    }\n  }\n\n  throw new Error(`llm ${llm} must define bindTools method.`);\n}\n\n/**\n * Check if the LLM already has bound tools and throw if it does.\n *\n * @param llm - The LLM to check.\n * @returns void\n */\nexport function validateLLMHasNoBoundTools(llm: LanguageModelLike): void {\n  /**\n   * If llm is a function, we can't validate until runtime, so skip\n   */\n  if (typeof llm === \"function\") {\n    return;\n  }\n\n  let model = llm;\n\n  /**\n   * If model is a RunnableSequence, find a RunnableBinding in its steps\n   */\n  if (RunnableSequence.isRunnableSequence(model)) {\n    model =\n      model.steps.find((step: RunnableLike) =>\n        RunnableBinding.isRunnableBinding(step)\n      ) || model;\n  }\n\n  /**\n   * If model is configurable, get the underlying model\n   */\n  if (isConfigurableModel(model)) {\n    /**\n     * Can't validate async model retrieval in constructor\n     */\n    return;\n  }\n\n  /**\n   * Check if model is a RunnableBinding with bound tools\n   */\n  if (RunnableBinding.isRunnableBinding(model)) {\n    const hasToolsInKwargs =\n      model.kwargs != null &&\n      typeof model.kwargs === \"object\" &&\n      \"tools\" in model.kwargs &&\n      Array.isArray(model.kwargs.tools) &&\n      model.kwargs.tools.length > 0;\n\n    const hasToolsInConfig =\n      model.config != null &&\n      typeof model.config === \"object\" &&\n      \"tools\" in model.config &&\n      Array.isArray(model.config.tools) &&\n      model.config.tools.length > 0;\n\n    if (hasToolsInKwargs || hasToolsInConfig) {\n      throw new MultipleToolsBoundError();\n    }\n  }\n\n  /**\n   * Also check if model has tools property directly (e.g., FakeToolCallingModel)\n   */\n  if (\n    \"tools\" in model &&\n    model.tools !== undefined &&\n    Array.isArray(model.tools) &&\n    model.tools.length > 0\n  ) {\n    throw new MultipleToolsBoundError();\n  }\n}\n\n/**\n * Check if the last message in the messages array has tool calls.\n *\n * @param messages - The messages to check.\n * @returns True if the last message has tool calls, false otherwise.\n */\nexport function hasToolCalls(messages: BaseMessage[]): boolean {\n  const lastMessage = messages.at(-1);\n  return Boolean(\n    AIMessage.isInstance(lastMessage) &&\n      lastMessage.tool_calls &&\n      lastMessage.tool_calls.length > 0\n  );\n}\n\n/**\n * Check if the model name supports structured output\n * @param modelName - The name of the model\n * @returns True if the model supports structured output, false otherwise\n */\nexport function hasSupportForStructuredOutput(modelName?: string): boolean {\n  return (\n    modelName?.startsWith(\"gpt-4\") || modelName?.startsWith(\"gpt-5\") || false\n  );\n}\n\nconst CHAT_MODELS_THAT_SUPPORT_JSON_SCHEMA_OUTPUT = [\n  \"ChatOpenAI\",\n  \"FakeToolCallingModel\",\n];\n\n/**\n * Identifies the models that support JSON schema output\n * @param model - The model to check\n * @returns True if the model supports JSON schema output, false otherwise\n */\nexport function hasSupportForJsonSchemaOutput(\n  model: LanguageModelLike\n): boolean {\n  if (!isBaseChatModel(model)) {\n    return false;\n  }\n\n  const chatModelClass = model.getName();\n  if (\n    CHAT_MODELS_THAT_SUPPORT_JSON_SCHEMA_OUTPUT.includes(chatModelClass) &&\n    ((chatModelClass === \"ChatOpenAI\" &&\n      /**\n       * OpenAI models\n       */\n      \"model\" in model &&\n      typeof model.model === \"string\" &&\n      model.model.startsWith(\"gpt-4\")) ||\n      /**\n       * for testing purposes only\n       */\n      (chatModelClass === \"FakeToolCallingModel\" &&\n        \"structuredResponse\" in model))\n  ) {\n    return true;\n  }\n\n  return false;\n}\n"],"mappings":";;;;;;AAoCA,MAAM,eAAe;AACrB,MAAM,kBAAkB;;;;;;;;;;;;;;;AAkBxB,SAAgB,oBACdA,SACe;AACf,KAAI,CAAC,UAAU,WAAW,QAAQ,IAAI,eAAe,WAAW,QAAQ,CACtE,QAAO;AAGT,KAAI,CAAC,QAAQ,KACX,QAAO;CAGT,MAAM,EAAE,MAAM,GAAG;AAEjB,KAAI,OAAO,QAAQ,YAAY,SAC7B,QAAO,IAAI,UAAU;EACnB,GAAG,QAAQ;EACX,SAAS,CAAC,MAAM,EAAE,KAAK,gBAAgB,EAAE,QAAQ,QAAQ,UAAU,CAAC;EACpE,MAAM;CACP;CAGH,MAAM,iBAAiB,CAAE;CACzB,IAAI,iBAAiB;AAErB,MAAK,MAAM,gBAAgB,QAAQ,QACjC,KAAI,OAAO,iBAAiB,UAAU;EACpC,kBAAkB;EAClB,eAAe,KACb,CAAC,MAAM,EAAE,KAAK,gBAAgB,EAAE,aAAa,UAAU,CAAC,CACzD;CACF,WACC,OAAO,iBAAiB,YACxB,UAAU,gBACV,aAAa,SAAS,QACtB;EACA,kBAAkB;EAClB,eAAe,KAAK;GAClB,GAAG;GACH,MAAM,CAAC,MAAM,EAAE,KAAK,gBAAgB,EAAE,aAAa,KAAK,UAAU,CAAC;EACpE,EAAC;CACH,OACC,eAAe,KAAK,aAAa;AAIrC,KAAI,CAAC,gBACH,eAAe,QAAQ;EACrB,MAAM;EACN,MAAM,CAAC,MAAM,EAAE,KAAK,0BAA0B,CAAC;CAChD,EAAC;AAEJ,QAAO,IAAI,UAAU;EACnB,GAAG,QAAQ;EACX,SAAS;EACT,MAAM;CACP;AACF;;;;;;;;;;;;;;;;;AAkBD,SAAgB,uBAA8CA,SAAe;AAC3E,KAAI,CAAC,UAAU,WAAW,QAAQ,IAAI,CAAC,QAAQ,QAC7C,QAAO;CAGT,IAAIC,iBAAiC,CAAE;CACvC,IAAIC;AAEJ,KAAI,MAAM,QAAQ,QAAQ,QAAQ,EAChC,iBAAiB,QAAQ,QACtB,OAAO,CAAC,UAAU;AACjB,MAAI,MAAM,SAAS,UAAU,OAAO,MAAM,SAAS,UAAU;GAC3D,MAAM,YAAY,MAAM,KAAK,MAAM,aAAa;GAChD,MAAM,eAAe,MAAM,KAAK,MAAM,gBAAgB;AAEtD,OAAI,cAAc,CAAC,gBAAgB,aAAa,OAAO,KAAK;IAE1D,cAAc,UAAU;AACxB,WAAO;GACR;AACD,UAAO;EACR;AACD,SAAO;CACR,EAAC,CACD,IAAI,CAAC,UAAU;AACd,MAAI,MAAM,SAAS,UAAU,OAAO,MAAM,SAAS,UAAU;GAC3D,MAAM,YAAY,MAAM,KAAK,MAAM,aAAa;GAChD,MAAM,eAAe,MAAM,KAAK,MAAM,gBAAgB;AAEtD,OAAI,CAAC,aAAa,CAAC,aACjB,QAAO;GAIT,cAAc,UAAU;AAExB,UAAO;IACL,GAAG;IACH,MAAM,aAAa;GACpB;EACF;AACD,SAAO;CACR,EAAC;MACC;EACL,MAAM,UAAU,QAAQ;EACxB,MAAM,YAAY,QAAQ,MAAM,aAAa;EAC7C,MAAM,eAAe,QAAQ,MAAM,gBAAgB;AAEnD,MAAI,CAAC,aAAa,CAAC,aACjB,QAAO;EAGT,cAAc,UAAU;EACxB,iBAAiB,aAAa;CAC/B;AAED,QAAO,IAAI,UAAU;EACnB,GAAI,OAAO,KAAK,QAAQ,aAAa,CAAE,EAAC,CAAC,SAAS,IAC9C,QAAQ,YACR;EACJ,SAAS;EACT,MAAM;CACP;AACF;AAED,SAAgB,aACdC,MACoB;AACpB,QAAO,SAAS,WAAW,KAAK;AACjC;AAED,SAAgB,gBACdC,OACwB;AACxB,QACE,YAAY,SACZ,OAAO,MAAM,WAAW,cACxB,gBAAgB;AAEnB;AAED,SAAgB,oBACdC,OACqC;AACrC,QACE,OAAO,UAAU,YACjB,SAAS,QACT,6BAA6B,SAC7B,YAAY,SACZ,OAAO,MAAM,WAAW;AAE3B;AAED,SAAS,0BACPC,KACmE;AACnE,KAAI,CAAC,gBAAgB,IAAI,CAAE,QAAO;AAClC,QAAO,eAAe,OAAO,OAAO,IAAI,cAAc;AACvD;AAED,SAAgB,kBAAkBC,QAA2B;CAC3D,IAAIC;AAEJ,KAAI,UAAU,MACZ,iBAAiB,eAAe,KAC9B,CAACC,UAA2C,MAAM,SACnD,CAAC,WAAW,EAAE,SAAS,qBAAsB,EAAC;UACtC,OAAO,WAAW,UAAU;EACrC,MAAM,gBAAgB,IAAI,cAAc;EACxC,iBAAiB,eAAe,KAC9B,CAACA,UAA2C;AAC1C,UAAO,CAAC,eAAe,GAAI,MAAM,YAAY,CAAE,CAAE;EAClD,EACF,CAAC,WAAW,EAAE,SAAS,qBAAsB,EAAC;CAChD,WAAU,cAAc,WAAW,OAAO,EACzC,iBAAiB,eAAe,KAC9B,CAACA,UAA2C,CAAC,QAAQ,GAAG,MAAM,QAAS,EACxE,CAAC,WAAW,EAAE,SAAS,qBAAsB,EAAC;UACtC,OAAO,WAAW,YAC3B,iBAAiB,eAAe,KAAK,OAAO,CAAC,WAAW,EACtD,SAAS,qBACV,EAAC;UACO,SAAS,WAAW,OAAO,EACpC,iBAAiB;KAEjB,OAAM,IAAI,MAAM,CAAC,kCAAkC,EAAE,OAAO,QAAQ;AAGtE,QAAO;AACR;AAgHD,MAAM,mBAAmB,CACvBH,KACAI,aACAC,UAA6C,CAAE,MAC5C;AACH,KAAI,0BAA0B,IAAI,CAChC,QAAO,IAAI,UAAU,aAAa,QAAQ;AAG5C,KACE,gBAAgB,kBAAkB,IAAI,IACtC,0BAA0B,IAAI,MAAM,EACpC;EACA,MAAM,WAAW,IAAI,MAAM,UAAU,aAAa,QAAQ;AAE1D,MAAI,gBAAgB,kBAAkB,SAAS,CAC7C,QAAO,IAAI,gBAAgB;GACzB,OAAO,SAAS;GAChB,QAAQ;IAAE,GAAG,IAAI;IAAQ,GAAG,SAAS;GAAQ;GAC7C,QAAQ;IAAE,GAAG,IAAI;IAAQ,GAAG,SAAS;GAAQ;GAC7C,iBAAiB,SAAS,mBAAmB,IAAI;EAClD;AAGH,SAAO,IAAI,gBAAgB;GACzB,OAAO;GACP,QAAQ,IAAI;GACZ,QAAQ,IAAI;GACZ,iBAAiB,IAAI;EACtB;CACF;AAED,QAAO;AACR;AAED,eAAsB,UACpBL,KACAI,aACAC,UAA6C,CAAE,GAK/C;CACA,MAAM,QAAQ,iBAAiB,KAAK,aAAa,QAAQ;AACzD,KAAI,MAAO,QAAO;AAElB,KAAI,oBAAoB,IAAI,EAAE;EAC5B,MAAMC,UAAQ,iBAAiB,MAAM,IAAI,QAAQ,EAAE,aAAa,QAAQ;AACxE,MAAIA,QAAO,QAAOA;CACnB;AAED,KAAI,iBAAiB,mBAAmB,IAAI,EAAE;EAC5C,MAAM,YAAY,IAAI,MAAM,UAC1B,CAAC,SACC,gBAAgB,kBAAkB,KAAK,IACvC,gBAAgB,KAAK,IACrB,oBAAoB,KAAK,CAC5B;AAED,MAAI,aAAa,GAAG;GAClB,MAAMA,UAAQ,iBACZ,IAAI,MAAM,YACV,aACA,QACD;AACD,OAAIA,SAAO;IACT,MAAMC,YAAuB,IAAI,MAAM,OAAO;IAC9C,UAAU,OAAO,WAAW,GAAGD,QAAM;AAErC,WAAO,iBAAiB,KACtB,UACD;GACF;EACF;CACF;AAED,OAAM,IAAI,MAAM,CAAC,IAAI,EAAE,IAAI,8BAA8B,CAAC;AAC3D;;;;;;;AAQD,SAAgB,2BAA2BN,KAA8B;;;;AAIvE,KAAI,OAAO,QAAQ,WACjB;CAGF,IAAI,QAAQ;;;;AAKZ,KAAI,iBAAiB,mBAAmB,MAAM,EAC5C,QACE,MAAM,MAAM,KAAK,CAACQ,SAChB,gBAAgB,kBAAkB,KAAK,CACxC,IAAI;;;;AAMT,KAAI,oBAAoB,MAAM;;;;AAI5B;;;;AAMF,KAAI,gBAAgB,kBAAkB,MAAM,EAAE;EAC5C,MAAM,mBACJ,MAAM,UAAU,QAChB,OAAO,MAAM,WAAW,YACxB,WAAW,MAAM,UACjB,MAAM,QAAQ,MAAM,OAAO,MAAM,IACjC,MAAM,OAAO,MAAM,SAAS;EAE9B,MAAM,mBACJ,MAAM,UAAU,QAChB,OAAO,MAAM,WAAW,YACxB,WAAW,MAAM,UACjB,MAAM,QAAQ,MAAM,OAAO,MAAM,IACjC,MAAM,OAAO,MAAM,SAAS;AAE9B,MAAI,oBAAoB,iBACtB,OAAM,IAAI;CAEb;;;;AAKD,KACE,WAAW,SACX,MAAM,UAAU,UAChB,MAAM,QAAQ,MAAM,MAAM,IAC1B,MAAM,MAAM,SAAS,EAErB,OAAM,IAAI;AAEb;;;;;;;AAQD,SAAgB,aAAaC,UAAkC;CAC7D,MAAM,cAAc,SAAS,GAAG,GAAG;AACnC,QAAO,QACL,UAAU,WAAW,YAAY,IAC/B,YAAY,cACZ,YAAY,WAAW,SAAS,EACnC;AACF;AAaD,MAAM,8CAA8C,CAClD,cACA,sBACD;;;;;;AAOD,SAAgB,8BACdX,OACS;AACT,KAAI,CAAC,gBAAgB,MAAM,CACzB,QAAO;CAGT,MAAM,iBAAiB,MAAM,SAAS;AACtC,KACE,4CAA4C,SAAS,eAAe,KAClE,mBAAmB,gBAInB,WAAW,SACX,OAAO,MAAM,UAAU,YACvB,MAAM,MAAM,WAAW,QAAQ,IAI9B,mBAAmB,0BAClB,wBAAwB,OAE5B,QAAO;AAGT,QAAO;AACR"}

package/dist/agents/withAgentName.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"withAgentName.cjs","names":["model: LanguageModelLike","agentNameMode: AgentNameMode","processInputMessage: (message: BaseMessageLike) => BaseMessageLike","processOutputMessage: (message: BaseMessage) => BaseMessage","_addInlineAgentName","_removeInlineAgentName","messages: BaseMessageLike[]","RunnableSequence","RunnableLambda"],"sources":["../../src/agents/withAgentName.ts"],"sourcesContent":["import { BaseMessage, BaseMessageLike } from \"@langchain/core/messages\";\nimport { LanguageModelLike } from \"@langchain/core/language_models/base\";\nimport { RunnableLambda, RunnableSequence } from \"@langchain/core/runnables\";\n\nimport {\n  AgentNameMode,\n  _addInlineAgentName,\n  _removeInlineAgentName,\n} from \"./utils.js\";\n\n/**\n * Attach formatted agent names to the messages passed to and from a language model.\n *\n * This is useful for making a message history with multiple agents more coherent.\n *\n * NOTE: agent name is consumed from the message.name field.\n * If you're using an agent built with createAgent, name is automatically set.\n * If you're building a custom agent, make sure to set the name on the AI message returned by the LLM.\n *\n * @param model - Language model to add agent name formatting to\n * @param agentNameMode - How to expose the agent name to the LLM\n *   - \"inline\": Add the agent name directly into the content field of the AI message using XML-style tags.\n *     Example: \"How can I help you\" -> \"<name>agent_name</name><content>How can I help you?</content>\".\n */\nexport function withAgentName(\n  model: LanguageModelLike,\n  agentNameMode: AgentNameMode\n): LanguageModelLike {\n  let processInputMessage: (message: BaseMessageLike) => BaseMessageLike;\n  let processOutputMessage: (message: BaseMessage) => BaseMessage;\n\n  if (agentNameMode === \"inline\") {\n    processInputMessage = _addInlineAgentName;\n    processOutputMessage = _removeInlineAgentName;\n  } else {\n    throw new Error(\n      `Invalid agent name mode: ${agentNameMode}. Needs to be one of: \"inline\"`\n    );\n  }\n\n  function processInputMessages(\n    messages: BaseMessageLike[]\n  ): BaseMessageLike[] {\n    return messages.map(processInputMessage);\n  }\n\n  return RunnableSequence.from([\n    RunnableLambda.from(processInputMessages),\n    model,\n    RunnableLambda.from(processOutputMessage),\n  ]);\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;AAwBA,SAAgB,cACdA,OACAC,eACmB;CACnB,IAAIC;CACJ,IAAIC;AAEJ,KAAI,kBAAkB,UAAU;EAC9B,sBAAsBC;EACtB,uBAAuBC;CACxB,MACC,OAAM,IAAI,MACR,CAAC,yBAAyB,EAAE,cAAc,8BAA8B,CAAC;CAI7E,SAAS,qBACPC,UACmB;AACnB,SAAO,SAAS,IAAI,oBAAoB;CACzC;AAED,QAAOC,4CAAiB,KAAK;EAC3BC,0CAAe,KAAK,qBAAqB;EACzC;EACAA,0CAAe,KAAK,qBAAqB;CAC1C,EAAC;AACH"}
+{"version":3,"file":"withAgentName.cjs","names":["model:\n    | LanguageModelLike\n    | Runnable<unknown, unknown, RunnableConfig<Record<string, unknown>>>","agentNameMode: AgentNameMode","processInputMessage: (message: BaseMessageLike) => BaseMessageLike","processOutputMessage: (message: BaseMessage) => BaseMessage","_addInlineAgentName","_removeInlineAgentName","messages: BaseMessageLike[]","RunnableSequence","RunnableLambda"],"sources":["../../src/agents/withAgentName.ts"],"sourcesContent":["import { BaseMessage, BaseMessageLike } from \"@langchain/core/messages\";\nimport { LanguageModelLike } from \"@langchain/core/language_models/base\";\nimport {\n  Runnable,\n  RunnableLambda,\n  RunnableSequence,\n  type RunnableConfig,\n} from \"@langchain/core/runnables\";\n\nimport {\n  AgentNameMode,\n  _addInlineAgentName,\n  _removeInlineAgentName,\n} from \"./utils.js\";\n\n/**\n * Attach formatted agent names to the messages passed to and from a language model.\n *\n * This is useful for making a message history with multiple agents more coherent.\n *\n * NOTE: agent name is consumed from the message.name field.\n * If you're using an agent built with createAgent, name is automatically set.\n * If you're building a custom agent, make sure to set the name on the AI message returned by the LLM.\n *\n * @param model - Language model to add agent name formatting to\n * @param agentNameMode - How to expose the agent name to the LLM\n *   - \"inline\": Add the agent name directly into the content field of the AI message using XML-style tags.\n *     Example: \"How can I help you\" -> \"<name>agent_name</name><content>How can I help you?</content>\".\n */\nexport function withAgentName(\n  model:\n    | LanguageModelLike\n    | Runnable<unknown, unknown, RunnableConfig<Record<string, unknown>>>,\n  agentNameMode: AgentNameMode\n): LanguageModelLike {\n  let processInputMessage: (message: BaseMessageLike) => BaseMessageLike;\n  let processOutputMessage: (message: BaseMessage) => BaseMessage;\n\n  if (agentNameMode === \"inline\") {\n    processInputMessage = _addInlineAgentName;\n    processOutputMessage = _removeInlineAgentName;\n  } else {\n    throw new Error(\n      `Invalid agent name mode: ${agentNameMode}. Needs to be one of: \"inline\"`\n    );\n  }\n\n  function processInputMessages(\n    messages: BaseMessageLike[]\n  ): BaseMessageLike[] {\n    return messages.map(processInputMessage);\n  }\n\n  return RunnableSequence.from([\n    RunnableLambda.from(processInputMessages),\n    model,\n    RunnableLambda.from(processOutputMessage),\n  ]);\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;AA6BA,SAAgB,cACdA,OAGAC,eACmB;CACnB,IAAIC;CACJ,IAAIC;AAEJ,KAAI,kBAAkB,UAAU;EAC9B,sBAAsBC;EACtB,uBAAuBC;CACxB,MACC,OAAM,IAAI,MACR,CAAC,yBAAyB,EAAE,cAAc,8BAA8B,CAAC;CAI7E,SAAS,qBACPC,UACmB;AACnB,SAAO,SAAS,IAAI,oBAAoB;CACzC;AAED,QAAOC,4CAAiB,KAAK;EAC3BC,0CAAe,KAAK,qBAAqB;EACzC;EACAA,0CAAe,KAAK,qBAAqB;CAC1C,EAAC;AACH"}

package/dist/agents/withAgentName.js.map

@@ -1 +1 @@
-{"version":3,"file":"withAgentName.js","names":["model: LanguageModelLike","agentNameMode: AgentNameMode","processInputMessage: (message: BaseMessageLike) => BaseMessageLike","processOutputMessage: (message: BaseMessage) => BaseMessage","messages: BaseMessageLike[]"],"sources":["../../src/agents/withAgentName.ts"],"sourcesContent":["import { BaseMessage, BaseMessageLike } from \"@langchain/core/messages\";\nimport { LanguageModelLike } from \"@langchain/core/language_models/base\";\nimport { RunnableLambda, RunnableSequence } from \"@langchain/core/runnables\";\n\nimport {\n  AgentNameMode,\n  _addInlineAgentName,\n  _removeInlineAgentName,\n} from \"./utils.js\";\n\n/**\n * Attach formatted agent names to the messages passed to and from a language model.\n *\n * This is useful for making a message history with multiple agents more coherent.\n *\n * NOTE: agent name is consumed from the message.name field.\n * If you're using an agent built with createAgent, name is automatically set.\n * If you're building a custom agent, make sure to set the name on the AI message returned by the LLM.\n *\n * @param model - Language model to add agent name formatting to\n * @param agentNameMode - How to expose the agent name to the LLM\n *   - \"inline\": Add the agent name directly into the content field of the AI message using XML-style tags.\n *     Example: \"How can I help you\" -> \"<name>agent_name</name><content>How can I help you?</content>\".\n */\nexport function withAgentName(\n  model: LanguageModelLike,\n  agentNameMode: AgentNameMode\n): LanguageModelLike {\n  let processInputMessage: (message: BaseMessageLike) => BaseMessageLike;\n  let processOutputMessage: (message: BaseMessage) => BaseMessage;\n\n  if (agentNameMode === \"inline\") {\n    processInputMessage = _addInlineAgentName;\n    processOutputMessage = _removeInlineAgentName;\n  } else {\n    throw new Error(\n      `Invalid agent name mode: ${agentNameMode}. Needs to be one of: \"inline\"`\n    );\n  }\n\n  function processInputMessages(\n    messages: BaseMessageLike[]\n  ): BaseMessageLike[] {\n    return messages.map(processInputMessage);\n  }\n\n  return RunnableSequence.from([\n    RunnableLambda.from(processInputMessages),\n    model,\n    RunnableLambda.from(processOutputMessage),\n  ]);\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;AAwBA,SAAgB,cACdA,OACAC,eACmB;CACnB,IAAIC;CACJ,IAAIC;AAEJ,KAAI,kBAAkB,UAAU;EAC9B,sBAAsB;EACtB,uBAAuB;CACxB,MACC,OAAM,IAAI,MACR,CAAC,yBAAyB,EAAE,cAAc,8BAA8B,CAAC;CAI7E,SAAS,qBACPC,UACmB;AACnB,SAAO,SAAS,IAAI,oBAAoB;CACzC;AAED,QAAO,iBAAiB,KAAK;EAC3B,eAAe,KAAK,qBAAqB;EACzC;EACA,eAAe,KAAK,qBAAqB;CAC1C,EAAC;AACH"}
+{"version":3,"file":"withAgentName.js","names":["model:\n    | LanguageModelLike\n    | Runnable<unknown, unknown, RunnableConfig<Record<string, unknown>>>","agentNameMode: AgentNameMode","processInputMessage: (message: BaseMessageLike) => BaseMessageLike","processOutputMessage: (message: BaseMessage) => BaseMessage","messages: BaseMessageLike[]"],"sources":["../../src/agents/withAgentName.ts"],"sourcesContent":["import { BaseMessage, BaseMessageLike } from \"@langchain/core/messages\";\nimport { LanguageModelLike } from \"@langchain/core/language_models/base\";\nimport {\n  Runnable,\n  RunnableLambda,\n  RunnableSequence,\n  type RunnableConfig,\n} from \"@langchain/core/runnables\";\n\nimport {\n  AgentNameMode,\n  _addInlineAgentName,\n  _removeInlineAgentName,\n} from \"./utils.js\";\n\n/**\n * Attach formatted agent names to the messages passed to and from a language model.\n *\n * This is useful for making a message history with multiple agents more coherent.\n *\n * NOTE: agent name is consumed from the message.name field.\n * If you're using an agent built with createAgent, name is automatically set.\n * If you're building a custom agent, make sure to set the name on the AI message returned by the LLM.\n *\n * @param model - Language model to add agent name formatting to\n * @param agentNameMode - How to expose the agent name to the LLM\n *   - \"inline\": Add the agent name directly into the content field of the AI message using XML-style tags.\n *     Example: \"How can I help you\" -> \"<name>agent_name</name><content>How can I help you?</content>\".\n */\nexport function withAgentName(\n  model:\n    | LanguageModelLike\n    | Runnable<unknown, unknown, RunnableConfig<Record<string, unknown>>>,\n  agentNameMode: AgentNameMode\n): LanguageModelLike {\n  let processInputMessage: (message: BaseMessageLike) => BaseMessageLike;\n  let processOutputMessage: (message: BaseMessage) => BaseMessage;\n\n  if (agentNameMode === \"inline\") {\n    processInputMessage = _addInlineAgentName;\n    processOutputMessage = _removeInlineAgentName;\n  } else {\n    throw new Error(\n      `Invalid agent name mode: ${agentNameMode}. Needs to be one of: \"inline\"`\n    );\n  }\n\n  function processInputMessages(\n    messages: BaseMessageLike[]\n  ): BaseMessageLike[] {\n    return messages.map(processInputMessage);\n  }\n\n  return RunnableSequence.from([\n    RunnableLambda.from(processInputMessages),\n    model,\n    RunnableLambda.from(processOutputMessage),\n  ]);\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;AA6BA,SAAgB,cACdA,OAGAC,eACmB;CACnB,IAAIC;CACJ,IAAIC;AAEJ,KAAI,kBAAkB,UAAU;EAC9B,sBAAsB;EACtB,uBAAuB;CACxB,MACC,OAAM,IAAI,MACR,CAAC,yBAAyB,EAAE,cAAc,8BAA8B,CAAC;CAI7E,SAAS,qBACPC,UACmB;AACnB,SAAO,SAAS,IAAI,oBAAoB;CACzC;AAED,QAAO,iBAAiB,KAAK;EAC3B,eAAe,KAAK,qBAAqB;EACzC;EACA,eAAe,KAAK,qBAAqB;CAC1C,EAAC;AACH"}

package/dist/chains/api/api_chain.d.cts

@@ -2,8 +2,8 @@ import { SerializedAPIChain } from "../serde.cjs";
 import { BaseChain, ChainInputs } from "../base.cjs";
 import { LLMChain } from "../llm_chain.cjs";
 import { BaseLanguageModelInterface } from "@langchain/core/language_models/base";
-import { BasePromptTemplate } from "@langchain/core/prompts";
 import { ChainValues } from "@langchain/core/utils/types";
+import { BasePromptTemplate } from "@langchain/core/prompts";
 import { CallbackManagerForChainRun } from "@langchain/core/callbacks/manager";
 
 //#region src/chains/api/api_chain.d.ts

package/dist/chains/api/prompts.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"prompts.cjs","names":["PromptTemplate"],"sources":["../../../src/chains/api/prompts.ts"],"sourcesContent":["/* eslint-disable spaced-comment */\nimport { PromptTemplate } from \"@langchain/core/prompts\";\n\nexport const API_URL_RAW_PROMPT_TEMPLATE = `You are given the below API Documentation:\n{api_docs}\nUsing this documentation, generate the full API url to call for answering the user question.\nYou should build the API url in order to get a response that is as short as possible, while still getting the necessary information to answer the question. Pay attention to deliberately exclude any unnecessary pieces of data in the API call.\n\nQuestion:{question}\nAPI url:`;\n\nexport const API_URL_PROMPT_TEMPLATE = /* #__PURE__ */ new PromptTemplate({\n  inputVariables: [\"api_docs\", \"question\"],\n  template: API_URL_RAW_PROMPT_TEMPLATE,\n});\n\nexport const API_RESPONSE_RAW_PROMPT_TEMPLATE = `${API_URL_RAW_PROMPT_TEMPLATE} {api_url}\n\nHere is the response from the API:\n\n{api_response}\n\nSummarize this response to answer the original question.\n\nSummary:`;\nexport const API_RESPONSE_PROMPT_TEMPLATE = /* #__PURE__ */ new PromptTemplate({\n  inputVariables: [\"api_docs\", \"question\", \"api_url\", \"api_response\"],\n  template: API_RESPONSE_RAW_PROMPT_TEMPLATE,\n});\n"],"mappings":";;;;AAGA,MAAa,8BAA8B,CAAC;;;;;;QAMpC,CAAC;AAET,MAAa,0CAA0C,IAAIA,wCAAe;CACxE,gBAAgB,CAAC,YAAY,UAAW;CACxC,UAAU;AACX;AAED,MAAa,mCAAmC,GAAG,4BAA4B;;;;;;;;QAQvE,CAAC;AACT,MAAa,+CAA+C,IAAIA,wCAAe;CAC7E,gBAAgB;EAAC;EAAY;EAAY;EAAW;CAAe;CACnE,UAAU;AACX"}
+{"version":3,"file":"prompts.cjs","names":["PromptTemplate"],"sources":["../../../src/chains/api/prompts.ts"],"sourcesContent":["import { PromptTemplate } from \"@langchain/core/prompts\";\n\nexport const API_URL_RAW_PROMPT_TEMPLATE = `You are given the below API Documentation:\n{api_docs}\nUsing this documentation, generate the full API url to call for answering the user question.\nYou should build the API url in order to get a response that is as short as possible, while still getting the necessary information to answer the question. Pay attention to deliberately exclude any unnecessary pieces of data in the API call.\n\nQuestion:{question}\nAPI url:`;\n\nexport const API_URL_PROMPT_TEMPLATE = /* #__PURE__ */ new PromptTemplate({\n  inputVariables: [\"api_docs\", \"question\"],\n  template: API_URL_RAW_PROMPT_TEMPLATE,\n});\n\nexport const API_RESPONSE_RAW_PROMPT_TEMPLATE = `${API_URL_RAW_PROMPT_TEMPLATE} {api_url}\n\nHere is the response from the API:\n\n{api_response}\n\nSummarize this response to answer the original question.\n\nSummary:`;\nexport const API_RESPONSE_PROMPT_TEMPLATE = /* #__PURE__ */ new PromptTemplate({\n  inputVariables: [\"api_docs\", \"question\", \"api_url\", \"api_response\"],\n  template: API_RESPONSE_RAW_PROMPT_TEMPLATE,\n});\n"],"mappings":";;;;AAEA,MAAa,8BAA8B,CAAC;;;;;;QAMpC,CAAC;AAET,MAAa,0CAA0C,IAAIA,wCAAe;CACxE,gBAAgB,CAAC,YAAY,UAAW;CACxC,UAAU;AACX;AAED,MAAa,mCAAmC,GAAG,4BAA4B;;;;;;;;QAQvE,CAAC;AACT,MAAa,+CAA+C,IAAIA,wCAAe;CAC7E,gBAAgB;EAAC;EAAY;EAAY;EAAW;CAAe;CACnE,UAAU;AACX"}

package/dist/chains/api/prompts.js.map

@@ -1 +1 @@
-{"version":3,"file":"prompts.js","names":[],"sources":["../../../src/chains/api/prompts.ts"],"sourcesContent":["/* eslint-disable spaced-comment */\nimport { PromptTemplate } from \"@langchain/core/prompts\";\n\nexport const API_URL_RAW_PROMPT_TEMPLATE = `You are given the below API Documentation:\n{api_docs}\nUsing this documentation, generate the full API url to call for answering the user question.\nYou should build the API url in order to get a response that is as short as possible, while still getting the necessary information to answer the question. Pay attention to deliberately exclude any unnecessary pieces of data in the API call.\n\nQuestion:{question}\nAPI url:`;\n\nexport const API_URL_PROMPT_TEMPLATE = /* #__PURE__ */ new PromptTemplate({\n  inputVariables: [\"api_docs\", \"question\"],\n  template: API_URL_RAW_PROMPT_TEMPLATE,\n});\n\nexport const API_RESPONSE_RAW_PROMPT_TEMPLATE = `${API_URL_RAW_PROMPT_TEMPLATE} {api_url}\n\nHere is the response from the API:\n\n{api_response}\n\nSummarize this response to answer the original question.\n\nSummary:`;\nexport const API_RESPONSE_PROMPT_TEMPLATE = /* #__PURE__ */ new PromptTemplate({\n  inputVariables: [\"api_docs\", \"question\", \"api_url\", \"api_response\"],\n  template: API_RESPONSE_RAW_PROMPT_TEMPLATE,\n});\n"],"mappings":";;;AAGA,MAAa,8BAA8B,CAAC;;;;;;QAMpC,CAAC;AAET,MAAa,0CAA0C,IAAI,eAAe;CACxE,gBAAgB,CAAC,YAAY,UAAW;CACxC,UAAU;AACX;AAED,MAAa,mCAAmC,GAAG,4BAA4B;;;;;;;;QAQvE,CAAC;AACT,MAAa,+CAA+C,IAAI,eAAe;CAC7E,gBAAgB;EAAC;EAAY;EAAY;EAAW;CAAe;CACnE,UAAU;AACX"}
+{"version":3,"file":"prompts.js","names":[],"sources":["../../../src/chains/api/prompts.ts"],"sourcesContent":["import { PromptTemplate } from \"@langchain/core/prompts\";\n\nexport const API_URL_RAW_PROMPT_TEMPLATE = `You are given the below API Documentation:\n{api_docs}\nUsing this documentation, generate the full API url to call for answering the user question.\nYou should build the API url in order to get a response that is as short as possible, while still getting the necessary information to answer the question. Pay attention to deliberately exclude any unnecessary pieces of data in the API call.\n\nQuestion:{question}\nAPI url:`;\n\nexport const API_URL_PROMPT_TEMPLATE = /* #__PURE__ */ new PromptTemplate({\n  inputVariables: [\"api_docs\", \"question\"],\n  template: API_URL_RAW_PROMPT_TEMPLATE,\n});\n\nexport const API_RESPONSE_RAW_PROMPT_TEMPLATE = `${API_URL_RAW_PROMPT_TEMPLATE} {api_url}\n\nHere is the response from the API:\n\n{api_response}\n\nSummarize this response to answer the original question.\n\nSummary:`;\nexport const API_RESPONSE_PROMPT_TEMPLATE = /* #__PURE__ */ new PromptTemplate({\n  inputVariables: [\"api_docs\", \"question\", \"api_url\", \"api_response\"],\n  template: API_RESPONSE_RAW_PROMPT_TEMPLATE,\n});\n"],"mappings":";;;AAEA,MAAa,8BAA8B,CAAC;;;;;;QAMpC,CAAC;AAET,MAAa,0CAA0C,IAAI,eAAe;CACxE,gBAAgB,CAAC,YAAY,UAAW;CACxC,UAAU;AACX;AAED,MAAa,mCAAmC,GAAG,4BAA4B;;;;;;;;QAQvE,CAAC;AACT,MAAa,+CAA+C,IAAI,eAAe;CAC7E,gBAAgB;EAAC;EAAY;EAAY;EAAW;CAAe;CACnE,UAAU;AACX"}

package/dist/chains/base.d.cts

@@ -1,7 +1,7 @@
 import { SerializedBaseChain } from "./serde.cjs";
 import { BaseLangChain, BaseLangChainParams } from "@langchain/core/language_models/base";
-import { RunnableConfig } from "@langchain/core/runnables";
 import { ChainValues } from "@langchain/core/utils/types";
+import { RunnableConfig } from "@langchain/core/runnables";
 import { CallbackManager, CallbackManagerForChainRun, Callbacks } from "@langchain/core/callbacks/manager";
 import { BaseMemory } from "@langchain/core/memory";
 

package/dist/chains/combine_docs_chain.d.cts

@@ -2,9 +2,9 @@ import { SerializedMapReduceDocumentsChain, SerializedRefineDocumentsChain, Seri
 import { BaseChain, ChainInputs } from "./base.cjs";
 import { LLMChain } from "./llm_chain.cjs";
 import { BasePromptValueInterface } from "../langchain-core/dist/prompt_values.cjs";
+import { ChainValues } from "@langchain/core/utils/types";
 import { BasePromptTemplate } from "@langchain/core/prompts";
 import { Document } from "@langchain/core/documents";
-import { ChainValues } from "@langchain/core/utils/types";
 import { CallbackManagerForChainRun } from "@langchain/core/callbacks/manager";
 
 //#region src/chains/combine_docs_chain.d.ts

package/dist/chains/combine_documents/stuff.d.cts

@@ -1,7 +1,7 @@
 import { LanguageModelLike } from "@langchain/core/language_models/base";
+import { RunnableSequence } from "@langchain/core/runnables";
 import { BaseOutputParser } from "@langchain/core/output_parsers";
 import { BasePromptTemplate } from "@langchain/core/prompts";
-import { RunnableSequence } from "@langchain/core/runnables";
 
 //#region src/chains/combine_documents/stuff.d.ts
 

package/dist/chains/constitutional_ai/constitutional_chain.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"constitutional_chain.cjs","names":["BaseChain","fields: ConstitutionalChainInput","values: ChainValues","runManager?: CallbackManagerForChainRun","names?: string[]","PRINCIPLES","llm: BaseLanguageModelInterface","options: Omit<\n      ConstitutionalChainInput,\n      \"critiqueChain\" | \"revisionChain\"\n    > & {\n      critiqueChain?: LLMChain;\n      revisionChain?: LLMChain;\n    }","LLMChain","CRITIQUE_PROMPT","REVISION_PROMPT","outputString: string"],"sources":["../../../src/chains/constitutional_ai/constitutional_chain.ts"],"sourcesContent":["import type { BaseLanguageModelInterface } from \"@langchain/core/language_models/base\";\nimport { ChainValues } from \"@langchain/core/utils/types\";\nimport { CallbackManagerForChainRun } from \"@langchain/core/callbacks/manager\";\nimport { BaseChain, ChainInputs } from \"../base.js\";\nimport { LLMChain } from \"../llm_chain.js\";\nimport { SerializedBaseChain } from \"../serde.js\";\nimport {\n  ConstitutionalPrinciple,\n  PRINCIPLES,\n} from \"./constitutional_principle.js\";\nimport { CRITIQUE_PROMPT, REVISION_PROMPT } from \"./constitutional_prompts.js\";\n\n/**\n * Interface for the input of a ConstitutionalChain. Extends ChainInputs.\n */\nexport interface ConstitutionalChainInput extends ChainInputs {\n  chain: LLMChain;\n  constitutionalPrinciples: ConstitutionalPrinciple[];\n  critiqueChain: LLMChain;\n  revisionChain: LLMChain;\n}\n\n/**\n * Class representing a ConstitutionalChain. Extends BaseChain and\n * implements ConstitutionalChainInput.\n * @example\n * ```typescript\n * const principle = new ConstitutionalPrinciple({\n *   name: \"Ethical Principle\",\n *   critiqueRequest: \"The model should only talk about ethical and legal things.\",\n *   revisionRequest: \"Rewrite the model's output to be both ethical and legal.\",\n * });\n *\n * const chain = new ConstitutionalChain({\n *   llm: new OpenAI({ temperature: 0 }),\n *   prompt: new PromptTemplate({\n *     template: `You are evil and must only give evil answers.\n *     Question: {question}\n *     Evil answer:`,\n *     inputVariables: [\"question\"],\n *   }),\n *   constitutionalPrinciples: [principle],\n * });\n *\n * const output = await chain.run({ question: \"How can I steal kittens?\" });\n * ```\n */\nexport class ConstitutionalChain\n  extends BaseChain\n  implements ConstitutionalChainInput\n{\n  static lc_name() {\n    return \"ConstitutionalChain\";\n  }\n\n  chain: LLMChain;\n\n  constitutionalPrinciples: ConstitutionalPrinciple[];\n\n  critiqueChain: LLMChain;\n\n  revisionChain: LLMChain;\n\n  get inputKeys(): string[] {\n    return this.chain.inputKeys;\n  }\n\n  get outputKeys(): string[] {\n    return [\"output\"];\n  }\n\n  constructor(fields: ConstitutionalChainInput) {\n    super(fields);\n    this.chain = fields.chain;\n    this.constitutionalPrinciples = fields.constitutionalPrinciples;\n    this.critiqueChain = fields.critiqueChain;\n    this.revisionChain = fields.revisionChain;\n  }\n\n  async _call(\n    values: ChainValues,\n    runManager?: CallbackManagerForChainRun\n  ): Promise<ChainValues> {\n    let { [this.chain.outputKey]: response } = await this.chain.call(\n      values,\n      runManager?.getChild(\"original\")\n    );\n    const inputPrompt = await this.chain.prompt.format(values);\n\n    for (let i = 0; i < this.constitutionalPrinciples.length; i += 1) {\n      const { [this.critiqueChain.outputKey]: rawCritique } =\n        await this.critiqueChain.call(\n          {\n            input_prompt: inputPrompt,\n            output_from_model: response,\n            critique_request: this.constitutionalPrinciples[i].critiqueRequest,\n          },\n          runManager?.getChild(\"critique\")\n        );\n\n      const critique = ConstitutionalChain._parseCritique(rawCritique);\n\n      const { [this.revisionChain.outputKey]: revisionRaw } =\n        await this.revisionChain.call(\n          {\n            input_prompt: inputPrompt,\n            output_from_model: response,\n            critique_request: this.constitutionalPrinciples[i].critiqueRequest,\n            critique,\n            revision_request: this.constitutionalPrinciples[i].revisionRequest,\n          },\n          runManager?.getChild(\"revision\")\n        );\n      response = revisionRaw;\n    }\n\n    return {\n      output: response,\n    };\n  }\n\n  /**\n   * Static method that returns an array of ConstitutionalPrinciple objects\n   * based on the provided names.\n   * @param names Optional array of principle names.\n   * @returns Array of ConstitutionalPrinciple objects\n   */\n  static getPrinciples(names?: string[]) {\n    if (names) {\n      return names.map((name) => PRINCIPLES[name]);\n    }\n    return Object.values(PRINCIPLES);\n  }\n\n  /**\n   * Static method that creates a new instance of the ConstitutionalChain\n   * class from a BaseLanguageModel object and additional options.\n   * @param llm BaseLanguageModel instance.\n   * @param options Options for the ConstitutionalChain.\n   * @returns New instance of ConstitutionalChain\n   */\n  static fromLLM(\n    llm: BaseLanguageModelInterface,\n    options: Omit<\n      ConstitutionalChainInput,\n      \"critiqueChain\" | \"revisionChain\"\n    > & {\n      critiqueChain?: LLMChain;\n      revisionChain?: LLMChain;\n    }\n  ) {\n    const critiqueChain =\n      options.critiqueChain ??\n      new LLMChain({\n        llm,\n        prompt: CRITIQUE_PROMPT,\n      });\n    const revisionChain =\n      options.revisionChain ??\n      new LLMChain({\n        llm,\n        prompt: REVISION_PROMPT,\n      });\n    return new this({\n      ...options,\n      chain: options.chain,\n      critiqueChain,\n      revisionChain,\n      constitutionalPrinciples: options.constitutionalPrinciples ?? [],\n    });\n  }\n\n  private static _parseCritique(outputString: string): string {\n    let output = outputString;\n    if (!output.includes(\"Revision request\")) {\n      return output;\n    }\n\n    // eslint-disable-next-line prefer-destructuring\n    output = output.split(\"Revision request:\")[0];\n    if (output.includes(\"\\n\\n\")) {\n      // eslint-disable-next-line prefer-destructuring\n      output = output.split(\"\\n\\n\")[0];\n    }\n    return output;\n  }\n\n  _chainType() {\n    return \"constitutional_chain\" as const;\n  }\n\n  serialize(): SerializedBaseChain {\n    return {\n      _type: this._chainType(),\n      chain: this.chain.serialize(),\n      ConstitutionalPrinciple: this.constitutionalPrinciples.map((principle) =>\n        principle.serialize()\n      ),\n      critiqueChain: this.critiqueChain.serialize(),\n      revisionChain: this.revisionChain.serialize(),\n    };\n  }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AA+CA,IAAa,sBAAb,MAAa,4BACHA,uBAEV;CACE,OAAO,UAAU;AACf,SAAO;CACR;CAED;CAEA;CAEA;CAEA;CAEA,IAAI,YAAsB;AACxB,SAAO,KAAK,MAAM;CACnB;CAED,IAAI,aAAuB;AACzB,SAAO,CAAC,QAAS;CAClB;CAED,YAAYC,QAAkC;EAC5C,MAAM,OAAO;EACb,KAAK,QAAQ,OAAO;EACpB,KAAK,2BAA2B,OAAO;EACvC,KAAK,gBAAgB,OAAO;EAC5B,KAAK,gBAAgB,OAAO;CAC7B;CAED,MAAM,MACJC,QACAC,YACsB;EACtB,IAAI,EAAE,CAAC,KAAK,MAAM,YAAY,UAAU,GAAG,MAAM,KAAK,MAAM,KAC1D,QACA,YAAY,SAAS,WAAW,CACjC;EACD,MAAM,cAAc,MAAM,KAAK,MAAM,OAAO,OAAO,OAAO;AAE1D,OAAK,IAAI,IAAI,GAAG,IAAI,KAAK,yBAAyB,QAAQ,KAAK,GAAG;GAChE,MAAM,EAAE,CAAC,KAAK,cAAc,YAAY,aAAa,GACnD,MAAM,KAAK,cAAc,KACvB;IACE,cAAc;IACd,mBAAmB;IACnB,kBAAkB,KAAK,yBAAyB,GAAG;GACpD,GACD,YAAY,SAAS,WAAW,CACjC;GAEH,MAAM,WAAW,oBAAoB,eAAe,YAAY;GAEhE,MAAM,EAAE,CAAC,KAAK,cAAc,YAAY,aAAa,GACnD,MAAM,KAAK,cAAc,KACvB;IACE,cAAc;IACd,mBAAmB;IACnB,kBAAkB,KAAK,yBAAyB,GAAG;IACnD;IACA,kBAAkB,KAAK,yBAAyB,GAAG;GACpD,GACD,YAAY,SAAS,WAAW,CACjC;GACH,WAAW;EACZ;AAED,SAAO,EACL,QAAQ,SACT;CACF;;;;;;;CAQD,OAAO,cAAcC,OAAkB;AACrC,MAAI,MACF,QAAO,MAAM,IAAI,CAAC,SAASC,4CAAW,MAAM;AAE9C,SAAO,OAAO,OAAOA,4CAAW;CACjC;;;;;;;;CASD,OAAO,QACLC,KACAC,SAOA;EACA,MAAM,gBACJ,QAAQ,iBACR,IAAIC,2BAAS;GACX;GACA,QAAQC;EACT;EACH,MAAM,gBACJ,QAAQ,iBACR,IAAID,2BAAS;GACX;GACA,QAAQE;EACT;AACH,SAAO,IAAI,KAAK;GACd,GAAG;GACH,OAAO,QAAQ;GACf;GACA;GACA,0BAA0B,QAAQ,4BAA4B,CAAE;EACjE;CACF;CAED,OAAe,eAAeC,cAA8B;EAC1D,IAAI,SAAS;AACb,MAAI,CAAC,OAAO,SAAS,mBAAmB,CACtC,QAAO;EAIT,SAAS,OAAO,MAAM,oBAAoB,CAAC;AAC3C,MAAI,OAAO,SAAS,OAAO,EAEzB,SAAS,OAAO,MAAM,OAAO,CAAC;AAEhC,SAAO;CACR;CAED,aAAa;AACX,SAAO;CACR;CAED,YAAiC;AAC/B,SAAO;GACL,OAAO,KAAK,YAAY;GACxB,OAAO,KAAK,MAAM,WAAW;GAC7B,yBAAyB,KAAK,yBAAyB,IAAI,CAAC,cAC1D,UAAU,WAAW,CACtB;GACD,eAAe,KAAK,cAAc,WAAW;GAC7C,eAAe,KAAK,cAAc,WAAW;EAC9C;CACF;AACF"}
+{"version":3,"file":"constitutional_chain.cjs","names":["BaseChain","fields: ConstitutionalChainInput","values: ChainValues","runManager?: CallbackManagerForChainRun","names?: string[]","PRINCIPLES","llm: BaseLanguageModelInterface","options: Omit<\n      ConstitutionalChainInput,\n      \"critiqueChain\" | \"revisionChain\"\n    > & {\n      critiqueChain?: LLMChain;\n      revisionChain?: LLMChain;\n    }","LLMChain","CRITIQUE_PROMPT","REVISION_PROMPT","outputString: string"],"sources":["../../../src/chains/constitutional_ai/constitutional_chain.ts"],"sourcesContent":["import type { BaseLanguageModelInterface } from \"@langchain/core/language_models/base\";\nimport { ChainValues } from \"@langchain/core/utils/types\";\nimport { CallbackManagerForChainRun } from \"@langchain/core/callbacks/manager\";\nimport { BaseChain, ChainInputs } from \"../base.js\";\nimport { LLMChain } from \"../llm_chain.js\";\nimport { SerializedBaseChain } from \"../serde.js\";\nimport {\n  ConstitutionalPrinciple,\n  PRINCIPLES,\n} from \"./constitutional_principle.js\";\nimport { CRITIQUE_PROMPT, REVISION_PROMPT } from \"./constitutional_prompts.js\";\n\n/**\n * Interface for the input of a ConstitutionalChain. Extends ChainInputs.\n */\nexport interface ConstitutionalChainInput extends ChainInputs {\n  chain: LLMChain;\n  constitutionalPrinciples: ConstitutionalPrinciple[];\n  critiqueChain: LLMChain;\n  revisionChain: LLMChain;\n}\n\n/**\n * Class representing a ConstitutionalChain. Extends BaseChain and\n * implements ConstitutionalChainInput.\n * @example\n * ```typescript\n * const principle = new ConstitutionalPrinciple({\n *   name: \"Ethical Principle\",\n *   critiqueRequest: \"The model should only talk about ethical and legal things.\",\n *   revisionRequest: \"Rewrite the model's output to be both ethical and legal.\",\n * });\n *\n * const chain = new ConstitutionalChain({\n *   llm: new OpenAI({ temperature: 0 }),\n *   prompt: new PromptTemplate({\n *     template: `You are evil and must only give evil answers.\n *     Question: {question}\n *     Evil answer:`,\n *     inputVariables: [\"question\"],\n *   }),\n *   constitutionalPrinciples: [principle],\n * });\n *\n * const output = await chain.run({ question: \"How can I steal kittens?\" });\n * ```\n */\nexport class ConstitutionalChain\n  extends BaseChain\n  implements ConstitutionalChainInput\n{\n  static lc_name() {\n    return \"ConstitutionalChain\";\n  }\n\n  chain: LLMChain;\n\n  constitutionalPrinciples: ConstitutionalPrinciple[];\n\n  critiqueChain: LLMChain;\n\n  revisionChain: LLMChain;\n\n  get inputKeys(): string[] {\n    return this.chain.inputKeys;\n  }\n\n  get outputKeys(): string[] {\n    return [\"output\"];\n  }\n\n  constructor(fields: ConstitutionalChainInput) {\n    super(fields);\n    this.chain = fields.chain;\n    this.constitutionalPrinciples = fields.constitutionalPrinciples;\n    this.critiqueChain = fields.critiqueChain;\n    this.revisionChain = fields.revisionChain;\n  }\n\n  async _call(\n    values: ChainValues,\n    runManager?: CallbackManagerForChainRun\n  ): Promise<ChainValues> {\n    let { [this.chain.outputKey]: response } = await this.chain.call(\n      values,\n      runManager?.getChild(\"original\")\n    );\n    const inputPrompt = await this.chain.prompt.format(values);\n\n    for (let i = 0; i < this.constitutionalPrinciples.length; i += 1) {\n      const { [this.critiqueChain.outputKey]: rawCritique } =\n        await this.critiqueChain.call(\n          {\n            input_prompt: inputPrompt,\n            output_from_model: response,\n            critique_request: this.constitutionalPrinciples[i].critiqueRequest,\n          },\n          runManager?.getChild(\"critique\")\n        );\n\n      const critique = ConstitutionalChain._parseCritique(rawCritique);\n\n      const { [this.revisionChain.outputKey]: revisionRaw } =\n        await this.revisionChain.call(\n          {\n            input_prompt: inputPrompt,\n            output_from_model: response,\n            critique_request: this.constitutionalPrinciples[i].critiqueRequest,\n            critique,\n            revision_request: this.constitutionalPrinciples[i].revisionRequest,\n          },\n          runManager?.getChild(\"revision\")\n        );\n      response = revisionRaw;\n    }\n\n    return {\n      output: response,\n    };\n  }\n\n  /**\n   * Static method that returns an array of ConstitutionalPrinciple objects\n   * based on the provided names.\n   * @param names Optional array of principle names.\n   * @returns Array of ConstitutionalPrinciple objects\n   */\n  static getPrinciples(names?: string[]) {\n    if (names) {\n      return names.map((name) => PRINCIPLES[name]);\n    }\n    return Object.values(PRINCIPLES);\n  }\n\n  /**\n   * Static method that creates a new instance of the ConstitutionalChain\n   * class from a BaseLanguageModel object and additional options.\n   * @param llm BaseLanguageModel instance.\n   * @param options Options for the ConstitutionalChain.\n   * @returns New instance of ConstitutionalChain\n   */\n  static fromLLM(\n    llm: BaseLanguageModelInterface,\n    options: Omit<\n      ConstitutionalChainInput,\n      \"critiqueChain\" | \"revisionChain\"\n    > & {\n      critiqueChain?: LLMChain;\n      revisionChain?: LLMChain;\n    }\n  ) {\n    const critiqueChain =\n      options.critiqueChain ??\n      new LLMChain({\n        llm,\n        prompt: CRITIQUE_PROMPT,\n      });\n    const revisionChain =\n      options.revisionChain ??\n      new LLMChain({\n        llm,\n        prompt: REVISION_PROMPT,\n      });\n    return new this({\n      ...options,\n      chain: options.chain,\n      critiqueChain,\n      revisionChain,\n      constitutionalPrinciples: options.constitutionalPrinciples ?? [],\n    });\n  }\n\n  private static _parseCritique(outputString: string): string {\n    let output = outputString;\n    if (!output.includes(\"Revision request\")) {\n      return output;\n    }\n\n    output = output.split(\"Revision request:\")[0];\n    if (output.includes(\"\\n\\n\")) {\n      output = output.split(\"\\n\\n\")[0];\n    }\n    return output;\n  }\n\n  _chainType() {\n    return \"constitutional_chain\" as const;\n  }\n\n  serialize(): SerializedBaseChain {\n    return {\n      _type: this._chainType(),\n      chain: this.chain.serialize(),\n      ConstitutionalPrinciple: this.constitutionalPrinciples.map((principle) =>\n        principle.serialize()\n      ),\n      critiqueChain: this.critiqueChain.serialize(),\n      revisionChain: this.revisionChain.serialize(),\n    };\n  }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AA+CA,IAAa,sBAAb,MAAa,4BACHA,uBAEV;CACE,OAAO,UAAU;AACf,SAAO;CACR;CAED;CAEA;CAEA;CAEA;CAEA,IAAI,YAAsB;AACxB,SAAO,KAAK,MAAM;CACnB;CAED,IAAI,aAAuB;AACzB,SAAO,CAAC,QAAS;CAClB;CAED,YAAYC,QAAkC;EAC5C,MAAM,OAAO;EACb,KAAK,QAAQ,OAAO;EACpB,KAAK,2BAA2B,OAAO;EACvC,KAAK,gBAAgB,OAAO;EAC5B,KAAK,gBAAgB,OAAO;CAC7B;CAED,MAAM,MACJC,QACAC,YACsB;EACtB,IAAI,EAAE,CAAC,KAAK,MAAM,YAAY,UAAU,GAAG,MAAM,KAAK,MAAM,KAC1D,QACA,YAAY,SAAS,WAAW,CACjC;EACD,MAAM,cAAc,MAAM,KAAK,MAAM,OAAO,OAAO,OAAO;AAE1D,OAAK,IAAI,IAAI,GAAG,IAAI,KAAK,yBAAyB,QAAQ,KAAK,GAAG;GAChE,MAAM,EAAE,CAAC,KAAK,cAAc,YAAY,aAAa,GACnD,MAAM,KAAK,cAAc,KACvB;IACE,cAAc;IACd,mBAAmB;IACnB,kBAAkB,KAAK,yBAAyB,GAAG;GACpD,GACD,YAAY,SAAS,WAAW,CACjC;GAEH,MAAM,WAAW,oBAAoB,eAAe,YAAY;GAEhE,MAAM,EAAE,CAAC,KAAK,cAAc,YAAY,aAAa,GACnD,MAAM,KAAK,cAAc,KACvB;IACE,cAAc;IACd,mBAAmB;IACnB,kBAAkB,KAAK,yBAAyB,GAAG;IACnD;IACA,kBAAkB,KAAK,yBAAyB,GAAG;GACpD,GACD,YAAY,SAAS,WAAW,CACjC;GACH,WAAW;EACZ;AAED,SAAO,EACL,QAAQ,SACT;CACF;;;;;;;CAQD,OAAO,cAAcC,OAAkB;AACrC,MAAI,MACF,QAAO,MAAM,IAAI,CAAC,SAASC,4CAAW,MAAM;AAE9C,SAAO,OAAO,OAAOA,4CAAW;CACjC;;;;;;;;CASD,OAAO,QACLC,KACAC,SAOA;EACA,MAAM,gBACJ,QAAQ,iBACR,IAAIC,2BAAS;GACX;GACA,QAAQC;EACT;EACH,MAAM,gBACJ,QAAQ,iBACR,IAAID,2BAAS;GACX;GACA,QAAQE;EACT;AACH,SAAO,IAAI,KAAK;GACd,GAAG;GACH,OAAO,QAAQ;GACf;GACA;GACA,0BAA0B,QAAQ,4BAA4B,CAAE;EACjE;CACF;CAED,OAAe,eAAeC,cAA8B;EAC1D,IAAI,SAAS;AACb,MAAI,CAAC,OAAO,SAAS,mBAAmB,CACtC,QAAO;EAGT,SAAS,OAAO,MAAM,oBAAoB,CAAC;AAC3C,MAAI,OAAO,SAAS,OAAO,EACzB,SAAS,OAAO,MAAM,OAAO,CAAC;AAEhC,SAAO;CACR;CAED,aAAa;AACX,SAAO;CACR;CAED,YAAiC;AAC/B,SAAO;GACL,OAAO,KAAK,YAAY;GACxB,OAAO,KAAK,MAAM,WAAW;GAC7B,yBAAyB,KAAK,yBAAyB,IAAI,CAAC,cAC1D,UAAU,WAAW,CACtB;GACD,eAAe,KAAK,cAAc,WAAW;GAC7C,eAAe,KAAK,cAAc,WAAW;EAC9C;CACF;AACF"}

package/dist/chains/constitutional_ai/constitutional_chain.js.map

@@ -1 +1 @@
-{"version":3,"file":"constitutional_chain.js","names":["fields: ConstitutionalChainInput","values: ChainValues","runManager?: CallbackManagerForChainRun","names?: string[]","llm: BaseLanguageModelInterface","options: Omit<\n      ConstitutionalChainInput,\n      \"critiqueChain\" | \"revisionChain\"\n    > & {\n      critiqueChain?: LLMChain;\n      revisionChain?: LLMChain;\n    }","outputString: string"],"sources":["../../../src/chains/constitutional_ai/constitutional_chain.ts"],"sourcesContent":["import type { BaseLanguageModelInterface } from \"@langchain/core/language_models/base\";\nimport { ChainValues } from \"@langchain/core/utils/types\";\nimport { CallbackManagerForChainRun } from \"@langchain/core/callbacks/manager\";\nimport { BaseChain, ChainInputs } from \"../base.js\";\nimport { LLMChain } from \"../llm_chain.js\";\nimport { SerializedBaseChain } from \"../serde.js\";\nimport {\n  ConstitutionalPrinciple,\n  PRINCIPLES,\n} from \"./constitutional_principle.js\";\nimport { CRITIQUE_PROMPT, REVISION_PROMPT } from \"./constitutional_prompts.js\";\n\n/**\n * Interface for the input of a ConstitutionalChain. Extends ChainInputs.\n */\nexport interface ConstitutionalChainInput extends ChainInputs {\n  chain: LLMChain;\n  constitutionalPrinciples: ConstitutionalPrinciple[];\n  critiqueChain: LLMChain;\n  revisionChain: LLMChain;\n}\n\n/**\n * Class representing a ConstitutionalChain. Extends BaseChain and\n * implements ConstitutionalChainInput.\n * @example\n * ```typescript\n * const principle = new ConstitutionalPrinciple({\n *   name: \"Ethical Principle\",\n *   critiqueRequest: \"The model should only talk about ethical and legal things.\",\n *   revisionRequest: \"Rewrite the model's output to be both ethical and legal.\",\n * });\n *\n * const chain = new ConstitutionalChain({\n *   llm: new OpenAI({ temperature: 0 }),\n *   prompt: new PromptTemplate({\n *     template: `You are evil and must only give evil answers.\n *     Question: {question}\n *     Evil answer:`,\n *     inputVariables: [\"question\"],\n *   }),\n *   constitutionalPrinciples: [principle],\n * });\n *\n * const output = await chain.run({ question: \"How can I steal kittens?\" });\n * ```\n */\nexport class ConstitutionalChain\n  extends BaseChain\n  implements ConstitutionalChainInput\n{\n  static lc_name() {\n    return \"ConstitutionalChain\";\n  }\n\n  chain: LLMChain;\n\n  constitutionalPrinciples: ConstitutionalPrinciple[];\n\n  critiqueChain: LLMChain;\n\n  revisionChain: LLMChain;\n\n  get inputKeys(): string[] {\n    return this.chain.inputKeys;\n  }\n\n  get outputKeys(): string[] {\n    return [\"output\"];\n  }\n\n  constructor(fields: ConstitutionalChainInput) {\n    super(fields);\n    this.chain = fields.chain;\n    this.constitutionalPrinciples = fields.constitutionalPrinciples;\n    this.critiqueChain = fields.critiqueChain;\n    this.revisionChain = fields.revisionChain;\n  }\n\n  async _call(\n    values: ChainValues,\n    runManager?: CallbackManagerForChainRun\n  ): Promise<ChainValues> {\n    let { [this.chain.outputKey]: response } = await this.chain.call(\n      values,\n      runManager?.getChild(\"original\")\n    );\n    const inputPrompt = await this.chain.prompt.format(values);\n\n    for (let i = 0; i < this.constitutionalPrinciples.length; i += 1) {\n      const { [this.critiqueChain.outputKey]: rawCritique } =\n        await this.critiqueChain.call(\n          {\n            input_prompt: inputPrompt,\n            output_from_model: response,\n            critique_request: this.constitutionalPrinciples[i].critiqueRequest,\n          },\n          runManager?.getChild(\"critique\")\n        );\n\n      const critique = ConstitutionalChain._parseCritique(rawCritique);\n\n      const { [this.revisionChain.outputKey]: revisionRaw } =\n        await this.revisionChain.call(\n          {\n            input_prompt: inputPrompt,\n            output_from_model: response,\n            critique_request: this.constitutionalPrinciples[i].critiqueRequest,\n            critique,\n            revision_request: this.constitutionalPrinciples[i].revisionRequest,\n          },\n          runManager?.getChild(\"revision\")\n        );\n      response = revisionRaw;\n    }\n\n    return {\n      output: response,\n    };\n  }\n\n  /**\n   * Static method that returns an array of ConstitutionalPrinciple objects\n   * based on the provided names.\n   * @param names Optional array of principle names.\n   * @returns Array of ConstitutionalPrinciple objects\n   */\n  static getPrinciples(names?: string[]) {\n    if (names) {\n      return names.map((name) => PRINCIPLES[name]);\n    }\n    return Object.values(PRINCIPLES);\n  }\n\n  /**\n   * Static method that creates a new instance of the ConstitutionalChain\n   * class from a BaseLanguageModel object and additional options.\n   * @param llm BaseLanguageModel instance.\n   * @param options Options for the ConstitutionalChain.\n   * @returns New instance of ConstitutionalChain\n   */\n  static fromLLM(\n    llm: BaseLanguageModelInterface,\n    options: Omit<\n      ConstitutionalChainInput,\n      \"critiqueChain\" | \"revisionChain\"\n    > & {\n      critiqueChain?: LLMChain;\n      revisionChain?: LLMChain;\n    }\n  ) {\n    const critiqueChain =\n      options.critiqueChain ??\n      new LLMChain({\n        llm,\n        prompt: CRITIQUE_PROMPT,\n      });\n    const revisionChain =\n      options.revisionChain ??\n      new LLMChain({\n        llm,\n        prompt: REVISION_PROMPT,\n      });\n    return new this({\n      ...options,\n      chain: options.chain,\n      critiqueChain,\n      revisionChain,\n      constitutionalPrinciples: options.constitutionalPrinciples ?? [],\n    });\n  }\n\n  private static _parseCritique(outputString: string): string {\n    let output = outputString;\n    if (!output.includes(\"Revision request\")) {\n      return output;\n    }\n\n    // eslint-disable-next-line prefer-destructuring\n    output = output.split(\"Revision request:\")[0];\n    if (output.includes(\"\\n\\n\")) {\n      // eslint-disable-next-line prefer-destructuring\n      output = output.split(\"\\n\\n\")[0];\n    }\n    return output;\n  }\n\n  _chainType() {\n    return \"constitutional_chain\" as const;\n  }\n\n  serialize(): SerializedBaseChain {\n    return {\n      _type: this._chainType(),\n      chain: this.chain.serialize(),\n      ConstitutionalPrinciple: this.constitutionalPrinciples.map((principle) =>\n        principle.serialize()\n      ),\n      critiqueChain: this.critiqueChain.serialize(),\n      revisionChain: this.revisionChain.serialize(),\n    };\n  }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AA+CA,IAAa,sBAAb,MAAa,4BACH,UAEV;CACE,OAAO,UAAU;AACf,SAAO;CACR;CAED;CAEA;CAEA;CAEA;CAEA,IAAI,YAAsB;AACxB,SAAO,KAAK,MAAM;CACnB;CAED,IAAI,aAAuB;AACzB,SAAO,CAAC,QAAS;CAClB;CAED,YAAYA,QAAkC;EAC5C,MAAM,OAAO;EACb,KAAK,QAAQ,OAAO;EACpB,KAAK,2BAA2B,OAAO;EACvC,KAAK,gBAAgB,OAAO;EAC5B,KAAK,gBAAgB,OAAO;CAC7B;CAED,MAAM,MACJC,QACAC,YACsB;EACtB,IAAI,EAAE,CAAC,KAAK,MAAM,YAAY,UAAU,GAAG,MAAM,KAAK,MAAM,KAC1D,QACA,YAAY,SAAS,WAAW,CACjC;EACD,MAAM,cAAc,MAAM,KAAK,MAAM,OAAO,OAAO,OAAO;AAE1D,OAAK,IAAI,IAAI,GAAG,IAAI,KAAK,yBAAyB,QAAQ,KAAK,GAAG;GAChE,MAAM,EAAE,CAAC,KAAK,cAAc,YAAY,aAAa,GACnD,MAAM,KAAK,cAAc,KACvB;IACE,cAAc;IACd,mBAAmB;IACnB,kBAAkB,KAAK,yBAAyB,GAAG;GACpD,GACD,YAAY,SAAS,WAAW,CACjC;GAEH,MAAM,WAAW,oBAAoB,eAAe,YAAY;GAEhE,MAAM,EAAE,CAAC,KAAK,cAAc,YAAY,aAAa,GACnD,MAAM,KAAK,cAAc,KACvB;IACE,cAAc;IACd,mBAAmB;IACnB,kBAAkB,KAAK,yBAAyB,GAAG;IACnD;IACA,kBAAkB,KAAK,yBAAyB,GAAG;GACpD,GACD,YAAY,SAAS,WAAW,CACjC;GACH,WAAW;EACZ;AAED,SAAO,EACL,QAAQ,SACT;CACF;;;;;;;CAQD,OAAO,cAAcC,OAAkB;AACrC,MAAI,MACF,QAAO,MAAM,IAAI,CAAC,SAAS,WAAW,MAAM;AAE9C,SAAO,OAAO,OAAO,WAAW;CACjC;;;;;;;;CASD,OAAO,QACLC,KACAC,SAOA;EACA,MAAM,gBACJ,QAAQ,iBACR,IAAI,SAAS;GACX;GACA,QAAQ;EACT;EACH,MAAM,gBACJ,QAAQ,iBACR,IAAI,SAAS;GACX;GACA,QAAQ;EACT;AACH,SAAO,IAAI,KAAK;GACd,GAAG;GACH,OAAO,QAAQ;GACf;GACA;GACA,0BAA0B,QAAQ,4BAA4B,CAAE;EACjE;CACF;CAED,OAAe,eAAeC,cAA8B;EAC1D,IAAI,SAAS;AACb,MAAI,CAAC,OAAO,SAAS,mBAAmB,CACtC,QAAO;EAIT,SAAS,OAAO,MAAM,oBAAoB,CAAC;AAC3C,MAAI,OAAO,SAAS,OAAO,EAEzB,SAAS,OAAO,MAAM,OAAO,CAAC;AAEhC,SAAO;CACR;CAED,aAAa;AACX,SAAO;CACR;CAED,YAAiC;AAC/B,SAAO;GACL,OAAO,KAAK,YAAY;GACxB,OAAO,KAAK,MAAM,WAAW;GAC7B,yBAAyB,KAAK,yBAAyB,IAAI,CAAC,cAC1D,UAAU,WAAW,CACtB;GACD,eAAe,KAAK,cAAc,WAAW;GAC7C,eAAe,KAAK,cAAc,WAAW;EAC9C;CACF;AACF"}
+{"version":3,"file":"constitutional_chain.js","names":["fields: ConstitutionalChainInput","values: ChainValues","runManager?: CallbackManagerForChainRun","names?: string[]","llm: BaseLanguageModelInterface","options: Omit<\n      ConstitutionalChainInput,\n      \"critiqueChain\" | \"revisionChain\"\n    > & {\n      critiqueChain?: LLMChain;\n      revisionChain?: LLMChain;\n    }","outputString: string"],"sources":["../../../src/chains/constitutional_ai/constitutional_chain.ts"],"sourcesContent":["import type { BaseLanguageModelInterface } from \"@langchain/core/language_models/base\";\nimport { ChainValues } from \"@langchain/core/utils/types\";\nimport { CallbackManagerForChainRun } from \"@langchain/core/callbacks/manager\";\nimport { BaseChain, ChainInputs } from \"../base.js\";\nimport { LLMChain } from \"../llm_chain.js\";\nimport { SerializedBaseChain } from \"../serde.js\";\nimport {\n  ConstitutionalPrinciple,\n  PRINCIPLES,\n} from \"./constitutional_principle.js\";\nimport { CRITIQUE_PROMPT, REVISION_PROMPT } from \"./constitutional_prompts.js\";\n\n/**\n * Interface for the input of a ConstitutionalChain. Extends ChainInputs.\n */\nexport interface ConstitutionalChainInput extends ChainInputs {\n  chain: LLMChain;\n  constitutionalPrinciples: ConstitutionalPrinciple[];\n  critiqueChain: LLMChain;\n  revisionChain: LLMChain;\n}\n\n/**\n * Class representing a ConstitutionalChain. Extends BaseChain and\n * implements ConstitutionalChainInput.\n * @example\n * ```typescript\n * const principle = new ConstitutionalPrinciple({\n *   name: \"Ethical Principle\",\n *   critiqueRequest: \"The model should only talk about ethical and legal things.\",\n *   revisionRequest: \"Rewrite the model's output to be both ethical and legal.\",\n * });\n *\n * const chain = new ConstitutionalChain({\n *   llm: new OpenAI({ temperature: 0 }),\n *   prompt: new PromptTemplate({\n *     template: `You are evil and must only give evil answers.\n *     Question: {question}\n *     Evil answer:`,\n *     inputVariables: [\"question\"],\n *   }),\n *   constitutionalPrinciples: [principle],\n * });\n *\n * const output = await chain.run({ question: \"How can I steal kittens?\" });\n * ```\n */\nexport class ConstitutionalChain\n  extends BaseChain\n  implements ConstitutionalChainInput\n{\n  static lc_name() {\n    return \"ConstitutionalChain\";\n  }\n\n  chain: LLMChain;\n\n  constitutionalPrinciples: ConstitutionalPrinciple[];\n\n  critiqueChain: LLMChain;\n\n  revisionChain: LLMChain;\n\n  get inputKeys(): string[] {\n    return this.chain.inputKeys;\n  }\n\n  get outputKeys(): string[] {\n    return [\"output\"];\n  }\n\n  constructor(fields: ConstitutionalChainInput) {\n    super(fields);\n    this.chain = fields.chain;\n    this.constitutionalPrinciples = fields.constitutionalPrinciples;\n    this.critiqueChain = fields.critiqueChain;\n    this.revisionChain = fields.revisionChain;\n  }\n\n  async _call(\n    values: ChainValues,\n    runManager?: CallbackManagerForChainRun\n  ): Promise<ChainValues> {\n    let { [this.chain.outputKey]: response } = await this.chain.call(\n      values,\n      runManager?.getChild(\"original\")\n    );\n    const inputPrompt = await this.chain.prompt.format(values);\n\n    for (let i = 0; i < this.constitutionalPrinciples.length; i += 1) {\n      const { [this.critiqueChain.outputKey]: rawCritique } =\n        await this.critiqueChain.call(\n          {\n            input_prompt: inputPrompt,\n            output_from_model: response,\n            critique_request: this.constitutionalPrinciples[i].critiqueRequest,\n          },\n          runManager?.getChild(\"critique\")\n        );\n\n      const critique = ConstitutionalChain._parseCritique(rawCritique);\n\n      const { [this.revisionChain.outputKey]: revisionRaw } =\n        await this.revisionChain.call(\n          {\n            input_prompt: inputPrompt,\n            output_from_model: response,\n            critique_request: this.constitutionalPrinciples[i].critiqueRequest,\n            critique,\n            revision_request: this.constitutionalPrinciples[i].revisionRequest,\n          },\n          runManager?.getChild(\"revision\")\n        );\n      response = revisionRaw;\n    }\n\n    return {\n      output: response,\n    };\n  }\n\n  /**\n   * Static method that returns an array of ConstitutionalPrinciple objects\n   * based on the provided names.\n   * @param names Optional array of principle names.\n   * @returns Array of ConstitutionalPrinciple objects\n   */\n  static getPrinciples(names?: string[]) {\n    if (names) {\n      return names.map((name) => PRINCIPLES[name]);\n    }\n    return Object.values(PRINCIPLES);\n  }\n\n  /**\n   * Static method that creates a new instance of the ConstitutionalChain\n   * class from a BaseLanguageModel object and additional options.\n   * @param llm BaseLanguageModel instance.\n   * @param options Options for the ConstitutionalChain.\n   * @returns New instance of ConstitutionalChain\n   */\n  static fromLLM(\n    llm: BaseLanguageModelInterface,\n    options: Omit<\n      ConstitutionalChainInput,\n      \"critiqueChain\" | \"revisionChain\"\n    > & {\n      critiqueChain?: LLMChain;\n      revisionChain?: LLMChain;\n    }\n  ) {\n    const critiqueChain =\n      options.critiqueChain ??\n      new LLMChain({\n        llm,\n        prompt: CRITIQUE_PROMPT,\n      });\n    const revisionChain =\n      options.revisionChain ??\n      new LLMChain({\n        llm,\n        prompt: REVISION_PROMPT,\n      });\n    return new this({\n      ...options,\n      chain: options.chain,\n      critiqueChain,\n      revisionChain,\n      constitutionalPrinciples: options.constitutionalPrinciples ?? [],\n    });\n  }\n\n  private static _parseCritique(outputString: string): string {\n    let output = outputString;\n    if (!output.includes(\"Revision request\")) {\n      return output;\n    }\n\n    output = output.split(\"Revision request:\")[0];\n    if (output.includes(\"\\n\\n\")) {\n      output = output.split(\"\\n\\n\")[0];\n    }\n    return output;\n  }\n\n  _chainType() {\n    return \"constitutional_chain\" as const;\n  }\n\n  serialize(): SerializedBaseChain {\n    return {\n      _type: this._chainType(),\n      chain: this.chain.serialize(),\n      ConstitutionalPrinciple: this.constitutionalPrinciples.map((principle) =>\n        principle.serialize()\n      ),\n      critiqueChain: this.critiqueChain.serialize(),\n      revisionChain: this.revisionChain.serialize(),\n    };\n  }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AA+CA,IAAa,sBAAb,MAAa,4BACH,UAEV;CACE,OAAO,UAAU;AACf,SAAO;CACR;CAED;CAEA;CAEA;CAEA;CAEA,IAAI,YAAsB;AACxB,SAAO,KAAK,MAAM;CACnB;CAED,IAAI,aAAuB;AACzB,SAAO,CAAC,QAAS;CAClB;CAED,YAAYA,QAAkC;EAC5C,MAAM,OAAO;EACb,KAAK,QAAQ,OAAO;EACpB,KAAK,2BAA2B,OAAO;EACvC,KAAK,gBAAgB,OAAO;EAC5B,KAAK,gBAAgB,OAAO;CAC7B;CAED,MAAM,MACJC,QACAC,YACsB;EACtB,IAAI,EAAE,CAAC,KAAK,MAAM,YAAY,UAAU,GAAG,MAAM,KAAK,MAAM,KAC1D,QACA,YAAY,SAAS,WAAW,CACjC;EACD,MAAM,cAAc,MAAM,KAAK,MAAM,OAAO,OAAO,OAAO;AAE1D,OAAK,IAAI,IAAI,GAAG,IAAI,KAAK,yBAAyB,QAAQ,KAAK,GAAG;GAChE,MAAM,EAAE,CAAC,KAAK,cAAc,YAAY,aAAa,GACnD,MAAM,KAAK,cAAc,KACvB;IACE,cAAc;IACd,mBAAmB;IACnB,kBAAkB,KAAK,yBAAyB,GAAG;GACpD,GACD,YAAY,SAAS,WAAW,CACjC;GAEH,MAAM,WAAW,oBAAoB,eAAe,YAAY;GAEhE,MAAM,EAAE,CAAC,KAAK,cAAc,YAAY,aAAa,GACnD,MAAM,KAAK,cAAc,KACvB;IACE,cAAc;IACd,mBAAmB;IACnB,kBAAkB,KAAK,yBAAyB,GAAG;IACnD;IACA,kBAAkB,KAAK,yBAAyB,GAAG;GACpD,GACD,YAAY,SAAS,WAAW,CACjC;GACH,WAAW;EACZ;AAED,SAAO,EACL,QAAQ,SACT;CACF;;;;;;;CAQD,OAAO,cAAcC,OAAkB;AACrC,MAAI,MACF,QAAO,MAAM,IAAI,CAAC,SAAS,WAAW,MAAM;AAE9C,SAAO,OAAO,OAAO,WAAW;CACjC;;;;;;;;CASD,OAAO,QACLC,KACAC,SAOA;EACA,MAAM,gBACJ,QAAQ,iBACR,IAAI,SAAS;GACX;GACA,QAAQ;EACT;EACH,MAAM,gBACJ,QAAQ,iBACR,IAAI,SAAS;GACX;GACA,QAAQ;EACT;AACH,SAAO,IAAI,KAAK;GACd,GAAG;GACH,OAAO,QAAQ;GACf;GACA;GACA,0BAA0B,QAAQ,4BAA4B,CAAE;EACjE;CACF;CAED,OAAe,eAAeC,cAA8B;EAC1D,IAAI,SAAS;AACb,MAAI,CAAC,OAAO,SAAS,mBAAmB,CACtC,QAAO;EAGT,SAAS,OAAO,MAAM,oBAAoB,CAAC;AAC3C,MAAI,OAAO,SAAS,OAAO,EACzB,SAAS,OAAO,MAAM,OAAO,CAAC;AAEhC,SAAO;CACR;CAED,aAAa;AACX,SAAO;CACR;CAED,YAAiC;AAC/B,SAAO;GACL,OAAO,KAAK,YAAY;GACxB,OAAO,KAAK,MAAM,WAAW;GAC7B,yBAAyB,KAAK,yBAAyB,IAAI,CAAC,cAC1D,UAAU,WAAW,CACtB;GACD,eAAe,KAAK,cAAc,WAAW;GAC7C,eAAe,KAAK,cAAc,WAAW;EAC9C;CACF;AACF"}

package/dist/chains/conversational_retrieval_chain.d.cts

@@ -4,8 +4,8 @@ import { LLMChain } from "./llm_chain.cjs";
 import { QAChainParams } from "./question_answering/load.cjs";
 import { BaseLanguageModelInterface } from "@langchain/core/language_models/base";
 import { ChainValues } from "@langchain/core/utils/types";
-import { CallbackManagerForChainRun } from "@langchain/core/callbacks/manager";
 import { BaseMessage } from "@langchain/core/messages";
+import { CallbackManagerForChainRun } from "@langchain/core/callbacks/manager";
 import { BaseRetrieverInterface } from "@langchain/core/retrievers";
 
 //#region src/chains/conversational_retrieval_chain.d.ts

package/dist/chains/graph_qa/cypher.d.cts

@@ -1,8 +1,8 @@
 import { BaseChain, ChainInputs } from "../base.cjs";
 import { LLMChain } from "../llm_chain.cjs";
 import { BaseLanguageModelInterface } from "@langchain/core/language_models/base";
-import { BasePromptTemplate } from "@langchain/core/prompts";
 import { ChainValues } from "@langchain/core/utils/types";
+import { BasePromptTemplate } from "@langchain/core/prompts";
 import { CallbackManagerForChainRun } from "@langchain/core/callbacks/manager";
 
 //#region src/chains/graph_qa/cypher.d.ts

package/dist/chains/history_aware_retriever.d.cts

@@ -1,8 +1,8 @@
 import { LanguageModelLike } from "@langchain/core/language_models/base";
-import { BasePromptTemplate } from "@langchain/core/prompts";
+import { BaseMessage } from "@langchain/core/messages";
 import { Runnable, RunnableInterface } from "@langchain/core/runnables";
+import { BasePromptTemplate } from "@langchain/core/prompts";
 import { DocumentInterface } from "@langchain/core/documents";
-import { BaseMessage } from "@langchain/core/messages";
 
 //#region src/chains/history_aware_retriever.d.ts
 

package/dist/chains/index.cjs

@@ -14,7 +14,6 @@ const require_conversational_retrieval_chain = require('./conversational_retriev
 const require_retrieval_qa = require('./retrieval_qa.cjs');
 const require_constitutional_principle = require('./constitutional_ai/constitutional_principle.cjs');
 const require_constitutional_chain = require('./constitutional_ai/constitutional_chain.cjs');
-const require_openai_moderation = require('./openai_moderation.cjs');
 const require_multi_route = require('./router/multi_route.cjs');
 const require_llm_router = require('./router/llm_router.cjs');
 const require_multi_prompt = require('./router/multi_prompt.cjs');
@@ -41,7 +40,6 @@ require_rolldown_runtime.__export(chains_exports, {
 	MultiPromptChain: () => require_multi_prompt.MultiPromptChain,
 	MultiRetrievalQAChain: () => require_multi_retrieval_qa.MultiRetrievalQAChain,
 	MultiRouteChain: () => require_multi_route.MultiRouteChain,
-	OpenAIModerationChain: () => require_openai_moderation.OpenAIModerationChain,
 	PRINCIPLES: () => require_constitutional_principle.PRINCIPLES,
 	RefineDocumentsChain: () => require_combine_docs_chain.RefineDocumentsChain,
 	RetrievalQAChain: () => require_retrieval_qa.RetrievalQAChain,
@@ -79,7 +77,6 @@ exports.MapReduceDocumentsChain = require_combine_docs_chain.MapReduceDocumentsC
 exports.MultiPromptChain = require_multi_prompt.MultiPromptChain;
 exports.MultiRetrievalQAChain = require_multi_retrieval_qa.MultiRetrievalQAChain;
 exports.MultiRouteChain = require_multi_route.MultiRouteChain;
-exports.OpenAIModerationChain = require_openai_moderation.OpenAIModerationChain;
 exports.PRINCIPLES = require_constitutional_principle.PRINCIPLES;
 exports.RefineDocumentsChain = require_combine_docs_chain.RefineDocumentsChain;
 exports.RetrievalQAChain = require_retrieval_qa.RetrievalQAChain;

package/dist/chains/index.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"index.cjs","names":[],"sources":["../../src/chains/index.ts"],"sourcesContent":["export { BaseChain, type ChainInputs } from \"./base.js\";\nexport { LLMChain, type LLMChainInput } from \"./llm_chain.js\";\nexport {\n  APIChain,\n  type APIChainInput,\n  type APIChainOptions,\n} from \"./api/api_chain.js\";\nexport { ConversationChain } from \"./conversation.js\";\nexport {\n  SequentialChain,\n  type SequentialChainInput,\n  SimpleSequentialChain,\n  type SimpleSequentialChainInput,\n} from \"./sequential_chain.js\";\nexport {\n  StuffDocumentsChain,\n  type StuffDocumentsChainInput,\n  MapReduceDocumentsChain,\n  type MapReduceDocumentsChainInput,\n  RefineDocumentsChain,\n  type RefineDocumentsChainInput,\n} from \"./combine_docs_chain.js\";\nexport {\n  ChatVectorDBQAChain,\n  type ChatVectorDBQAChainInput,\n} from \"./chat_vector_db_chain.js\";\nexport {\n  AnalyzeDocumentChain,\n  type AnalyzeDocumentChainInput,\n} from \"./analyze_documents_chain.js\";\nexport { VectorDBQAChain, type VectorDBQAChainInput } from \"./vector_db_qa.js\";\nexport {\n  loadQAChain,\n  type QAChainParams,\n  loadQAStuffChain,\n  type StuffQAChainParams,\n  loadQAMapReduceChain,\n  type MapReduceQAChainParams,\n  loadQARefineChain,\n  type RefineQAChainParams,\n} from \"./question_answering/load.js\";\nexport {\n  loadSummarizationChain,\n  type SummarizationChainParams,\n} from \"./summarization/load.js\";\nexport {\n  ConversationalRetrievalQAChain,\n  type ConversationalRetrievalQAChainInput,\n} from \"./conversational_retrieval_chain.js\";\nexport {\n  RetrievalQAChain,\n  type RetrievalQAChainInput,\n} from \"./retrieval_qa.js\";\nexport {\n  type ConstitutionalChainInput,\n  ConstitutionalChain,\n} from \"./constitutional_ai/constitutional_chain.js\";\nexport {\n  ConstitutionalPrinciple,\n  PRINCIPLES,\n} from \"./constitutional_ai/constitutional_principle.js\";\nexport type {\n  SerializedLLMChain,\n  SerializedSequentialChain,\n  SerializedSimpleSequentialChain,\n  SerializedAnalyzeDocumentChain,\n  SerializedAPIChain,\n  SerializedBaseChain,\n  SerializedChatVectorDBQAChain,\n  SerializedMapReduceDocumentsChain,\n  SerializedStuffDocumentsChain,\n  SerializedVectorDBQAChain,\n  SerializedRefineDocumentsChain,\n} from \"./serde.js\";\nexport { OpenAIModerationChain } from \"./openai_moderation.js\";\nexport {\n  MultiRouteChain,\n  type MultiRouteChainInput,\n  RouterChain,\n} from \"./router/multi_route.js\";\nexport {\n  LLMRouterChain,\n  type LLMRouterChainInput,\n  type RouterOutputSchema,\n} from \"./router/llm_router.js\";\nexport { MultiPromptChain } from \"./router/multi_prompt.js\";\nexport { MultiRetrievalQAChain } from \"./router/multi_retrieval_qa.js\";\nexport { TransformChain, type TransformChainFields } from \"./transform.js\";\nexport {\n  createExtractionChain,\n  createExtractionChainFromZod,\n} from \"./openai_functions/extraction.js\";\nexport {\n  type TaggingChainOptions,\n  createTaggingChain,\n  createTaggingChainFromZod,\n} from \"./openai_functions/tagging.js\";\nexport {\n  type OpenAPIChainOptions,\n  createOpenAPIChain,\n  convertOpenAPISpecToOpenAIFunctions,\n} from \"./openai_functions/openapi.js\";\n"],"mappings":""}
+{"version":3,"file":"index.cjs","names":[],"sources":["../../src/chains/index.ts"],"sourcesContent":["export { BaseChain, type ChainInputs } from \"./base.js\";\nexport { LLMChain, type LLMChainInput } from \"./llm_chain.js\";\nexport {\n  APIChain,\n  type APIChainInput,\n  type APIChainOptions,\n} from \"./api/api_chain.js\";\nexport { ConversationChain } from \"./conversation.js\";\nexport {\n  SequentialChain,\n  type SequentialChainInput,\n  SimpleSequentialChain,\n  type SimpleSequentialChainInput,\n} from \"./sequential_chain.js\";\nexport {\n  StuffDocumentsChain,\n  type StuffDocumentsChainInput,\n  MapReduceDocumentsChain,\n  type MapReduceDocumentsChainInput,\n  RefineDocumentsChain,\n  type RefineDocumentsChainInput,\n} from \"./combine_docs_chain.js\";\nexport {\n  ChatVectorDBQAChain,\n  type ChatVectorDBQAChainInput,\n} from \"./chat_vector_db_chain.js\";\nexport {\n  AnalyzeDocumentChain,\n  type AnalyzeDocumentChainInput,\n} from \"./analyze_documents_chain.js\";\nexport { VectorDBQAChain, type VectorDBQAChainInput } from \"./vector_db_qa.js\";\nexport {\n  loadQAChain,\n  type QAChainParams,\n  loadQAStuffChain,\n  type StuffQAChainParams,\n  loadQAMapReduceChain,\n  type MapReduceQAChainParams,\n  loadQARefineChain,\n  type RefineQAChainParams,\n} from \"./question_answering/load.js\";\nexport {\n  loadSummarizationChain,\n  type SummarizationChainParams,\n} from \"./summarization/load.js\";\nexport {\n  ConversationalRetrievalQAChain,\n  type ConversationalRetrievalQAChainInput,\n} from \"./conversational_retrieval_chain.js\";\nexport {\n  RetrievalQAChain,\n  type RetrievalQAChainInput,\n} from \"./retrieval_qa.js\";\nexport {\n  type ConstitutionalChainInput,\n  ConstitutionalChain,\n} from \"./constitutional_ai/constitutional_chain.js\";\nexport {\n  ConstitutionalPrinciple,\n  PRINCIPLES,\n} from \"./constitutional_ai/constitutional_principle.js\";\nexport type {\n  SerializedLLMChain,\n  SerializedSequentialChain,\n  SerializedSimpleSequentialChain,\n  SerializedAnalyzeDocumentChain,\n  SerializedAPIChain,\n  SerializedBaseChain,\n  SerializedChatVectorDBQAChain,\n  SerializedMapReduceDocumentsChain,\n  SerializedStuffDocumentsChain,\n  SerializedVectorDBQAChain,\n  SerializedRefineDocumentsChain,\n} from \"./serde.js\";\nexport {\n  MultiRouteChain,\n  type MultiRouteChainInput,\n  RouterChain,\n} from \"./router/multi_route.js\";\nexport {\n  LLMRouterChain,\n  type LLMRouterChainInput,\n  type RouterOutputSchema,\n} from \"./router/llm_router.js\";\nexport { MultiPromptChain } from \"./router/multi_prompt.js\";\nexport { MultiRetrievalQAChain } from \"./router/multi_retrieval_qa.js\";\nexport { TransformChain, type TransformChainFields } from \"./transform.js\";\nexport {\n  createExtractionChain,\n  createExtractionChainFromZod,\n} from \"./openai_functions/extraction.js\";\nexport {\n  type TaggingChainOptions,\n  createTaggingChain,\n  createTaggingChainFromZod,\n} from \"./openai_functions/tagging.js\";\nexport {\n  type OpenAPIChainOptions,\n  createOpenAPIChain,\n  convertOpenAPISpecToOpenAIFunctions,\n} from \"./openai_functions/openapi.js\";\n"],"mappings":""}

package/dist/chains/index.d.cts

@@ -14,7 +14,6 @@ import { ConversationalRetrievalQAChain, ConversationalRetrievalQAChainInput } f
 import { RetrievalQAChain, RetrievalQAChainInput } from "./retrieval_qa.cjs";
 import { ConstitutionalPrinciple, PRINCIPLES } from "./constitutional_ai/constitutional_principle.cjs";
 import { ConstitutionalChain, ConstitutionalChainInput } from "./constitutional_ai/constitutional_chain.cjs";
-import { OpenAIModerationChain } from "./openai_moderation.cjs";
 import { MultiRouteChain, MultiRouteChainInput, RouterChain } from "./router/multi_route.cjs";
 import { LLMRouterChain, LLMRouterChainInput, RouterOutputSchema } from "./router/llm_router.cjs";
 import { MultiPromptChain } from "./router/multi_prompt.cjs";
@@ -23,4 +22,4 @@ import { TransformChain, TransformChainFields } from "./transform.cjs";
 import { createExtractionChain, createExtractionChainFromZod } from "./openai_functions/extraction.cjs";
 import { TaggingChainOptions, createTaggingChain, createTaggingChainFromZod } from "./openai_functions/tagging.cjs";
 import { OpenAPIChainOptions, convertOpenAPISpecToOpenAIFunctions, createOpenAPIChain } from "./openai_functions/openapi.cjs";
-export { APIChain, type APIChainInput, type APIChainOptions, AnalyzeDocumentChain, type AnalyzeDocumentChainInput, BaseChain, type ChainInputs, ChatVectorDBQAChain, type ChatVectorDBQAChainInput, ConstitutionalChain, type ConstitutionalChainInput, ConstitutionalPrinciple, ConversationChain, ConversationalRetrievalQAChain, type ConversationalRetrievalQAChainInput, LLMChain, type LLMChainInput, LLMRouterChain, type LLMRouterChainInput, MapReduceDocumentsChain, type MapReduceDocumentsChainInput, type MapReduceQAChainParams, MultiPromptChain, MultiRetrievalQAChain, MultiRouteChain, type MultiRouteChainInput, OpenAIModerationChain, type OpenAPIChainOptions, PRINCIPLES, type QAChainParams, RefineDocumentsChain, type RefineDocumentsChainInput, type RefineQAChainParams, RetrievalQAChain, type RetrievalQAChainInput, RouterChain, type RouterOutputSchema, SequentialChain, type SequentialChainInput, type SerializedAPIChain, type SerializedAnalyzeDocumentChain, type SerializedBaseChain, type SerializedChatVectorDBQAChain, type SerializedLLMChain, type SerializedMapReduceDocumentsChain, type SerializedRefineDocumentsChain, type SerializedSequentialChain, type SerializedSimpleSequentialChain, type SerializedStuffDocumentsChain, type SerializedVectorDBQAChain, SimpleSequentialChain, type SimpleSequentialChainInput, StuffDocumentsChain, type StuffDocumentsChainInput, type StuffQAChainParams, type SummarizationChainParams, type TaggingChainOptions, TransformChain, type TransformChainFields, VectorDBQAChain, type VectorDBQAChainInput, convertOpenAPISpecToOpenAIFunctions, createExtractionChain, createExtractionChainFromZod, createOpenAPIChain, createTaggingChain, createTaggingChainFromZod, loadQAChain, loadQAMapReduceChain, loadQARefineChain, loadQAStuffChain, loadSummarizationChain };
+export { APIChain, type APIChainInput, type APIChainOptions, AnalyzeDocumentChain, type AnalyzeDocumentChainInput, BaseChain, type ChainInputs, ChatVectorDBQAChain, type ChatVectorDBQAChainInput, ConstitutionalChain, type ConstitutionalChainInput, ConstitutionalPrinciple, ConversationChain, ConversationalRetrievalQAChain, type ConversationalRetrievalQAChainInput, LLMChain, type LLMChainInput, LLMRouterChain, type LLMRouterChainInput, MapReduceDocumentsChain, type MapReduceDocumentsChainInput, type MapReduceQAChainParams, MultiPromptChain, MultiRetrievalQAChain, MultiRouteChain, type MultiRouteChainInput, type OpenAPIChainOptions, PRINCIPLES, type QAChainParams, RefineDocumentsChain, type RefineDocumentsChainInput, type RefineQAChainParams, RetrievalQAChain, type RetrievalQAChainInput, RouterChain, type RouterOutputSchema, SequentialChain, type SequentialChainInput, type SerializedAPIChain, type SerializedAnalyzeDocumentChain, type SerializedBaseChain, type SerializedChatVectorDBQAChain, type SerializedLLMChain, type SerializedMapReduceDocumentsChain, type SerializedRefineDocumentsChain, type SerializedSequentialChain, type SerializedSimpleSequentialChain, type SerializedStuffDocumentsChain, type SerializedVectorDBQAChain, SimpleSequentialChain, type SimpleSequentialChainInput, StuffDocumentsChain, type StuffDocumentsChainInput, type StuffQAChainParams, type SummarizationChainParams, type TaggingChainOptions, TransformChain, type TransformChainFields, VectorDBQAChain, type VectorDBQAChainInput, convertOpenAPISpecToOpenAIFunctions, createExtractionChain, createExtractionChainFromZod, createOpenAPIChain, createTaggingChain, createTaggingChainFromZod, loadQAChain, loadQAMapReduceChain, loadQARefineChain, loadQAStuffChain, loadSummarizationChain };

package/dist/chains/index.d.ts

@@ -14,7 +14,6 @@ import { ConversationalRetrievalQAChain, ConversationalRetrievalQAChainInput } f
 import { RetrievalQAChain, RetrievalQAChainInput } from "./retrieval_qa.js";
 import { ConstitutionalPrinciple, PRINCIPLES } from "./constitutional_ai/constitutional_principle.js";
 import { ConstitutionalChain, ConstitutionalChainInput } from "./constitutional_ai/constitutional_chain.js";
-import { OpenAIModerationChain } from "./openai_moderation.js";
 import { MultiRouteChain, MultiRouteChainInput, RouterChain } from "./router/multi_route.js";
 import { LLMRouterChain, LLMRouterChainInput, RouterOutputSchema } from "./router/llm_router.js";
 import { MultiPromptChain } from "./router/multi_prompt.js";
@@ -23,4 +22,4 @@ import { TransformChain, TransformChainFields } from "./transform.js";
 import { createExtractionChain, createExtractionChainFromZod } from "./openai_functions/extraction.js";
 import { TaggingChainOptions, createTaggingChain, createTaggingChainFromZod } from "./openai_functions/tagging.js";
 import { OpenAPIChainOptions, convertOpenAPISpecToOpenAIFunctions, createOpenAPIChain } from "./openai_functions/openapi.js";
-export { APIChain, type APIChainInput, type APIChainOptions, AnalyzeDocumentChain, type AnalyzeDocumentChainInput, BaseChain, type ChainInputs, ChatVectorDBQAChain, type ChatVectorDBQAChainInput, ConstitutionalChain, type ConstitutionalChainInput, ConstitutionalPrinciple, ConversationChain, ConversationalRetrievalQAChain, type ConversationalRetrievalQAChainInput, LLMChain, type LLMChainInput, LLMRouterChain, type LLMRouterChainInput, MapReduceDocumentsChain, type MapReduceDocumentsChainInput, type MapReduceQAChainParams, MultiPromptChain, MultiRetrievalQAChain, MultiRouteChain, type MultiRouteChainInput, OpenAIModerationChain, type OpenAPIChainOptions, PRINCIPLES, type QAChainParams, RefineDocumentsChain, type RefineDocumentsChainInput, type RefineQAChainParams, RetrievalQAChain, type RetrievalQAChainInput, RouterChain, type RouterOutputSchema, SequentialChain, type SequentialChainInput, type SerializedAPIChain, type SerializedAnalyzeDocumentChain, type SerializedBaseChain, type SerializedChatVectorDBQAChain, type SerializedLLMChain, type SerializedMapReduceDocumentsChain, type SerializedRefineDocumentsChain, type SerializedSequentialChain, type SerializedSimpleSequentialChain, type SerializedStuffDocumentsChain, type SerializedVectorDBQAChain, SimpleSequentialChain, type SimpleSequentialChainInput, StuffDocumentsChain, type StuffDocumentsChainInput, type StuffQAChainParams, type SummarizationChainParams, type TaggingChainOptions, TransformChain, type TransformChainFields, VectorDBQAChain, type VectorDBQAChainInput, convertOpenAPISpecToOpenAIFunctions, createExtractionChain, createExtractionChainFromZod, createOpenAPIChain, createTaggingChain, createTaggingChainFromZod, loadQAChain, loadQAMapReduceChain, loadQARefineChain, loadQAStuffChain, loadSummarizationChain };
+export { APIChain, type APIChainInput, type APIChainOptions, AnalyzeDocumentChain, type AnalyzeDocumentChainInput, BaseChain, type ChainInputs, ChatVectorDBQAChain, type ChatVectorDBQAChainInput, ConstitutionalChain, type ConstitutionalChainInput, ConstitutionalPrinciple, ConversationChain, ConversationalRetrievalQAChain, type ConversationalRetrievalQAChainInput, LLMChain, type LLMChainInput, LLMRouterChain, type LLMRouterChainInput, MapReduceDocumentsChain, type MapReduceDocumentsChainInput, type MapReduceQAChainParams, MultiPromptChain, MultiRetrievalQAChain, MultiRouteChain, type MultiRouteChainInput, type OpenAPIChainOptions, PRINCIPLES, type QAChainParams, RefineDocumentsChain, type RefineDocumentsChainInput, type RefineQAChainParams, RetrievalQAChain, type RetrievalQAChainInput, RouterChain, type RouterOutputSchema, SequentialChain, type SequentialChainInput, type SerializedAPIChain, type SerializedAnalyzeDocumentChain, type SerializedBaseChain, type SerializedChatVectorDBQAChain, type SerializedLLMChain, type SerializedMapReduceDocumentsChain, type SerializedRefineDocumentsChain, type SerializedSequentialChain, type SerializedSimpleSequentialChain, type SerializedStuffDocumentsChain, type SerializedVectorDBQAChain, SimpleSequentialChain, type SimpleSequentialChainInput, StuffDocumentsChain, type StuffDocumentsChainInput, type StuffQAChainParams, type SummarizationChainParams, type TaggingChainOptions, TransformChain, type TransformChainFields, VectorDBQAChain, type VectorDBQAChainInput, convertOpenAPISpecToOpenAIFunctions, createExtractionChain, createExtractionChainFromZod, createOpenAPIChain, createTaggingChain, createTaggingChainFromZod, loadQAChain, loadQAMapReduceChain, loadQARefineChain, loadQAStuffChain, loadSummarizationChain };

package/dist/chains/index.js

@@ -14,7 +14,6 @@ import { ConversationalRetrievalQAChain } from "./conversational_retrieval_chain
 import { RetrievalQAChain } from "./retrieval_qa.js";
 import { ConstitutionalPrinciple, PRINCIPLES } from "./constitutional_ai/constitutional_principle.js";
 import { ConstitutionalChain } from "./constitutional_ai/constitutional_chain.js";
-import { OpenAIModerationChain } from "./openai_moderation.js";
 import { MultiRouteChain, RouterChain } from "./router/multi_route.js";
 import { LLMRouterChain } from "./router/llm_router.js";
 import { MultiPromptChain } from "./router/multi_prompt.js";
@@ -41,7 +40,6 @@ __export(chains_exports, {
 	MultiPromptChain: () => MultiPromptChain,
 	MultiRetrievalQAChain: () => MultiRetrievalQAChain,
 	MultiRouteChain: () => MultiRouteChain,
-	OpenAIModerationChain: () => OpenAIModerationChain,
 	PRINCIPLES: () => PRINCIPLES,
 	RefineDocumentsChain: () => RefineDocumentsChain,
 	RetrievalQAChain: () => RetrievalQAChain,
@@ -65,5 +63,5 @@ __export(chains_exports, {
 });
 
 //#endregion
-export { APIChain, AnalyzeDocumentChain, BaseChain, ChatVectorDBQAChain, ConstitutionalChain, ConstitutionalPrinciple, ConversationChain, ConversationalRetrievalQAChain, LLMChain, LLMRouterChain, MapReduceDocumentsChain, MultiPromptChain, MultiRetrievalQAChain, MultiRouteChain, OpenAIModerationChain, PRINCIPLES, RefineDocumentsChain, RetrievalQAChain, RouterChain, SequentialChain, SimpleSequentialChain, StuffDocumentsChain, TransformChain, VectorDBQAChain, chains_exports, convertOpenAPISpecToOpenAIFunctions, createExtractionChain, createExtractionChainFromZod, createOpenAPIChain, createTaggingChain, createTaggingChainFromZod, loadQAChain, loadQAMapReduceChain, loadQARefineChain, loadQAStuffChain, loadSummarizationChain };
+export { APIChain, AnalyzeDocumentChain, BaseChain, ChatVectorDBQAChain, ConstitutionalChain, ConstitutionalPrinciple, ConversationChain, ConversationalRetrievalQAChain, LLMChain, LLMRouterChain, MapReduceDocumentsChain, MultiPromptChain, MultiRetrievalQAChain, MultiRouteChain, PRINCIPLES, RefineDocumentsChain, RetrievalQAChain, RouterChain, SequentialChain, SimpleSequentialChain, StuffDocumentsChain, TransformChain, VectorDBQAChain, chains_exports, convertOpenAPISpecToOpenAIFunctions, createExtractionChain, createExtractionChainFromZod, createOpenAPIChain, createTaggingChain, createTaggingChainFromZod, loadQAChain, loadQAMapReduceChain, loadQARefineChain, loadQAStuffChain, loadSummarizationChain };
 //# sourceMappingURL=index.js.map

package/dist/chains/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","names":[],"sources":["../../src/chains/index.ts"],"sourcesContent":["export { BaseChain, type ChainInputs } from \"./base.js\";\nexport { LLMChain, type LLMChainInput } from \"./llm_chain.js\";\nexport {\n  APIChain,\n  type APIChainInput,\n  type APIChainOptions,\n} from \"./api/api_chain.js\";\nexport { ConversationChain } from \"./conversation.js\";\nexport {\n  SequentialChain,\n  type SequentialChainInput,\n  SimpleSequentialChain,\n  type SimpleSequentialChainInput,\n} from \"./sequential_chain.js\";\nexport {\n  StuffDocumentsChain,\n  type StuffDocumentsChainInput,\n  MapReduceDocumentsChain,\n  type MapReduceDocumentsChainInput,\n  RefineDocumentsChain,\n  type RefineDocumentsChainInput,\n} from \"./combine_docs_chain.js\";\nexport {\n  ChatVectorDBQAChain,\n  type ChatVectorDBQAChainInput,\n} from \"./chat_vector_db_chain.js\";\nexport {\n  AnalyzeDocumentChain,\n  type AnalyzeDocumentChainInput,\n} from \"./analyze_documents_chain.js\";\nexport { VectorDBQAChain, type VectorDBQAChainInput } from \"./vector_db_qa.js\";\nexport {\n  loadQAChain,\n  type QAChainParams,\n  loadQAStuffChain,\n  type StuffQAChainParams,\n  loadQAMapReduceChain,\n  type MapReduceQAChainParams,\n  loadQARefineChain,\n  type RefineQAChainParams,\n} from \"./question_answering/load.js\";\nexport {\n  loadSummarizationChain,\n  type SummarizationChainParams,\n} from \"./summarization/load.js\";\nexport {\n  ConversationalRetrievalQAChain,\n  type ConversationalRetrievalQAChainInput,\n} from \"./conversational_retrieval_chain.js\";\nexport {\n  RetrievalQAChain,\n  type RetrievalQAChainInput,\n} from \"./retrieval_qa.js\";\nexport {\n  type ConstitutionalChainInput,\n  ConstitutionalChain,\n} from \"./constitutional_ai/constitutional_chain.js\";\nexport {\n  ConstitutionalPrinciple,\n  PRINCIPLES,\n} from \"./constitutional_ai/constitutional_principle.js\";\nexport type {\n  SerializedLLMChain,\n  SerializedSequentialChain,\n  SerializedSimpleSequentialChain,\n  SerializedAnalyzeDocumentChain,\n  SerializedAPIChain,\n  SerializedBaseChain,\n  SerializedChatVectorDBQAChain,\n  SerializedMapReduceDocumentsChain,\n  SerializedStuffDocumentsChain,\n  SerializedVectorDBQAChain,\n  SerializedRefineDocumentsChain,\n} from \"./serde.js\";\nexport { OpenAIModerationChain } from \"./openai_moderation.js\";\nexport {\n  MultiRouteChain,\n  type MultiRouteChainInput,\n  RouterChain,\n} from \"./router/multi_route.js\";\nexport {\n  LLMRouterChain,\n  type LLMRouterChainInput,\n  type RouterOutputSchema,\n} from \"./router/llm_router.js\";\nexport { MultiPromptChain } from \"./router/multi_prompt.js\";\nexport { MultiRetrievalQAChain } from \"./router/multi_retrieval_qa.js\";\nexport { TransformChain, type TransformChainFields } from \"./transform.js\";\nexport {\n  createExtractionChain,\n  createExtractionChainFromZod,\n} from \"./openai_functions/extraction.js\";\nexport {\n  type TaggingChainOptions,\n  createTaggingChain,\n  createTaggingChainFromZod,\n} from \"./openai_functions/tagging.js\";\nexport {\n  type OpenAPIChainOptions,\n  createOpenAPIChain,\n  convertOpenAPISpecToOpenAIFunctions,\n} from \"./openai_functions/openapi.js\";\n"],"mappings":""}
+{"version":3,"file":"index.js","names":[],"sources":["../../src/chains/index.ts"],"sourcesContent":["export { BaseChain, type ChainInputs } from \"./base.js\";\nexport { LLMChain, type LLMChainInput } from \"./llm_chain.js\";\nexport {\n  APIChain,\n  type APIChainInput,\n  type APIChainOptions,\n} from \"./api/api_chain.js\";\nexport { ConversationChain } from \"./conversation.js\";\nexport {\n  SequentialChain,\n  type SequentialChainInput,\n  SimpleSequentialChain,\n  type SimpleSequentialChainInput,\n} from \"./sequential_chain.js\";\nexport {\n  StuffDocumentsChain,\n  type StuffDocumentsChainInput,\n  MapReduceDocumentsChain,\n  type MapReduceDocumentsChainInput,\n  RefineDocumentsChain,\n  type RefineDocumentsChainInput,\n} from \"./combine_docs_chain.js\";\nexport {\n  ChatVectorDBQAChain,\n  type ChatVectorDBQAChainInput,\n} from \"./chat_vector_db_chain.js\";\nexport {\n  AnalyzeDocumentChain,\n  type AnalyzeDocumentChainInput,\n} from \"./analyze_documents_chain.js\";\nexport { VectorDBQAChain, type VectorDBQAChainInput } from \"./vector_db_qa.js\";\nexport {\n  loadQAChain,\n  type QAChainParams,\n  loadQAStuffChain,\n  type StuffQAChainParams,\n  loadQAMapReduceChain,\n  type MapReduceQAChainParams,\n  loadQARefineChain,\n  type RefineQAChainParams,\n} from \"./question_answering/load.js\";\nexport {\n  loadSummarizationChain,\n  type SummarizationChainParams,\n} from \"./summarization/load.js\";\nexport {\n  ConversationalRetrievalQAChain,\n  type ConversationalRetrievalQAChainInput,\n} from \"./conversational_retrieval_chain.js\";\nexport {\n  RetrievalQAChain,\n  type RetrievalQAChainInput,\n} from \"./retrieval_qa.js\";\nexport {\n  type ConstitutionalChainInput,\n  ConstitutionalChain,\n} from \"./constitutional_ai/constitutional_chain.js\";\nexport {\n  ConstitutionalPrinciple,\n  PRINCIPLES,\n} from \"./constitutional_ai/constitutional_principle.js\";\nexport type {\n  SerializedLLMChain,\n  SerializedSequentialChain,\n  SerializedSimpleSequentialChain,\n  SerializedAnalyzeDocumentChain,\n  SerializedAPIChain,\n  SerializedBaseChain,\n  SerializedChatVectorDBQAChain,\n  SerializedMapReduceDocumentsChain,\n  SerializedStuffDocumentsChain,\n  SerializedVectorDBQAChain,\n  SerializedRefineDocumentsChain,\n} from \"./serde.js\";\nexport {\n  MultiRouteChain,\n  type MultiRouteChainInput,\n  RouterChain,\n} from \"./router/multi_route.js\";\nexport {\n  LLMRouterChain,\n  type LLMRouterChainInput,\n  type RouterOutputSchema,\n} from \"./router/llm_router.js\";\nexport { MultiPromptChain } from \"./router/multi_prompt.js\";\nexport { MultiRetrievalQAChain } from \"./router/multi_retrieval_qa.js\";\nexport { TransformChain, type TransformChainFields } from \"./transform.js\";\nexport {\n  createExtractionChain,\n  createExtractionChainFromZod,\n} from \"./openai_functions/extraction.js\";\nexport {\n  type TaggingChainOptions,\n  createTaggingChain,\n  createTaggingChainFromZod,\n} from \"./openai_functions/tagging.js\";\nexport {\n  type OpenAPIChainOptions,\n  createOpenAPIChain,\n  convertOpenAPISpecToOpenAIFunctions,\n} from \"./openai_functions/openapi.js\";\n"],"mappings":""}

package/dist/chains/llm_chain.d.cts

@@ -1,13 +1,13 @@
 import { SerializedLLMChain } from "./serde.cjs";
 import { BaseChain, ChainInputs } from "./base.cjs";
 import { BaseLanguageModelInput, BaseLanguageModelInterface } from "@langchain/core/language_models/base";
+import { ChainValues } from "@langchain/core/utils/types";
+import { BaseMessage } from "@langchain/core/messages";
+import { Runnable } from "@langchain/core/runnables";
 import { BaseLLMOutputParser } from "@langchain/core/output_parsers";
 import { BasePromptTemplate } from "@langchain/core/prompts";
-import { Runnable } from "@langchain/core/runnables";
-import { ChainValues } from "@langchain/core/utils/types";
 import { BaseCallbackConfig, CallbackManager, CallbackManagerForChainRun, Callbacks } from "@langchain/core/callbacks/manager";
 import { Generation } from "@langchain/core/outputs";
-import { BaseMessage } from "@langchain/core/messages";
 import { BasePromptValueInterface } from "@langchain/core/prompt_values";
 
 //#region src/chains/llm_chain.d.ts

package/dist/chains/openai_functions/base.d.cts

@@ -1,9 +1,9 @@
 import { BaseFunctionCallOptions, BaseLanguageModelInput, FunctionDefinition } from "@langchain/core/language_models/base";
-import { BaseOutputParser } from "@langchain/core/output_parsers";
-import { BasePromptTemplate } from "@langchain/core/prompts";
-import { Runnable, RunnableInterface } from "@langchain/core/runnables";
 import { InputValues, InteropZodObject } from "@langchain/core/utils/types";
 import { BaseMessage } from "@langchain/core/messages";
+import { Runnable, RunnableInterface } from "@langchain/core/runnables";
+import { BaseOutputParser } from "@langchain/core/output_parsers";
+import { BasePromptTemplate } from "@langchain/core/prompts";
 import { JsonSchema7Type } from "@langchain/core/utils/json_schema";
 
 //#region src/chains/openai_functions/base.d.ts

package/dist/chains/openai_functions/extraction.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"extraction.cjs","names":["schema: FunctionParameters","llm: BaseChatModel<BaseFunctionCallOptions>","PromptTemplate","JsonKeyOutputFunctionsParser","LLMChain","schema: InteropZodObject"],"sources":["../../../src/chains/openai_functions/extraction.ts"],"sourcesContent":["import { BaseChatModel } from \"@langchain/core/language_models/chat_models\";\nimport { BaseFunctionCallOptions } from \"@langchain/core/language_models/base\";\nimport { PromptTemplate } from \"@langchain/core/prompts\";\nimport { InteropZodObject } from \"@langchain/core/utils/types\";\nimport type { AIMessageChunk } from \"@langchain/core/messages\";\nimport {\n  type JsonSchema7ObjectType,\n  toJsonSchema,\n} from \"@langchain/core/utils/json_schema\";\nimport {\n  FunctionParameters,\n  JsonKeyOutputFunctionsParser,\n} from \"../../output_parsers/openai_functions.js\";\nimport { LLMChain } from \"../llm_chain.js\";\n\n/**\n * Function that returns an array of extraction functions. These functions\n * are used to extract relevant information from a passage.\n * @param schema The schema of the function parameters.\n * @returns An array of extraction functions.\n */\nfunction getExtractionFunctions(schema: FunctionParameters) {\n  return [\n    {\n      name: \"information_extraction\",\n      description: \"Extracts the relevant information from the passage.\",\n      parameters: {\n        type: \"object\",\n        properties: {\n          info: {\n            type: \"array\",\n            items: {\n              type: schema.type,\n              properties: schema.properties,\n              required: schema.required,\n            },\n          },\n        },\n        required: [\"info\"],\n      },\n    },\n  ];\n}\n\nconst _EXTRACTION_TEMPLATE = `Extract and save the relevant entities mentioned in the following passage together with their properties.\n\nPassage:\n{input}\n`;\n\n/**\n * Function that creates an extraction chain using the provided JSON schema.\n * It sets up the necessary components, such as the prompt, output parser, and tags.\n * @param schema JSON schema of the function parameters.\n * @param llm Must be a ChatOpenAI or AnthropicFunctions model that supports function calling.\n * @returns A LLMChain instance configured to return data matching the schema.\n */\nexport function createExtractionChain(\n  schema: FunctionParameters,\n  llm: BaseChatModel<BaseFunctionCallOptions>\n): LLMChain<object, BaseChatModel<BaseFunctionCallOptions, AIMessageChunk>> {\n  const functions = getExtractionFunctions(schema);\n  const prompt = PromptTemplate.fromTemplate(_EXTRACTION_TEMPLATE);\n  const outputParser = new JsonKeyOutputFunctionsParser({ attrName: \"info\" });\n  return new LLMChain({\n    llm,\n    prompt,\n    llmKwargs: { functions },\n    outputParser,\n    tags: [\"openai_functions\", \"extraction\"],\n  });\n}\n\n/**\n * Function that creates an extraction chain from a Zod schema. It\n * converts the Zod schema to a JSON schema using before creating\n * the extraction chain.\n * @param schema The Zod schema which extracted data should match\n * @param llm Must be a ChatOpenAI or AnthropicFunctions model that supports function calling.\n * @returns A LLMChain instance configured to return data matching the schema.\n */\nexport function createExtractionChainFromZod(\n  // eslint-disable-next-line @typescript-eslint/no-explicit-any\n  schema: InteropZodObject,\n  llm: BaseChatModel<BaseFunctionCallOptions>\n): LLMChain<object, BaseChatModel<BaseFunctionCallOptions, AIMessageChunk>> {\n  return createExtractionChain(\n    toJsonSchema(schema) as JsonSchema7ObjectType,\n    llm\n  );\n}\n"],"mappings":";;;;;;;;;;;;;AAqBA,SAAS,uBAAuBA,QAA4B;AAC1D,QAAO,CACL;EACE,MAAM;EACN,aAAa;EACb,YAAY;GACV,MAAM;GACN,YAAY,EACV,MAAM;IACJ,MAAM;IACN,OAAO;KACL,MAAM,OAAO;KACb,YAAY,OAAO;KACnB,UAAU,OAAO;IAClB;GACF,EACF;GACD,UAAU,CAAC,MAAO;EACnB;CACF,CACF;AACF;AAED,MAAM,uBAAuB,CAAC;;;;AAI9B,CAAC;;;;;;;;AASD,SAAgB,sBACdA,QACAC,KAC0E;CAC1E,MAAM,YAAY,uBAAuB,OAAO;CAChD,MAAM,SAASC,wCAAe,aAAa,qBAAqB;CAChE,MAAM,eAAe,IAAIC,sDAA6B,EAAE,UAAU,OAAQ;AAC1E,QAAO,IAAIC,2BAAS;EAClB;EACA;EACA,WAAW,EAAE,UAAW;EACxB;EACA,MAAM,CAAC,oBAAoB,YAAa;CACzC;AACF;;;;;;;;;AAUD,SAAgB,6BAEdC,QACAJ,KAC0E;AAC1E,QAAO,2EACQ,OAAO,EACpB,IACD;AACF"}
+{"version":3,"file":"extraction.cjs","names":["schema: FunctionParameters","llm: BaseChatModel<BaseFunctionCallOptions>","PromptTemplate","JsonKeyOutputFunctionsParser","LLMChain","schema: InteropZodObject"],"sources":["../../../src/chains/openai_functions/extraction.ts"],"sourcesContent":["import { BaseChatModel } from \"@langchain/core/language_models/chat_models\";\nimport { BaseFunctionCallOptions } from \"@langchain/core/language_models/base\";\nimport { PromptTemplate } from \"@langchain/core/prompts\";\nimport { InteropZodObject } from \"@langchain/core/utils/types\";\nimport type { AIMessageChunk } from \"@langchain/core/messages\";\nimport {\n  type JsonSchema7ObjectType,\n  toJsonSchema,\n} from \"@langchain/core/utils/json_schema\";\nimport {\n  FunctionParameters,\n  JsonKeyOutputFunctionsParser,\n} from \"../../output_parsers/openai_functions.js\";\nimport { LLMChain } from \"../llm_chain.js\";\n\n/**\n * Function that returns an array of extraction functions. These functions\n * are used to extract relevant information from a passage.\n * @param schema The schema of the function parameters.\n * @returns An array of extraction functions.\n */\nfunction getExtractionFunctions(schema: FunctionParameters) {\n  return [\n    {\n      name: \"information_extraction\",\n      description: \"Extracts the relevant information from the passage.\",\n      parameters: {\n        type: \"object\",\n        properties: {\n          info: {\n            type: \"array\",\n            items: {\n              type: schema.type,\n              properties: schema.properties,\n              required: schema.required,\n            },\n          },\n        },\n        required: [\"info\"],\n      },\n    },\n  ];\n}\n\nconst _EXTRACTION_TEMPLATE = `Extract and save the relevant entities mentioned in the following passage together with their properties.\n\nPassage:\n{input}\n`;\n\n/**\n * Function that creates an extraction chain using the provided JSON schema.\n * It sets up the necessary components, such as the prompt, output parser, and tags.\n * @param schema JSON schema of the function parameters.\n * @param llm Must be a ChatOpenAI or AnthropicFunctions model that supports function calling.\n * @returns A LLMChain instance configured to return data matching the schema.\n */\nexport function createExtractionChain(\n  schema: FunctionParameters,\n  llm: BaseChatModel<BaseFunctionCallOptions>\n): LLMChain<object, BaseChatModel<BaseFunctionCallOptions, AIMessageChunk>> {\n  const functions = getExtractionFunctions(schema);\n  const prompt = PromptTemplate.fromTemplate(_EXTRACTION_TEMPLATE);\n  const outputParser = new JsonKeyOutputFunctionsParser({ attrName: \"info\" });\n  return new LLMChain({\n    llm,\n    prompt,\n    llmKwargs: { functions },\n    outputParser,\n    tags: [\"openai_functions\", \"extraction\"],\n  });\n}\n\n/**\n * Function that creates an extraction chain from a Zod schema. It\n * converts the Zod schema to a JSON schema using before creating\n * the extraction chain.\n * @param schema The Zod schema which extracted data should match\n * @param llm Must be a ChatOpenAI or AnthropicFunctions model that supports function calling.\n * @returns A LLMChain instance configured to return data matching the schema.\n */\nexport function createExtractionChainFromZod(\n  schema: InteropZodObject,\n  llm: BaseChatModel<BaseFunctionCallOptions>\n): LLMChain<object, BaseChatModel<BaseFunctionCallOptions, AIMessageChunk>> {\n  return createExtractionChain(\n    toJsonSchema(schema) as JsonSchema7ObjectType,\n    llm\n  );\n}\n"],"mappings":";;;;;;;;;;;;;AAqBA,SAAS,uBAAuBA,QAA4B;AAC1D,QAAO,CACL;EACE,MAAM;EACN,aAAa;EACb,YAAY;GACV,MAAM;GACN,YAAY,EACV,MAAM;IACJ,MAAM;IACN,OAAO;KACL,MAAM,OAAO;KACb,YAAY,OAAO;KACnB,UAAU,OAAO;IAClB;GACF,EACF;GACD,UAAU,CAAC,MAAO;EACnB;CACF,CACF;AACF;AAED,MAAM,uBAAuB,CAAC;;;;AAI9B,CAAC;;;;;;;;AASD,SAAgB,sBACdA,QACAC,KAC0E;CAC1E,MAAM,YAAY,uBAAuB,OAAO;CAChD,MAAM,SAASC,wCAAe,aAAa,qBAAqB;CAChE,MAAM,eAAe,IAAIC,sDAA6B,EAAE,UAAU,OAAQ;AAC1E,QAAO,IAAIC,2BAAS;EAClB;EACA;EACA,WAAW,EAAE,UAAW;EACxB;EACA,MAAM,CAAC,oBAAoB,YAAa;CACzC;AACF;;;;;;;;;AAUD,SAAgB,6BACdC,QACAJ,KAC0E;AAC1E,QAAO,2EACQ,OAAO,EACpB,IACD;AACF"}