@langchain/classic 1.0.0-alpha.1

package/dist/output_parsers/structured.js

@@ -0,0 +1,161 @@
+import { BaseOutputParser, OutputParserException } from "@langchain/core/output_parsers";
+import { interopParseAsync } from "@langchain/core/utils/types";
+import { toJsonSchema } from "@langchain/core/utils/json_schema";
+import { z } from "zod/v3";
+
+//#region src/output_parsers/structured.ts
+var StructuredOutputParser = class extends BaseOutputParser {
+	static lc_name() {
+		return "StructuredOutputParser";
+	}
+	lc_namespace = [
+		"langchain",
+		"output_parsers",
+		"structured"
+	];
+	toJSON() {
+		return this.toJSONNotImplemented();
+	}
+	constructor(schema) {
+		super(schema);
+		this.schema = schema;
+	}
+	/**
+	* Creates a new StructuredOutputParser from a Zod schema.
+	* @param schema The Zod schema which the output should match
+	* @returns A new instance of StructuredOutputParser.
+	*/
+	static fromZodSchema(schema) {
+		return new this(schema);
+	}
+	/**
+	* Creates a new StructuredOutputParser from a set of names and
+	* descriptions.
+	* @param schemas An object where each key is a name and each value is a description
+	* @returns A new instance of StructuredOutputParser.
+	*/
+	static fromNamesAndDescriptions(schemas) {
+		const zodSchema = z.object(Object.fromEntries(Object.entries(schemas).map(([name, description]) => [name, z.string().describe(description)])));
+		return new this(zodSchema);
+	}
+	/**
+	* Returns a markdown code snippet with a JSON object formatted according
+	* to the schema.
+	* @param options Optional. The options for formatting the instructions
+	* @returns A markdown code snippet with a JSON object formatted according to the schema.
+	*/
+	getFormatInstructions() {
+		return `You must format your output as a JSON value that adheres to a given "JSON Schema" instance.
+
+"JSON Schema" is a declarative language that allows you to annotate and validate JSON documents.
+
+For example, the example "JSON Schema" instance {{"properties": {{"foo": {{"description": "a list of test words", "type": "array", "items": {{"type": "string"}}}}}}, "required": ["foo"]}}}}
+would match an object with one required property, "foo". The "type" property specifies "foo" must be an "array", and the "description" property semantically describes it as "a list of test words". The items within "foo" must be strings.
+Thus, the object {{"foo": ["bar", "baz"]}} is a well-formatted instance of this example "JSON Schema". The object {{"properties": {{"foo": ["bar", "baz"]}}}} is not well-formatted.
+
+Your output will be parsed and type-checked according to the provided schema instance, so make sure all fields in your output match the schema exactly and there are no trailing commas!
+
+Here is the JSON Schema instance your output must adhere to. Include the enclosing markdown codeblock:
+\`\`\`json
+${JSON.stringify(toJsonSchema(this.schema))}
+\`\`\`
+`;
+	}
+	/**
+	* Parses the given text according to the schema.
+	* @param text The text to parse
+	* @returns The parsed output.
+	*/
+	async parse(text) {
+		try {
+			const json = text.includes("```") ? text.trim().split(/```(?:json)?/)[1] : text.trim();
+			return await interopParseAsync(this.schema, JSON.parse(json));
+		} catch {
+			try {
+				return await interopParseAsync(this.schema, JSON.parse(text.trim()));
+			} catch (e2) {
+				throw new OutputParserException(`Failed to parse. Text: "${text}". Error: ${e2}`, text);
+			}
+		}
+	}
+};
+/**
+* A specific type of `StructuredOutputParser` that parses JSON data
+* formatted as a markdown code snippet.
+*/
+var JsonMarkdownStructuredOutputParser = class extends StructuredOutputParser {
+	static lc_name() {
+		return "JsonMarkdownStructuredOutputParser";
+	}
+	getFormatInstructions(options) {
+		const interpolationDepth = options?.interpolationDepth ?? 1;
+		if (interpolationDepth < 1) throw new Error("f string interpolation depth must be at least 1");
+		return `Return a markdown code snippet with a JSON object formatted to look like:\n\`\`\`json\n${this._schemaToInstruction(toJsonSchema(this.schema)).replaceAll("{", "{".repeat(interpolationDepth)).replaceAll("}", "}".repeat(interpolationDepth))}\n\`\`\``;
+	}
+	_schemaToInstruction(schemaInput, indent = 2) {
+		const schema = schemaInput;
+		if ("type" in schema) {
+			let nullable = false;
+			let type;
+			if (Array.isArray(schema.type)) {
+				const nullIdx = schema.type.findIndex((type$1) => type$1 === "null");
+				if (nullIdx !== -1) {
+					nullable = true;
+					schema.type.splice(nullIdx, 1);
+				}
+				type = schema.type.join(" | ");
+			} else type = schema.type;
+			if (schema.type === "object" && schema.properties) {
+				const description$1 = schema.description ? ` // ${schema.description}` : "";
+				const properties = Object.entries(schema.properties).map(([key, value]) => {
+					const isOptional = schema.required?.includes(key) ? "" : " (optional)";
+					return `${" ".repeat(indent)}"${key}": ${this._schemaToInstruction(value, indent + 2)}${isOptional}`;
+				}).join("\n");
+				return `{\n${properties}\n${" ".repeat(indent - 2)}}${description$1}`;
+			}
+			if (schema.type === "array" && schema.items) {
+				const description$1 = schema.description ? ` // ${schema.description}` : "";
+				return `array[\n${" ".repeat(indent)}${this._schemaToInstruction(schema.items, indent + 2)}\n${" ".repeat(indent - 2)}] ${description$1}`;
+			}
+			const isNullable = nullable ? " (nullable)" : "";
+			const description = schema.description ? ` // ${schema.description}` : "";
+			return `${type}${description}${isNullable}`;
+		}
+		if ("anyOf" in schema) return schema.anyOf.map((s) => this._schemaToInstruction(s, indent)).join(`\n${" ".repeat(indent - 2)}`);
+		throw new Error("unsupported schema type");
+	}
+	static fromZodSchema(schema) {
+		return new this(schema);
+	}
+	static fromNamesAndDescriptions(schemas) {
+		const zodSchema = z.object(Object.fromEntries(Object.entries(schemas).map(([name, description]) => [name, z.string().describe(description)])));
+		return new this(zodSchema);
+	}
+};
+/**
+* A type of `StructuredOutputParser` that handles asymmetric input and
+* output schemas.
+*/
+var AsymmetricStructuredOutputParser = class extends BaseOutputParser {
+	structuredInputParser;
+	constructor({ inputSchema }) {
+		super(...arguments);
+		this.structuredInputParser = new JsonMarkdownStructuredOutputParser(inputSchema);
+	}
+	async parse(text) {
+		let parsedInput;
+		try {
+			parsedInput = await this.structuredInputParser.parse(text);
+		} catch (e) {
+			throw new OutputParserException(`Failed to parse. Text: "${text}". Error: ${e}`, text);
+		}
+		return this.outputProcessor(parsedInput);
+	}
+	getFormatInstructions() {
+		return this.structuredInputParser.getFormatInstructions();
+	}
+};
+
+//#endregion
+export { AsymmetricStructuredOutputParser, JsonMarkdownStructuredOutputParser, StructuredOutputParser };
+//# sourceMappingURL=structured.js.map

package/dist/output_parsers/structured.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"structured.js","names":["schema: T","schemas: S","text: string","options?: JsonMarkdownFormatInstructionsOptions","schemaInput: JsonSchema7Type","type: string","type","description"],"sources":["../../src/output_parsers/structured.ts"],"sourcesContent":["import { z } from \"zod/v3\";\nimport {\n  toJsonSchema,\n  type JsonSchema7Type,\n  type JsonSchema7ArrayType,\n  type JsonSchema7ObjectType,\n  type JsonSchema7StringType,\n  type JsonSchema7NumberType,\n  type JsonSchema7NullableType,\n} from \"@langchain/core/utils/json_schema\";\nimport {\n  BaseOutputParser,\n  FormatInstructionsOptions,\n  OutputParserException,\n} from \"@langchain/core/output_parsers\";\nimport {\n  interopParseAsync,\n  type InferInteropZodOutput,\n  type InteropZodType,\n} from \"@langchain/core/utils/types\";\n\nexport type JsonMarkdownStructuredOutputParserInput = {\n  interpolationDepth?: number;\n};\n\nexport interface JsonMarkdownFormatInstructionsOptions\n  extends FormatInstructionsOptions {\n  interpolationDepth?: number;\n}\n\nexport class StructuredOutputParser<\n  T extends InteropZodType\n> extends BaseOutputParser<InferInteropZodOutput<T>> {\n  static lc_name() {\n    return \"StructuredOutputParser\";\n  }\n\n  lc_namespace = [\"langchain\", \"output_parsers\", \"structured\"];\n\n  toJSON() {\n    return this.toJSONNotImplemented();\n  }\n\n  constructor(public schema: T) {\n    super(schema);\n  }\n\n  /**\n   * Creates a new StructuredOutputParser from a Zod schema.\n   * @param schema The Zod schema which the output should match\n   * @returns A new instance of StructuredOutputParser.\n   */\n  static fromZodSchema<T extends InteropZodType>(schema: T) {\n    return new this(schema);\n  }\n\n  /**\n   * Creates a new StructuredOutputParser from a set of names and\n   * descriptions.\n   * @param schemas An object where each key is a name and each value is a description\n   * @returns A new instance of StructuredOutputParser.\n   */\n  static fromNamesAndDescriptions<S extends { [key: string]: string }>(\n    schemas: S\n  ) {\n    const zodSchema = z.object(\n      Object.fromEntries(\n        Object.entries(schemas).map(\n          ([name, description]) =>\n            [name, z.string().describe(description)] as const\n        )\n      )\n    );\n\n    return new this(zodSchema);\n  }\n\n  /**\n   * Returns a markdown code snippet with a JSON object formatted according\n   * to the schema.\n   * @param options Optional. The options for formatting the instructions\n   * @returns A markdown code snippet with a JSON object formatted according to the schema.\n   */\n  getFormatInstructions(): string {\n    return `You must format your output as a JSON value that adheres to a given \"JSON Schema\" instance.\n\n\"JSON Schema\" is a declarative language that allows you to annotate and validate JSON documents.\n\nFor example, the example \"JSON Schema\" instance {{\"properties\": {{\"foo\": {{\"description\": \"a list of test words\", \"type\": \"array\", \"items\": {{\"type\": \"string\"}}}}}}, \"required\": [\"foo\"]}}}}\nwould match an object with one required property, \"foo\". The \"type\" property specifies \"foo\" must be an \"array\", and the \"description\" property semantically describes it as \"a list of test words\". The items within \"foo\" must be strings.\nThus, the object {{\"foo\": [\"bar\", \"baz\"]}} is a well-formatted instance of this example \"JSON Schema\". The object {{\"properties\": {{\"foo\": [\"bar\", \"baz\"]}}}} is not well-formatted.\n\nYour output will be parsed and type-checked according to the provided schema instance, so make sure all fields in your output match the schema exactly and there are no trailing commas!\n\nHere is the JSON Schema instance your output must adhere to. Include the enclosing markdown codeblock:\n\\`\\`\\`json\n${JSON.stringify(toJsonSchema(this.schema))}\n\\`\\`\\`\n`;\n  }\n\n  /**\n   * Parses the given text according to the schema.\n   * @param text The text to parse\n   * @returns The parsed output.\n   */\n  async parse(text: string): Promise<InferInteropZodOutput<T>> {\n    try {\n      const json = text.includes(\"```\")\n        ? text.trim().split(/```(?:json)?/)[1]\n        : text.trim();\n      return await interopParseAsync(this.schema, JSON.parse(json));\n    } catch {\n      try {\n        return await interopParseAsync(this.schema, JSON.parse(text.trim()));\n      } catch (e2) {\n        throw new OutputParserException(\n          `Failed to parse. Text: \"${text}\". Error: ${e2}`,\n          text\n        );\n      }\n    }\n  }\n}\n\n/**\n * A specific type of `StructuredOutputParser` that parses JSON data\n * formatted as a markdown code snippet.\n */\nexport class JsonMarkdownStructuredOutputParser<\n  T extends InteropZodType\n> extends StructuredOutputParser<T> {\n  static lc_name() {\n    return \"JsonMarkdownStructuredOutputParser\";\n  }\n\n  getFormatInstructions(\n    options?: JsonMarkdownFormatInstructionsOptions\n  ): string {\n    const interpolationDepth = options?.interpolationDepth ?? 1;\n    if (interpolationDepth < 1) {\n      throw new Error(\"f string interpolation depth must be at least 1\");\n    }\n\n    return `Return a markdown code snippet with a JSON object formatted to look like:\\n\\`\\`\\`json\\n${this._schemaToInstruction(\n      toJsonSchema(this.schema)\n    )\n      .replaceAll(\"{\", \"{\".repeat(interpolationDepth))\n      .replaceAll(\"}\", \"}\".repeat(interpolationDepth))}\\n\\`\\`\\``;\n  }\n\n  private _schemaToInstruction(\n    schemaInput: JsonSchema7Type,\n    indent = 2\n  ): string {\n    const schema = schemaInput as Extract<\n      JsonSchema7Type,\n      | JsonSchema7ObjectType\n      | JsonSchema7ArrayType\n      | JsonSchema7StringType\n      | JsonSchema7NumberType\n      | JsonSchema7NullableType\n    >;\n\n    if (\"type\" in schema) {\n      let nullable = false;\n      let type: string;\n      if (Array.isArray(schema.type)) {\n        const nullIdx = schema.type.findIndex((type) => type === \"null\");\n        if (nullIdx !== -1) {\n          nullable = true;\n          schema.type.splice(nullIdx, 1);\n        }\n        type = schema.type.join(\" | \") as string;\n      } else {\n        type = schema.type;\n      }\n\n      if (schema.type === \"object\" && schema.properties) {\n        const description = schema.description\n          ? ` // ${schema.description}`\n          : \"\";\n        const properties = Object.entries(schema.properties)\n          .map(([key, value]) => {\n            const isOptional = schema.required?.includes(key)\n              ? \"\"\n              : \" (optional)\";\n            return `${\" \".repeat(indent)}\"${key}\": ${this._schemaToInstruction(\n              value,\n              indent + 2\n            )}${isOptional}`;\n          })\n          .join(\"\\n\");\n        return `{\\n${properties}\\n${\" \".repeat(indent - 2)}}${description}`;\n      }\n      if (schema.type === \"array\" && schema.items) {\n        const description = schema.description\n          ? ` // ${schema.description}`\n          : \"\";\n        return `array[\\n${\" \".repeat(indent)}${this._schemaToInstruction(\n          schema.items,\n          indent + 2\n        )}\\n${\" \".repeat(indent - 2)}] ${description}`;\n      }\n      const isNullable = nullable ? \" (nullable)\" : \"\";\n      const description = schema.description ? ` // ${schema.description}` : \"\";\n      return `${type}${description}${isNullable}`;\n    }\n\n    if (\"anyOf\" in schema) {\n      return schema.anyOf\n        .map((s) => this._schemaToInstruction(s, indent))\n        .join(`\\n${\" \".repeat(indent - 2)}`);\n    }\n\n    throw new Error(\"unsupported schema type\");\n  }\n\n  static fromZodSchema<T extends InteropZodType>(schema: T) {\n    return new this<T>(schema);\n  }\n\n  static fromNamesAndDescriptions<S extends { [key: string]: string }>(\n    schemas: S\n  ) {\n    const zodSchema = z.object(\n      Object.fromEntries(\n        Object.entries(schemas).map(\n          ([name, description]) =>\n            [name, z.string().describe(description)] as const\n        )\n      )\n    );\n\n    return new this<typeof zodSchema>(zodSchema);\n  }\n}\n\nexport interface AsymmetricStructuredOutputParserFields<\n  T extends InteropZodType\n> {\n  inputSchema: T;\n}\n\n/**\n * A type of `StructuredOutputParser` that handles asymmetric input and\n * output schemas.\n */\nexport abstract class AsymmetricStructuredOutputParser<\n  T extends InteropZodType,\n  Y = unknown\n> extends BaseOutputParser<Y> {\n  private structuredInputParser: JsonMarkdownStructuredOutputParser<T>;\n\n  constructor({ inputSchema }: AsymmetricStructuredOutputParserFields<T>) {\n    super(...arguments);\n    this.structuredInputParser = new JsonMarkdownStructuredOutputParser(\n      inputSchema\n    );\n  }\n\n  /**\n   * Processes the parsed input into the desired output format. Must be\n   * implemented by subclasses.\n   * @param input The parsed input\n   * @returns The processed output.\n   */\n  abstract outputProcessor(input: InferInteropZodOutput<T>): Promise<Y>;\n\n  async parse(text: string): Promise<Y> {\n    let parsedInput;\n    try {\n      parsedInput = await this.structuredInputParser.parse(text);\n    } catch (e) {\n      throw new OutputParserException(\n        `Failed to parse. Text: \"${text}\". Error: ${e}`,\n        text\n      );\n    }\n\n    return this.outputProcessor(parsedInput);\n  }\n\n  getFormatInstructions(): string {\n    return this.structuredInputParser.getFormatInstructions();\n  }\n}\n"],"mappings":";;;;;;AA8BA,IAAa,yBAAb,cAEU,iBAA2C;CACnD,OAAO,UAAU;AACf,SAAO;CACR;CAED,eAAe;EAAC;EAAa;EAAkB;CAAa;CAE5D,SAAS;AACP,SAAO,KAAK,sBAAsB;CACnC;CAED,YAAmBA,QAAW;EAC5B,MAAM,OAAO;EADI;CAElB;;;;;;CAOD,OAAO,cAAwCA,QAAW;AACxD,SAAO,IAAI,KAAK;CACjB;;;;;;;CAQD,OAAO,yBACLC,SACA;EACA,MAAM,YAAY,EAAE,OAClB,OAAO,YACL,OAAO,QAAQ,QAAQ,CAAC,IACtB,CAAC,CAAC,MAAM,YAAY,KAClB,CAAC,MAAM,EAAE,QAAQ,CAAC,SAAS,YAAY,AAAC,EAC3C,CACF,CACF;AAED,SAAO,IAAI,KAAK;CACjB;;;;;;;CAQD,wBAAgC;AAC9B,SAAO,CAAC;;;;;;;;;;;;AAYZ,EAAE,KAAK,UAAU,aAAa,KAAK,OAAO,CAAC,CAAC;;AAE5C,CAAC;CACE;;;;;;CAOD,MAAM,MAAMC,MAAiD;AAC3D,MAAI;GACF,MAAM,OAAO,KAAK,SAAS,MAAM,GAC7B,KAAK,MAAM,CAAC,MAAM,eAAe,CAAC,KAClC,KAAK,MAAM;AACf,UAAO,MAAM,kBAAkB,KAAK,QAAQ,KAAK,MAAM,KAAK,CAAC;EAC9D,QAAO;AACN,OAAI;AACF,WAAO,MAAM,kBAAkB,KAAK,QAAQ,KAAK,MAAM,KAAK,MAAM,CAAC,CAAC;GACrE,SAAQ,IAAI;AACX,UAAM,IAAI,sBACR,CAAC,wBAAwB,EAAE,KAAK,UAAU,EAAE,IAAI,EAChD;GAEH;EACF;CACF;AACF;;;;;AAMD,IAAa,qCAAb,cAEU,uBAA0B;CAClC,OAAO,UAAU;AACf,SAAO;CACR;CAED,sBACEC,SACQ;EACR,MAAM,qBAAqB,SAAS,sBAAsB;AAC1D,MAAI,qBAAqB,EACvB,OAAM,IAAI,MAAM;AAGlB,SAAO,CAAC,uFAAuF,EAAE,KAAK,qBACpG,aAAa,KAAK,OAAO,CAC1B,CACE,WAAW,KAAK,IAAI,OAAO,mBAAmB,CAAC,CAC/C,WAAW,KAAK,IAAI,OAAO,mBAAmB,CAAC,CAAC,QAAQ,CAAC;CAC7D;CAED,AAAQ,qBACNC,aACA,SAAS,GACD;EACR,MAAM,SAAS;AASf,MAAI,UAAU,QAAQ;GACpB,IAAI,WAAW;GACf,IAAIC;AACJ,OAAI,MAAM,QAAQ,OAAO,KAAK,EAAE;IAC9B,MAAM,UAAU,OAAO,KAAK,UAAU,CAACC,WAASA,WAAS,OAAO;AAChE,QAAI,YAAY,IAAI;KAClB,WAAW;KACX,OAAO,KAAK,OAAO,SAAS,EAAE;IAC/B;IACD,OAAO,OAAO,KAAK,KAAK,MAAM;GAC/B,OACC,OAAO,OAAO;AAGhB,OAAI,OAAO,SAAS,YAAY,OAAO,YAAY;IACjD,MAAMC,gBAAc,OAAO,cACvB,CAAC,IAAI,EAAE,OAAO,aAAa,GAC3B;IACJ,MAAM,aAAa,OAAO,QAAQ,OAAO,WAAW,CACjD,IAAI,CAAC,CAAC,KAAK,MAAM,KAAK;KACrB,MAAM,aAAa,OAAO,UAAU,SAAS,IAAI,GAC7C,KACA;AACJ,YAAO,GAAG,IAAI,OAAO,OAAO,CAAC,CAAC,EAAE,IAAI,GAAG,EAAE,KAAK,qBAC5C,OACA,SAAS,EACV,GAAG,YAAY;IACjB,EAAC,CACD,KAAK,KAAK;AACb,WAAO,CAAC,GAAG,EAAE,WAAW,EAAE,EAAE,IAAI,OAAO,SAAS,EAAE,CAAC,CAAC,EAAEA,eAAa;GACpE;AACD,OAAI,OAAO,SAAS,WAAW,OAAO,OAAO;IAC3C,MAAMA,gBAAc,OAAO,cACvB,CAAC,IAAI,EAAE,OAAO,aAAa,GAC3B;AACJ,WAAO,CAAC,QAAQ,EAAE,IAAI,OAAO,OAAO,GAAG,KAAK,qBAC1C,OAAO,OACP,SAAS,EACV,CAAC,EAAE,EAAE,IAAI,OAAO,SAAS,EAAE,CAAC,EAAE,EAAEA,eAAa;GAC/C;GACD,MAAM,aAAa,WAAW,gBAAgB;GAC9C,MAAM,cAAc,OAAO,cAAc,CAAC,IAAI,EAAE,OAAO,aAAa,GAAG;AACvE,UAAO,GAAG,OAAO,cAAc,YAAY;EAC5C;AAED,MAAI,WAAW,OACb,QAAO,OAAO,MACX,IAAI,CAAC,MAAM,KAAK,qBAAqB,GAAG,OAAO,CAAC,CAChD,KAAK,CAAC,EAAE,EAAE,IAAI,OAAO,SAAS,EAAE,EAAE,CAAC;AAGxC,QAAM,IAAI,MAAM;CACjB;CAED,OAAO,cAAwCP,QAAW;AACxD,SAAO,IAAI,KAAQ;CACpB;CAED,OAAO,yBACLC,SACA;EACA,MAAM,YAAY,EAAE,OAClB,OAAO,YACL,OAAO,QAAQ,QAAQ,CAAC,IACtB,CAAC,CAAC,MAAM,YAAY,KAClB,CAAC,MAAM,EAAE,QAAQ,CAAC,SAAS,YAAY,AAAC,EAC3C,CACF,CACF;AAED,SAAO,IAAI,KAAuB;CACnC;AACF;;;;;AAYD,IAAsB,mCAAtB,cAGU,iBAAoB;CAC5B,AAAQ;CAER,YAAY,EAAE,aAAwD,EAAE;EACtE,MAAM,GAAG,UAAU;EACnB,KAAK,wBAAwB,IAAI,mCAC/B;CAEH;CAUD,MAAM,MAAMC,MAA0B;EACpC,IAAI;AACJ,MAAI;GACF,cAAc,MAAM,KAAK,sBAAsB,MAAM,KAAK;EAC3D,SAAQ,GAAG;AACV,SAAM,IAAI,sBACR,CAAC,wBAAwB,EAAE,KAAK,UAAU,EAAE,GAAG,EAC/C;EAEH;AAED,SAAO,KAAK,gBAAgB,YAAY;CACzC;CAED,wBAAgC;AAC9B,SAAO,KAAK,sBAAsB,uBAAuB;CAC1D;AACF"}

package/dist/retrievers/contextual_compression.cjs

@@ -0,0 +1,53 @@
+const require_rolldown_runtime = require('../_virtual/rolldown_runtime.cjs');
+const __langchain_core_retrievers = require_rolldown_runtime.__toESM(require("@langchain/core/retrievers"));
+
+//#region src/retrievers/contextual_compression.ts
+var contextual_compression_exports = {};
+require_rolldown_runtime.__export(contextual_compression_exports, { ContextualCompressionRetriever: () => ContextualCompressionRetriever });
+/**
+* A retriever that wraps a base retriever and compresses the results. It
+* retrieves relevant documents based on a given query and then compresses
+* these documents using a specified document compressor.
+* @example
+* ```typescript
+* const retriever = new ContextualCompressionRetriever({
+*   baseCompressor: new LLMChainExtractor(),
+*   baseRetriever: new HNSWLib().asRetriever(),
+* });
+* const retrievedDocs = await retriever.invoke(
+*   "What did the speaker say about Justice Breyer?",
+* );
+* ```
+*/
+var ContextualCompressionRetriever = class extends __langchain_core_retrievers.BaseRetriever {
+	static lc_name() {
+		return "ContextualCompressionRetriever";
+	}
+	lc_namespace = [
+		"langchain",
+		"retrievers",
+		"contextual_compression"
+	];
+	baseCompressor;
+	baseRetriever;
+	constructor(fields) {
+		super(fields);
+		this.baseCompressor = fields.baseCompressor;
+		this.baseRetriever = fields.baseRetriever;
+	}
+	async _getRelevantDocuments(query, runManager) {
+		const docs = await this.baseRetriever.invoke(query, runManager?.getChild("base_retriever"));
+		const compressedDocs = await this.baseCompressor.compressDocuments(docs, query, runManager?.getChild("base_compressor"));
+		return compressedDocs;
+	}
+};
+
+//#endregion
+exports.ContextualCompressionRetriever = ContextualCompressionRetriever;
+Object.defineProperty(exports, 'contextual_compression_exports', {
+  enumerable: true,
+  get: function () {
+    return contextual_compression_exports;
+  }
+});
+//# sourceMappingURL=contextual_compression.cjs.map

package/dist/retrievers/contextual_compression.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"contextual_compression.cjs","names":["BaseRetriever","fields: ContextualCompressionRetrieverArgs","query: string","runManager?: CallbackManagerForRetrieverRun"],"sources":["../../src/retrievers/contextual_compression.ts"],"sourcesContent":["import {\n  BaseRetriever,\n  type BaseRetrieverInput,\n  type BaseRetrieverInterface,\n} from \"@langchain/core/retrievers\";\nimport type { DocumentInterface } from \"@langchain/core/documents\";\nimport { CallbackManagerForRetrieverRun } from \"@langchain/core/callbacks/manager\";\nimport { BaseDocumentCompressor } from \"./document_compressors/index.js\";\n\n/**\n * Interface for the arguments required to construct a\n * ContextualCompressionRetriever. It extends the BaseRetrieverInput\n * interface with two additional fields: baseCompressor and baseRetriever.\n */\nexport interface ContextualCompressionRetrieverArgs extends BaseRetrieverInput {\n  baseCompressor: BaseDocumentCompressor;\n  baseRetriever: BaseRetrieverInterface;\n}\n\n/**\n * A retriever that wraps a base retriever and compresses the results. It\n * retrieves relevant documents based on a given query and then compresses\n * these documents using a specified document compressor.\n * @example\n * ```typescript\n * const retriever = new ContextualCompressionRetriever({\n *   baseCompressor: new LLMChainExtractor(),\n *   baseRetriever: new HNSWLib().asRetriever(),\n * });\n * const retrievedDocs = await retriever.invoke(\n *   \"What did the speaker say about Justice Breyer?\",\n * );\n * ```\n */\nexport class ContextualCompressionRetriever extends BaseRetriever {\n  static lc_name() {\n    return \"ContextualCompressionRetriever\";\n  }\n\n  lc_namespace = [\"langchain\", \"retrievers\", \"contextual_compression\"];\n\n  baseCompressor: BaseDocumentCompressor;\n\n  baseRetriever: BaseRetrieverInterface;\n\n  constructor(fields: ContextualCompressionRetrieverArgs) {\n    super(fields);\n\n    this.baseCompressor = fields.baseCompressor;\n    this.baseRetriever = fields.baseRetriever;\n  }\n\n  async _getRelevantDocuments(\n    query: string,\n    runManager?: CallbackManagerForRetrieverRun\n  ): Promise<DocumentInterface[]> {\n    const docs = await this.baseRetriever.invoke(\n      query,\n      runManager?.getChild(\"base_retriever\")\n    );\n    const compressedDocs = await this.baseCompressor.compressDocuments(\n      docs,\n      query,\n      runManager?.getChild(\"base_compressor\")\n    );\n    return compressedDocs;\n  }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;AAkCA,IAAa,iCAAb,cAAoDA,0CAAc;CAChE,OAAO,UAAU;AACf,SAAO;CACR;CAED,eAAe;EAAC;EAAa;EAAc;CAAyB;CAEpE;CAEA;CAEA,YAAYC,QAA4C;EACtD,MAAM,OAAO;EAEb,KAAK,iBAAiB,OAAO;EAC7B,KAAK,gBAAgB,OAAO;CAC7B;CAED,MAAM,sBACJC,OACAC,YAC8B;EAC9B,MAAM,OAAO,MAAM,KAAK,cAAc,OACpC,OACA,YAAY,SAAS,iBAAiB,CACvC;EACD,MAAM,iBAAiB,MAAM,KAAK,eAAe,kBAC/C,MACA,OACA,YAAY,SAAS,kBAAkB,CACxC;AACD,SAAO;CACR;AACF"}

package/dist/retrievers/contextual_compression.d.cts

@@ -0,0 +1,42 @@
+import { BaseDocumentCompressor } from "./document_compressors/index.cjs";
+import { CallbackManagerForRetrieverRun } from "@langchain/core/callbacks/manager";
+import { BaseRetriever, BaseRetrieverInput, BaseRetrieverInterface } from "@langchain/core/retrievers";
+import { DocumentInterface } from "@langchain/core/documents";
+
+//#region src/retrievers/contextual_compression.d.ts
+
+/**
+ * Interface for the arguments required to construct a
+ * ContextualCompressionRetriever. It extends the BaseRetrieverInput
+ * interface with two additional fields: baseCompressor and baseRetriever.
+ */
+interface ContextualCompressionRetrieverArgs extends BaseRetrieverInput {
+  baseCompressor: BaseDocumentCompressor;
+  baseRetriever: BaseRetrieverInterface;
+}
+/**
+ * A retriever that wraps a base retriever and compresses the results. It
+ * retrieves relevant documents based on a given query and then compresses
+ * these documents using a specified document compressor.
+ * @example
+ * ```typescript
+ * const retriever = new ContextualCompressionRetriever({
+ *   baseCompressor: new LLMChainExtractor(),
+ *   baseRetriever: new HNSWLib().asRetriever(),
+ * });
+ * const retrievedDocs = await retriever.invoke(
+ *   "What did the speaker say about Justice Breyer?",
+ * );
+ * ```
+ */
+declare class ContextualCompressionRetriever extends BaseRetriever {
+  static lc_name(): string;
+  lc_namespace: string[];
+  baseCompressor: BaseDocumentCompressor;
+  baseRetriever: BaseRetrieverInterface;
+  constructor(fields: ContextualCompressionRetrieverArgs);
+  _getRelevantDocuments(query: string, runManager?: CallbackManagerForRetrieverRun): Promise<DocumentInterface[]>;
+}
+//#endregion
+export { ContextualCompressionRetriever, ContextualCompressionRetrieverArgs };
+//# sourceMappingURL=contextual_compression.d.cts.map

package/dist/retrievers/contextual_compression.d.cts.map

@@ -0,0 +1 @@
+{"version":3,"file":"contextual_compression.d.cts","names":["BaseRetriever","BaseRetrieverInput","BaseRetrieverInterface","DocumentInterface","CallbackManagerForRetrieverRun","BaseDocumentCompressor","ContextualCompressionRetrieverArgs","ContextualCompressionRetriever","Promise"],"sources":["../../src/retrievers/contextual_compression.d.ts"],"sourcesContent":["import { BaseRetriever, type BaseRetrieverInput, type BaseRetrieverInterface } from \"@langchain/core/retrievers\";\nimport type { DocumentInterface } from \"@langchain/core/documents\";\nimport { CallbackManagerForRetrieverRun } from \"@langchain/core/callbacks/manager\";\nimport { BaseDocumentCompressor } from \"./document_compressors/index.js\";\n/**\n * Interface for the arguments required to construct a\n * ContextualCompressionRetriever. It extends the BaseRetrieverInput\n * interface with two additional fields: baseCompressor and baseRetriever.\n */\nexport interface ContextualCompressionRetrieverArgs extends BaseRetrieverInput {\n    baseCompressor: BaseDocumentCompressor;\n    baseRetriever: BaseRetrieverInterface;\n}\n/**\n * A retriever that wraps a base retriever and compresses the results. It\n * retrieves relevant documents based on a given query and then compresses\n * these documents using a specified document compressor.\n * @example\n * ```typescript\n * const retriever = new ContextualCompressionRetriever({\n *   baseCompressor: new LLMChainExtractor(),\n *   baseRetriever: new HNSWLib().asRetriever(),\n * });\n * const retrievedDocs = await retriever.invoke(\n *   \"What did the speaker say about Justice Breyer?\",\n * );\n * ```\n */\nexport declare class ContextualCompressionRetriever extends BaseRetriever {\n    static lc_name(): string;\n    lc_namespace: string[];\n    baseCompressor: BaseDocumentCompressor;\n    baseRetriever: BaseRetrieverInterface;\n    constructor(fields: ContextualCompressionRetrieverArgs);\n    _getRelevantDocuments(query: string, runManager?: CallbackManagerForRetrieverRun): Promise<DocumentInterface[]>;\n}\n"],"mappings":";;;;;;;;;AASA;;;AAEmBE,UAFFI,kCAAAA,SAA2CL,kBAEzCC,CAAAA;EAAsB,cAFmBD,EACxCI,sBADwCJ;EAAkB,aAAA,EAE3DC,sBAF2D;AAmB9E;;;;;;;;;AAAyE;;;;;;;cAApDK,8BAAAA,SAAuCP,aAAAA;;;kBAGxCK;iBACDH;sBACKI;oDAC8BF,iCAAiCI,QAAQL"}

package/dist/retrievers/contextual_compression.d.ts

@@ -0,0 +1,42 @@
+import { BaseDocumentCompressor } from "./document_compressors/index.js";
+import { DocumentInterface } from "@langchain/core/documents";
+import { CallbackManagerForRetrieverRun } from "@langchain/core/callbacks/manager";
+import { BaseRetriever, BaseRetrieverInput, BaseRetrieverInterface } from "@langchain/core/retrievers";
+
+//#region src/retrievers/contextual_compression.d.ts
+
+/**
+ * Interface for the arguments required to construct a
+ * ContextualCompressionRetriever. It extends the BaseRetrieverInput
+ * interface with two additional fields: baseCompressor and baseRetriever.
+ */
+interface ContextualCompressionRetrieverArgs extends BaseRetrieverInput {
+  baseCompressor: BaseDocumentCompressor;
+  baseRetriever: BaseRetrieverInterface;
+}
+/**
+ * A retriever that wraps a base retriever and compresses the results. It
+ * retrieves relevant documents based on a given query and then compresses
+ * these documents using a specified document compressor.
+ * @example
+ * ```typescript
+ * const retriever = new ContextualCompressionRetriever({
+ *   baseCompressor: new LLMChainExtractor(),
+ *   baseRetriever: new HNSWLib().asRetriever(),
+ * });
+ * const retrievedDocs = await retriever.invoke(
+ *   "What did the speaker say about Justice Breyer?",
+ * );
+ * ```
+ */
+declare class ContextualCompressionRetriever extends BaseRetriever {
+  static lc_name(): string;
+  lc_namespace: string[];
+  baseCompressor: BaseDocumentCompressor;
+  baseRetriever: BaseRetrieverInterface;
+  constructor(fields: ContextualCompressionRetrieverArgs);
+  _getRelevantDocuments(query: string, runManager?: CallbackManagerForRetrieverRun): Promise<DocumentInterface[]>;
+}
+//#endregion
+export { ContextualCompressionRetriever, ContextualCompressionRetrieverArgs };
+//# sourceMappingURL=contextual_compression.d.ts.map

package/dist/retrievers/contextual_compression.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"contextual_compression.d.ts","names":["BaseRetriever","BaseRetrieverInput","BaseRetrieverInterface","DocumentInterface","CallbackManagerForRetrieverRun","BaseDocumentCompressor","ContextualCompressionRetrieverArgs","ContextualCompressionRetriever","Promise"],"sources":["../../src/retrievers/contextual_compression.d.ts"],"sourcesContent":["import { BaseRetriever, type BaseRetrieverInput, type BaseRetrieverInterface } from \"@langchain/core/retrievers\";\nimport type { DocumentInterface } from \"@langchain/core/documents\";\nimport { CallbackManagerForRetrieverRun } from \"@langchain/core/callbacks/manager\";\nimport { BaseDocumentCompressor } from \"./document_compressors/index.js\";\n/**\n * Interface for the arguments required to construct a\n * ContextualCompressionRetriever. It extends the BaseRetrieverInput\n * interface with two additional fields: baseCompressor and baseRetriever.\n */\nexport interface ContextualCompressionRetrieverArgs extends BaseRetrieverInput {\n    baseCompressor: BaseDocumentCompressor;\n    baseRetriever: BaseRetrieverInterface;\n}\n/**\n * A retriever that wraps a base retriever and compresses the results. It\n * retrieves relevant documents based on a given query and then compresses\n * these documents using a specified document compressor.\n * @example\n * ```typescript\n * const retriever = new ContextualCompressionRetriever({\n *   baseCompressor: new LLMChainExtractor(),\n *   baseRetriever: new HNSWLib().asRetriever(),\n * });\n * const retrievedDocs = await retriever.invoke(\n *   \"What did the speaker say about Justice Breyer?\",\n * );\n * ```\n */\nexport declare class ContextualCompressionRetriever extends BaseRetriever {\n    static lc_name(): string;\n    lc_namespace: string[];\n    baseCompressor: BaseDocumentCompressor;\n    baseRetriever: BaseRetrieverInterface;\n    constructor(fields: ContextualCompressionRetrieverArgs);\n    _getRelevantDocuments(query: string, runManager?: CallbackManagerForRetrieverRun): Promise<DocumentInterface[]>;\n}\n"],"mappings":";;;;;;;;;AASA;;;AAEmBE,UAFFI,kCAAAA,SAA2CL,kBAEzCC,CAAAA;EAAsB,cAFmBD,EACxCI,sBADwCJ;EAAkB,aAAA,EAE3DC,sBAF2D;AAmB9E;;;;;;;;;AAAyE;;;;;;;cAApDK,8BAAAA,SAAuCP,aAAAA;;;kBAGxCK;iBACDH;sBACKI;oDAC8BF,iCAAiCI,QAAQL"}

