langchain 0.0.76 → 0.0.78

package/dist/retrievers/self_query/translator.cjs

@@ -0,0 +1,72 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.BasicTranslator = exports.BaseTranslator = void 0;
+const ir_js_1 = require("../../chains/query_constructor/ir.cjs");
+class BaseTranslator extends ir_js_1.Visitor {
+}
+exports.BaseTranslator = BaseTranslator;
+class BasicTranslator extends BaseTranslator {
+    constructor() {
+        super(...arguments);
+        Object.defineProperty(this, "allowedOperators", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: [ir_js_1.Operators.and, ir_js_1.Operators.or]
+        });
+        Object.defineProperty(this, "allowedComparators", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: [
+                ir_js_1.Comparators.eq,
+                ir_js_1.Comparators.neq,
+                ir_js_1.Comparators.gt,
+                ir_js_1.Comparators.gte,
+                ir_js_1.Comparators.lt,
+                ir_js_1.Comparators.lte,
+            ]
+        });
+    }
+    formatFunction(func) {
+        if (func in ir_js_1.Comparators) {
+            if (this.allowedComparators.length > 0 &&
+                this.allowedComparators.indexOf(func) === -1) {
+                throw new Error(`Comparator ${func} not allowed. Allowed operators: ${this.allowedComparators.join(", ")}`);
+            }
+        }
+        else if (func in ir_js_1.Operators) {
+            if (this.allowedOperators.length > 0 &&
+                this.allowedOperators.indexOf(func) === -1) {
+                throw new Error(`Operator ${func} not allowed. Allowed operators: ${this.allowedOperators.join(", ")}`);
+            }
+        }
+        else {
+            throw new Error("Unknown comparator or operator");
+        }
+        return `$${func}`;
+    }
+    visitOperation(operation) {
+        const args = operation.args?.map((arg) => arg.accept(this));
+        return {
+            [this.formatFunction(operation.operator)]: args,
+        };
+    }
+    visitComparison(comparison) {
+        return {
+            [comparison.attribute]: {
+                [this.formatFunction(comparison.comparator)]: comparison.value,
+            },
+        };
+    }
+    visitStructuredQuery(query) {
+        let nextArg = {};
+        if (query.filter) {
+            nextArg = {
+                filter: query.filter.accept(this),
+            };
+        }
+        return nextArg;
+    }
+}
+exports.BasicTranslator = BasicTranslator;

package/dist/retrievers/self_query/translator.d.ts

@@ -0,0 +1,14 @@
+import { Comparator, Comparison, Operation, Operator, StructuredQuery, Visitor, VisitorComparisonResult, VisitorOperationResult, VisitorStructuredQueryResult } from "../../chains/query_constructor/ir.js";
+export declare abstract class BaseTranslator extends Visitor {
+    abstract allowedOperators: Operator[];
+    abstract allowedComparators: Comparator[];
+    abstract formatFunction(func: Operator | Comparator): string;
+}
+export declare class BasicTranslator extends BaseTranslator {
+    allowedOperators: Operator[];
+    allowedComparators: Comparator[];
+    formatFunction(func: Operator | Comparator): string;
+    visitOperation(operation: Operation): VisitorOperationResult;
+    visitComparison(comparison: Comparison): VisitorComparisonResult;
+    visitStructuredQuery(query: StructuredQuery): VisitorStructuredQueryResult;
+}

package/dist/retrievers/self_query/translator.js

