@langchain/classic 1.0.0-alpha.1

package/dist/chains/openai_functions/tagging.js

@@ -0,0 +1,70 @@
+import { LLMChain } from "../llm_chain.js";
+import { JsonOutputFunctionsParser } from "../../output_parsers/openai_functions.js";
+import { PromptTemplate } from "@langchain/core/prompts";
+import { toJsonSchema } from "@langchain/core/utils/json_schema";
+
+//#region src/chains/openai_functions/tagging.ts
+/**
+* Function that returns an array of tagging functions. These functions
+* are used to extract relevant information from a passage.
+* @param schema The schema defining the structure of function parameters.
+* @returns An array of tagging functions.
+*/
+function getTaggingFunctions(schema) {
+	return [{
+		name: "information_extraction",
+		description: "Extracts the relevant information from the passage.",
+		parameters: schema
+	}];
+}
+const TAGGING_TEMPLATE = `Extract the desired information from the following passage.
+
+Passage:
+{input}
+`;
+/**
+* Function that creates a tagging chain using the provided schema,
+* LLM, and options. It constructs the LLM with the necessary
+* functions, prompt, output parser, and tags.
+* @param schema The schema defining the structure of function parameters.
+* @param llm LLM to use in the chain. Must support function calling.
+* @param options Options for creating the tagging chain.
+* @returns A new instance of LLMChain configured for tagging.
+*
+* @deprecated
+* Switch to expression language: https://js.langchain.com/docs/expression_language/
+* Will be removed in 0.2.0
+*/
+function createTaggingChain(schema, llm, options = {}) {
+	const { prompt = PromptTemplate.fromTemplate(TAGGING_TEMPLATE),...rest } = options;
+	const functions = getTaggingFunctions(schema);
+	const outputParser = new JsonOutputFunctionsParser();
+	return new LLMChain({
+		llm,
+		prompt,
+		llmKwargs: { functions },
+		outputParser,
+		tags: ["openai_functions", "tagging"],
+		...rest
+	});
+}
+/**
+* Function that creates a tagging chain from a Zod schema. It converts
+* the Zod schema to a JSON schema using the zodToJsonSchema function and
+* then calls createTaggingChain with the converted schema.
+* @param schema The Zod schema which extracted data should match.
+* @param llm LLM to use in the chain. Must support function calling.
+* @param options Options for creating the tagging chain.
+* @returns A new instance of LLMChain configured for tagging.
+*
+* @deprecated
+* Switch to expression language: https://js.langchain.com/docs/expression_language/
+* Will be removed in 0.2.0
+*/
+function createTaggingChainFromZod(schema, llm, options) {
+	return createTaggingChain(toJsonSchema(schema), llm, options);
+}
+
+//#endregion
+export { createTaggingChain, createTaggingChainFromZod };
+//# sourceMappingURL=tagging.js.map

package/dist/chains/openai_functions/tagging.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"tagging.js","names":["schema: FunctionParameters","llm: BaseChatModel<BaseFunctionCallOptions>","options: TaggingChainOptions","schema: InteropZodObject","options?: TaggingChainOptions"],"sources":["../../../src/chains/openai_functions/tagging.ts"],"sourcesContent":["import { BaseChatModel } from \"@langchain/core/language_models/chat_models\";\nimport { BaseFunctionCallOptions } from \"@langchain/core/language_models/base\";\nimport { AIMessageChunk } from \"@langchain/core/messages\";\nimport { PromptTemplate } from \"@langchain/core/prompts\";\nimport { InteropZodObject } from \"@langchain/core/utils/types\";\nimport {\n  type JsonSchema7ObjectType,\n  toJsonSchema,\n} from \"@langchain/core/utils/json_schema\";\nimport {\n  FunctionParameters,\n  JsonOutputFunctionsParser,\n} from \"../../output_parsers/openai_functions.js\";\nimport { LLMChain, LLMChainInput } from \"../llm_chain.js\";\n\n/**\n * Type representing the options for creating a tagging chain.\n */\nexport type TaggingChainOptions = {\n  prompt?: PromptTemplate;\n} & Omit<LLMChainInput<object>, \"prompt\" | \"llm\">;\n\n/**\n * Function that returns an array of tagging functions. These functions\n * are used to extract relevant information from a passage.\n * @param schema The schema defining the structure of function parameters.\n * @returns An array of tagging functions.\n */\nfunction getTaggingFunctions(schema: FunctionParameters) {\n  return [\n    {\n      name: \"information_extraction\",\n      description: \"Extracts the relevant information from the passage.\",\n      parameters: schema,\n    },\n  ];\n}\n\nconst TAGGING_TEMPLATE = `Extract the desired information from the following passage.\n\nPassage:\n{input}\n`;\n\n/**\n * Function that creates a tagging chain using the provided schema,\n * LLM, and options. It constructs the LLM with the necessary\n * functions, prompt, output parser, and tags.\n * @param schema The schema defining the structure of function parameters.\n * @param llm LLM to use in the chain. Must support function calling.\n * @param options Options for creating the tagging chain.\n * @returns A new instance of LLMChain configured for tagging.\n *\n * @deprecated\n * Switch to expression language: https://js.langchain.com/docs/expression_language/\n * Will be removed in 0.2.0\n */\nexport function createTaggingChain(\n  schema: FunctionParameters,\n  llm: BaseChatModel<BaseFunctionCallOptions>,\n  options: TaggingChainOptions = {}\n): LLMChain<object, BaseChatModel<BaseFunctionCallOptions, AIMessageChunk>> {\n  const { prompt = PromptTemplate.fromTemplate(TAGGING_TEMPLATE), ...rest } =\n    options;\n  const functions = getTaggingFunctions(schema);\n  const outputParser = new JsonOutputFunctionsParser();\n  return new LLMChain({\n    llm,\n    prompt,\n    llmKwargs: { functions },\n    outputParser,\n    tags: [\"openai_functions\", \"tagging\"],\n    ...rest,\n  });\n}\n\n/**\n * Function that creates a tagging chain from a Zod schema. It converts\n * the Zod schema to a JSON schema using the zodToJsonSchema function and\n * then calls createTaggingChain with the converted schema.\n * @param schema The Zod schema which extracted data should match.\n * @param llm LLM to use in the chain. Must support function calling.\n * @param options Options for creating the tagging chain.\n * @returns A new instance of LLMChain configured for tagging.\n *\n * @deprecated\n * Switch to expression language: https://js.langchain.com/docs/expression_language/\n * Will be removed in 0.2.0\n */\nexport function createTaggingChainFromZod(\n  schema: InteropZodObject,\n  llm: BaseChatModel<BaseFunctionCallOptions>,\n  options?: TaggingChainOptions\n): LLMChain<object, BaseChatModel<BaseFunctionCallOptions, AIMessageChunk>> {\n  return createTaggingChain(\n    toJsonSchema(schema) as JsonSchema7ObjectType,\n    llm,\n    options\n  );\n}\n"],"mappings":";;;;;;;;;;;;AA4BA,SAAS,oBAAoBA,QAA4B;AACvD,QAAO,CACL;EACE,MAAM;EACN,aAAa;EACb,YAAY;CACb,CACF;AACF;AAED,MAAM,mBAAmB,CAAC;;;;AAI1B,CAAC;;;;;;;;;;;;;;AAeD,SAAgB,mBACdA,QACAC,KACAC,UAA+B,CAAE,GACyC;CAC1E,MAAM,EAAE,SAAS,eAAe,aAAa,iBAAiB,CAAE,GAAG,MAAM,GACvE;CACF,MAAM,YAAY,oBAAoB,OAAO;CAC7C,MAAM,eAAe,IAAI;AACzB,QAAO,IAAI,SAAS;EAClB;EACA;EACA,WAAW,EAAE,UAAW;EACxB;EACA,MAAM,CAAC,oBAAoB,SAAU;EACrC,GAAG;CACJ;AACF;;;;;;;;;;;;;;AAeD,SAAgB,0BACdC,QACAF,KACAG,SAC0E;AAC1E,QAAO,mBACL,aAAa,OAAO,EACpB,KACA,QACD;AACF"}

package/dist/chains/query_constructor/index.cjs