package/dist/retrievers/contextual_compression.js

@@ -0,0 +1,47 @@
+import { __export } from "../_virtual/rolldown_runtime.js";
+import { BaseRetriever } from "@langchain/core/retrievers";
+
+//#region src/retrievers/contextual_compression.ts
+var contextual_compression_exports = {};
+__export(contextual_compression_exports, { ContextualCompressionRetriever: () => ContextualCompressionRetriever });
+/**
+* A retriever that wraps a base retriever and compresses the results. It
+* retrieves relevant documents based on a given query and then compresses
+* these documents using a specified document compressor.
+* @example
+* ```typescript
+* const retriever = new ContextualCompressionRetriever({
+*   baseCompressor: new LLMChainExtractor(),
+*   baseRetriever: new HNSWLib().asRetriever(),
+* });
+* const retrievedDocs = await retriever.invoke(
+*   "What did the speaker say about Justice Breyer?",
+* );
+* ```
+*/
+var ContextualCompressionRetriever = class extends BaseRetriever {
+	static lc_name() {
+		return "ContextualCompressionRetriever";
+	}
+	lc_namespace = [
+		"langchain",
+		"retrievers",
+		"contextual_compression"
+	];
+	baseCompressor;
+	baseRetriever;
+	constructor(fields) {
+		super(fields);
+		this.baseCompressor = fields.baseCompressor;
+		this.baseRetriever = fields.baseRetriever;
+	}
+	async _getRelevantDocuments(query, runManager) {
+		const docs = await this.baseRetriever.invoke(query, runManager?.getChild("base_retriever"));
+		const compressedDocs = await this.baseCompressor.compressDocuments(docs, query, runManager?.getChild("base_compressor"));
+		return compressedDocs;
+	}
+};
+
+//#endregion
+export { ContextualCompressionRetriever, contextual_compression_exports };
+//# sourceMappingURL=contextual_compression.js.map

