langchain 0.1.26 → 0.1.28

package/dist/agents/react/index.cjs

@@ -5,6 +5,7 @@ const runnables_1 = require("@langchain/core/runnables");
 const render_js_1 = require("../../tools/render.cjs");
 const log_js_1 = require("../format_scratchpad/log.cjs");
 const output_parser_js_1 = require("./output_parser.cjs");
+const agent_js_1 = require("../agent.cjs");
 /**
  * Create an agent that uses ReAct prompting.
  * @param params Params required to create the agent. Includes an LLM, tools, and prompt.
@@ -48,7 +49,7 @@ const output_parser_js_1 = require("./output_parser.cjs");
  * });
  * ```
  */
-async function createReactAgent({ llm, tools, prompt, }) {
+async function createReactAgent({ llm, tools, prompt, streamRunnable, }) {
     const missingVariables = ["tools", "tool_names", "agent_scratchpad"].filter((v) => !prompt.inputVariables.includes(v));
     if (missingVariables.length > 0) {
         throw new Error(`Provided prompt is missing required input variables: ${JSON.stringify(missingVariables)}`);
@@ -72,6 +73,10 @@ async function createReactAgent({ llm, tools, prompt, }) {
             toolNames,
         }),
     ]);
-    return agent;
+    return new agent_js_1.RunnableSingleActionAgent({
+        runnable: agent,
+        defaultRunName: "ReactAgent",
+        streamRunnable,
+    });
 }
 exports.createReactAgent = createReactAgent;

package/dist/agents/react/index.d.ts

@@ -1,8 +1,7 @@
 import type { ToolInterface } from "@langchain/core/tools";
 import { BasePromptTemplate } from "@langchain/core/prompts";
 import type { BaseLanguageModelInterface } from "@langchain/core/language_models/base";
-import { RunnableSequence } from "@langchain/core/runnables";
-import { AgentStep } from "@langchain/core/agents";
+import { RunnableSingleActionAgent } from "../agent.js";
 /**
  * Params used by the createXmlAgent function.
  */
@@ -16,6 +15,11 @@ export type CreateReactAgentParams = {
      * `tools`, `tool_names`, and `agent_scratchpad`.
      */
     prompt: BasePromptTemplate;
+    /**
+     * Whether to invoke the underlying model in streaming mode,
+     * allowing streaming of intermediate steps. Defaults to true.
+     */
+    streamRunnable?: boolean;
 };
 /**
  * Create an agent that uses ReAct prompting.
@@ -60,6 +64,4 @@ export type CreateReactAgentParams = {
  * });
  * ```
  */
-export declare function createReactAgent({ llm, tools, prompt, }: CreateReactAgentParams): Promise<RunnableSequence<{
-    steps: AgentStep[];
-}, import("@langchain/core/agents").AgentAction | import("@langchain/core/agents").AgentFinish>>;
+export declare function createReactAgent({ llm, tools, prompt, streamRunnable, }: CreateReactAgentParams): Promise<RunnableSingleActionAgent>;

package/dist/agents/react/index.js

@@ -2,6 +2,7 @@ import { RunnablePassthrough, RunnableSequence, } from "@langchain/core/runnable
 import { renderTextDescription } from "../../tools/render.js";
 import { formatLogToString } from "../format_scratchpad/log.js";
 import { ReActSingleInputOutputParser } from "./output_parser.js";
+import { RunnableSingleActionAgent } from "../agent.js";
 /**
  * Create an agent that uses ReAct prompting.
  * @param params Params required to create the agent. Includes an LLM, tools, and prompt.
@@ -45,7 +46,7 @@ import { ReActSingleInputOutputParser } from "./output_parser.js";
  * });
  * ```
  */
-export async function createReactAgent({ llm, tools, prompt, }) {
+export async function createReactAgent({ llm, tools, prompt, streamRunnable, }) {
     const missingVariables = ["tools", "tool_names", "agent_scratchpad"].filter((v) => !prompt.inputVariables.includes(v));
     if (missingVariables.length > 0) {
         throw new Error(`Provided prompt is missing required input variables: ${JSON.stringify(missingVariables)}`);
@@ -69,5 +70,9 @@ export async function createReactAgent({ llm, tools, prompt, }) {
             toolNames,
         }),
     ]);
