langchain 0.0.178 → 0.0.180

package/dist/output_parsers/json.cjs

@@ -0,0 +1,77 @@
+"use strict";
+// Adapted from https://github.com/KillianLucas/open-interpreter/blob/main/interpreter/utils/parse_partial_json.py
+// MIT License
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.parsePartialJson = void 0;
+function parsePartialJson(s) {
+    // Attempt to parse the string as-is.
+    try {
+        return JSON.parse(s);
+    }
+    catch (error) {
+        // Pass
+    }
+    // Initialize variables.
+    let new_s = "";
+    const stack = [];
+    let isInsideString = false;
+    let escaped = false;
+    // Process each character in the string one at a time.
+    for (let char of s) {
+        if (isInsideString) {
+            if (char === '"' && !escaped) {
+                isInsideString = false;
+            }
+            else if (char === "\n" && !escaped) {
+                char = "\\n"; // Replace the newline character with the escape sequence.
+            }
+            else if (char === "\\") {
+                escaped = !escaped;
+            }
+            else {
+                escaped = false;
+            }
+        }
+        else {
+            if (char === '"') {
+                isInsideString = true;
+                escaped = false;
+            }
+            else if (char === "{") {
+                stack.push("}");
+            }
+            else if (char === "[") {
+                stack.push("]");
+            }
+            else if (char === "}" || char === "]") {
+                if (stack && stack[stack.length - 1] === char) {
+                    stack.pop();
+                }
+                else {
+                    // Mismatched closing character; the input is malformed.
+                    return null;
+                }
+            }
+        }
+        // Append the processed character to the new string.
+        new_s += char;
+    }
+    // If we're still inside a string at the end of processing,
+    // we need to close the string.
+    if (isInsideString) {
+        new_s += '"';
+    }
+    // Close any remaining open structures in the reverse order that they were opened.
+    for (let i = stack.length - 1; i >= 0; i -= 1) {
+        new_s += stack[i];
+    }
+    // Attempt to parse the modified string as JSON.
+    try {
+        return JSON.parse(new_s);
+    }
+    catch (error) {
+        // If we still can't parse the string as JSON, return null to indicate failure.
+        return null;
+    }
+}
+exports.parsePartialJson = parsePartialJson;

package/dist/output_parsers/json.d.ts

@@ -0,0 +1 @@
+export declare function parsePartialJson(s: string): any;

package/dist/output_parsers/json.js

@@ -0,0 +1,73 @@
+// Adapted from https://github.com/KillianLucas/open-interpreter/blob/main/interpreter/utils/parse_partial_json.py
+// MIT License
+export function parsePartialJson(s) {
+    // Attempt to parse the string as-is.
+    try {
+        return JSON.parse(s);
+    }
+    catch (error) {
+        // Pass
+    }
+    // Initialize variables.
+    let new_s = "";
+    const stack = [];
+    let isInsideString = false;
+    let escaped = false;
+    // Process each character in the string one at a time.
+    for (let char of s) {
+        if (isInsideString) {
+            if (char === '"' && !escaped) {
+                isInsideString = false;
+            }
+            else if (char === "\n" && !escaped) {
+                char = "\\n"; // Replace the newline character with the escape sequence.
+            }
+            else if (char === "\\") {
+                escaped = !escaped;
+            }
+            else {
+                escaped = false;
+            }
+        }
+        else {
+            if (char === '"') {
+                isInsideString = true;
+                escaped = false;
+            }
+            else if (char === "{") {
+                stack.push("}");
+            }
+            else if (char === "[") {
+                stack.push("]");
+            }
+            else if (char === "}" || char === "]") {
+                if (stack && stack[stack.length - 1] === char) {
+                    stack.pop();
+                }
+                else {
+                    // Mismatched closing character; the input is malformed.
+                    return null;
+                }
+            }
+        }
+        // Append the processed character to the new string.
+        new_s += char;
+    }
+    // If we're still inside a string at the end of processing,
+    // we need to close the string.
+    if (isInsideString) {
+        new_s += '"';
+    }
+    // Close any remaining open structures in the reverse order that they were opened.
+    for (let i = stack.length - 1; i >= 0; i -= 1) {
+        new_s += stack[i];
+    }
+    // Attempt to parse the modified string as JSON.
+    try {
+        return JSON.parse(new_s);
+    }
+    catch (error) {
+        // If we still can't parse the string as JSON, return null to indicate failure.
+        return null;
+    }
+}