package/dist/retrievers/contextual_compression.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"contextual_compression.js","names":["fields: ContextualCompressionRetrieverArgs","query: string","runManager?: CallbackManagerForRetrieverRun"],"sources":["../../src/retrievers/contextual_compression.ts"],"sourcesContent":["import {\n  BaseRetriever,\n  type BaseRetrieverInput,\n  type BaseRetrieverInterface,\n} from \"@langchain/core/retrievers\";\nimport type { DocumentInterface } from \"@langchain/core/documents\";\nimport { CallbackManagerForRetrieverRun } from \"@langchain/core/callbacks/manager\";\nimport { BaseDocumentCompressor } from \"./document_compressors/index.js\";\n\n/**\n * Interface for the arguments required to construct a\n * ContextualCompressionRetriever. It extends the BaseRetrieverInput\n * interface with two additional fields: baseCompressor and baseRetriever.\n */\nexport interface ContextualCompressionRetrieverArgs extends BaseRetrieverInput {\n  baseCompressor: BaseDocumentCompressor;\n  baseRetriever: BaseRetrieverInterface;\n}\n\n/**\n * A retriever that wraps a base retriever and compresses the results. It\n * retrieves relevant documents based on a given query and then compresses\n * these documents using a specified document compressor.\n * @example\n * ```typescript\n * const retriever = new ContextualCompressionRetriever({\n *   baseCompressor: new LLMChainExtractor(),\n *   baseRetriever: new HNSWLib().asRetriever(),\n * });\n * const retrievedDocs = await retriever.invoke(\n *   \"What did the speaker say about Justice Breyer?\",\n * );\n * ```\n */\nexport class ContextualCompressionRetriever extends BaseRetriever {\n  static lc_name() {\n    return \"ContextualCompressionRetriever\";\n  }\n\n  lc_namespace = [\"langchain\", \"retrievers\", \"contextual_compression\"];\n\n  baseCompressor: BaseDocumentCompressor;\n\n  baseRetriever: BaseRetrieverInterface;\n\n  constructor(fields: ContextualCompressionRetrieverArgs) {\n    super(fields);\n\n    this.baseCompressor = fields.baseCompressor;\n    this.baseRetriever = fields.baseRetriever;\n  }\n\n  async _getRelevantDocuments(\n    query: string,\n    runManager?: CallbackManagerForRetrieverRun\n  ): Promise<DocumentInterface[]> {\n    const docs = await this.baseRetriever.invoke(\n      query,\n      runManager?.getChild(\"base_retriever\")\n    );\n    const compressedDocs = await this.baseCompressor.compressDocuments(\n      docs,\n      query,\n      runManager?.getChild(\"base_compressor\")\n    );\n    return compressedDocs;\n  }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;AAkCA,IAAa,iCAAb,cAAoD,cAAc;CAChE,OAAO,UAAU;AACf,SAAO;CACR;CAED,eAAe;EAAC;EAAa;EAAc;CAAyB;CAEpE;CAEA;CAEA,YAAYA,QAA4C;EACtD,MAAM,OAAO;EAEb,KAAK,iBAAiB,OAAO;EAC7B,KAAK,gBAAgB,OAAO;CAC7B;CAED,MAAM,sBACJC,OACAC,YAC8B;EAC9B,MAAM,OAAO,MAAM,KAAK,cAAc,OACpC,OACA,YAAY,SAAS,iBAAiB,CACvC;EACD,MAAM,iBAAiB,MAAM,KAAK,eAAe,kBAC/C,MACA,OACA,YAAY,SAAS,kBAAkB,CACxC;AACD,SAAO;CACR;AACF"}