@@ -0,0 +1,145 @@
+const require_rolldown_runtime = require('../../_virtual/rolldown_runtime.cjs');
+const require_structured = require('../../output_parsers/structured.cjs');
+const require_parser = require('./parser.cjs');
+const require_prompt = require('./prompt.cjs');
+const __langchain_core_prompts = require_rolldown_runtime.__toESM(require("@langchain/core/prompts"));
+const zod_v3 = require_rolldown_runtime.__toESM(require("zod/v3"));
+const __langchain_core_structured_query = require_rolldown_runtime.__toESM(require("@langchain/core/structured_query"));
+
+//#region src/chains/query_constructor/index.ts
+var query_constructor_exports = {};
+require_rolldown_runtime.__export(query_constructor_exports, {
+	AttributeInfo: () => AttributeInfo,
+	DEFAULT_EXAMPLES: () => require_prompt.DEFAULT_EXAMPLES,
+	DEFAULT_PREFIX: () => require_prompt.DEFAULT_PREFIX,
+	DEFAULT_SCHEMA: () => require_prompt.DEFAULT_SCHEMA,
+	DEFAULT_SUFFIX: () => require_prompt.DEFAULT_SUFFIX,
+	EXAMPLE_PROMPT: () => require_prompt.EXAMPLE_PROMPT,
+	QueryTransformer: () => require_parser.QueryTransformer,
+	StructuredQueryOutputParser: () => StructuredQueryOutputParser,
+	formatAttributeInfo: () => formatAttributeInfo,
+	loadQueryConstructorRunnable: () => loadQueryConstructorRunnable
+});
+/**
+* A simple data structure that holds information about an attribute. It
+* is typically used to provide metadata about attributes in other classes
+* or data structures within the LangChain framework.
+*/
+var AttributeInfo = class {
+	constructor(name, type, description) {
+		this.name = name;
+		this.type = type;
+		this.description = description;
+	}
+};
+const queryInputSchema = /* @__PURE__ */ zod_v3.z.object({
+	query: /* @__PURE__ */ zod_v3.z.string().describe("text string to compare to document contents"),
+	filter: /* @__PURE__ */ zod_v3.z.string().optional().describe("logical condition statement for filtering documents")
+});
+/**
+* A class that extends AsymmetricStructuredOutputParser to parse
+* structured query output.
+*/
+var StructuredQueryOutputParser = class StructuredQueryOutputParser extends require_structured.AsymmetricStructuredOutputParser {
+	lc_namespace = [
+		"langchain",
+		"chains",
+		"query_constructor"
+	];
+	queryTransformer;
+	constructor(fields) {
+		super({
+			...fields,
+			inputSchema: queryInputSchema
+		});
+		const { allowedComparators, allowedOperators } = fields;
+		this.queryTransformer = new require_parser.QueryTransformer(allowedComparators, allowedOperators);
+	}
+	/**
+	* Processes the output of a structured query.
+	* @param query The query string.
+	* @param filter The filter condition.
+	* @returns A Promise that resolves to a StructuredQuery instance.
+	*/
+	async outputProcessor({ query, filter }) {
+		let myQuery = query;
+		if (myQuery.length === 0) myQuery = " ";
+		if (filter === "NO_FILTER" || filter === void 0) return new __langchain_core_structured_query.StructuredQuery(query);
+		else {
+			const parsedFilter = await this.queryTransformer.parse(filter);
+			return new __langchain_core_structured_query.StructuredQuery(query, parsedFilter);
+		}
+	}
+	/**
+	* Creates a new StructuredQueryOutputParser instance from the provided
+	* components.
+	* @param allowedComparators An array of allowed Comparator instances.
+	* @param allowedOperators An array of allowed Operator instances.
+	* @returns A new StructuredQueryOutputParser instance.
+	*/
+	static fromComponents(allowedComparators = [], allowedOperators = []) {
+		return new StructuredQueryOutputParser({
+			allowedComparators,
+			allowedOperators
+		});
+	}
+};
+function formatAttributeInfo(info) {
+	const infoObj = info.reduce((acc, attr) => {
+		acc[attr.name] = {
+			type: attr.type,
+			description: attr.description
+		};
+		return acc;
+	}, {});
+	return JSON.stringify(infoObj, null, 2).replaceAll("{", "{{").replaceAll("}", "}}");
+}
+const defaultExample = require_prompt.DEFAULT_EXAMPLES.map((EXAMPLE) => EXAMPLE);
+function _getPrompt(documentContents, attributeInfo, allowedComparators, allowedOperators, examples = defaultExample) {
+	const myAllowedComparators = allowedComparators ?? Object.values(__langchain_core_structured_query.Comparators);
+	const myAllowedOperators = allowedOperators ?? Object.values(__langchain_core_structured_query.Operators);
+	const attributeJSON = formatAttributeInfo(attributeInfo);
+	const schema = (0, __langchain_core_prompts.interpolateFString)(require_prompt.DEFAULT_SCHEMA, {
+		allowed_comparators: myAllowedComparators.join(" | "),
+		allowed_operators: myAllowedOperators.join(" | ")
+	});
+	const prefix = (0, __langchain_core_prompts.interpolateFString)(require_prompt.DEFAULT_PREFIX, { schema });
+	const suffix = (0, __langchain_core_prompts.interpolateFString)(require_prompt.DEFAULT_SUFFIX, {
+		i: examples.length + 1,
+		content: documentContents,
+		attributes: attributeJSON
+	});
+	const outputParser = StructuredQueryOutputParser.fromComponents(allowedComparators, allowedOperators);
+	return new __langchain_core_prompts.FewShotPromptTemplate({
+		examples,
+		examplePrompt: require_prompt.EXAMPLE_PROMPT,
+		inputVariables: ["query"],
+		suffix,
+		prefix,
+		outputParser
+	});
+}
+function loadQueryConstructorRunnable(opts) {
+	const prompt = _getPrompt(opts.documentContents, opts.attributeInfo, opts.allowedComparators, opts.allowedOperators, opts.examples);
+	const outputParser = StructuredQueryOutputParser.fromComponents(opts.allowedComparators, opts.allowedOperators);
+	return prompt.pipe(opts.llm).pipe(outputParser);
+}
+
+//#endregion
+exports.AttributeInfo = AttributeInfo;
+exports.DEFAULT_EXAMPLES = require_prompt.DEFAULT_EXAMPLES;
+exports.DEFAULT_PREFIX = require_prompt.DEFAULT_PREFIX;
+exports.DEFAULT_SCHEMA = require_prompt.DEFAULT_SCHEMA;
+exports.DEFAULT_SUFFIX = require_prompt.DEFAULT_SUFFIX;
+exports.EXAMPLE_PROMPT = require_prompt.EXAMPLE_PROMPT;
+exports.QueryTransformer = require_parser.QueryTransformer;
+exports.StructuredQueryOutputParser = StructuredQueryOutputParser;
+exports.formatAttributeInfo = formatAttributeInfo;
+exports.loadQueryConstructorRunnable = loadQueryConstructorRunnable;
+Object.defineProperty(exports, 'query_constructor_exports', {
+  enumerable: true,
+  get: function () {
+    return query_constructor_exports;
+  }
+});
+//# sourceMappingURL=index.cjs.map