package/dist/output_parsers/openai_functions.cjs

@@ -2,6 +2,8 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.JsonKeyOutputFunctionsParser = exports.JsonOutputFunctionsParser = exports.OutputFunctionsParser = void 0;
 const output_parser_js_1 = require("../schema/output_parser.cjs");
+const index_js_1 = require("../util/fast-json-patch/index.cjs");
+const json_js_1 = require("./json.cjs");
 /**
  * Class for parsing the output of an LLM. Can be configured to return
  * only the arguments of the function call in the output.
@@ -63,12 +65,12 @@ exports.OutputFunctionsParser = OutputFunctionsParser;
  * Class for parsing the output of an LLM into a JSON object. Uses an
  * instance of `OutputFunctionsParser` to parse the output.
  */
-class JsonOutputFunctionsParser extends output_parser_js_1.BaseLLMOutputParser {
+class JsonOutputFunctionsParser extends output_parser_js_1.BaseCumulativeTransformOutputParser {
     static lc_name() {
         return "JsonOutputFunctionsParser";
     }
     constructor(config) {
-        super();
+        super(config);
         Object.defineProperty(this, "lc_namespace", {
             enumerable: true,
             configurable: true,
@@ -96,6 +98,31 @@ class JsonOutputFunctionsParser extends output_parser_js_1.BaseLLMOutputParser {
         this.argsOnly = config?.argsOnly ?? this.argsOnly;
         this.outputParser = new OutputFunctionsParser(config);
     }
+    _diff(prev, next) {
+        if (!next) {
+            return undefined;
+        }
+        const ops = (0, index_js_1.compare)(prev ?? {}, next);
+        return ops;
+    }
+    async parsePartialResult(generations) {
+        const generation = generations[0];
+        if (!generation.message) {
+            return undefined;
+        }
+        const { message } = generation;
+        const functionCall = message.additional_kwargs.function_call;
+        if (!functionCall) {
+            return undefined;
+        }
+        if (this.argsOnly) {
+            return (0, json_js_1.parsePartialJson)(functionCall.arguments);
+        }
+        return {
+            ...functionCall,
+            arguments: (0, json_js_1.parsePartialJson)(functionCall.arguments),
+        };
+    }
     /**
      * Parses the output and returns a JSON object. If `argsOnly` is true,
      * only the arguments of the function call are returned.
@@ -114,6 +141,14 @@ class JsonOutputFunctionsParser extends output_parser_js_1.BaseLLMOutputParser {
         parsedResult.arguments = JSON.parse(parsedResult.arguments);
         return parsedResult;
     }
+    // This method would be called by the default implementation of `parse_result`
+    // but we're overriding that method so it's not needed.
+    async parse(_text) {
+        throw new Error("Not implemented.");
+    }
+    getFormatInstructions() {
+        return "";
+    }
 }
 exports.JsonOutputFunctionsParser = JsonOutputFunctionsParser;
 /**

package/dist/output_parsers/openai_functions.d.ts

@@ -1,7 +1,8 @@
 import { JsonSchema7ObjectType } from "zod-to-json-schema/src/parsers/object.js";
 import { ChatGeneration, Generation } from "../schema/index.js";
 import { Optional } from "../types/type-utils.js";
-import { BaseLLMOutputParser } from "../schema/output_parser.js";
+import { BaseCumulativeTransformOutputParser, type BaseCumulativeTransformOutputParserInput, BaseLLMOutputParser } from "../schema/output_parser.js";
+import { type Operation as JSONPatchOperation } from "../util/fast-json-patch/index.js";
 /**
  * Represents optional parameters for a function in a JSON Schema.
  */
@@ -16,7 +17,7 @@ export declare class OutputFunctionsParser extends BaseLLMOutputParser<string> {
     lc_serializable: boolean;
     argsOnly: boolean;
     constructor(config?: {
-        argsOnly: boolean;
+        argsOnly?: boolean;
     });
     /**
      * Parses the output and returns a string representation of the function
@@ -30,15 +31,17 @@ export declare class OutputFunctionsParser extends BaseLLMOutputParser<string> {
  * Class for parsing the output of an LLM into a JSON object. Uses an
  * instance of `OutputFunctionsParser` to parse the output.
  */
-export declare class JsonOutputFunctionsParser extends BaseLLMOutputParser<object> {
+export declare class JsonOutputFunctionsParser extends BaseCumulativeTransformOutputParser<object> {
     static lc_name(): string;
     lc_namespace: string[];
     lc_serializable: boolean;
     outputParser: OutputFunctionsParser;
     argsOnly: boolean;
     constructor(config?: {
-        argsOnly: boolean;
-    });
+        argsOnly?: boolean;
+    } & BaseCumulativeTransformOutputParserInput);
+    protected _diff(prev: JSONPatchOperation | undefined, next: JSONPatchOperation): object | undefined;
+    parsePartialResult(generations: ChatGeneration[]): Promise<object | undefined>;
     /**
      * Parses the output and returns a JSON object. If `argsOnly` is true,
      * only the arguments of the function call are returned.
@@ -46,6 +49,8 @@ export declare class JsonOutputFunctionsParser extends BaseLLMOutputParser<objec
      * @returns A JSON object representation of the function call or its arguments.
      */
     parseResult(generations: Generation[] | ChatGeneration[]): Promise<object>;
+    parse(_text: string): Promise<object>;
+    getFormatInstructions(): string;
 }
 /**
  * Class for parsing the output of an LLM into a JSON object and returning

package/dist/output_parsers/openai_functions.js

@@ -1,4 +1,6 @@
-import { BaseLLMOutputParser } from "../schema/output_parser.js";
+import { BaseCumulativeTransformOutputParser, BaseLLMOutputParser, } from "../schema/output_parser.js";
+import { compare, } from "../util/fast-json-patch/index.js";
+import { parsePartialJson } from "./json.js";
 /**
  * Class for parsing the output of an LLM. Can be configured to return
  * only the arguments of the function call in the output.
@@ -59,12 +61,12 @@ export class OutputFunctionsParser extends BaseLLMOutputParser {
  * Class for parsing the output of an LLM into a JSON object. Uses an
  * instance of `OutputFunctionsParser` to parse the output.
  */
-export class JsonOutputFunctionsParser extends BaseLLMOutputParser {
+export class JsonOutputFunctionsParser extends BaseCumulativeTransformOutputParser {
     static lc_name() {
         return "JsonOutputFunctionsParser";
     }
     constructor(config) {
-        super();
+        super(config);
         Object.defineProperty(this, "lc_namespace", {
             enumerable: true,
             configurable: true,
@@ -92,6 +94,31 @@ export class JsonOutputFunctionsParser extends BaseLLMOutputParser {
         this.argsOnly = config?.argsOnly ?? this.argsOnly;
         this.outputParser = new OutputFunctionsParser(config);
     }
+    _diff(prev, next) {
+        if (!next) {
+            return undefined;
+        }
+        const ops = compare(prev ?? {}, next);
+        return ops;
+    }
+    async parsePartialResult(generations) {
+        const generation = generations[0];
+        if (!generation.message) {
+            return undefined;
+        }
+        const { message } = generation;
+        const functionCall = message.additional_kwargs.function_call;
+        if (!functionCall) {
+            return undefined;
+        }
+        if (this.argsOnly) {
+            return parsePartialJson(functionCall.arguments);
+        }
+        return {
+            ...functionCall,
+            arguments: parsePartialJson(functionCall.arguments),
+        };
+    }
     /**
      * Parses the output and returns a JSON object. If `argsOnly` is true,
      * only the arguments of the function call are returned.
@@ -110,6 +137,14 @@ export class JsonOutputFunctionsParser extends BaseLLMOutputParser {
         parsedResult.arguments = JSON.parse(parsedResult.arguments);
         return parsedResult;
     }
+    // This method would be called by the default implementation of `parse_result`
+    // but we're overriding that method so it's not needed.
+    async parse(_text) {
+        throw new Error("Not implemented.");
+    }
+    getFormatInstructions() {
+        return "";
+    }
 }
 /**
  * Class for parsing the output of an LLM into a JSON object and returning

package/dist/schema/index.cjs

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.Docstore = exports.BaseEntityStore = exports.BaseFileStore = exports.BaseCache = exports.BaseListChatMessageHistory = exports.BaseChatMessageHistory = exports.BasePromptValue = exports.mapStoredMessageToChatMessage = exports.ChatGenerationChunk = exports.ChatMessageChunk = exports.coerceMessageLikeToMessage = exports.isBaseMessage = exports.ChatMessage = exports.FunctionMessageChunk = exports.FunctionMessage = exports.SystemChatMessage = exports.AIChatMessage = exports.HumanChatMessage = exports.BaseChatMessage = exports.SystemMessageChunk = exports.SystemMessage = exports.AIMessageChunk = exports.AIMessage = exports.HumanMessageChunk = exports.HumanMessage = exports.BaseMessageChunk = exports.BaseMessage = exports.GenerationChunk = exports.RUN_KEY = void 0;
+exports.Docstore = exports.BaseEntityStore = exports.BaseFileStore = exports.BaseCache = exports.BaseListChatMessageHistory = exports.BaseChatMessageHistory = exports.BasePromptValue = exports.mapStoredMessageToChatMessage = exports.ChatGenerationChunk = exports.ChatMessageChunk = exports.coerceMessageLikeToMessage = exports.isBaseMessageChunk = exports.isBaseMessage = exports.ChatMessage = exports.FunctionMessageChunk = exports.FunctionMessage = exports.SystemChatMessage = exports.AIChatMessage = exports.HumanChatMessage = exports.BaseChatMessage = exports.SystemMessageChunk = exports.SystemMessage = exports.AIMessageChunk = exports.AIMessage = exports.HumanMessageChunk = exports.HumanMessage = exports.BaseMessageChunk = exports.BaseMessage = exports.GenerationChunk = exports.RUN_KEY = void 0;
 const serializable_js_1 = require("../load/serializable.cjs");
 exports.RUN_KEY = "__run";
 /**
@@ -105,6 +105,33 @@ class BaseMessage extends serializable_js_1.Serializable {
                 .kwargs,
         };
     }
+    toChunk() {
+        const type = this._getType();
+        if (type === "human") {
+            // eslint-disable-next-line @typescript-eslint/no-use-before-define
+            return new HumanMessageChunk({ ...this });
+        }
+        else if (type === "ai") {
+            // eslint-disable-next-line @typescript-eslint/no-use-before-define
+            return new AIMessageChunk({ ...this });
+        }
+        else if (type === "system") {
+            // eslint-disable-next-line @typescript-eslint/no-use-before-define
+            return new SystemMessageChunk({ ...this });
+        }
+        else if (type === "function") {
+            // eslint-disable-next-line @typescript-eslint/no-use-before-define
+            return new FunctionMessageChunk({ ...this });
+            // eslint-disable-next-line @typescript-eslint/no-use-before-define
+        }
+        else if (ChatMessage.isInstance(this)) {
+            // eslint-disable-next-line @typescript-eslint/no-use-before-define
+            return new ChatMessageChunk({ ...this });
+        }
+        else {
+            throw new Error("Unknown message type.");
+        }
+    }
 }
 exports.BaseMessage = BaseMessage;
 /**
@@ -326,6 +353,11 @@ function isBaseMessage(messageLike) {
     return typeof messageLike?._getType === "function";
 }
 exports.isBaseMessage = isBaseMessage;
+function isBaseMessageChunk(messageLike) {
+    return (isBaseMessage(messageLike) &&
+        typeof messageLike.concat === "function");
+}
+exports.isBaseMessageChunk = isBaseMessageChunk;
 function coerceMessageLikeToMessage(messageLike) {
     if (typeof messageLike === "string") {
         return new HumanMessage(messageLike);

package/dist/schema/index.d.ts

@@ -103,6 +103,7 @@ export declare abstract class BaseMessage extends Serializable implements BaseMe
     /** @deprecated */
     kwargs?: Record<string, unknown>);
     toDict(): StoredMessage;
+    toChunk(): BaseMessageChunk;
 }
 /**
  * Represents a chunk of a message, which can be concatenated with other
@@ -219,6 +220,7 @@ export type BaseMessageLike = BaseMessage | [
     string
 ] | string;
 export declare function isBaseMessage(messageLike?: unknown): messageLike is BaseMessage;
+export declare function isBaseMessageChunk(messageLike?: unknown): messageLike is BaseMessageChunk;
 export declare function coerceMessageLikeToMessage(messageLike: BaseMessageLike): BaseMessage;
 /**
  * Represents a chunk of a chat message, which can be concatenated with
@@ -243,7 +245,7 @@ export declare class ChatGenerationChunk extends GenerationChunk implements Chat
     constructor(fields: ChatGenerationChunkFields);
     concat(chunk: ChatGenerationChunk): ChatGenerationChunk;
 }
-export declare function mapStoredMessageToChatMessage(message: StoredMessage): HumanMessage | AIMessage | SystemMessage | FunctionMessage | ChatMessage;
+export declare function mapStoredMessageToChatMessage(message: StoredMessage): ChatMessage | HumanMessage | AIMessage | SystemMessage | FunctionMessage;
 export interface ChatResult {
     generations: ChatGeneration[];
     llmOutput?: Record<string, any>;

package/dist/schema/index.js

@@ -101,6 +101,33 @@ export class BaseMessage extends Serializable {
                 .kwargs,
         };
     }
+    toChunk() {
+        const type = this._getType();
+        if (type === "human") {
+            // eslint-disable-next-line @typescript-eslint/no-use-before-define
+            return new HumanMessageChunk({ ...this });
+        }
+        else if (type === "ai") {
+            // eslint-disable-next-line @typescript-eslint/no-use-before-define
+            return new AIMessageChunk({ ...this });
+        }
+        else if (type === "system") {
+            // eslint-disable-next-line @typescript-eslint/no-use-before-define
+            return new SystemMessageChunk({ ...this });
+        }
+        else if (type === "function") {
+            // eslint-disable-next-line @typescript-eslint/no-use-before-define
+            return new FunctionMessageChunk({ ...this });
+            // eslint-disable-next-line @typescript-eslint/no-use-before-define
+        }
+        else if (ChatMessage.isInstance(this)) {
+            // eslint-disable-next-line @typescript-eslint/no-use-before-define
+            return new ChatMessageChunk({ ...this });
+        }
+        else {
+            throw new Error("Unknown message type.");
+        }
+    }
 }
 /**
  * Represents a chunk of a message, which can be concatenated with other
@@ -310,6 +337,10 @@ export class ChatMessage extends BaseMessage {
 export function isBaseMessage(messageLike) {
     return typeof messageLike?._getType === "function";
 }
+export function isBaseMessageChunk(messageLike) {
+    return (isBaseMessage(messageLike) &&
+        typeof messageLike.concat === "function");
+}
 export function coerceMessageLikeToMessage(messageLike) {
     if (typeof messageLike === "string") {
         return new HumanMessage(messageLike);

package/dist/schema/output_parser.cjs

@@ -1,13 +1,15 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.OutputParserException = exports.BytesOutputParser = exports.StringOutputParser = exports.BaseTransformOutputParser = exports.BaseOutputParser = exports.BaseLLMOutputParser = void 0;
-const index_js_1 = require("./runnable/index.cjs");
+exports.OutputParserException = exports.BytesOutputParser = exports.StringOutputParser = exports.BaseCumulativeTransformOutputParser = exports.BaseTransformOutputParser = exports.BaseOutputParser = exports.BaseLLMOutputParser = void 0;
+const index_js_1 = require("./index.cjs");
+const index_js_2 = require("./runnable/index.cjs");
+const index_js_3 = require("../util/@cfworker/json-schema/index.cjs");
 /**
  * Abstract base class for parsing the output of a Large Language Model
  * (LLM) call. It provides methods for parsing the result of an LLM call
  * and invoking the parser with a given input.
  */
-class BaseLLMOutputParser extends index_js_1.Runnable {
+class BaseLLMOutputParser extends index_js_2.Runnable {
     /**
      * Parses the result of an LLM call with a given prompt. By default, it
      * simply calls `parseResult`.
@@ -86,6 +88,64 @@ class BaseTransformOutputParser extends BaseOutputParser {
     }
 }
 exports.BaseTransformOutputParser = BaseTransformOutputParser;
+/**
+ * A base class for output parsers that can handle streaming input. It
+ * extends the `BaseTransformOutputParser` class and provides a method for
+ * converting parsed outputs into a diff format.
+ */
+class BaseCumulativeTransformOutputParser extends BaseTransformOutputParser {
+    constructor(fields) {
+        super(fields);
+        Object.defineProperty(this, "diff", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: false
+        });
+        this.diff = fields?.diff ?? this.diff;
+    }
+    async *_transform(inputGenerator) {
+        let prevParsed;
+        let accGen;
+        for await (const chunk of inputGenerator) {
+            let chunkGen;
+            if ((0, index_js_1.isBaseMessageChunk)(chunk)) {
+                chunkGen = new index_js_1.ChatGenerationChunk({
+                    message: chunk,
+                    text: chunk.content,
+                });
+            }
+            else if ((0, index_js_1.isBaseMessage)(chunk)) {
+                chunkGen = new index_js_1.ChatGenerationChunk({
+                    message: chunk.toChunk(),
+                    text: chunk.content,
+                });
+            }
+            else {
+                chunkGen = new index_js_1.GenerationChunk({ text: chunk });
+            }
+            if (accGen === undefined) {
+                accGen = chunkGen;
+            }
+            else {
+                accGen = accGen.concat(chunkGen);
+            }
+            const parsed = await this.parsePartialResult([accGen]);
+            if (parsed !== undefined &&
+                parsed !== null &&
+                !(0, index_js_3.deepCompareStrict)(parsed, prevParsed)) {
+                if (this.diff) {
+                    yield this._diff(prevParsed, parsed);
+                }
+                else {
+                    yield parsed;
+                }
+                prevParsed = parsed;
+            }
+        }
+    }
+}
+exports.BaseCumulativeTransformOutputParser = BaseCumulativeTransformOutputParser;
 /**
  * OutputParser that parses LLMResult into the top likely string.
  */

package/dist/schema/output_parser.d.ts

@@ -77,7 +77,7 @@ export declare abstract class BaseOutputParser<T = unknown> extends BaseLLMOutpu
  * Class to parse the output of an LLM call that also allows streaming inputs.
  */
 export declare abstract class BaseTransformOutputParser<T = unknown> extends BaseOutputParser<T> {
-    _transform(inputGenerator: AsyncGenerator<string | BaseMessage>): AsyncGenerator<T>;
+    protected _transform(inputGenerator: AsyncGenerator<string | BaseMessage>): AsyncGenerator<T>;
     /**
      * Transforms an asynchronous generator of input into an asynchronous
      * generator of parsed output.
@@ -87,6 +87,21 @@ export declare abstract class BaseTransformOutputParser<T = unknown> extends Bas
      */
     transform(inputGenerator: AsyncGenerator<string | BaseMessage>, options: BaseCallbackConfig): AsyncGenerator<T>;
 }
+export type BaseCumulativeTransformOutputParserInput = {
+    diff?: boolean;
+};
+/**
+ * A base class for output parsers that can handle streaming input. It
+ * extends the `BaseTransformOutputParser` class and provides a method for
+ * converting parsed outputs into a diff format.
+ */
+export declare abstract class BaseCumulativeTransformOutputParser<T = unknown> extends BaseTransformOutputParser<T> {
+    protected diff: boolean;
+    constructor(fields?: BaseCumulativeTransformOutputParserInput);
+    protected abstract _diff(prev: any | undefined, next: any): any;
+    abstract parsePartialResult(generations: Generation[] | ChatGeneration[]): Promise<T | undefined>;
+    protected _transform(inputGenerator: AsyncGenerator<string | BaseMessage>): AsyncGenerator<T>;
+}
 /**
  * OutputParser that parses LLMResult into the top likely string.
  */

package/dist/schema/output_parser.js

@@ -1,4 +1,6 @@
+import { isBaseMessage, isBaseMessageChunk, ChatGenerationChunk, GenerationChunk, } from "./index.js";
 import { Runnable } from "./runnable/index.js";
+import { deepCompareStrict } from "../util/@cfworker/json-schema/index.js";
 /**
  * Abstract base class for parsing the output of a Large Language Model
  * (LLM) call. It provides methods for parsing the result of an LLM call
@@ -80,6 +82,63 @@ export class BaseTransformOutputParser extends BaseOutputParser {
         });
     }
 }
+/**
+ * A base class for output parsers that can handle streaming input. It
+ * extends the `BaseTransformOutputParser` class and provides a method for
+ * converting parsed outputs into a diff format.
+ */
+export class BaseCumulativeTransformOutputParser extends BaseTransformOutputParser {
+    constructor(fields) {
+        super(fields);
+        Object.defineProperty(this, "diff", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: false
+        });
+        this.diff = fields?.diff ?? this.diff;
+    }
+    async *_transform(inputGenerator) {
+        let prevParsed;
+        let accGen;
+        for await (const chunk of inputGenerator) {
+            let chunkGen;
+            if (isBaseMessageChunk(chunk)) {
+                chunkGen = new ChatGenerationChunk({
+                    message: chunk,
+                    text: chunk.content,
+                });
+            }
+            else if (isBaseMessage(chunk)) {
+                chunkGen = new ChatGenerationChunk({
+                    message: chunk.toChunk(),
+                    text: chunk.content,
+                });
+            }
+            else {
+                chunkGen = new GenerationChunk({ text: chunk });
+            }
+            if (accGen === undefined) {
+                accGen = chunkGen;
+            }
+            else {
+                accGen = accGen.concat(chunkGen);
+            }
+            const parsed = await this.parsePartialResult([accGen]);
+            if (parsed !== undefined &&
+                parsed !== null &&
+                !deepCompareStrict(parsed, prevParsed)) {
+                if (this.diff) {
+                    yield this._diff(prevParsed, parsed);
+                }
+                else {
+                    yield parsed;
+                }
+                prevParsed = parsed;
+            }
+        }
+    }
+}
 /**
  * OutputParser that parses LLMResult into the top likely string.
  */