package/dist/retrievers/document_compressors/chain_extract.cjs

@@ -0,0 +1,103 @@
+const require_rolldown_runtime = require('../../_virtual/rolldown_runtime.cjs');
+const require_llm_chain = require('../../chains/llm_chain.cjs');
+const require_retrievers_document_compressors_index = require('./index.cjs');
+const require_chain_extract_prompt = require('./chain_extract_prompt.cjs');
+const __langchain_core_documents = require_rolldown_runtime.__toESM(require("@langchain/core/documents"));
+const __langchain_core_prompts = require_rolldown_runtime.__toESM(require("@langchain/core/prompts"));
+const __langchain_core_output_parsers = require_rolldown_runtime.__toESM(require("@langchain/core/output_parsers"));
+
+//#region src/retrievers/document_compressors/chain_extract.ts
+var chain_extract_exports = {};
+require_rolldown_runtime.__export(chain_extract_exports, { LLMChainExtractor: () => LLMChainExtractor });
+function defaultGetInput(query, doc) {
+	return {
+		question: query,
+		context: doc.pageContent
+	};
+}
+var NoOutputParser = class extends __langchain_core_output_parsers.BaseOutputParser {
+	lc_namespace = [
+		"langchain",
+		"retrievers",
+		"document_compressors",
+		"chain_extract"
+	];
+	noOutputStr = "NO_OUTPUT";
+	parse(text) {
+		const cleanedText = text.trim();
+		if (cleanedText === this.noOutputStr) return Promise.resolve("");
+		return Promise.resolve(cleanedText);
+	}
+	getFormatInstructions() {
+		throw new Error("Method not implemented.");
+	}
+};
+function getDefaultChainPrompt() {
+	const outputParser = new NoOutputParser();
+	const template = require_chain_extract_prompt.PROMPT_TEMPLATE(outputParser.noOutputStr);
+	return new __langchain_core_prompts.PromptTemplate({
+		template,
+		inputVariables: ["question", "context"],
+		outputParser
+	});
+}
+/**
+* A class that uses an LLM chain to extract relevant parts of documents.
+* It extends the BaseDocumentCompressor class.
+*/
+var LLMChainExtractor = class LLMChainExtractor extends require_retrievers_document_compressors_index.BaseDocumentCompressor {
+	llmChain;
+	getInput = defaultGetInput;
+	constructor({ llmChain, getInput }) {
+		super();
+		this.llmChain = llmChain;
+		this.getInput = getInput;
+	}
+	/**
+	* Compresses a list of documents based on the output of an LLM chain.
+	* @param documents The list of documents to be compressed.
+	* @param query The query to be used for document compression.
+	* @returns A list of compressed documents.
+	*/
+	async compressDocuments(documents, query) {
+		const compressedDocs = await Promise.all(documents.map(async (doc) => {
+			const input = this.getInput(query, doc);
+			const output = await this.llmChain.predict(input);
+			return output.length > 0 ? new __langchain_core_documents.Document({
+				pageContent: output,
+				metadata: doc.metadata
+			}) : void 0;
+		}));
+		return compressedDocs.filter((doc) => doc !== void 0);
+	}
+	/**
+	* Creates a new instance of LLMChainExtractor from a given LLM, prompt
+	* template, and getInput function.
+	* @param llm The BaseLanguageModel instance used for document extraction.
+	* @param prompt The PromptTemplate instance used for document extraction.
+	* @param getInput A function used for constructing the chain input from the query and a Document.
+	* @returns A new instance of LLMChainExtractor.
+	*/
+	static fromLLM(llm, prompt, getInput) {
+		const _prompt = prompt || getDefaultChainPrompt();
+		const _getInput = getInput || defaultGetInput;
+		const llmChain = new require_llm_chain.LLMChain({
+			llm,
+			prompt: _prompt
+		});
+		return new LLMChainExtractor({
+			llmChain,
+			getInput: _getInput
+		});
+	}
+};
+
+//#endregion
+exports.LLMChainExtractor = LLMChainExtractor;
+Object.defineProperty(exports, 'chain_extract_exports', {
+  enumerable: true,
+  get: function () {
+    return chain_extract_exports;
+  }
+});
+//# sourceMappingURL=chain_extract.cjs.map