@@ -0,0 +1,67 @@
+import { Comparators, Operators, Visitor, } from "../../chains/query_constructor/ir.js";
+export class BaseTranslator extends Visitor {
+}
+export class BasicTranslator extends BaseTranslator {
+    constructor() {
+        super(...arguments);
+        Object.defineProperty(this, "allowedOperators", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: [Operators.and, Operators.or]
+        });
+        Object.defineProperty(this, "allowedComparators", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: [
+                Comparators.eq,
+                Comparators.neq,
+                Comparators.gt,
+                Comparators.gte,
+                Comparators.lt,
+                Comparators.lte,
+            ]
+        });
+    }
+    formatFunction(func) {
+        if (func in Comparators) {
+            if (this.allowedComparators.length > 0 &&
+                this.allowedComparators.indexOf(func) === -1) {
+                throw new Error(`Comparator ${func} not allowed. Allowed operators: ${this.allowedComparators.join(", ")}`);
+            }
+        }
+        else if (func in Operators) {
+            if (this.allowedOperators.length > 0 &&
+                this.allowedOperators.indexOf(func) === -1) {
+                throw new Error(`Operator ${func} not allowed. Allowed operators: ${this.allowedOperators.join(", ")}`);
+            }
+        }
+        else {
+            throw new Error("Unknown comparator or operator");
+        }
+        return `$${func}`;
+    }
+    visitOperation(operation) {
+        const args = operation.args?.map((arg) => arg.accept(this));
+        return {
+            [this.formatFunction(operation.operator)]: args,
+        };
+    }
+    visitComparison(comparison) {
+        return {
+            [comparison.attribute]: {
+                [this.formatFunction(comparison.comparator)]: comparison.value,
+            },
+        };
+    }
+    visitStructuredQuery(query) {
+        let nextArg = {};
+        if (query.filter) {
+            nextArg = {
+                filter: query.filter.accept(this),
+            };
+        }
+        return nextArg;
+    }
+}

package/dist/schema/query_constructor.cjs

@@ -0,0 +1,26 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.AttributeInfo = void 0;
+class AttributeInfo {
+    constructor(name, type, description) {
+        Object.defineProperty(this, "name", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: name
+        });
+        Object.defineProperty(this, "type", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: type
+        });
+        Object.defineProperty(this, "description", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: description
+        });
+    }
+}
+exports.AttributeInfo = AttributeInfo;

package/dist/schema/query_constructor.d.ts

@@ -0,0 +1,6 @@
+export declare class AttributeInfo {
+    name: string;
+    type: string;
+    description: string;
+    constructor(name: string, type: string, description: string);
+}

package/dist/schema/query_constructor.js

@@ -0,0 +1,22 @@
+export class AttributeInfo {
+    constructor(name, type, description) {
+        Object.defineProperty(this, "name", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: name
+        });
+        Object.defineProperty(this, "type", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: type
+        });
+        Object.defineProperty(this, "description", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: description
+        });
+    }
+}

package/dist/text_splitter.cjs