package/dist/chains/query_constructor/index.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.cjs","names":["name: string","type: string","description: string","z","AsymmetricStructuredOutputParser","fields: {\n    allowedComparators: Comparator[];\n    allowedOperators: Operator[];\n  }","QueryTransformer","StructuredQuery","allowedComparators: Comparator[]","allowedOperators: Operator[]","info: AttributeInfo[]","DEFAULT_EXAMPLES","documentContents: string","attributeInfo: AttributeInfo[]","allowedComparators?: Comparator[]","allowedOperators?: Operator[]","examples: InputValues[]","myAllowedComparators: Comparator[]","Comparators","myAllowedOperators: Operator[]","Operators","DEFAULT_SCHEMA","DEFAULT_PREFIX","DEFAULT_SUFFIX","FewShotPromptTemplate","EXAMPLE_PROMPT","opts: QueryConstructorRunnableOptions"],"sources":["../../../src/chains/query_constructor/index.ts"],"sourcesContent":["import { z } from \"zod/v3\";\nimport type { BaseLanguageModelInterface } from \"@langchain/core/language_models/base\";\nimport {\n  Example,\n  interpolateFString,\n  FewShotPromptTemplate,\n} from \"@langchain/core/prompts\";\nimport {\n  InferInteropZodOutput,\n  InputValues,\n} from \"@langchain/core/utils/types\";\nimport {\n  Comparator,\n  Comparators,\n  Operator,\n  Operators,\n  StructuredQuery,\n} from \"@langchain/core/structured_query\";\nimport { Runnable, RunnableConfig } from \"@langchain/core/runnables\";\nimport { QueryTransformer, TraverseType } from \"./parser.js\";\nimport {\n  DEFAULT_EXAMPLES,\n  DEFAULT_PREFIX,\n  DEFAULT_SCHEMA,\n  DEFAULT_SUFFIX,\n  EXAMPLE_PROMPT,\n} from \"./prompt.js\";\nimport { AsymmetricStructuredOutputParser } from \"../../output_parsers/structured.js\";\n\n/**\n * A simple data structure that holds information about an attribute. It\n * is typically used to provide metadata about attributes in other classes\n * or data structures within the LangChain framework.\n */\nexport class AttributeInfo {\n  constructor(\n    public name: string,\n    public type: string,\n    public description: string\n  ) {}\n}\n\nexport { QueryTransformer, type TraverseType };\nexport {\n  DEFAULT_EXAMPLES,\n  DEFAULT_PREFIX,\n  DEFAULT_SCHEMA,\n  DEFAULT_SUFFIX,\n  EXAMPLE_PROMPT,\n};\n\nconst queryInputSchema = /* #__PURE__ */ z.object({\n  query: /* #__PURE__ */ z\n    .string()\n    .describe(\"text string to compare to document contents\"),\n  filter: /* #__PURE__ */ z\n    .string()\n    .optional()\n    .describe(\"logical condition statement for filtering documents\"),\n});\n\n/**\n * A class that extends AsymmetricStructuredOutputParser to parse\n * structured query output.\n */\nexport class StructuredQueryOutputParser extends AsymmetricStructuredOutputParser<\n  typeof queryInputSchema,\n  StructuredQuery\n> {\n  lc_namespace = [\"langchain\", \"chains\", \"query_constructor\"];\n\n  queryTransformer: QueryTransformer;\n\n  constructor(fields: {\n    allowedComparators: Comparator[];\n    allowedOperators: Operator[];\n  }) {\n    super({ ...fields, inputSchema: queryInputSchema });\n\n    const { allowedComparators, allowedOperators } = fields;\n    this.queryTransformer = new QueryTransformer(\n      allowedComparators,\n      allowedOperators\n    );\n  }\n\n  /**\n   * Processes the output of a structured query.\n   * @param query The query string.\n   * @param filter The filter condition.\n   * @returns A Promise that resolves to a StructuredQuery instance.\n   */\n  async outputProcessor({\n    query,\n    filter,\n  }: InferInteropZodOutput<typeof queryInputSchema>): Promise<StructuredQuery> {\n    let myQuery = query;\n    if (myQuery.length === 0) {\n      myQuery = \" \";\n    }\n    if (filter === \"NO_FILTER\" || filter === undefined) {\n      return new StructuredQuery(query);\n    } else {\n      const parsedFilter = await this.queryTransformer.parse(filter);\n      return new StructuredQuery(query, parsedFilter);\n    }\n  }\n\n  /**\n   * Creates a new StructuredQueryOutputParser instance from the provided\n   * components.\n   * @param allowedComparators An array of allowed Comparator instances.\n   * @param allowedOperators An array of allowed Operator instances.\n   * @returns A new StructuredQueryOutputParser instance.\n   */\n  static fromComponents(\n    allowedComparators: Comparator[] = [],\n    allowedOperators: Operator[] = []\n  ) {\n    return new StructuredQueryOutputParser({\n      allowedComparators,\n      allowedOperators,\n    });\n  }\n}\n\nexport function formatAttributeInfo(info: AttributeInfo[]) {\n  const infoObj = info.reduce((acc, attr) => {\n    acc[attr.name] = {\n      type: attr.type,\n      description: attr.description,\n    };\n    return acc;\n  }, {} as { [name: string]: { type: string; description: string } });\n\n  return JSON.stringify(infoObj, null, 2)\n    .replaceAll(\"{\", \"{{\")\n    .replaceAll(\"}\", \"}}\");\n}\n\nconst defaultExample = DEFAULT_EXAMPLES.map((EXAMPLE) => EXAMPLE as Example);\n\nfunction _getPrompt(\n  documentContents: string,\n  attributeInfo: AttributeInfo[],\n  allowedComparators?: Comparator[],\n  allowedOperators?: Operator[],\n  examples: InputValues[] = defaultExample\n) {\n  const myAllowedComparators: Comparator[] =\n    allowedComparators ?? Object.values(Comparators);\n  const myAllowedOperators: Operator[] =\n    allowedOperators ?? Object.values(Operators);\n  const attributeJSON = formatAttributeInfo(attributeInfo);\n  const schema = interpolateFString(DEFAULT_SCHEMA, {\n    allowed_comparators: myAllowedComparators.join(\" | \"),\n    allowed_operators: myAllowedOperators.join(\" | \"),\n  });\n  const prefix = interpolateFString(DEFAULT_PREFIX, {\n    schema,\n  });\n  const suffix = interpolateFString(DEFAULT_SUFFIX, {\n    i: examples.length + 1,\n    content: documentContents,\n    attributes: attributeJSON,\n  });\n\n  const outputParser = StructuredQueryOutputParser.fromComponents(\n    allowedComparators,\n    allowedOperators\n  );\n\n  return new FewShotPromptTemplate({\n    examples,\n    examplePrompt: EXAMPLE_PROMPT,\n    inputVariables: [\"query\"],\n    suffix,\n    prefix,\n    outputParser,\n  });\n}\n\n/**\n * A type that represents options for the query constructor chain.\n */\nexport type QueryConstructorRunnableOptions = {\n  llm: BaseLanguageModelInterface;\n  documentContents: string;\n  attributeInfo: AttributeInfo[];\n  examples?: InputValues[];\n  allowedComparators?: Comparator[];\n  allowedOperators?: Operator[];\n};\n\n/** @deprecated */\nexport type QueryConstructorChainOptions = QueryConstructorRunnableOptions;\n\nexport function loadQueryConstructorRunnable(\n  opts: QueryConstructorRunnableOptions\n  // eslint-disable-next-line @typescript-eslint/no-explicit-any\n): Runnable<any, StructuredQuery, RunnableConfig<Record<string, any>>> {\n  const prompt = _getPrompt(\n    opts.documentContents,\n    opts.attributeInfo,\n    opts.allowedComparators,\n    opts.allowedOperators,\n    opts.examples\n  );\n  const outputParser = StructuredQueryOutputParser.fromComponents(\n    opts.allowedComparators,\n    opts.allowedOperators\n  );\n  return prompt.pipe(opts.llm).pipe(outputParser);\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;AAkCA,IAAa,gBAAb,MAA2B;CACzB,YACSA,MACAC,MACAC,aACP;EAHO;EACA;EACA;CACL;AACL;AAWD,MAAM,mCAAmCC,SAAE,OAAO;CAChD,uBAAuBA,SACpB,QAAQ,CACR,SAAS,8CAA8C;CAC1D,wBAAwBA,SACrB,QAAQ,CACR,UAAU,CACV,SAAS,sDAAsD;AACnE,EAAC;;;;;AAMF,IAAa,8BAAb,MAAa,oCAAoCC,oDAG/C;CACA,eAAe;EAAC;EAAa;EAAU;CAAoB;CAE3D;CAEA,YAAYC,QAGT;EACD,MAAM;GAAE,GAAG;GAAQ,aAAa;EAAkB,EAAC;EAEnD,MAAM,EAAE,oBAAoB,kBAAkB,GAAG;EACjD,KAAK,mBAAmB,IAAIC,gCAC1B,oBACA;CAEH;;;;;;;CAQD,MAAM,gBAAgB,EACpB,OACA,QAC+C,EAA4B;EAC3E,IAAI,UAAU;AACd,MAAI,QAAQ,WAAW,GACrB,UAAU;AAEZ,MAAI,WAAW,eAAe,WAAW,OACvC,QAAO,IAAIC,kDAAgB;OACtB;GACL,MAAM,eAAe,MAAM,KAAK,iBAAiB,MAAM,OAAO;AAC9D,UAAO,IAAIA,kDAAgB,OAAO;EACnC;CACF;;;;;;;;CASD,OAAO,eACLC,qBAAmC,CAAE,GACrCC,mBAA+B,CAAE,GACjC;AACA,SAAO,IAAI,4BAA4B;GACrC;GACA;EACD;CACF;AACF;AAED,SAAgB,oBAAoBC,MAAuB;CACzD,MAAM,UAAU,KAAK,OAAO,CAAC,KAAK,SAAS;EACzC,IAAI,KAAK,QAAQ;GACf,MAAM,KAAK;GACX,aAAa,KAAK;EACnB;AACD,SAAO;CACR,GAAE,CAAE,EAA8D;AAEnE,QAAO,KAAK,UAAU,SAAS,MAAM,EAAE,CACpC,WAAW,KAAK,KAAK,CACrB,WAAW,KAAK,KAAK;AACzB;AAED,MAAM,iBAAiBC,gCAAiB,IAAI,CAAC,YAAY,QAAmB;AAE5E,SAAS,WACPC,kBACAC,eACAC,oBACAC,kBACAC,WAA0B,gBAC1B;CACA,MAAMC,uBACJ,sBAAsB,OAAO,OAAOC,8CAAY;CAClD,MAAMC,qBACJ,oBAAoB,OAAO,OAAOC,4CAAU;CAC9C,MAAM,gBAAgB,oBAAoB,cAAc;CACxD,MAAM,0DAA4BC,+BAAgB;EAChD,qBAAqB,qBAAqB,KAAK,MAAM;EACrD,mBAAmB,mBAAmB,KAAK,MAAM;CAClD,EAAC;CACF,MAAM,0DAA4BC,+BAAgB,EAChD,OACD,EAAC;CACF,MAAM,0DAA4BC,+BAAgB;EAChD,GAAG,SAAS,SAAS;EACrB,SAAS;EACT,YAAY;CACb,EAAC;CAEF,MAAM,eAAe,4BAA4B,eAC/C,oBACA,iBACD;AAED,QAAO,IAAIC,+CAAsB;EAC/B;EACA,eAAeC;EACf,gBAAgB,CAAC,OAAQ;EACzB;EACA;EACA;CACD;AACF;AAiBD,SAAgB,6BACdC,MAEqE;CACrE,MAAM,SAAS,WACb,KAAK,kBACL,KAAK,eACL,KAAK,oBACL,KAAK,kBACL,KAAK,SACN;CACD,MAAM,eAAe,4BAA4B,eAC/C,KAAK,oBACL,KAAK,iBACN;AACD,QAAO,OAAO,KAAK,KAAK,IAAI,CAAC,KAAK,aAAa;AAChD"}