package/dist/retrievers/document_compressors/chain_extract.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"chain_extract.cjs","names":["query: string","doc: DocumentInterface","BaseOutputParser","text: string","PROMPT_TEMPLATE","PromptTemplate","BaseDocumentCompressor","documents: DocumentInterface[]","Document","llm: BaseLanguageModelInterface","prompt?: PromptTemplate","getInput?: (\n      query: string,\n      doc: DocumentInterface\n    ) => Record<string, unknown>","LLMChain"],"sources":["../../../src/retrievers/document_compressors/chain_extract.ts"],"sourcesContent":["import type { BaseLanguageModelInterface } from \"@langchain/core/language_models/base\";\nimport { type DocumentInterface, Document } from \"@langchain/core/documents\";\nimport { PromptTemplate } from \"@langchain/core/prompts\";\nimport { BaseOutputParser } from \"@langchain/core/output_parsers\";\nimport { LLMChain } from \"../../chains/llm_chain.js\";\nimport { BaseDocumentCompressor } from \"./index.js\";\nimport { PROMPT_TEMPLATE } from \"./chain_extract_prompt.js\";\n\nfunction defaultGetInput(\n  query: string,\n  doc: DocumentInterface\n): Record<string, unknown> {\n  return { question: query, context: doc.pageContent };\n}\n\nclass NoOutputParser extends BaseOutputParser<string> {\n  lc_namespace = [\n    \"langchain\",\n    \"retrievers\",\n    \"document_compressors\",\n    \"chain_extract\",\n  ];\n\n  noOutputStr = \"NO_OUTPUT\";\n\n  parse(text: string): Promise<string> {\n    const cleanedText = text.trim();\n    if (cleanedText === this.noOutputStr) {\n      return Promise.resolve(\"\");\n    }\n    return Promise.resolve(cleanedText);\n  }\n\n  getFormatInstructions(): string {\n    throw new Error(\"Method not implemented.\");\n  }\n}\n\nfunction getDefaultChainPrompt(): PromptTemplate {\n  const outputParser = new NoOutputParser();\n  const template = PROMPT_TEMPLATE(outputParser.noOutputStr);\n  return new PromptTemplate({\n    template,\n    inputVariables: [\"question\", \"context\"],\n    outputParser,\n  });\n}\n\n/**\n * Interface for the arguments required to create an instance of\n * LLMChainExtractor.\n */\nexport interface LLMChainExtractorArgs {\n  llmChain: LLMChain;\n  getInput: (query: string, doc: DocumentInterface) => Record<string, unknown>;\n}\n\n/**\n * A class that uses an LLM chain to extract relevant parts of documents.\n * It extends the BaseDocumentCompressor class.\n */\nexport class LLMChainExtractor extends BaseDocumentCompressor {\n  llmChain: LLMChain;\n\n  getInput: (query: string, doc: DocumentInterface) => Record<string, unknown> =\n    defaultGetInput;\n\n  constructor({ llmChain, getInput }: LLMChainExtractorArgs) {\n    super();\n    this.llmChain = llmChain;\n    this.getInput = getInput;\n  }\n\n  /**\n   * Compresses a list of documents based on the output of an LLM chain.\n   * @param documents The list of documents to be compressed.\n   * @param query The query to be used for document compression.\n   * @returns A list of compressed documents.\n   */\n  async compressDocuments(\n    documents: DocumentInterface[],\n    query: string\n  ): Promise<DocumentInterface[]> {\n    const compressedDocs = await Promise.all(\n      documents.map(async (doc) => {\n        const input = this.getInput(query, doc);\n        const output = await this.llmChain.predict(input);\n        return output.length > 0\n          ? new Document({\n              pageContent: output,\n              metadata: doc.metadata,\n            })\n          : undefined;\n      })\n    );\n    return compressedDocs.filter((doc): doc is Document => doc !== undefined);\n  }\n\n  /**\n   * Creates a new instance of LLMChainExtractor from a given LLM, prompt\n   * template, and getInput function.\n   * @param llm The BaseLanguageModel instance used for document extraction.\n   * @param prompt The PromptTemplate instance used for document extraction.\n   * @param getInput A function used for constructing the chain input from the query and a Document.\n   * @returns A new instance of LLMChainExtractor.\n   */\n  static fromLLM(\n    llm: BaseLanguageModelInterface,\n    prompt?: PromptTemplate,\n    getInput?: (\n      query: string,\n      doc: DocumentInterface\n    ) => Record<string, unknown>\n  ): LLMChainExtractor {\n    const _prompt = prompt || getDefaultChainPrompt();\n    const _getInput = getInput || defaultGetInput;\n    const llmChain = new LLMChain({ llm, prompt: _prompt });\n    return new LLMChainExtractor({ llmChain, getInput: _getInput });\n  }\n}\n"],"mappings":";;;;;;;;;;;AAQA,SAAS,gBACPA,OACAC,KACyB;AACzB,QAAO;EAAE,UAAU;EAAO,SAAS,IAAI;CAAa;AACrD;AAED,IAAM,iBAAN,cAA6BC,iDAAyB;CACpD,eAAe;EACb;EACA;EACA;EACA;CACD;CAED,cAAc;CAEd,MAAMC,MAA+B;EACnC,MAAM,cAAc,KAAK,MAAM;AAC/B,MAAI,gBAAgB,KAAK,YACvB,QAAO,QAAQ,QAAQ,GAAG;AAE5B,SAAO,QAAQ,QAAQ,YAAY;CACpC;CAED,wBAAgC;AAC9B,QAAM,IAAI,MAAM;CACjB;AACF;AAED,SAAS,wBAAwC;CAC/C,MAAM,eAAe,IAAI;CACzB,MAAM,WAAWC,6CAAgB,aAAa,YAAY;AAC1D,QAAO,IAAIC,wCAAe;EACxB;EACA,gBAAgB,CAAC,YAAY,SAAU;EACvC;CACD;AACF;;;;;AAeD,IAAa,oBAAb,MAAa,0BAA0BC,qEAAuB;CAC5D;CAEA,WACE;CAEF,YAAY,EAAE,UAAU,UAAiC,EAAE;EACzD,OAAO;EACP,KAAK,WAAW;EAChB,KAAK,WAAW;CACjB;;;;;;;CAQD,MAAM,kBACJC,WACAP,OAC8B;EAC9B,MAAM,iBAAiB,MAAM,QAAQ,IACnC,UAAU,IAAI,OAAO,QAAQ;GAC3B,MAAM,QAAQ,KAAK,SAAS,OAAO,IAAI;GACvC,MAAM,SAAS,MAAM,KAAK,SAAS,QAAQ,MAAM;AACjD,UAAO,OAAO,SAAS,IACnB,IAAIQ,oCAAS;IACX,aAAa;IACb,UAAU,IAAI;GACf,KACD;EACL,EAAC,CACH;AACD,SAAO,eAAe,OAAO,CAAC,QAAyB,QAAQ,OAAU;CAC1E;;;;;;;;;CAUD,OAAO,QACLC,KACAC,QACAC,UAImB;EACnB,MAAM,UAAU,UAAU,uBAAuB;EACjD,MAAM,YAAY,YAAY;EAC9B,MAAM,WAAW,IAAIC,2BAAS;GAAE;GAAK,QAAQ;EAAS;AACtD,SAAO,IAAI,kBAAkB;GAAE;GAAU,UAAU;EAAW;CAC/D;AACF"}