@@ -80,7 +80,8 @@ class TextSplitter {
         let total = 0;
         for (const d of splits) {
             const _len = d.length;
-            if (total + _len >= this.chunkSize) {
+            if (total + _len + (currentDoc.length > 0 ? separator.length : 0) >
+                this.chunkSize) {
                 if (total > this.chunkSize) {
                     console.warn(`Created a chunk of size ${total}, +
 which is longer than the specified ${this.chunkSize}`);

package/dist/text_splitter.js

@@ -77,7 +77,8 @@ export class TextSplitter {
         let total = 0;
         for (const d of splits) {
             const _len = d.length;
-            if (total + _len >= this.chunkSize) {
+            if (total + _len + (currentDoc.length > 0 ? separator.length : 0) >
+                this.chunkSize) {
                 if (total > this.chunkSize) {
                     console.warn(`Created a chunk of size ${total}, +
 which is longer than the specified ${this.chunkSize}`);

package/dist/tools/dynamic.cjs

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.DynamicTool = void 0;
+exports.DynamicStructuredTool = exports.DynamicTool = void 0;
 const base_js_1 = require("./base.cjs");
 /**
  * A tool that can be created dynamically from a function, name, and description.
@@ -37,3 +37,41 @@ class DynamicTool extends base_js_1.Tool {
     }
 }
 exports.DynamicTool = DynamicTool;
+class DynamicStructuredTool extends base_js_1.StructuredTool {
+    constructor(fields) {
+        super(fields);
+        Object.defineProperty(this, "name", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "description", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "func", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "schema", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        this.name = fields.name;
+        this.description = fields.description;
+        this.func = fields.func;
+        this.returnDirect = fields.returnDirect ?? this.returnDirect;
+        this.schema = fields.schema;
+    }
+    _call(arg, runManager) {
+        return this.func(arg, runManager);
+    }
+}
+exports.DynamicStructuredTool = DynamicStructuredTool;

package/dist/tools/dynamic.d.ts

@@ -1,13 +1,20 @@
+import { z } from "zod";
 import { CallbackManagerForToolRun, Callbacks } from "../callbacks/manager.js";
-import { Tool } from "./base.js";
-export interface DynamicToolInput {
+import { StructuredTool, Tool } from "./base.js";
+export interface BaseDynamicToolInput {
     name: string;
     description: string;
-    func: (input: string, runManager?: CallbackManagerForToolRun) => Promise<string>;
     returnDirect?: boolean;
     verbose?: boolean;
     callbacks?: Callbacks;
 }
+export interface DynamicToolInput extends BaseDynamicToolInput {
+    func: (input: string, runManager?: CallbackManagerForToolRun) => Promise<string>;
+}
+export interface DynamicStructuredToolInput<T extends z.ZodObject<any, any, any, any> = z.ZodObject<any, any, any, any>> extends BaseDynamicToolInput {
+    func: (input: z.infer<T>, runManager?: CallbackManagerForToolRun) => Promise<string>;
+    schema: T;
+}
 /**
  * A tool that can be created dynamically from a function, name, and description.
  */
@@ -19,3 +26,11 @@ export declare class DynamicTool extends Tool {
     /** @ignore */
     _call(input: string, runManager?: CallbackManagerForToolRun): Promise<string>;
 }
+export declare class DynamicStructuredTool<T extends z.ZodObject<any, any, any, any> = z.ZodObject<any, any, any, any>> extends StructuredTool {
+    name: string;
+    description: string;
+    func: DynamicStructuredToolInput["func"];
+    schema: T;
+    constructor(fields: DynamicStructuredToolInput<T>);
+    protected _call(arg: z.output<T>, runManager?: CallbackManagerForToolRun): Promise<string>;
+}

package/dist/tools/dynamic.js

@@ -1,4 +1,4 @@
-import { Tool } from "./base.js";
+import { StructuredTool, Tool } from "./base.js";
 /**
  * A tool that can be created dynamically from a function, name, and description.
  */
@@ -33,3 +33,40 @@ export class DynamicTool extends Tool {
         return this.func(input, runManager);
     }
 }
+export class DynamicStructuredTool extends StructuredTool {
+    constructor(fields) {
+        super(fields);
+        Object.defineProperty(this, "name", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "description", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "func", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "schema", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        this.name = fields.name;
+        this.description = fields.description;
+        this.func = fields.func;
+        this.returnDirect = fields.returnDirect ?? this.returnDirect;
+        this.schema = fields.schema;
+    }
+    _call(arg, runManager) {
+        return this.func(arg, runManager);
+    }
+}

package/dist/tools/index.cjs

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.WriteFileTool = exports.ReadFileTool = exports.AIPluginTool = exports.Serper = exports.ZapierNLAWrapper = exports.ZapierNLARunAction = exports.VectorStoreQATool = exports.RequestsPostTool = exports.RequestsGetTool = exports.JsonGetValueTool = exports.JsonListKeysTool = exports.JsonSpec = exports.QueryCheckerTool = exports.ListTablesSqlTool = exports.InfoSqlTool = exports.QuerySqlTool = exports.ChainTool = exports.IFTTTWebhook = exports.DynamicTool = exports.StructuredTool = exports.Tool = exports.BingSerpAPI = exports.DadJokeAPI = exports.SerpAPI = void 0;
+exports.WriteFileTool = exports.ReadFileTool = exports.AIPluginTool = exports.Serper = exports.ZapierNLAWrapper = exports.ZapierNLARunAction = exports.VectorStoreQATool = exports.RequestsPostTool = exports.RequestsGetTool = exports.JsonGetValueTool = exports.JsonListKeysTool = exports.JsonSpec = exports.QueryCheckerTool = exports.ListTablesSqlTool = exports.InfoSqlTool = exports.QuerySqlTool = exports.ChainTool = exports.IFTTTWebhook = exports.DynamicStructuredTool = exports.DynamicTool = exports.StructuredTool = exports.Tool = exports.BingSerpAPI = exports.DadJokeAPI = exports.SerpAPI = void 0;
 var serpapi_js_1 = require("./serpapi.cjs");
 Object.defineProperty(exports, "SerpAPI", { enumerable: true, get: function () { return serpapi_js_1.SerpAPI; } });
 var dadjokeapi_js_1 = require("./dadjokeapi.cjs");
@@ -12,6 +12,7 @@ Object.defineProperty(exports, "Tool", { enumerable: true, get: function () { re
 Object.defineProperty(exports, "StructuredTool", { enumerable: true, get: function () { return base_js_1.StructuredTool; } });
 var dynamic_js_1 = require("./dynamic.cjs");
 Object.defineProperty(exports, "DynamicTool", { enumerable: true, get: function () { return dynamic_js_1.DynamicTool; } });
+Object.defineProperty(exports, "DynamicStructuredTool", { enumerable: true, get: function () { return dynamic_js_1.DynamicStructuredTool; } });
 var IFTTTWebhook_js_1 = require("./IFTTTWebhook.cjs");
 Object.defineProperty(exports, "IFTTTWebhook", { enumerable: true, get: function () { return IFTTTWebhook_js_1.IFTTTWebhook; } });
 var chain_js_1 = require("./chain.cjs");

package/dist/tools/index.d.ts

@@ -2,7 +2,7 @@ export { SerpAPI, SerpAPIParameters } from "./serpapi.js";
 export { DadJokeAPI } from "./dadjokeapi.js";
 export { BingSerpAPI } from "./bingserpapi.js";
 export { Tool, ToolParams, StructuredTool } from "./base.js";
-export { DynamicTool, DynamicToolInput } from "./dynamic.js";
+export { DynamicTool, DynamicToolInput, DynamicStructuredTool, DynamicStructuredToolInput, } from "./dynamic.js";
 export { IFTTTWebhook } from "./IFTTTWebhook.js";
 export { ChainTool, ChainToolInput } from "./chain.js";
 export { QuerySqlTool, InfoSqlTool, ListTablesSqlTool, QueryCheckerTool, } from "./sql.js";

package/dist/tools/index.js

@@ -2,7 +2,7 @@ export { SerpAPI } from "./serpapi.js";
 export { DadJokeAPI } from "./dadjokeapi.js";
 export { BingSerpAPI } from "./bingserpapi.js";
 export { Tool, StructuredTool } from "./base.js";
-export { DynamicTool } from "./dynamic.js";
+export { DynamicTool, DynamicStructuredTool, } from "./dynamic.js";
 export { IFTTTWebhook } from "./IFTTTWebhook.js";
 export { ChainTool } from "./chain.js";
 export { QuerySqlTool, InfoSqlTool, ListTablesSqlTool, QueryCheckerTool, } from "./sql.js";

package/dist/tools/json.cjs

@@ -27,7 +27,9 @@ class JsonSpec {
         const pointer = jsonpointer_1.default.compile(input);
         const res = pointer.get(this.obj);
         if (typeof res === "object" && !Array.isArray(res) && res !== null) {
-            return Object.keys(res).join(", ");
+            return Object.keys(res)
+                .map((i) => i.replaceAll("~", "~0").replaceAll("/", "~1"))
+                .join(", ");
         }
         throw new Error(`Value at ${input} is not a dictionary, get the value directly instead.`);
     }

package/dist/tools/json.js

@@ -21,7 +21,9 @@ export class JsonSpec {
         const pointer = jsonpointer.compile(input);
         const res = pointer.get(this.obj);
         if (typeof res === "object" && !Array.isArray(res) && res !== null) {
-            return Object.keys(res).join(", ");
+            return Object.keys(res)
+                .map((i) => i.replaceAll("~", "~0").replaceAll("/", "~1"))
+                .join(", ");
         }
         throw new Error(`Value at ${input} is not a dictionary, get the value directly instead.`);
     }

package/dist/tools/sql.cjs

@@ -25,7 +25,7 @@ class QuerySqlTool extends base_js_1.Tool {
             configurable: true,
             writable: true,
             value: `Input to this tool is a detailed and correct SQL query, output is a result from the database.
-  If the query is not correct, an error message will be returned. 
+  If the query is not correct, an error message will be returned.
   If an error is returned, rewrite the query, check the query, and try again.`
         });
         this.db = db;
@@ -62,7 +62,7 @@ class InfoSqlTool extends base_js_1.Tool {
             writable: true,
             value: `Input to this tool is a comma-separated list of tables, output is the schema and sample rows for those tables.
     Be sure that the tables actually exist by calling list-tables-sql first!
-    
+
     Example Input: "table1, table2, table3.`
         });
         this.db = db;
@@ -115,7 +115,7 @@ class ListTablesSqlTool extends base_js_1.Tool {
 }
 exports.ListTablesSqlTool = ListTablesSqlTool;
 class QueryCheckerTool extends base_js_1.Tool {
-    constructor(llmChain) {
+    constructor(llmChainOrOptions) {
         super();
         Object.defineProperty(this, "name", {
             enumerable: true,
@@ -154,16 +154,22 @@ If there are any of the above mistakes, rewrite the query. If there are no mista
             value: `Use this tool to double check if your query is correct before executing it.
     Always use this tool before executing a query with query-sql!`
         });
-        if (llmChain) {
-            this.llmChain = llmChain;
+        if (typeof llmChainOrOptions?._chainType === "function") {
+            this.llmChain = llmChainOrOptions;
         }
         else {
-            const model = new openai_js_1.OpenAI({ temperature: 0 });
-            const prompt = new prompt_js_1.PromptTemplate({
-                template: this.template,
-                inputVariables: ["query"],
-            });
-            this.llmChain = new llm_chain_js_1.LLMChain({ llm: model, prompt });
+            const options = llmChainOrOptions;
+            if (options?.llmChain !== undefined) {
+                this.llmChain = options.llmChain;
+            }
+            else {
+                const prompt = new prompt_js_1.PromptTemplate({
+                    template: this.template,
+                    inputVariables: ["query"],
+                });
+                const llm = options?.llm ?? new openai_js_1.OpenAI({ temperature: 0 });
+                this.llmChain = new llm_chain_js_1.LLMChain({ llm, prompt });
+            }
         }
     }
     /** @ignore */

package/dist/tools/sql.d.ts

@@ -1,6 +1,7 @@
 import { Tool } from "./base.js";
 import { LLMChain } from "../chains/llm_chain.js";
 import type { SqlDatabase } from "../sql_db.js";
+import { BaseLanguageModel } from "../base_language/index.js";
 interface SqlTool {
     db: SqlDatabase;
 }
@@ -28,11 +29,16 @@ export declare class ListTablesSqlTool extends Tool implements SqlTool {
     _call(_: string): Promise<string>;
     description: string;
 }
+type QueryCheckerToolArgs = {
+    llmChain?: LLMChain;
+    llm?: BaseLanguageModel;
+    _chainType?: never;
+};
 export declare class QueryCheckerTool extends Tool {
     name: string;
     template: string;
     llmChain: LLMChain;
-    constructor(llmChain?: LLMChain);
+    constructor(llmChainOrOptions?: LLMChain | QueryCheckerToolArgs);
     /** @ignore */
     _call(input: string): Promise<string>;
     description: string;

package/dist/tools/sql.js

@@ -22,7 +22,7 @@ export class QuerySqlTool extends Tool {
             configurable: true,
             writable: true,
             value: `Input to this tool is a detailed and correct SQL query, output is a result from the database.
-  If the query is not correct, an error message will be returned. 
+  If the query is not correct, an error message will be returned.
   If an error is returned, rewrite the query, check the query, and try again.`
         });
         this.db = db;
@@ -58,7 +58,7 @@ export class InfoSqlTool extends Tool {
             writable: true,
             value: `Input to this tool is a comma-separated list of tables, output is the schema and sample rows for those tables.
     Be sure that the tables actually exist by calling list-tables-sql first!
-    
+
     Example Input: "table1, table2, table3.`
         });
         this.db = db;
@@ -109,7 +109,7 @@ export class ListTablesSqlTool extends Tool {
     }
 }
 export class QueryCheckerTool extends Tool {
-    constructor(llmChain) {
+    constructor(llmChainOrOptions) {
         super();
         Object.defineProperty(this, "name", {
             enumerable: true,
@@ -148,16 +148,22 @@ If there are any of the above mistakes, rewrite the query. If there are no mista
             value: `Use this tool to double check if your query is correct before executing it.
     Always use this tool before executing a query with query-sql!`
         });
-        if (llmChain) {
-            this.llmChain = llmChain;
+        if (typeof llmChainOrOptions?._chainType === "function") {
+            this.llmChain = llmChainOrOptions;
         }
         else {
-            const model = new OpenAI({ temperature: 0 });
-            const prompt = new PromptTemplate({
-                template: this.template,
-                inputVariables: ["query"],
-            });
-            this.llmChain = new LLMChain({ llm: model, prompt });
+            const options = llmChainOrOptions;
+            if (options?.llmChain !== undefined) {
+                this.llmChain = options.llmChain;
+            }
+            else {
+                const prompt = new PromptTemplate({
+                    template: this.template,
+                    inputVariables: ["query"],
+                });
+                const llm = options?.llm ?? new OpenAI({ temperature: 0 });
+                this.llmChain = new LLMChain({ llm, prompt });
+            }
         }
     }
     /** @ignore */

package/dist/util/event-source-parse.cjs

@@ -16,9 +16,18 @@ exports.EventStreamContentType = "text/event-stream";
  */
 async function getBytes(stream, onChunk) {
     const reader = stream.getReader();
-    let result;
-    // eslint-disable-next-line no-cond-assign
-    while (!(result = await reader.read()).done) {
+    // CHANGED: Introduced a "flush" mechanism to process potential pending messages when the stream ends.
+    //          This change is essential to ensure that we capture every last piece of information from streams,
+    //          such as those from Azure OpenAI, which may not terminate with a blank line. Without this
+    //          mechanism, we risk ignoring a possibly significant last message.
+    //          See https://github.com/hwchase17/langchainjs/issues/1299 for details.
+    // eslint-disable-next-line no-constant-condition
+    while (true) {
+        const result = await reader.read();
+        if (result.done) {
+            onChunk(new Uint8Array(), true);
+            break;
+        }
         onChunk(result.value);
     }
 }
@@ -35,7 +44,11 @@ function getLines(onLine) {
     let fieldLength; // length of the `field` portion of the line
     let discardTrailingNewline = false;
     // return a function that can process each incoming byte chunk:
-    return function onChunk(arr) {
+    return function onChunk(arr, flush) {
+        if (flush) {
+            onLine(arr, 0, true);
+            return;
+        }
         if (buffer === undefined) {
             buffer = arr;
             position = 0;
@@ -107,7 +120,14 @@ function getMessages(onMessage, onId, onRetry) {
     let message = newMessage();
     const decoder = new TextDecoder();
     // return a function that can process each incoming line buffer:
-    return function onLine(line, fieldLength) {
+    return function onLine(line, fieldLength, flush) {
+        if (flush) {
+            if (!isEmpty(message)) {
+                onMessage?.(message);
+                message = newMessage();
+            }
+            return;
+        }
         if (line.length === 0) {
             // empty line denotes end of message. Trigger the callback and start a new message:
             onMessage?.(message);
@@ -163,3 +183,9 @@ function newMessage() {
         retry: undefined,
     };
 }
+function isEmpty(message) {
+    return (message.data === "" &&
+        message.event === "" &&
+        message.id === "" &&
+        message.retry === undefined);
+}

package/dist/util/event-source-parse.d.ts

@@ -19,14 +19,14 @@ export interface EventSourceMessage {
  * @param onChunk A function that will be called on each new byte chunk in the stream.
  * @returns {Promise<void>} A promise that will be resolved when the stream closes.
  */
-export declare function getBytes(stream: ReadableStream<Uint8Array>, onChunk: (arr: Uint8Array) => void): Promise<void>;
+export declare function getBytes(stream: ReadableStream<Uint8Array>, onChunk: (arr: Uint8Array, flush?: boolean) => void): Promise<void>;
 /**
  * Parses arbitary byte chunks into EventSource line buffers.
  * Each line should be of the format "field: value" and ends with \r, \n, or \r\n.
  * @param onLine A function that will be called on each new EventSource line.
  * @returns A function that should be called for each incoming byte chunk.
  */
-export declare function getLines(onLine: (line: Uint8Array, fieldLength: number) => void): (arr: Uint8Array) => void;
+export declare function getLines(onLine: (line: Uint8Array, fieldLength: number, flush?: boolean) => void): (arr: Uint8Array, flush?: boolean) => void;
 /**
  * Parses line buffers into EventSourceMessages.
  * @param onId A function that will be called on each `id` field.
@@ -34,4 +34,4 @@ export declare function getLines(onLine: (line: Uint8Array, fieldLength: number)
  * @param onMessage A function that will be called on each message.
  * @returns A function that should be called for each incoming line buffer.
  */
-export declare function getMessages(onMessage?: (msg: EventSourceMessage) => void, onId?: (id: string) => void, onRetry?: (retry: number) => void): (line: Uint8Array, fieldLength: number) => void;
+export declare function getMessages(onMessage?: (msg: EventSourceMessage) => void, onId?: (id: string) => void, onRetry?: (retry: number) => void): (line: Uint8Array, fieldLength: number, flush?: boolean) => void;