package/dist/chains/query_constructor/index.d.cts

@@ -0,0 +1,82 @@
+import { QueryTransformer, TraverseType } from "./parser.cjs";
+import { DEFAULT_EXAMPLES, DEFAULT_PREFIX, DEFAULT_SCHEMA, DEFAULT_SUFFIX, EXAMPLE_PROMPT } from "./prompt.cjs";
+import { AsymmetricStructuredOutputParser } from "../../output_parsers/structured.cjs";
+import { Runnable, RunnableConfig } from "@langchain/core/runnables";
+import { InferInteropZodOutput, InputValues } from "@langchain/core/utils/types";
+import { BaseLanguageModelInterface } from "@langchain/core/language_models/base";
+import { z } from "zod/v3";
+import { Comparator, Operator, StructuredQuery } from "@langchain/core/structured_query";
+
+//#region src/chains/query_constructor/index.d.ts
+
+/**
+ * A simple data structure that holds information about an attribute. It
+ * is typically used to provide metadata about attributes in other classes
+ * or data structures within the LangChain framework.
+ */
+declare class AttributeInfo {
+  name: string;
+  type: string;
+  description: string;
+  constructor(name: string, type: string, description: string);
+}
+declare const queryInputSchema: z.ZodObject<{
+  query: z.ZodString;
+  filter: z.ZodOptional<z.ZodString>;
+}, "strip", z.ZodTypeAny, {
+  query: string;
+  filter?: string | undefined;
+}, {
+  query: string;
+  filter?: string | undefined;
+}>;
+/**
+ * A class that extends AsymmetricStructuredOutputParser to parse
+ * structured query output.
+ */
+declare class StructuredQueryOutputParser extends AsymmetricStructuredOutputParser<typeof queryInputSchema, StructuredQuery> {
+  lc_namespace: string[];
+  queryTransformer: QueryTransformer;
+  constructor(fields: {
+    allowedComparators: Comparator[];
+    allowedOperators: Operator[];
+  });
+  /**
+   * Processes the output of a structured query.
+   * @param query The query string.
+   * @param filter The filter condition.
+   * @returns A Promise that resolves to a StructuredQuery instance.
+   */
+  outputProcessor({
+    query,
+    filter
+  }: InferInteropZodOutput<typeof queryInputSchema>): Promise<StructuredQuery>;
+  /**
+   * Creates a new StructuredQueryOutputParser instance from the provided
+   * components.
+   * @param allowedComparators An array of allowed Comparator instances.
+   * @param allowedOperators An array of allowed Operator instances.
+   * @returns A new StructuredQueryOutputParser instance.
+   */
+  static fromComponents(allowedComparators?: Comparator[], allowedOperators?: Operator[]): StructuredQueryOutputParser;
+}
+declare function formatAttributeInfo(info: AttributeInfo[]): string;
+/**
+ * A type that represents options for the query constructor chain.
+ */
+type QueryConstructorRunnableOptions = {
+  llm: BaseLanguageModelInterface;
+  documentContents: string;
+  attributeInfo: AttributeInfo[];
+  examples?: InputValues[];
+  allowedComparators?: Comparator[];
+  allowedOperators?: Operator[];
+};
+/** @deprecated */
+type QueryConstructorChainOptions = QueryConstructorRunnableOptions;
+declare function loadQueryConstructorRunnable(opts: QueryConstructorRunnableOptions
+// eslint-disable-next-line @typescript-eslint/no-explicit-any
+): Runnable<any, StructuredQuery, RunnableConfig<Record<string, any>>>;
+//#endregion
+export { AttributeInfo, DEFAULT_EXAMPLES, DEFAULT_PREFIX, DEFAULT_SCHEMA, DEFAULT_SUFFIX, EXAMPLE_PROMPT, QueryConstructorChainOptions, QueryConstructorRunnableOptions, QueryTransformer, StructuredQueryOutputParser, type TraverseType, formatAttributeInfo, loadQueryConstructorRunnable };
+//# sourceMappingURL=index.d.cts.map