package/dist/retrievers/document_compressors/chain_extract.d.cts

@@ -0,0 +1,47 @@
+import { LLMChain } from "../../chains/llm_chain.cjs";
+import { BaseDocumentCompressor } from "./index.cjs";
+import { PromptTemplate } from "@langchain/core/prompts";
+import { BaseLanguageModelInterface } from "@langchain/core/language_models/base";
+import { DocumentInterface } from "@langchain/core/documents";
+
+//#region src/retrievers/document_compressors/chain_extract.d.ts
+
+/**
+ * Interface for the arguments required to create an instance of
+ * LLMChainExtractor.
+ */
+interface LLMChainExtractorArgs {
+  llmChain: LLMChain;
+  getInput: (query: string, doc: DocumentInterface) => Record<string, unknown>;
+}
+/**
+ * A class that uses an LLM chain to extract relevant parts of documents.
+ * It extends the BaseDocumentCompressor class.
+ */
+declare class LLMChainExtractor extends BaseDocumentCompressor {
+  llmChain: LLMChain;
+  getInput: (query: string, doc: DocumentInterface) => Record<string, unknown>;
+  constructor({
+    llmChain,
+    getInput
+  }: LLMChainExtractorArgs);
+  /**
+   * Compresses a list of documents based on the output of an LLM chain.
+   * @param documents The list of documents to be compressed.
+   * @param query The query to be used for document compression.
+   * @returns A list of compressed documents.
+   */
+  compressDocuments(documents: DocumentInterface[], query: string): Promise<DocumentInterface[]>;
+  /**
+   * Creates a new instance of LLMChainExtractor from a given LLM, prompt
+   * template, and getInput function.
+   * @param llm The BaseLanguageModel instance used for document extraction.
+   * @param prompt The PromptTemplate instance used for document extraction.
+   * @param getInput A function used for constructing the chain input from the query and a Document.
+   * @returns A new instance of LLMChainExtractor.
+   */
+  static fromLLM(llm: BaseLanguageModelInterface, prompt?: PromptTemplate, getInput?: (query: string, doc: DocumentInterface) => Record<string, unknown>): LLMChainExtractor;
+}
+//#endregion
+export { LLMChainExtractor, LLMChainExtractorArgs };
+//# sourceMappingURL=chain_extract.d.cts.map