-    return agent;
+    return new RunnableSingleActionAgent({
+        runnable: agent,
+        defaultRunName: "ReactAgent",
+        streamRunnable,
+    });
 }

package/dist/agents/structured_chat/index.cjs

@@ -210,7 +210,7 @@ exports.StructuredChatAgent = StructuredChatAgent;
  * });
  * ```
  */
-async function createStructuredChatAgent({ llm, tools, prompt, }) {
+async function createStructuredChatAgent({ llm, tools, prompt, streamRunnable, }) {
     const missingVariables = ["tools", "tool_names", "agent_scratchpad"].filter((v) => !prompt.inputVariables.includes(v));
     if (missingVariables.length > 0) {
         throw new Error(`Provided prompt is missing required input variables: ${JSON.stringify(missingVariables)}`);
@@ -234,6 +234,10 @@ async function createStructuredChatAgent({ llm, tools, prompt, }) {
             toolNames,
         }),
     ]);
-    return agent;
+    return new agent_js_1.RunnableSingleActionAgent({
+        runnable: agent,
+        defaultRunName: "StructuredChatAgent",
+        streamRunnable,
+    });
 }
 exports.createStructuredChatAgent = createStructuredChatAgent;

package/dist/agents/structured_chat/index.d.ts

@@ -1,11 +1,10 @@
 import type { StructuredToolInterface } from "@langchain/core/tools";
 import type { BaseLanguageModelInterface } from "@langchain/core/language_models/base";
-import { RunnableSequence } from "@langchain/core/runnables";
 import type { BasePromptTemplate } from "@langchain/core/prompts";
 import { BaseMessagePromptTemplate, ChatPromptTemplate } from "@langchain/core/prompts";
 import { AgentStep } from "@langchain/core/agents";
 import { Optional } from "../../types/type-utils.js";
-import { Agent, AgentArgs, OutputParserArgs } from "../agent.js";
+import { Agent, AgentArgs, OutputParserArgs, RunnableSingleActionAgent } from "../agent.js";
 import { AgentInput } from "../types.js";
 import { StructuredChatOutputParserWithRetries } from "./outputParser.js";
 /**
@@ -106,6 +105,11 @@ export type CreateStructuredChatAgentParams = {
      * `tools`, `tool_names`, and `agent_scratchpad`.
      */
     prompt: BasePromptTemplate;
+    /**
+     * Whether to invoke the underlying model in streaming mode,
+     * allowing streaming of intermediate steps. Defaults to true.
+     */
+    streamRunnable?: boolean;
 };
 /**
  * Create an agent aimed at supporting tools with multiple inputs.
@@ -163,6 +167,4 @@ export type CreateStructuredChatAgentParams = {
  * });
  * ```
  */
-export declare function createStructuredChatAgent({ llm, tools, prompt, }: CreateStructuredChatAgentParams): Promise<RunnableSequence<{
-    steps: AgentStep[];
-}, import("@langchain/core/agents").AgentAction | import("@langchain/core/agents").AgentFinish>>;
+export declare function createStructuredChatAgent({ llm, tools, prompt, streamRunnable, }: CreateStructuredChatAgentParams): Promise<RunnableSingleActionAgent>;

package/dist/agents/structured_chat/index.js

@@ -2,7 +2,7 @@ import { zodToJsonSchema } from "zod-to-json-schema";
 import { RunnablePassthrough, RunnableSequence, } from "@langchain/core/runnables";
 import { ChatPromptTemplate, HumanMessagePromptTemplate, SystemMessagePromptTemplate, PromptTemplate, } from "@langchain/core/prompts";
 import { LLMChain } from "../../chains/llm_chain.js";
-import { Agent } from "../agent.js";
+import { Agent, RunnableSingleActionAgent, } from "../agent.js";
 import { StructuredChatOutputParserWithRetries } from "./outputParser.js";
 import { FORMAT_INSTRUCTIONS, PREFIX, SUFFIX } from "./prompt.js";
 import { renderTextDescriptionAndArgs } from "../../tools/render.js";