package/dist/chains/query_constructor/index.d.cts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.cts","names":["z","BaseLanguageModelInterface","InferInteropZodOutput","InputValues","Comparator","Operator","StructuredQuery","Runnable","RunnableConfig","QueryTransformer","TraverseType","DEFAULT_EXAMPLES","DEFAULT_PREFIX","DEFAULT_SCHEMA","DEFAULT_SUFFIX","EXAMPLE_PROMPT","AsymmetricStructuredOutputParser","AttributeInfo","queryInputSchema","ZodString","ZodOptional","ZodTypeAny","ZodObject","StructuredQueryOutputParser","query","filter","Promise","formatAttributeInfo","QueryConstructorRunnableOptions","QueryConstructorChainOptions","loadQueryConstructorRunnable","Record"],"sources":["../../../src/chains/query_constructor/index.d.ts"],"sourcesContent":["import { z } from \"zod/v3\";\nimport type { BaseLanguageModelInterface } from \"@langchain/core/language_models/base\";\nimport { InferInteropZodOutput, InputValues } from \"@langchain/core/utils/types\";\nimport { Comparator, Operator, StructuredQuery } from \"@langchain/core/structured_query\";\nimport { Runnable, RunnableConfig } from \"@langchain/core/runnables\";\nimport { QueryTransformer, TraverseType } from \"./parser.js\";\nimport { DEFAULT_EXAMPLES, DEFAULT_PREFIX, DEFAULT_SCHEMA, DEFAULT_SUFFIX, EXAMPLE_PROMPT } from \"./prompt.js\";\nimport { AsymmetricStructuredOutputParser } from \"../../output_parsers/structured.js\";\n/**\n * A simple data structure that holds information about an attribute. It\n * is typically used to provide metadata about attributes in other classes\n * or data structures within the LangChain framework.\n */\nexport declare class AttributeInfo {\n    name: string;\n    type: string;\n    description: string;\n    constructor(name: string, type: string, description: string);\n}\nexport { QueryTransformer, type TraverseType };\nexport { DEFAULT_EXAMPLES, DEFAULT_PREFIX, DEFAULT_SCHEMA, DEFAULT_SUFFIX, EXAMPLE_PROMPT, };\ndeclare const queryInputSchema: z.ZodObject<{\n    query: z.ZodString;\n    filter: z.ZodOptional<z.ZodString>;\n}, \"strip\", z.ZodTypeAny, {\n    query: string;\n    filter?: string | undefined;\n}, {\n    query: string;\n    filter?: string | undefined;\n}>;\n/**\n * A class that extends AsymmetricStructuredOutputParser to parse\n * structured query output.\n */\nexport declare class StructuredQueryOutputParser extends AsymmetricStructuredOutputParser<typeof queryInputSchema, StructuredQuery> {\n    lc_namespace: string[];\n    queryTransformer: QueryTransformer;\n    constructor(fields: {\n        allowedComparators: Comparator[];\n        allowedOperators: Operator[];\n    });\n    /**\n     * Processes the output of a structured query.\n     * @param query The query string.\n     * @param filter The filter condition.\n     * @returns A Promise that resolves to a StructuredQuery instance.\n     */\n    outputProcessor({ query, filter }: InferInteropZodOutput<typeof queryInputSchema>): Promise<StructuredQuery>;\n    /**\n     * Creates a new StructuredQueryOutputParser instance from the provided\n     * components.\n     * @param allowedComparators An array of allowed Comparator instances.\n     * @param allowedOperators An array of allowed Operator instances.\n     * @returns A new StructuredQueryOutputParser instance.\n     */\n    static fromComponents(allowedComparators?: Comparator[], allowedOperators?: Operator[]): StructuredQueryOutputParser;\n}\nexport declare function formatAttributeInfo(info: AttributeInfo[]): string;\n/**\n * A type that represents options for the query constructor chain.\n */\nexport type QueryConstructorRunnableOptions = {\n    llm: BaseLanguageModelInterface;\n    documentContents: string;\n    attributeInfo: AttributeInfo[];\n    examples?: InputValues[];\n    allowedComparators?: Comparator[];\n    allowedOperators?: Operator[];\n};\n/** @deprecated */\nexport type QueryConstructorChainOptions = QueryConstructorRunnableOptions;\nexport declare function loadQueryConstructorRunnable(opts: QueryConstructorRunnableOptions\n// eslint-disable-next-line @typescript-eslint/no-explicit-any\n): Runnable<any, StructuredQuery, RunnableConfig<Record<string, any>>>;\n"],"mappings":";;;;;;;;;;;;;AAaA;AAO6F;;AAElFA,cATUiB,aAAAA,CASRE;EAAS,IACInB,EAAEmB,MAAAA;EAAS,IAAzBnB,EAAEoB,MAAAA;EAAW,WACXC,EAAAA,MAAAA;EAAU,WAHUC,CAAAA,IAAAA,EAAAA,MAAAA,EAAAA,IAAAA,EAAAA,MAAAA,EAAAA,WAAAA,EAAAA,MAAAA;AAAS;cAA7BJ,gBAcmFA,EAdjElB,CAAAA,CAAEsB,SAc+DJ,CAAAA;EAAgB,KAAEZ,EAbxGN,CAAAA,CAAEmB,SAasGb;EAAe,MAE5GG,EAdVT,CAAAA,CAAEoB,WAcQX,CAdIT,CAAAA,CAAEmB,SAcNV,CAAAA;CAAgB,EAAA,OAEVL,EAfhBJ,CAAAA,CAAEqB,UAecjB,EAAAA;EAAU,KACZC,EAAAA,MAAAA;EAAQ,MAQZmB,CAAAA,EAAAA,MAAAA,GAAAA,SAAAA;CAAK,EAAA;EAAQ,KAAiCN,EAAAA,MAAAA;EAAgB,MAA7ChB,CAAAA,EAAAA,MAAAA,GAAAA,SAAAA;CAAqB,CAAA;;;;;AAbHc,cAApCO,2BAAAA,SAAoCP,gCAAAA,CAAAA,OAAwCE,gBAAxCF,EAA0DV,eAA1DU,CAAAA,CAAAA;EAAgC,YAAA,EAAA,MAAA,EAAA;EAuBjEW,gBAAAA,EArBFlB,gBAqB4BQ;EAItCW,WAAAA,CAAAA,MAAAA,EAAAA;IAA+B,kBAAA,EAvBfxB,UAuBe,EAAA;IAClCH,gBAAAA,EAvBiBI,QAuBjBJ,EAAAA;EAA0B,CAAA;EAEH;;;AAGD;AAG/B;AACA;EAAoD,eAAA,CAAA;IAAA,KAAA;IAAA;EAAA,CAAA,EAxBbC,qBAwBa,CAAA,OAxBgBgB,gBAwBhB,CAAA,CAAA,EAxBoCQ,OAwBpC,CAxB4CpB,eAwB5C,CAAA;EAAA;;;;;AAEzC;;6CAlBoCF,iCAAiCC,aAAakB;;iBAErEI,mBAAAA,OAA0BV;;;;KAItCW,+BAAAA;OACH3B;;iBAEUgB;aACJd;uBACUC;qBACFC;;;KAGXwB,4BAAAA,GAA+BD;iBACnBE,4BAAAA,OAAmCF;;GAExDrB,cAAcD,iBAAiBE,eAAeuB"}

package/dist/chains/query_constructor/index.d.ts

@@ -0,0 +1,82 @@
+import { QueryTransformer, TraverseType } from "./parser.js";
+import { DEFAULT_EXAMPLES, DEFAULT_PREFIX, DEFAULT_SCHEMA, DEFAULT_SUFFIX, EXAMPLE_PROMPT } from "./prompt.js";
+import { AsymmetricStructuredOutputParser } from "../../output_parsers/structured.js";
+import { Runnable, RunnableConfig } from "@langchain/core/runnables";
+import { BaseLanguageModelInterface } from "@langchain/core/language_models/base";
+import { InferInteropZodOutput, InputValues } from "@langchain/core/utils/types";
+import { z } from "zod/v3";
+import { Comparator, Operator, StructuredQuery } from "@langchain/core/structured_query";
+
+//#region src/chains/query_constructor/index.d.ts
+
+/**
+ * A simple data structure that holds information about an attribute. It
+ * is typically used to provide metadata about attributes in other classes
+ * or data structures within the LangChain framework.
+ */
+declare class AttributeInfo {
+  name: string;
+  type: string;
+  description: string;
+  constructor(name: string, type: string, description: string);
+}
+declare const queryInputSchema: z.ZodObject<{
+  query: z.ZodString;
+  filter: z.ZodOptional<z.ZodString>;
+}, "strip", z.ZodTypeAny, {
+  query: string;
+  filter?: string | undefined;
+}, {
+  query: string;
+  filter?: string | undefined;
+}>;
+/**
+ * A class that extends AsymmetricStructuredOutputParser to parse
+ * structured query output.
+ */
+declare class StructuredQueryOutputParser extends AsymmetricStructuredOutputParser<typeof queryInputSchema, StructuredQuery> {
+  lc_namespace: string[];
+  queryTransformer: QueryTransformer;
+  constructor(fields: {
+    allowedComparators: Comparator[];
+    allowedOperators: Operator[];
+  });
+  /**
+   * Processes the output of a structured query.
+   * @param query The query string.
+   * @param filter The filter condition.
+   * @returns A Promise that resolves to a StructuredQuery instance.
+   */
+  outputProcessor({
+    query,
+    filter
+  }: InferInteropZodOutput<typeof queryInputSchema>): Promise<StructuredQuery>;
+  /**
+   * Creates a new StructuredQueryOutputParser instance from the provided
+   * components.
+   * @param allowedComparators An array of allowed Comparator instances.
+   * @param allowedOperators An array of allowed Operator instances.
+   * @returns A new StructuredQueryOutputParser instance.
+   */
+  static fromComponents(allowedComparators?: Comparator[], allowedOperators?: Operator[]): StructuredQueryOutputParser;
+}
+declare function formatAttributeInfo(info: AttributeInfo[]): string;
+/**
+ * A type that represents options for the query constructor chain.
+ */
+type QueryConstructorRunnableOptions = {
+  llm: BaseLanguageModelInterface;
+  documentContents: string;
+  attributeInfo: AttributeInfo[];
+  examples?: InputValues[];
+  allowedComparators?: Comparator[];
+  allowedOperators?: Operator[];
+};
+/** @deprecated */
+type QueryConstructorChainOptions = QueryConstructorRunnableOptions;
+declare function loadQueryConstructorRunnable(opts: QueryConstructorRunnableOptions
+// eslint-disable-next-line @typescript-eslint/no-explicit-any
+): Runnable<any, StructuredQuery, RunnableConfig<Record<string, any>>>;
+//#endregion
+export { AttributeInfo, DEFAULT_EXAMPLES, DEFAULT_PREFIX, DEFAULT_SCHEMA, DEFAULT_SUFFIX, EXAMPLE_PROMPT, QueryConstructorChainOptions, QueryConstructorRunnableOptions, QueryTransformer, StructuredQueryOutputParser, type TraverseType, formatAttributeInfo, loadQueryConstructorRunnable };
+//# sourceMappingURL=index.d.ts.map