package/dist/retrievers/document_compressors/chain_extract.d.cts.map

@@ -0,0 +1 @@
+{"version":3,"file":"chain_extract.d.cts","names":["BaseLanguageModelInterface","DocumentInterface","PromptTemplate","LLMChain","BaseDocumentCompressor","LLMChainExtractorArgs","Record","LLMChainExtractor","llmChain","getInput","Promise"],"sources":["../../../src/retrievers/document_compressors/chain_extract.d.ts"],"sourcesContent":["import type { BaseLanguageModelInterface } from \"@langchain/core/language_models/base\";\nimport { type DocumentInterface } from \"@langchain/core/documents\";\nimport { PromptTemplate } from \"@langchain/core/prompts\";\nimport { LLMChain } from \"../../chains/llm_chain.js\";\nimport { BaseDocumentCompressor } from \"./index.js\";\n/**\n * Interface for the arguments required to create an instance of\n * LLMChainExtractor.\n */\nexport interface LLMChainExtractorArgs {\n    llmChain: LLMChain;\n    getInput: (query: string, doc: DocumentInterface) => Record<string, unknown>;\n}\n/**\n * A class that uses an LLM chain to extract relevant parts of documents.\n * It extends the BaseDocumentCompressor class.\n */\nexport declare class LLMChainExtractor extends BaseDocumentCompressor {\n    llmChain: LLMChain;\n    getInput: (query: string, doc: DocumentInterface) => Record<string, unknown>;\n    constructor({ llmChain, getInput }: LLMChainExtractorArgs);\n    /**\n     * Compresses a list of documents based on the output of an LLM chain.\n     * @param documents The list of documents to be compressed.\n     * @param query The query to be used for document compression.\n     * @returns A list of compressed documents.\n     */\n    compressDocuments(documents: DocumentInterface[], query: string): Promise<DocumentInterface[]>;\n    /**\n     * Creates a new instance of LLMChainExtractor from a given LLM, prompt\n     * template, and getInput function.\n     * @param llm The BaseLanguageModel instance used for document extraction.\n     * @param prompt The PromptTemplate instance used for document extraction.\n     * @param getInput A function used for constructing the chain input from the query and a Document.\n     * @returns A new instance of LLMChainExtractor.\n     */\n    static fromLLM(llm: BaseLanguageModelInterface, prompt?: PromptTemplate, getInput?: (query: string, doc: DocumentInterface) => Record<string, unknown>): LLMChainExtractor;\n}\n"],"mappings":";;;;;;;;;;AASA;;AACcG,UADGE,qBAAAA,CACHF;EAAQ,QACaF,EADrBE,QACqBF;EAAiB,QAAKK,EAAAA,CAAAA,KAAAA,EAAAA,MAAAA,EAAAA,GAAAA,EAAtBL,iBAAsBK,EAAAA,GAAAA,MAAAA,CAAAA,MAAAA,EAAAA,OAAAA,CAAAA;AAAM;AAM/D;;;;AAEyDA,cAFpCC,iBAAAA,SAA0BH,sBAAAA,CAEUE;EAAM,QAC7CE,EAFJL,QAEIK;EAAQ,QAAEC,EAAAA,CAAAA,KAAAA,EAAAA,MAAAA,EAAAA,GAAAA,EADOR,iBACPQ,EAAAA,GAD6BH,MAC7BG,CAAAA,MAAAA,EAAAA,OAAAA,CAAAA;EAAQ,WAAIJ,CAAAA;IAAAA,QAAAA;IAAAA;EAAAA,CAAAA,EAAAA,qBAAAA;EAAqB;;;;;;EAgBiE,iBAAKC,CAAAA,SAAAA,EATlGL,iBASkGK,EAAAA,EAAAA,KAAAA,EAAAA,MAAAA,CAAAA,EAT7DI,OAS6DJ,CATrDL,iBASqDK,EAAAA,CAAAA;EAAM;;AAnBpE;;;;;;sBAmB7CN,qCAAqCE,gDAAgDD,sBAAsBK,0BAA0BC"}