@@ -206,7 +206,7 @@ export class StructuredChatAgent extends Agent {
  * });
  * ```
  */
-export async function createStructuredChatAgent({ llm, tools, prompt, }) {
+export async function createStructuredChatAgent({ llm, tools, prompt, streamRunnable, }) {
     const missingVariables = ["tools", "tool_names", "agent_scratchpad"].filter((v) => !prompt.inputVariables.includes(v));
     if (missingVariables.length > 0) {
         throw new Error(`Provided prompt is missing required input variables: ${JSON.stringify(missingVariables)}`);
@@ -230,5 +230,9 @@ export async function createStructuredChatAgent({ llm, tools, prompt, }) {
             toolNames,
         }),
     ]);
-    return agent;
+    return new RunnableSingleActionAgent({
+        runnable: agent,
+        defaultRunName: "StructuredChatAgent",
+        streamRunnable,
+    });
 }

package/dist/agents/types.d.ts

@@ -16,16 +16,34 @@ export interface AgentInput {
     allowedTools?: string[];
 }
 /**
- * Interface defining the input for creating an agent that uses runnables.
- * It includes the Runnable instance, and an optional list of stop strings.
+ * Interface defining the input for creating a single action agent
+ * that uses runnables.
  */
-export interface RunnableAgentInput {
+export interface RunnableSingleActionAgentInput {
+    runnable: Runnable<ChainValues & {
+        agent_scratchpad?: string | BaseMessage[];
+        stop?: string[];
+    }, AgentAction | AgentFinish>;
+    streamRunnable?: boolean;
+    defaultRunName?: string;
+}
+/**
+ * Interface defining the input for creating a multi-action agent that uses
+ * runnables. It includes the Runnable instance, and an optional list of
+ * stop strings.
+ */
+export interface RunnableMultiActionAgentInput {
     runnable: Runnable<ChainValues & {
         agent_scratchpad?: string | BaseMessage[];
         stop?: string[];
     }, AgentAction[] | AgentAction | AgentFinish>;
+    streamRunnable?: boolean;
+    defaultRunName?: string;
     stop?: string[];
 }
+/** @deprecated Renamed to RunnableMultiActionAgentInput. */
+export interface RunnableAgentInput extends RunnableMultiActionAgentInput {
+}
 /**
  * Abstract class representing an output parser specifically for agent
  * actions and finishes in LangChain. It extends the `BaseOutputParser`

package/dist/agents/xml/index.cjs

@@ -156,7 +156,7 @@ exports.XMLAgent = XMLAgent;
  * });
  * ```
  */
-async function createXmlAgent({ llm, tools, prompt, }) {
+async function createXmlAgent({ llm, tools, prompt, streamRunnable, }) {
     const missingVariables = ["tools", "agent_scratchpad"].filter((v) => !prompt.inputVariables.includes(v));
     if (missingVariables.length > 0) {
         throw new Error(`Provided prompt is missing required input variables: ${JSON.stringify(missingVariables)}`);
@@ -176,6 +176,10 @@ async function createXmlAgent({ llm, tools, prompt, }) {
         llmWithStop,
         new output_parser_js_1.XMLAgentOutputParser(),
     ]);
-    return agent;
+    return new agent_js_1.RunnableSingleActionAgent({
+        runnable: agent,
+        defaultRunName: "XMLAgent",
+        streamRunnable,
+    });
 }
 exports.createXmlAgent = createXmlAgent;

package/dist/agents/xml/index.d.ts

@@ -1,13 +1,12 @@
 import type { BaseLanguageModelInterface } from "@langchain/core/language_models/base";
 import type { ToolInterface } from "@langchain/core/tools";
-import { RunnableSequence } from "@langchain/core/runnables";
 import type { BasePromptTemplate } from "@langchain/core/prompts";
 import { AgentStep, AgentAction, AgentFinish } from "@langchain/core/agents";
 import { ChainValues } from "@langchain/core/utils/types";
 import { ChatPromptTemplate } from "@langchain/core/prompts";
 import { CallbackManager } from "@langchain/core/callbacks/manager";
 import { LLMChain } from "../../chains/llm_chain.js";
-import { AgentArgs, BaseSingleActionAgent } from "../agent.js";
+import { AgentArgs, BaseSingleActionAgent, RunnableSingleActionAgent } from "../agent.js";
 import { XMLAgentOutputParser } from "./output_parser.js";
 /**
  * Interface for the input to the XMLAgent class.
@@ -62,6 +61,11 @@ export type CreateXmlAgentParams = {
      * `tools` and `agent_scratchpad`.
      */
     prompt: BasePromptTemplate;
+    /**
+     * Whether to invoke the underlying model in streaming mode,
+     * allowing streaming of intermediate steps. Defaults to true.
+     */
+    streamRunnable?: boolean;
 };
 /**
  * Create an agent that uses XML to format its logic.
@@ -113,6 +117,4 @@ export type CreateXmlAgentParams = {
  * });
  * ```
  */