package/dist/chains/query_constructor/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","names":["z","BaseLanguageModelInterface","InferInteropZodOutput","InputValues","Comparator","Operator","StructuredQuery","Runnable","RunnableConfig","QueryTransformer","TraverseType","DEFAULT_EXAMPLES","DEFAULT_PREFIX","DEFAULT_SCHEMA","DEFAULT_SUFFIX","EXAMPLE_PROMPT","AsymmetricStructuredOutputParser","AttributeInfo","queryInputSchema","ZodString","ZodOptional","ZodTypeAny","ZodObject","StructuredQueryOutputParser","query","filter","Promise","formatAttributeInfo","QueryConstructorRunnableOptions","QueryConstructorChainOptions","loadQueryConstructorRunnable","Record"],"sources":["../../../src/chains/query_constructor/index.d.ts"],"sourcesContent":["import { z } from \"zod/v3\";\nimport type { BaseLanguageModelInterface } from \"@langchain/core/language_models/base\";\nimport { InferInteropZodOutput, InputValues } from \"@langchain/core/utils/types\";\nimport { Comparator, Operator, StructuredQuery } from \"@langchain/core/structured_query\";\nimport { Runnable, RunnableConfig } from \"@langchain/core/runnables\";\nimport { QueryTransformer, TraverseType } from \"./parser.js\";\nimport { DEFAULT_EXAMPLES, DEFAULT_PREFIX, DEFAULT_SCHEMA, DEFAULT_SUFFIX, EXAMPLE_PROMPT } from \"./prompt.js\";\nimport { AsymmetricStructuredOutputParser } from \"../../output_parsers/structured.js\";\n/**\n * A simple data structure that holds information about an attribute. It\n * is typically used to provide metadata about attributes in other classes\n * or data structures within the LangChain framework.\n */\nexport declare class AttributeInfo {\n    name: string;\n    type: string;\n    description: string;\n    constructor(name: string, type: string, description: string);\n}\nexport { QueryTransformer, type TraverseType };\nexport { DEFAULT_EXAMPLES, DEFAULT_PREFIX, DEFAULT_SCHEMA, DEFAULT_SUFFIX, EXAMPLE_PROMPT, };\ndeclare const queryInputSchema: z.ZodObject<{\n    query: z.ZodString;\n    filter: z.ZodOptional<z.ZodString>;\n}, \"strip\", z.ZodTypeAny, {\n    query: string;\n    filter?: string | undefined;\n}, {\n    query: string;\n    filter?: string | undefined;\n}>;\n/**\n * A class that extends AsymmetricStructuredOutputParser to parse\n * structured query output.\n */\nexport declare class StructuredQueryOutputParser extends AsymmetricStructuredOutputParser<typeof queryInputSchema, StructuredQuery> {\n    lc_namespace: string[];\n    queryTransformer: QueryTransformer;\n    constructor(fields: {\n        allowedComparators: Comparator[];\n        allowedOperators: Operator[];\n    });\n    /**\n     * Processes the output of a structured query.\n     * @param query The query string.\n     * @param filter The filter condition.\n     * @returns A Promise that resolves to a StructuredQuery instance.\n     */\n    outputProcessor({ query, filter }: InferInteropZodOutput<typeof queryInputSchema>): Promise<StructuredQuery>;\n    /**\n     * Creates a new StructuredQueryOutputParser instance from the provided\n     * components.\n     * @param allowedComparators An array of allowed Comparator instances.\n     * @param allowedOperators An array of allowed Operator instances.\n     * @returns A new StructuredQueryOutputParser instance.\n     */\n    static fromComponents(allowedComparators?: Comparator[], allowedOperators?: Operator[]): StructuredQueryOutputParser;\n}\nexport declare function formatAttributeInfo(info: AttributeInfo[]): string;\n/**\n * A type that represents options for the query constructor chain.\n */\nexport type QueryConstructorRunnableOptions = {\n    llm: BaseLanguageModelInterface;\n    documentContents: string;\n    attributeInfo: AttributeInfo[];\n    examples?: InputValues[];\n    allowedComparators?: Comparator[];\n    allowedOperators?: Operator[];\n};\n/** @deprecated */\nexport type QueryConstructorChainOptions = QueryConstructorRunnableOptions;\nexport declare function loadQueryConstructorRunnable(opts: QueryConstructorRunnableOptions\n// eslint-disable-next-line @typescript-eslint/no-explicit-any\n): Runnable<any, StructuredQuery, RunnableConfig<Record<string, any>>>;\n"],"mappings":";;;;;;;;;;;;;AAaA;AAO6F;;AAElFA,cATUiB,aAAAA,CASRE;EAAS,IACInB,EAAEmB,MAAAA;EAAS,IAAzBnB,EAAEoB,MAAAA;EAAW,WACXC,EAAAA,MAAAA;EAAU,WAHUC,CAAAA,IAAAA,EAAAA,MAAAA,EAAAA,IAAAA,EAAAA,MAAAA,EAAAA,WAAAA,EAAAA,MAAAA;AAAS;cAA7BJ,gBAcmFA,EAdjElB,CAAAA,CAAEsB,SAc+DJ,CAAAA;EAAgB,KAAEZ,EAbxGN,CAAAA,CAAEmB,SAasGb;EAAe,MAE5GG,EAdVT,CAAAA,CAAEoB,WAcQX,CAdIT,CAAAA,CAAEmB,SAcNV,CAAAA;CAAgB,EAAA,OAEVL,EAfhBJ,CAAAA,CAAEqB,UAecjB,EAAAA;EAAU,KACZC,EAAAA,MAAAA;EAAQ,MAQZmB,CAAAA,EAAAA,MAAAA,GAAAA,SAAAA;CAAK,EAAA;EAAQ,KAAiCN,EAAAA,MAAAA;EAAgB,MAA7ChB,CAAAA,EAAAA,MAAAA,GAAAA,SAAAA;CAAqB,CAAA;;;;;AAbHc,cAApCO,2BAAAA,SAAoCP,gCAAAA,CAAAA,OAAwCE,gBAAxCF,EAA0DV,eAA1DU,CAAAA,CAAAA;EAAgC,YAAA,EAAA,MAAA,EAAA;EAuBjEW,gBAAAA,EArBFlB,gBAqB4BQ;EAItCW,WAAAA,CAAAA,MAAAA,EAAAA;IAA+B,kBAAA,EAvBfxB,UAuBe,EAAA;IAClCH,gBAAAA,EAvBiBI,QAuBjBJ,EAAAA;EAA0B,CAAA;EAEH;;;AAGD;AAG/B;AACA;EAAoD,eAAA,CAAA;IAAA,KAAA;IAAA;EAAA,CAAA,EAxBbC,qBAwBa,CAAA,OAxBgBgB,gBAwBhB,CAAA,CAAA,EAxBoCQ,OAwBpC,CAxB4CpB,eAwB5C,CAAA;EAAA;;;;;AAEzC;;6CAlBoCF,iCAAiCC,aAAakB;;iBAErEI,mBAAAA,OAA0BV;;;;KAItCW,+BAAAA;OACH3B;;iBAEUgB;aACJd;uBACUC;qBACFC;;;KAGXwB,4BAAAA,GAA+BD;iBACnBE,4BAAAA,OAAmCF;;GAExDrB,cAAcD,iBAAiBE,eAAeuB"}

package/dist/chains/query_constructor/index.js