package/dist/retrievers/document_compressors/chain_extract.d.ts

@@ -0,0 +1,47 @@
+import { LLMChain } from "../../chains/llm_chain.js";
+import { BaseDocumentCompressor } from "./index.js";
+import { DocumentInterface } from "@langchain/core/documents";
+import { PromptTemplate } from "@langchain/core/prompts";
+import { BaseLanguageModelInterface } from "@langchain/core/language_models/base";
+
+//#region src/retrievers/document_compressors/chain_extract.d.ts
+
+/**
+ * Interface for the arguments required to create an instance of
+ * LLMChainExtractor.
+ */
+interface LLMChainExtractorArgs {
+  llmChain: LLMChain;
+  getInput: (query: string, doc: DocumentInterface) => Record<string, unknown>;
+}
+/**
+ * A class that uses an LLM chain to extract relevant parts of documents.
+ * It extends the BaseDocumentCompressor class.
+ */
+declare class LLMChainExtractor extends BaseDocumentCompressor {
+  llmChain: LLMChain;
+  getInput: (query: string, doc: DocumentInterface) => Record<string, unknown>;
+  constructor({
+    llmChain,
+    getInput
+  }: LLMChainExtractorArgs);
+  /**
+   * Compresses a list of documents based on the output of an LLM chain.
+   * @param documents The list of documents to be compressed.
+   * @param query The query to be used for document compression.
+   * @returns A list of compressed documents.
+   */
+  compressDocuments(documents: DocumentInterface[], query: string): Promise<DocumentInterface[]>;
+  /**
+   * Creates a new instance of LLMChainExtractor from a given LLM, prompt
+   * template, and getInput function.
+   * @param llm The BaseLanguageModel instance used for document extraction.
+   * @param prompt The PromptTemplate instance used for document extraction.
+   * @param getInput A function used for constructing the chain input from the query and a Document.
+   * @returns A new instance of LLMChainExtractor.
+   */
+  static fromLLM(llm: BaseLanguageModelInterface, prompt?: PromptTemplate, getInput?: (query: string, doc: DocumentInterface) => Record<string, unknown>): LLMChainExtractor;
+}
+//#endregion
+export { LLMChainExtractor, LLMChainExtractorArgs };
+//# sourceMappingURL=chain_extract.d.ts.map

package/dist/retrievers/document_compressors/chain_extract.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"chain_extract.d.ts","names":["BaseLanguageModelInterface","DocumentInterface","PromptTemplate","LLMChain","BaseDocumentCompressor","LLMChainExtractorArgs","Record","LLMChainExtractor","llmChain","getInput","Promise"],"sources":["../../../src/retrievers/document_compressors/chain_extract.d.ts"],"sourcesContent":["import type { BaseLanguageModelInterface } from \"@langchain/core/language_models/base\";\nimport { type DocumentInterface } from \"@langchain/core/documents\";\nimport { PromptTemplate } from \"@langchain/core/prompts\";\nimport { LLMChain } from \"../../chains/llm_chain.js\";\nimport { BaseDocumentCompressor } from \"./index.js\";\n/**\n * Interface for the arguments required to create an instance of\n * LLMChainExtractor.\n */\nexport interface LLMChainExtractorArgs {\n    llmChain: LLMChain;\n    getInput: (query: string, doc: DocumentInterface) => Record<string, unknown>;\n}\n/**\n * A class that uses an LLM chain to extract relevant parts of documents.\n * It extends the BaseDocumentCompressor class.\n */\nexport declare class LLMChainExtractor extends BaseDocumentCompressor {\n    llmChain: LLMChain;\n    getInput: (query: string, doc: DocumentInterface) => Record<string, unknown>;\n    constructor({ llmChain, getInput }: LLMChainExtractorArgs);\n    /**\n     * Compresses a list of documents based on the output of an LLM chain.\n     * @param documents The list of documents to be compressed.\n     * @param query The query to be used for document compression.\n     * @returns A list of compressed documents.\n     */\n    compressDocuments(documents: DocumentInterface[], query: string): Promise<DocumentInterface[]>;\n    /**\n     * Creates a new instance of LLMChainExtractor from a given LLM, prompt\n     * template, and getInput function.\n     * @param llm The BaseLanguageModel instance used for document extraction.\n     * @param prompt The PromptTemplate instance used for document extraction.\n     * @param getInput A function used for constructing the chain input from the query and a Document.\n     * @returns A new instance of LLMChainExtractor.\n     */\n    static fromLLM(llm: BaseLanguageModelInterface, prompt?: PromptTemplate, getInput?: (query: string, doc: DocumentInterface) => Record<string, unknown>): LLMChainExtractor;\n}\n"],"mappings":";;;;;;;;;;AASA;;AACcG,UADGE,qBAAAA,CACHF;EAAQ,QACaF,EADrBE,QACqBF;EAAiB,QAAKK,EAAAA,CAAAA,KAAAA,EAAAA,MAAAA,EAAAA,GAAAA,EAAtBL,iBAAsBK,EAAAA,GAAAA,MAAAA,CAAAA,MAAAA,EAAAA,OAAAA,CAAAA;AAAM;AAM/D;;;;AAEyDA,cAFpCC,iBAAAA,SAA0BH,sBAAAA,CAEUE;EAAM,QAC7CE,EAFJL,QAEIK;EAAQ,QAAEC,EAAAA,CAAAA,KAAAA,EAAAA,MAAAA,EAAAA,GAAAA,EADOR,iBACPQ,EAAAA,GAD6BH,MAC7BG,CAAAA,MAAAA,EAAAA,OAAAA,CAAAA;EAAQ,WAAIJ,CAAAA;IAAAA,QAAAA;IAAAA;EAAAA,CAAAA,EAAAA,qBAAAA;EAAqB;;;;;;EAgBiE,iBAAKC,CAAAA,SAAAA,EATlGL,iBASkGK,EAAAA,EAAAA,KAAAA,EAAAA,MAAAA,CAAAA,EAT7DI,OAS6DJ,CATrDL,iBASqDK,EAAAA,CAAAA;EAAM;;AAnBpE;;;;;;sBAmB7CN,qCAAqCE,gDAAgDD,sBAAsBK,0BAA0BC"}