-export declare function createXmlAgent({ llm, tools, prompt, }: CreateXmlAgentParams): Promise<RunnableSequence<{
-    steps: AgentStep[];
-}, AgentAction | AgentFinish>>;
+export declare function createXmlAgent({ llm, tools, prompt, streamRunnable, }: CreateXmlAgentParams): Promise<RunnableSingleActionAgent>;

package/dist/agents/xml/index.js

@@ -1,7 +1,7 @@
 import { RunnablePassthrough, RunnableSequence, } from "@langchain/core/runnables";
 import { AIMessagePromptTemplate, ChatPromptTemplate, HumanMessagePromptTemplate, } from "@langchain/core/prompts";
 import { LLMChain } from "../../chains/llm_chain.js";
-import { BaseSingleActionAgent } from "../agent.js";
+import { BaseSingleActionAgent, RunnableSingleActionAgent, } from "../agent.js";
 import { AGENT_INSTRUCTIONS } from "./prompt.js";
 import { XMLAgentOutputParser } from "./output_parser.js";
 import { renderTextDescription } from "../../tools/render.js";
@@ -152,7 +152,7 @@ export class XMLAgent extends BaseSingleActionAgent {
  * });
  * ```
  */
-export async function createXmlAgent({ llm, tools, prompt, }) {
+export async function createXmlAgent({ llm, tools, prompt, streamRunnable, }) {
     const missingVariables = ["tools", "agent_scratchpad"].filter((v) => !prompt.inputVariables.includes(v));
     if (missingVariables.length > 0) {
         throw new Error(`Provided prompt is missing required input variables: ${JSON.stringify(missingVariables)}`);
@@ -172,5 +172,9 @@ export async function createXmlAgent({ llm, tools, prompt, }) {
         llmWithStop,
         new XMLAgentOutputParser(),
     ]);
-    return agent;
+    return new RunnableSingleActionAgent({
+        runnable: agent,
+        defaultRunName: "XMLAgent",
+        streamRunnable,
+    });
 }

package/dist/document_loaders/fs/unstructured.cjs

@@ -116,6 +116,30 @@ class UnstructuredLoader extends base_js_1.BaseDocumentLoader {
             writable: true,
             value: void 0
         });
+        Object.defineProperty(this, "multiPageSections", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "combineUnderNChars", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "newAfterNChars", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "maxCharacters", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
         // Temporary shim to avoid breaking existing users
         // Remove when API keys are enforced by Unstructured and existing code will break anyway
         const isLegacySyntax = typeof optionsOrLegacyFilePath === "string";
@@ -138,6 +162,10 @@ class UnstructuredLoader extends base_js_1.BaseDocumentLoader {
             this.hiResModelName = options.hiResModelName;
             this.includePageBreaks = options.includePageBreaks;
             this.chunkingStrategy = options.chunkingStrategy;
+            this.multiPageSections = options.multiPageSections;
+            this.combineUnderNChars = options.combineUnderNChars;
+            this.newAfterNChars = options.newAfterNChars;
+            this.maxCharacters = options.maxCharacters;
         }
     }
     async _partition() {
@@ -177,6 +205,18 @@ class UnstructuredLoader extends base_js_1.BaseDocumentLoader {
         if (this.chunkingStrategy) {
             formData.append("chunking_strategy", this.chunkingStrategy);
         }
+        if (this.multiPageSections !== undefined) {
+            formData.append("multipage_sections", this.multiPageSections ? "true" : "false");
+        }
+        if (this.combineUnderNChars !== undefined) {
+            formData.append("combine_under_n_chars", String(this.combineUnderNChars));
+        }
+        if (this.newAfterNChars !== undefined) {
+            formData.append("new_after_n_chars", String(this.newAfterNChars));
+        }
+        if (this.maxCharacters !== undefined) {
+            formData.append("max_characters", String(this.maxCharacters));
+        }
         const headers = {
             "UNSTRUCTURED-API-KEY": this.apiKey ?? "",
         };

package/dist/document_loaders/fs/unstructured.d.ts

@@ -54,6 +54,10 @@ export type UnstructuredLoaderOptions = {
     hiResModelName?: StringWithAutocomplete<HiResModelName>;
     includePageBreaks?: boolean;
     chunkingStrategy?: StringWithAutocomplete<ChunkingStrategy>;
+    multiPageSections?: boolean;
+    combineUnderNChars?: number;
+    newAfterNChars?: number;
+    maxCharacters?: number;
 };
 type UnstructuredDirectoryLoaderOptions = UnstructuredLoaderOptions & {
     recursive?: boolean;
@@ -81,6 +85,10 @@ export declare class UnstructuredLoader extends BaseDocumentLoader {
     private hiResModelName?;
     private includePageBreaks?;
     private chunkingStrategy?;
+    private multiPageSections?;
+    private combineUnderNChars?;
+    private newAfterNChars?;
+    private maxCharacters?;
     constructor(filePathOrLegacyApiUrl: string, optionsOrLegacyFilePath?: UnstructuredLoaderOptions | string);
     _partition(): Promise<Element[]>;
     load(): Promise<Document[]>;

package/dist/document_loaders/fs/unstructured.js

@@ -112,6 +112,30 @@ export class UnstructuredLoader extends BaseDocumentLoader {
             writable: true,
             value: void 0
         });
+        Object.defineProperty(this, "multiPageSections", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "combineUnderNChars", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "newAfterNChars", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "maxCharacters", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
         // Temporary shim to avoid breaking existing users
         // Remove when API keys are enforced by Unstructured and existing code will break anyway
         const isLegacySyntax = typeof optionsOrLegacyFilePath === "string";
@@ -134,6 +158,10 @@ export class UnstructuredLoader extends BaseDocumentLoader {
             this.hiResModelName = options.hiResModelName;
             this.includePageBreaks = options.includePageBreaks;
             this.chunkingStrategy = options.chunkingStrategy;
+            this.multiPageSections = options.multiPageSections;
+            this.combineUnderNChars = options.combineUnderNChars;
+            this.newAfterNChars = options.newAfterNChars;
+            this.maxCharacters = options.maxCharacters;
         }
     }
     async _partition() {
@@ -173,6 +201,18 @@ export class UnstructuredLoader extends BaseDocumentLoader {
         if (this.chunkingStrategy) {
             formData.append("chunking_strategy", this.chunkingStrategy);
         }
+        if (this.multiPageSections !== undefined) {
+            formData.append("multipage_sections", this.multiPageSections ? "true" : "false");
+        }
+        if (this.combineUnderNChars !== undefined) {
+            formData.append("combine_under_n_chars", String(this.combineUnderNChars));
+        }
+        if (this.newAfterNChars !== undefined) {
+            formData.append("new_after_n_chars", String(this.newAfterNChars));
+        }
+        if (this.maxCharacters !== undefined) {
+            formData.append("max_characters", String(this.maxCharacters));
+        }
         const headers = {
             "UNSTRUCTURED-API-KEY": this.apiKey ?? "",
         };

package/dist/document_loaders/web/gitbook.cjs

@@ -23,6 +23,13 @@ class GitbookLoader extends cheerio_js_1.CheerioWebBaseLoader {
             writable: true,
             value: false
         });
+        Object.defineProperty(this, "baseUrl", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        this.baseUrl = webPath;
         this.webPath = path;
         this.shouldLoadAllPaths =
             params.shouldLoadAllPaths ?? this.shouldLoadAllPaths;
@@ -84,9 +91,10 @@ class GitbookLoader extends cheerio_js_1.CheerioWebBaseLoader {
             .map((element) => $(element).text());
         const documents = [];
         for (const url of urls) {
-            console.log(`Fetching text from ${url}`);
-            const html = await GitbookLoader._scrape(url, this.caller, this.timeout);
-            documents.push(...this.loadPath(html, url));
+            const buildUrl = url.includes(this.baseUrl) ? url : this.baseUrl + url;
+            console.log(`Fetching text from ${buildUrl}`);
+            const html = await GitbookLoader._scrape(buildUrl, this.caller, this.timeout);
+            documents.push(...this.loadPath(html, buildUrl));
         }
         console.log(`Fetched ${documents.length} documents.`);
         return documents;

package/dist/document_loaders/web/gitbook.d.ts

@@ -15,6 +15,7 @@ interface GitbookLoaderParams {
 export declare class GitbookLoader extends CheerioWebBaseLoader {
     webPath: string;
     shouldLoadAllPaths: boolean;
+    private readonly baseUrl;
     constructor(webPath: string, params?: GitbookLoaderParams);
     /**
      * Method that scrapes the web document using Cheerio and loads the

package/dist/document_loaders/web/gitbook.js

@@ -20,6 +20,13 @@ export class GitbookLoader extends CheerioWebBaseLoader {
             writable: true,
             value: false
         });
+        Object.defineProperty(this, "baseUrl", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        this.baseUrl = webPath;
         this.webPath = path;
         this.shouldLoadAllPaths =
             params.shouldLoadAllPaths ?? this.shouldLoadAllPaths;
@@ -81,9 +88,10 @@ export class GitbookLoader extends CheerioWebBaseLoader {
             .map((element) => $(element).text());
         const documents = [];
         for (const url of urls) {
-            console.log(`Fetching text from ${url}`);
-            const html = await GitbookLoader._scrape(url, this.caller, this.timeout);
-            documents.push(...this.loadPath(html, url));
+            const buildUrl = url.includes(this.baseUrl) ? url : this.baseUrl + url;
+            console.log(`Fetching text from ${buildUrl}`);
+            const html = await GitbookLoader._scrape(buildUrl, this.caller, this.timeout);
+            documents.push(...this.loadPath(html, buildUrl));
         }
         console.log(`Fetched ${documents.length} documents.`);
         return documents;

package/dist/output_parsers/fix.cjs

@@ -4,6 +4,9 @@ exports.OutputFixingParser = void 0;
 const output_parsers_1 = require("@langchain/core/output_parsers");
 const llm_chain_js_1 = require("../chains/llm_chain.cjs");
 const prompts_js_1 = require("./prompts.cjs");
+function isLLMChain(x) {
+    return (x.prompt !== undefined && x.llm !== undefined);
+}
 /**
  * Class that extends the BaseOutputParser to handle situations where the
  * initial parsing attempt fails. It contains a retryChain for retrying
@@ -70,13 +73,22 @@ class OutputFixingParser extends output_parsers_1.BaseOutputParser {
         catch (e) {
             // eslint-disable-next-line no-instanceof/no-instanceof
             if (e instanceof output_parsers_1.OutputParserException) {
-                const result = await this.retryChain.call({
+                const retryInput = {
                     instructions: this.parser.getFormatInstructions(),
                     completion,
                     error: e,
-                }, callbacks);
-                const newCompletion = result[this.retryChain.outputKey];
-                return this.parser.parse(newCompletion);
+                };
+                if (isLLMChain(this.retryChain)) {
+                    const result = await this.retryChain.call(retryInput, callbacks);
+                    const newCompletion = result[this.retryChain.outputKey];
+                    return this.parser.parse(newCompletion, callbacks);
+                }
+                else {
+                    const result = await this.retryChain.invoke(retryInput, {
+                        callbacks,
+                    });
+                    return result;
+                }
             }
             throw e;
         }

package/dist/output_parsers/fix.d.ts

@@ -1,8 +1,14 @@
 import type { BaseLanguageModelInterface } from "@langchain/core/language_models/base";
 import { Callbacks } from "@langchain/core/callbacks/manager";
-import { BaseOutputParser } from "@langchain/core/output_parsers";
+import { BaseOutputParser, OutputParserException } from "@langchain/core/output_parsers";
 import { BasePromptTemplate } from "@langchain/core/prompts";
+import { Runnable } from "@langchain/core/runnables";
 import { LLMChain } from "../chains/llm_chain.js";
+interface OutputFixingParserRetryInput {
+    instructions: string;
+    completion: string;
+    error: OutputParserException;
+}
 /**
  * Class that extends the BaseOutputParser to handle situations where the
  * initial parsing attempt fails. It contains a retryChain for retrying
@@ -13,7 +19,7 @@ export declare class OutputFixingParser<T> extends BaseOutputParser<T> {
     lc_namespace: string[];
     lc_serializable: boolean;
     parser: BaseOutputParser<T>;
-    retryChain: LLMChain;
+    retryChain: LLMChain | Runnable<OutputFixingParserRetryInput, T>;
     /**
      * Static method to create a new instance of OutputFixingParser using a
      * given language model, parser, and optional fields.
@@ -27,7 +33,7 @@ export declare class OutputFixingParser<T> extends BaseOutputParser<T> {
     }): OutputFixingParser<T>;
     constructor({ parser, retryChain, }: {
         parser: BaseOutputParser<T>;
-        retryChain: LLMChain;
+        retryChain: LLMChain | Runnable<OutputFixingParserRetryInput, T>;
     });
     /**
      * Method to parse the completion using the parser. If the initial parsing
@@ -44,3 +50,4 @@ export declare class OutputFixingParser<T> extends BaseOutputParser<T> {
      */
     getFormatInstructions(): string;
 }
+export {};

package/dist/output_parsers/fix.js

@@ -1,6 +1,9 @@
 import { BaseOutputParser, OutputParserException, } from "@langchain/core/output_parsers";
 import { LLMChain } from "../chains/llm_chain.js";
 import { NAIVE_FIX_PROMPT } from "./prompts.js";
+function isLLMChain(x) {
+    return (x.prompt !== undefined && x.llm !== undefined);
+}
 /**
  * Class that extends the BaseOutputParser to handle situations where the
  * initial parsing attempt fails. It contains a retryChain for retrying
@@ -67,13 +70,22 @@ export class OutputFixingParser extends BaseOutputParser {
         catch (e) {
             // eslint-disable-next-line no-instanceof/no-instanceof
             if (e instanceof OutputParserException) {
-                const result = await this.retryChain.call({
+                const retryInput = {
                     instructions: this.parser.getFormatInstructions(),
                     completion,
                     error: e,
-                }, callbacks);
-                const newCompletion = result[this.retryChain.outputKey];
-                return this.parser.parse(newCompletion);
+                };
+                if (isLLMChain(this.retryChain)) {
+                    const result = await this.retryChain.call(retryInput, callbacks);
+                    const newCompletion = result[this.retryChain.outputKey];
+                    return this.parser.parse(newCompletion, callbacks);
+                }
+                else {
+                    const result = await this.retryChain.invoke(retryInput, {
+                        callbacks,
+                    });
+                    return result;
+                }
             }
             throw e;
         }

package/dist/retrievers/parent_document.cjs

@@ -153,9 +153,10 @@ class ParentDocumentRetriever extends multi_vector_js_1.MultiVectorRetriever {
      * This can be false if and only if `ids` are provided. You may want
      *   to set this to False if the documents are already in the docstore
      *   and you don't want to re-add them.
+     * @param config.chunkHeaderOptions Object with options for adding Contextual chunk headers
      */
     async addDocuments(docs, config) {
-        const { ids, addToDocstore = true } = config ?? {};
+        const { ids, addToDocstore = true, childDocChunkHeaderOptions = {}, } = config ?? {};
         const parentDocs = this.parentSplitter
             ? await this.parentSplitter.splitDocuments(docs)
             : docs;
@@ -177,7 +178,7 @@ class ParentDocumentRetriever extends multi_vector_js_1.MultiVectorRetriever {
         for (let i = 0; i < parentDocs.length; i += 1) {
             const parentDoc = parentDocs[i];
             const parentDocId = parentDocIds[i];
-            const subDocs = await this.childSplitter.splitDocuments([parentDoc]);
+            const subDocs = await this.childSplitter.splitDocuments([parentDoc], childDocChunkHeaderOptions);
             const taggedSubDocs = subDocs.map((subDoc) => new documents_1.Document({
                 pageContent: subDoc.pageContent,
                 metadata: { ...subDoc.metadata, [this.idKey]: parentDocId },