@@ -0,0 +1,130 @@
+import { __export } from "../../_virtual/rolldown_runtime.js";
+import { AsymmetricStructuredOutputParser } from "../../output_parsers/structured.js";
+import { QueryTransformer } from "./parser.js";
+import { DEFAULT_EXAMPLES, DEFAULT_PREFIX, DEFAULT_SCHEMA, DEFAULT_SUFFIX, EXAMPLE_PROMPT } from "./prompt.js";
+import { FewShotPromptTemplate, interpolateFString } from "@langchain/core/prompts";
+import { z } from "zod/v3";
+import { Comparators, Operators, StructuredQuery } from "@langchain/core/structured_query";
+
+//#region src/chains/query_constructor/index.ts
+var query_constructor_exports = {};
+__export(query_constructor_exports, {
+	AttributeInfo: () => AttributeInfo,
+	DEFAULT_EXAMPLES: () => DEFAULT_EXAMPLES,
+	DEFAULT_PREFIX: () => DEFAULT_PREFIX,
+	DEFAULT_SCHEMA: () => DEFAULT_SCHEMA,
+	DEFAULT_SUFFIX: () => DEFAULT_SUFFIX,
+	EXAMPLE_PROMPT: () => EXAMPLE_PROMPT,
+	QueryTransformer: () => QueryTransformer,
+	StructuredQueryOutputParser: () => StructuredQueryOutputParser,
+	formatAttributeInfo: () => formatAttributeInfo,
+	loadQueryConstructorRunnable: () => loadQueryConstructorRunnable
+});
+/**
+* A simple data structure that holds information about an attribute. It
+* is typically used to provide metadata about attributes in other classes
+* or data structures within the LangChain framework.
+*/
+var AttributeInfo = class {
+	constructor(name, type, description) {
+		this.name = name;
+		this.type = type;
+		this.description = description;
+	}
+};
+const queryInputSchema = /* @__PURE__ */ z.object({
+	query: /* @__PURE__ */ z.string().describe("text string to compare to document contents"),
+	filter: /* @__PURE__ */ z.string().optional().describe("logical condition statement for filtering documents")
+});
+/**
+* A class that extends AsymmetricStructuredOutputParser to parse
+* structured query output.
+*/
+var StructuredQueryOutputParser = class StructuredQueryOutputParser extends AsymmetricStructuredOutputParser {
+	lc_namespace = [
+		"langchain",
+		"chains",
+		"query_constructor"
+	];
+	queryTransformer;
+	constructor(fields) {
+		super({
+			...fields,
+			inputSchema: queryInputSchema
+		});
+		const { allowedComparators, allowedOperators } = fields;
+		this.queryTransformer = new QueryTransformer(allowedComparators, allowedOperators);
+	}
+	/**
+	* Processes the output of a structured query.
+	* @param query The query string.
+	* @param filter The filter condition.
+	* @returns A Promise that resolves to a StructuredQuery instance.
+	*/
+	async outputProcessor({ query, filter }) {
+		let myQuery = query;
+		if (myQuery.length === 0) myQuery = " ";
+		if (filter === "NO_FILTER" || filter === void 0) return new StructuredQuery(query);
+		else {
+			const parsedFilter = await this.queryTransformer.parse(filter);
+			return new StructuredQuery(query, parsedFilter);
+		}
+	}
+	/**
+	* Creates a new StructuredQueryOutputParser instance from the provided
+	* components.
+	* @param allowedComparators An array of allowed Comparator instances.
+	* @param allowedOperators An array of allowed Operator instances.
+	* @returns A new StructuredQueryOutputParser instance.
+	*/
+	static fromComponents(allowedComparators = [], allowedOperators = []) {
+		return new StructuredQueryOutputParser({
+			allowedComparators,
+			allowedOperators
+		});
+	}
+};
+function formatAttributeInfo(info) {
+	const infoObj = info.reduce((acc, attr) => {
+		acc[attr.name] = {
+			type: attr.type,
+			description: attr.description
+		};
+		return acc;
+	}, {});
+	return JSON.stringify(infoObj, null, 2).replaceAll("{", "{{").replaceAll("}", "}}");
+}
+const defaultExample = DEFAULT_EXAMPLES.map((EXAMPLE) => EXAMPLE);
+function _getPrompt(documentContents, attributeInfo, allowedComparators, allowedOperators, examples = defaultExample) {
+	const myAllowedComparators = allowedComparators ?? Object.values(Comparators);
+	const myAllowedOperators = allowedOperators ?? Object.values(Operators);
+	const attributeJSON = formatAttributeInfo(attributeInfo);
+	const schema = interpolateFString(DEFAULT_SCHEMA, {
+		allowed_comparators: myAllowedComparators.join(" | "),
+		allowed_operators: myAllowedOperators.join(" | ")
+	});
+	const prefix = interpolateFString(DEFAULT_PREFIX, { schema });
+	const suffix = interpolateFString(DEFAULT_SUFFIX, {
+		i: examples.length + 1,
+		content: documentContents,
+		attributes: attributeJSON
+	});
+	const outputParser = StructuredQueryOutputParser.fromComponents(allowedComparators, allowedOperators);
+	return new FewShotPromptTemplate({
+		examples,
+		examplePrompt: EXAMPLE_PROMPT,
+		inputVariables: ["query"],
+		suffix,
+		prefix,
+		outputParser
+	});
+}
+function loadQueryConstructorRunnable(opts) {
+	const prompt = _getPrompt(opts.documentContents, opts.attributeInfo, opts.allowedComparators, opts.allowedOperators, opts.examples);
+	const outputParser = StructuredQueryOutputParser.fromComponents(opts.allowedComparators, opts.allowedOperators);
+	return prompt.pipe(opts.llm).pipe(outputParser);
+}
+
+//#endregion
+export { AttributeInfo, DEFAULT_EXAMPLES, DEFAULT_PREFIX, DEFAULT_SCHEMA, DEFAULT_SUFFIX, EXAMPLE_PROMPT, QueryTransformer, StructuredQueryOutputParser, formatAttributeInfo, loadQueryConstructorRunnable, query_constructor_exports };
+//# sourceMappingURL=index.js.map

package/dist/chains/query_constructor/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","names":["name: string","type: string","description: string","fields: {\n    allowedComparators: Comparator[];\n    allowedOperators: Operator[];\n  }","allowedComparators: Comparator[]","allowedOperators: Operator[]","info: AttributeInfo[]","documentContents: string","attributeInfo: AttributeInfo[]","allowedComparators?: Comparator[]","allowedOperators?: Operator[]","examples: InputValues[]","myAllowedComparators: Comparator[]","myAllowedOperators: Operator[]","opts: QueryConstructorRunnableOptions"],"sources":["../../../src/chains/query_constructor/index.ts"],"sourcesContent":["import { z } from \"zod/v3\";\nimport type { BaseLanguageModelInterface } from \"@langchain/core/language_models/base\";\nimport {\n  Example,\n  interpolateFString,\n  FewShotPromptTemplate,\n} from \"@langchain/core/prompts\";\nimport {\n  InferInteropZodOutput,\n  InputValues,\n} from \"@langchain/core/utils/types\";\nimport {\n  Comparator,\n  Comparators,\n  Operator,\n  Operators,\n  StructuredQuery,\n} from \"@langchain/core/structured_query\";\nimport { Runnable, RunnableConfig } from \"@langchain/core/runnables\";\nimport { QueryTransformer, TraverseType } from \"./parser.js\";\nimport {\n  DEFAULT_EXAMPLES,\n  DEFAULT_PREFIX,\n  DEFAULT_SCHEMA,\n  DEFAULT_SUFFIX,\n  EXAMPLE_PROMPT,\n} from \"./prompt.js\";\nimport { AsymmetricStructuredOutputParser } from \"../../output_parsers/structured.js\";\n\n/**\n * A simple data structure that holds information about an attribute. It\n * is typically used to provide metadata about attributes in other classes\n * or data structures within the LangChain framework.\n */\nexport class AttributeInfo {\n  constructor(\n    public name: string,\n    public type: string,\n    public description: string\n  ) {}\n}\n\nexport { QueryTransformer, type TraverseType };\nexport {\n  DEFAULT_EXAMPLES,\n  DEFAULT_PREFIX,\n  DEFAULT_SCHEMA,\n  DEFAULT_SUFFIX,\n  EXAMPLE_PROMPT,\n};\n\nconst queryInputSchema = /* #__PURE__ */ z.object({\n  query: /* #__PURE__ */ z\n    .string()\n    .describe(\"text string to compare to document contents\"),\n  filter: /* #__PURE__ */ z\n    .string()\n    .optional()\n    .describe(\"logical condition statement for filtering documents\"),\n});\n\n/**\n * A class that extends AsymmetricStructuredOutputParser to parse\n * structured query output.\n */\nexport class StructuredQueryOutputParser extends AsymmetricStructuredOutputParser<\n  typeof queryInputSchema,\n  StructuredQuery\n> {\n  lc_namespace = [\"langchain\", \"chains\", \"query_constructor\"];\n\n  queryTransformer: QueryTransformer;\n\n  constructor(fields: {\n    allowedComparators: Comparator[];\n    allowedOperators: Operator[];\n  }) {\n    super({ ...fields, inputSchema: queryInputSchema });\n\n    const { allowedComparators, allowedOperators } = fields;\n    this.queryTransformer = new QueryTransformer(\n      allowedComparators,\n      allowedOperators\n    );\n  }\n\n  /**\n   * Processes the output of a structured query.\n   * @param query The query string.\n   * @param filter The filter condition.\n   * @returns A Promise that resolves to a StructuredQuery instance.\n   */\n  async outputProcessor({\n    query,\n    filter,\n  }: InferInteropZodOutput<typeof queryInputSchema>): Promise<StructuredQuery> {\n    let myQuery = query;\n    if (myQuery.length === 0) {\n      myQuery = \" \";\n    }\n    if (filter === \"NO_FILTER\" || filter === undefined) {\n      return new StructuredQuery(query);\n    } else {\n      const parsedFilter = await this.queryTransformer.parse(filter);\n      return new StructuredQuery(query, parsedFilter);\n    }\n  }\n\n  /**\n   * Creates a new StructuredQueryOutputParser instance from the provided\n   * components.\n   * @param allowedComparators An array of allowed Comparator instances.\n   * @param allowedOperators An array of allowed Operator instances.\n   * @returns A new StructuredQueryOutputParser instance.\n   */\n  static fromComponents(\n    allowedComparators: Comparator[] = [],\n    allowedOperators: Operator[] = []\n  ) {\n    return new StructuredQueryOutputParser({\n      allowedComparators,\n      allowedOperators,\n    });\n  }\n}\n\nexport function formatAttributeInfo(info: AttributeInfo[]) {\n  const infoObj = info.reduce((acc, attr) => {\n    acc[attr.name] = {\n      type: attr.type,\n      description: attr.description,\n    };\n    return acc;\n  }, {} as { [name: string]: { type: string; description: string } });\n\n  return JSON.stringify(infoObj, null, 2)\n    .replaceAll(\"{\", \"{{\")\n    .replaceAll(\"}\", \"}}\");\n}\n\nconst defaultExample = DEFAULT_EXAMPLES.map((EXAMPLE) => EXAMPLE as Example);\n\nfunction _getPrompt(\n  documentContents: string,\n  attributeInfo: AttributeInfo[],\n  allowedComparators?: Comparator[],\n  allowedOperators?: Operator[],\n  examples: InputValues[] = defaultExample\n) {\n  const myAllowedComparators: Comparator[] =\n    allowedComparators ?? Object.values(Comparators);\n  const myAllowedOperators: Operator[] =\n    allowedOperators ?? Object.values(Operators);\n  const attributeJSON = formatAttributeInfo(attributeInfo);\n  const schema = interpolateFString(DEFAULT_SCHEMA, {\n    allowed_comparators: myAllowedComparators.join(\" | \"),\n    allowed_operators: myAllowedOperators.join(\" | \"),\n  });\n  const prefix = interpolateFString(DEFAULT_PREFIX, {\n    schema,\n  });\n  const suffix = interpolateFString(DEFAULT_SUFFIX, {\n    i: examples.length + 1,\n    content: documentContents,\n    attributes: attributeJSON,\n  });\n\n  const outputParser = StructuredQueryOutputParser.fromComponents(\n    allowedComparators,\n    allowedOperators\n  );\n\n  return new FewShotPromptTemplate({\n    examples,\n    examplePrompt: EXAMPLE_PROMPT,\n    inputVariables: [\"query\"],\n    suffix,\n    prefix,\n    outputParser,\n  });\n}\n\n/**\n * A type that represents options for the query constructor chain.\n */\nexport type QueryConstructorRunnableOptions = {\n  llm: BaseLanguageModelInterface;\n  documentContents: string;\n  attributeInfo: AttributeInfo[];\n  examples?: InputValues[];\n  allowedComparators?: Comparator[];\n  allowedOperators?: Operator[];\n};\n\n/** @deprecated */\nexport type QueryConstructorChainOptions = QueryConstructorRunnableOptions;\n\nexport function loadQueryConstructorRunnable(\n  opts: QueryConstructorRunnableOptions\n  // eslint-disable-next-line @typescript-eslint/no-explicit-any\n): Runnable<any, StructuredQuery, RunnableConfig<Record<string, any>>> {\n  const prompt = _getPrompt(\n    opts.documentContents,\n    opts.attributeInfo,\n    opts.allowedComparators,\n    opts.allowedOperators,\n    opts.examples\n  );\n  const outputParser = StructuredQueryOutputParser.fromComponents(\n    opts.allowedComparators,\n    opts.allowedOperators\n  );\n  return prompt.pipe(opts.llm).pipe(outputParser);\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;AAkCA,IAAa,gBAAb,MAA2B;CACzB,YACSA,MACAC,MACAC,aACP;EAHO;EACA;EACA;CACL;AACL;AAWD,MAAM,mCAAmC,EAAE,OAAO;CAChD,uBAAuB,EACpB,QAAQ,CACR,SAAS,8CAA8C;CAC1D,wBAAwB,EACrB,QAAQ,CACR,UAAU,CACV,SAAS,sDAAsD;AACnE,EAAC;;;;;AAMF,IAAa,8BAAb,MAAa,oCAAoC,iCAG/C;CACA,eAAe;EAAC;EAAa;EAAU;CAAoB;CAE3D;CAEA,YAAYC,QAGT;EACD,MAAM;GAAE,GAAG;GAAQ,aAAa;EAAkB,EAAC;EAEnD,MAAM,EAAE,oBAAoB,kBAAkB,GAAG;EACjD,KAAK,mBAAmB,IAAI,iBAC1B,oBACA;CAEH;;;;;;;CAQD,MAAM,gBAAgB,EACpB,OACA,QAC+C,EAA4B;EAC3E,IAAI,UAAU;AACd,MAAI,QAAQ,WAAW,GACrB,UAAU;AAEZ,MAAI,WAAW,eAAe,WAAW,OACvC,QAAO,IAAI,gBAAgB;OACtB;GACL,MAAM,eAAe,MAAM,KAAK,iBAAiB,MAAM,OAAO;AAC9D,UAAO,IAAI,gBAAgB,OAAO;EACnC;CACF;;;;;;;;CASD,OAAO,eACLC,qBAAmC,CAAE,GACrCC,mBAA+B,CAAE,GACjC;AACA,SAAO,IAAI,4BAA4B;GACrC;GACA;EACD;CACF;AACF;AAED,SAAgB,oBAAoBC,MAAuB;CACzD,MAAM,UAAU,KAAK,OAAO,CAAC,KAAK,SAAS;EACzC,IAAI,KAAK,QAAQ;GACf,MAAM,KAAK;GACX,aAAa,KAAK;EACnB;AACD,SAAO;CACR,GAAE,CAAE,EAA8D;AAEnE,QAAO,KAAK,UAAU,SAAS,MAAM,EAAE,CACpC,WAAW,KAAK,KAAK,CACrB,WAAW,KAAK,KAAK;AACzB;AAED,MAAM,iBAAiB,iBAAiB,IAAI,CAAC,YAAY,QAAmB;AAE5E,SAAS,WACPC,kBACAC,eACAC,oBACAC,kBACAC,WAA0B,gBAC1B;CACA,MAAMC,uBACJ,sBAAsB,OAAO,OAAO,YAAY;CAClD,MAAMC,qBACJ,oBAAoB,OAAO,OAAO,UAAU;CAC9C,MAAM,gBAAgB,oBAAoB,cAAc;CACxD,MAAM,SAAS,mBAAmB,gBAAgB;EAChD,qBAAqB,qBAAqB,KAAK,MAAM;EACrD,mBAAmB,mBAAmB,KAAK,MAAM;CAClD,EAAC;CACF,MAAM,SAAS,mBAAmB,gBAAgB,EAChD,OACD,EAAC;CACF,MAAM,SAAS,mBAAmB,gBAAgB;EAChD,GAAG,SAAS,SAAS;EACrB,SAAS;EACT,YAAY;CACb,EAAC;CAEF,MAAM,eAAe,4BAA4B,eAC/C,oBACA,iBACD;AAED,QAAO,IAAI,sBAAsB;EAC/B;EACA,eAAe;EACf,gBAAgB,CAAC,OAAQ;EACzB;EACA;EACA;CACD;AACF;AAiBD,SAAgB,6BACdC,MAEqE;CACrE,MAAM,SAAS,WACb,KAAK,kBACL,KAAK,eACL,KAAK,oBACL,KAAK,kBACL,KAAK,SACN;CACD,MAAM,eAAe,4BAA4B,eAC/C,KAAK,oBACL,KAAK,iBACN;AACD,QAAO,OAAO,KAAK,KAAK,IAAI,CAAC,KAAK,aAAa;AAChD"}

package/dist/chains/query_constructor/ir.cjs

@@ -0,0 +1,22 @@
+const require_rolldown_runtime = require('../../_virtual/rolldown_runtime.cjs');
+
+//#region src/chains/query_constructor/ir.ts
+var ir_exports = {};
+require_rolldown_runtime.__reExport(ir_exports, require("@langchain/core/structured_query"));
+
+//#endregion
+Object.defineProperty(exports, 'ir_exports', {
+  enumerable: true,
+  get: function () {
+    return ir_exports;
+  }
+});
+var __langchain_core_structured_query = require("@langchain/core/structured_query");
+Object.keys(__langchain_core_structured_query).forEach(function (k) {
+  if (k !== 'default' && !Object.prototype.hasOwnProperty.call(exports, k)) Object.defineProperty(exports, k, {
+    enumerable: true,
+    get: function () { return __langchain_core_structured_query[k]; }
+  });
+});
+
+//# sourceMappingURL=ir.cjs.map

package/dist/chains/query_constructor/ir.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"ir.cjs","names":[],"sources":["../../../src/chains/query_constructor/ir.ts"],"sourcesContent":["export * from \"@langchain/core/structured_query\";\n"],"mappings":""}

package/dist/chains/query_constructor/ir.d.cts

@@ -0,0 +1 @@
+export * from "@langchain/core/structured_query";

package/dist/chains/query_constructor/ir.d.ts

@@ -0,0 +1 @@
+export * from "@langchain/core/structured_query";

package/dist/chains/query_constructor/ir.js

@@ -0,0 +1,12 @@
+import { __reExport } from "../../_virtual/rolldown_runtime.js";
+
+export * from "@langchain/core/structured_query"
+
+//#region src/chains/query_constructor/ir.ts
+var ir_exports = {};
+import * as import___langchain_core_structured_query from "@langchain/core/structured_query";
+__reExport(ir_exports, import___langchain_core_structured_query);
+
+//#endregion
+export { ir_exports };
+//# sourceMappingURL=ir.js.map

package/dist/chains/query_constructor/ir.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ir.js","names":[],"sources":["../../../src/chains/query_constructor/ir.ts"],"sourcesContent":["export * from \"@langchain/core/structured_query\";\n"],"mappings":""}