langchain 1.0.0-alpha.5 → 1.0.0-alpha.6

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

@@ -1 +1 @@
-{"version":3,"file":"openapi.js","names":["url: string","pathParams: Record<string, string>","newParams: Record<string, string>","params: OpenAPIV3_1.ParameterObject[]","spec: OpenAPISpec","jsonSchema: JsonSchema7ObjectType","schema: OpenAPIV3_1.SchemaObject","nameToCallMap: Record<string, { method: string; url: string }>","operationParams: Record<string, OpenAPIV3_1.ParameterObject[]>","paramLocationToRequestArgNameMap: Record<string, string>","requestArgsSchema: Record<string, JsonSchema7ObjectType> & {\n        data?:\n          | JsonSchema7ObjectType\n          | {\n              anyOf?: JsonSchema7ObjectType[];\n            };\n      }","requestBodySchemas: Record<string, JsonSchema7ObjectType>","openAIFunction: OpenAIClient.Chat.ChatCompletionCreateParams.Function","name: string","requestArgs: Record<string, any>","options?: {\n        headers?: Record<string, string>;\n        params?: Record<string, string>;\n      }","filteredArgs: Record<string, any>","headers: Record<string, string>","config: { requestMethod: SimpleRequestChainExecutionMethod }","values: ChainValues","_runManager?: CallbackManagerForChainRun","spec: OpenAPIV3_1.Document | string","options: OpenAPIChainOptions","e"],"sources":["../../../src/chains/openai_functions/openapi.ts"],"sourcesContent":["import type { OpenAIClient } from \"@langchain/openai\";\nimport {\n  type JsonSchema7ObjectType,\n  type JsonSchema7ArrayType,\n  type JsonSchema7Type,\n} from \"@langchain/core/utils/json_schema\";\nimport type { OpenAPIV3_1 } from \"openapi-types\";\n\nimport { ChainValues } from \"@langchain/core/utils/types\";\nimport { ChatOpenAI } from \"@langchain/openai\";\nimport { BaseChatModel } from \"@langchain/core/language_models/chat_models\";\nimport { BaseFunctionCallOptions } from \"@langchain/core/language_models/base\";\nimport {\n  ChatPromptTemplate,\n  HumanMessagePromptTemplate,\n  BasePromptTemplate,\n} from \"@langchain/core/prompts\";\nimport { CallbackManagerForChainRun } from \"@langchain/core/callbacks/manager\";\nimport { OpenAPISpec } from \"../../util/openapi.js\";\nimport { BaseChain } from \"../base.js\";\nimport { LLMChain, LLMChainInput } from \"../llm_chain.js\";\nimport { SequentialChain } from \"../sequential_chain.js\";\nimport { JsonOutputFunctionsParser } from \"../../output_parsers/openai_functions.js\";\n\n/**\n * Type representing a function for executing OpenAPI requests.\n */\ntype OpenAPIExecutionMethod = (\n  name: string,\n  // eslint-disable-next-line @typescript-eslint/no-explicit-any\n  requestArgs: Record<string, any>,\n  options?: {\n    headers?: Record<string, string>;\n    params?: Record<string, string>;\n  }\n) => Promise<string>;\n\n/**\n * Formats a URL by replacing path parameters with their corresponding\n * values.\n * @param url The URL to format.\n * @param pathParams The path parameters to replace in the URL.\n * @returns The formatted URL.\n */\nfunction formatURL(url: string, pathParams: Record<string, string>): string {\n  const expectedPathParamNames = [...url.matchAll(/{(.*?)}/g)].map(\n    (match) => match[1]\n  );\n  const newParams: Record<string, string> = {};\n  for (const paramName of expectedPathParamNames) {\n    const cleanParamName = paramName.replace(/^\\.;/, \"\").replace(/\\*$/, \"\");\n    const value = pathParams[cleanParamName];\n    let formattedValue;\n    if (Array.isArray(value)) {\n      if (paramName.startsWith(\".\")) {\n        const separator = paramName.endsWith(\"*\") ? \".\" : \",\";\n        formattedValue = `.${value.join(separator)}`;\n      } else if (paramName.startsWith(\",\")) {\n        const separator = paramName.endsWith(\"*\") ? `${cleanParamName}=` : \",\";\n        formattedValue = `${cleanParamName}=${value.join(separator)}`;\n      } else {\n        formattedValue = value.join(\",\");\n      }\n    } else if (typeof value === \"object\") {\n      const kvSeparator = paramName.endsWith(\"*\") ? \"=\" : \",\";\n      const kvStrings = Object.entries(value).map(\n        ([k, v]) => k + kvSeparator + v\n      );\n      let entrySeparator;\n      if (paramName.startsWith(\".\")) {\n        entrySeparator = \".\";\n        formattedValue = \".\";\n      } else if (paramName.startsWith(\";\")) {\n        entrySeparator = \";\";\n        formattedValue = \";\";\n      } else {\n        entrySeparator = \",\";\n        formattedValue = \"\";\n      }\n      formattedValue += kvStrings.join(entrySeparator);\n    } else {\n      if (paramName.startsWith(\".\")) {\n        formattedValue = `.${value}`;\n      } else if (paramName.startsWith(\";\")) {\n        formattedValue = `;${cleanParamName}=${value}`;\n      } else {\n        formattedValue = value;\n      }\n    }\n    newParams[paramName] = formattedValue;\n  }\n  let formattedUrl = url;\n  for (const [key, newValue] of Object.entries(newParams)) {\n    formattedUrl = formattedUrl.replace(`{${key}}`, newValue);\n  }\n  return formattedUrl;\n}\n\n/**\n * Converts OpenAPI parameters to JSON schema format.\n * @param params The OpenAPI parameters to convert.\n * @param spec The OpenAPI specification that contains the parameters.\n * @returns The JSON schema representation of the OpenAPI parameters.\n */\nfunction convertOpenAPIParamsToJSONSchema(\n  params: OpenAPIV3_1.ParameterObject[],\n  spec: OpenAPISpec\n) {\n  return params.reduce(\n    (jsonSchema: JsonSchema7ObjectType, param) => {\n      let schema;\n      if (param.schema) {\n        schema = spec.getSchema(param.schema);\n        // eslint-disable-next-line no-param-reassign\n        jsonSchema.properties[param.name] = convertOpenAPISchemaToJSONSchema(\n          schema,\n          spec\n        );\n      } else if (param.content) {\n        const mediaTypeSchema = Object.values(param.content)[0].schema;\n        if (mediaTypeSchema) {\n          schema = spec.getSchema(mediaTypeSchema);\n        }\n        if (!schema) {\n          return jsonSchema;\n        }\n        if (schema.description === undefined) {\n          schema.description = param.description ?? \"\";\n        }\n        // eslint-disable-next-line no-param-reassign\n        jsonSchema.properties[param.name] = convertOpenAPISchemaToJSONSchema(\n          schema,\n          spec\n        );\n      } else {\n        return jsonSchema;\n      }\n      if (param.required && Array.isArray(jsonSchema.required)) {\n        jsonSchema.required.push(param.name);\n      }\n      return jsonSchema;\n    },\n    {\n      type: \"object\",\n      properties: {},\n      required: [],\n      additionalProperties: {},\n    }\n  );\n}\n\n// OpenAI throws errors on extraneous schema properties, e.g. if \"required\" is set on individual ones\n/**\n * Converts OpenAPI schemas to JSON schema format.\n * @param schema The OpenAPI schema to convert.\n * @param spec The OpenAPI specification that contains the schema.\n * @returns The JSON schema representation of the OpenAPI schema.\n */\nexport function convertOpenAPISchemaToJSONSchema(\n  schema: OpenAPIV3_1.SchemaObject,\n  spec: OpenAPISpec\n): JsonSchema7Type {\n  if (schema.type === \"object\") {\n    return Object.keys(schema.properties ?? {}).reduce(\n      (jsonSchema: JsonSchema7ObjectType, propertyName) => {\n        if (!schema.properties) {\n          return jsonSchema;\n        }\n        const openAPIProperty = spec.getSchema(schema.properties[propertyName]);\n        if (openAPIProperty.type === undefined) {\n          return jsonSchema;\n        }\n        // eslint-disable-next-line no-param-reassign\n        jsonSchema.properties[propertyName] = convertOpenAPISchemaToJSONSchema(\n          openAPIProperty,\n          spec\n        );\n        if (\n          (openAPIProperty.required ||\n            schema.required?.includes(propertyName)) &&\n          jsonSchema.required !== undefined\n        ) {\n          jsonSchema.required.push(propertyName);\n        }\n        return jsonSchema;\n      },\n      {\n        type: \"object\",\n        properties: {},\n        required: [],\n        additionalProperties: {},\n      }\n    );\n  }\n  if (schema.type === \"array\") {\n    return {\n      type: \"array\",\n      items: convertOpenAPISchemaToJSONSchema(schema.items ?? {}, spec),\n      minItems: schema.minItems,\n      maxItems: schema.maxItems,\n    } as JsonSchema7ArrayType;\n  }\n  return {\n    type: schema.type ?? \"string\",\n  } as JsonSchema7Type;\n}\n\n/**\n * Converts an OpenAPI specification to OpenAI functions.\n * @param spec The OpenAPI specification to convert.\n * @returns An object containing the OpenAI functions derived from the OpenAPI specification and a default execution method.\n */\nexport function convertOpenAPISpecToOpenAIFunctions(spec: OpenAPISpec): {\n  openAIFunctions: OpenAIClient.Chat.ChatCompletionCreateParams.Function[];\n  defaultExecutionMethod?: OpenAPIExecutionMethod;\n} {\n  if (!spec.document.paths) {\n    return { openAIFunctions: [] };\n  }\n  const openAIFunctions = [];\n  const nameToCallMap: Record<string, { method: string; url: string }> = {};\n  for (const path of Object.keys(spec.document.paths)) {\n    const pathParameters = spec.getParametersForPath(path);\n    for (const method of spec.getMethodsForPath(path)) {\n      const operation = spec.getOperation(path, method);\n      if (!operation) {\n        return { openAIFunctions: [] };\n      }\n      const operationParametersByLocation = pathParameters\n        .concat(spec.getParametersForOperation(operation))\n        .reduce(\n          (\n            operationParams: Record<string, OpenAPIV3_1.ParameterObject[]>,\n            param\n          ) => {\n            if (!operationParams[param.in]) {\n              // eslint-disable-next-line no-param-reassign\n              operationParams[param.in] = [];\n            }\n            operationParams[param.in].push(param);\n            return operationParams;\n          },\n          {}\n        );\n      const paramLocationToRequestArgNameMap: Record<string, string> = {\n        query: \"params\",\n        header: \"headers\",\n        cookie: \"cookies\",\n        path: \"path_params\",\n      };\n      const requestArgsSchema: Record<string, JsonSchema7ObjectType> & {\n        data?:\n          | JsonSchema7ObjectType\n          | {\n              anyOf?: JsonSchema7ObjectType[];\n            };\n      } = {};\n      for (const paramLocation of Object.keys(\n        paramLocationToRequestArgNameMap\n      )) {\n        if (operationParametersByLocation[paramLocation]) {\n          requestArgsSchema[paramLocationToRequestArgNameMap[paramLocation]] =\n            convertOpenAPIParamsToJSONSchema(\n              operationParametersByLocation[paramLocation],\n              spec\n            );\n        }\n      }\n      const requestBody = spec.getRequestBodyForOperation(operation);\n      if (requestBody?.content !== undefined) {\n        const requestBodySchemas: Record<string, JsonSchema7ObjectType> = {};\n        for (const [mediaType, mediaTypeObject] of Object.entries(\n          requestBody.content\n        )) {\n          if (mediaTypeObject.schema !== undefined) {\n            const schema = spec.getSchema(mediaTypeObject.schema);\n            requestBodySchemas[mediaType] = convertOpenAPISchemaToJSONSchema(\n              schema,\n              spec\n            ) as JsonSchema7ObjectType;\n          }\n        }\n        const mediaTypes = Object.keys(requestBodySchemas);\n        if (mediaTypes.length === 1) {\n          requestArgsSchema.data = requestBodySchemas[mediaTypes[0]];\n        } else if (mediaTypes.length > 1) {\n          requestArgsSchema.data = {\n            anyOf: Object.values(requestBodySchemas),\n          };\n        }\n      }\n      const openAIFunction: OpenAIClient.Chat.ChatCompletionCreateParams.Function =\n        {\n          name: OpenAPISpec.getCleanedOperationId(operation, path, method),\n          description: operation.description ?? operation.summary ?? \"\",\n          parameters: {\n            type: \"object\",\n            properties: requestArgsSchema,\n            // All remaining top-level parameters are required\n            required: Object.keys(requestArgsSchema),\n          },\n        };\n\n      openAIFunctions.push(openAIFunction);\n      const baseUrl = (spec.baseUrl ?? \"\").endsWith(\"/\")\n        ? (spec.baseUrl ?? \"\").slice(0, -1)\n        : spec.baseUrl ?? \"\";\n      nameToCallMap[openAIFunction.name] = {\n        method,\n        url: baseUrl + path,\n      };\n    }\n  }\n  return {\n    openAIFunctions,\n    defaultExecutionMethod: async (\n      name: string,\n      // eslint-disable-next-line @typescript-eslint/no-explicit-any\n      requestArgs: Record<string, any>,\n      options?: {\n        headers?: Record<string, string>;\n        params?: Record<string, string>;\n      }\n    ) => {\n      const {\n        headers: customHeaders,\n        params: customParams,\n        ...rest\n      } = options ?? {};\n      const { method, url } = nameToCallMap[name];\n      const requestParams = requestArgs.params ?? {};\n      const nonEmptyParams = Object.keys(requestParams).reduce(\n        // eslint-disable-next-line @typescript-eslint/no-explicit-any\n        (filteredArgs: Record<string, any>, argName) => {\n          if (\n            requestParams[argName] !== \"\" &&\n            requestParams[argName] !== null &&\n            requestParams[argName] !== undefined\n          ) {\n            // eslint-disable-next-line no-param-reassign\n            filteredArgs[argName] = requestParams[argName];\n          }\n          return filteredArgs;\n        },\n        {}\n      );\n      const queryString = new URLSearchParams({\n        ...nonEmptyParams,\n        ...customParams,\n      }).toString();\n      const pathParams = requestArgs.path_params;\n      const formattedUrl =\n        formatURL(url, pathParams) +\n        (queryString.length ? `?${queryString}` : \"\");\n      const headers: Record<string, string> = {};\n      let body;\n      if (requestArgs.data !== undefined) {\n        let contentType = \"text/plain\";\n        if (typeof requestArgs.data !== \"string\") {\n          if (typeof requestArgs.data === \"object\") {\n            contentType = \"application/json\";\n          }\n          body = JSON.stringify(requestArgs.data);\n        } else {\n          body = requestArgs.data;\n        }\n        headers[\"content-type\"] = contentType;\n      }\n      const response = await fetch(formattedUrl, {\n        ...requestArgs,\n        method,\n        headers: {\n          ...headers,\n          ...requestArgs.headers,\n          ...customHeaders,\n        },\n        body,\n        ...rest,\n      });\n      let output;\n      if (response.status < 200 || response.status > 299) {\n        output = `${response.status}: ${\n          response.statusText\n        } for ${name} called with ${JSON.stringify(queryString)}`;\n      } else {\n        output = await response.text();\n      }\n      return output;\n    },\n  };\n}\n\n/**\n * Type representing a function for executing simple requests.\n */\ntype SimpleRequestChainExecutionMethod = (\n  name: string,\n  // eslint-disable-next-line @typescript-eslint/no-explicit-any\n  requestArgs: Record<string, any>\n) => Promise<string>;\n\n/**\n * A chain for making simple API requests.\n */\nclass SimpleRequestChain extends BaseChain {\n  static lc_name() {\n    return \"SimpleRequestChain\";\n  }\n\n  private requestMethod: SimpleRequestChainExecutionMethod;\n\n  inputKey = \"function\";\n\n  outputKey = \"response\";\n\n  constructor(config: { requestMethod: SimpleRequestChainExecutionMethod }) {\n    super();\n    this.requestMethod = config.requestMethod;\n  }\n\n  get inputKeys() {\n    return [this.inputKey];\n  }\n\n  get outputKeys() {\n    return [this.outputKey];\n  }\n\n  _chainType() {\n    return \"simple_request_chain\" as const;\n  }\n\n  /** @ignore */\n  async _call(\n    values: ChainValues,\n    _runManager?: CallbackManagerForChainRun\n  ): Promise<ChainValues> {\n    const inputKeyValue = values[this.inputKey];\n    const methodName = inputKeyValue.name;\n    const args = inputKeyValue.arguments;\n    const response = await this.requestMethod(methodName, args);\n    return { [this.outputKey]: response };\n  }\n}\n\n/**\n * Type representing the options for creating an OpenAPI chain.\n */\nexport type OpenAPIChainOptions = {\n  llm?: BaseChatModel<BaseFunctionCallOptions>;\n  prompt?: BasePromptTemplate;\n  requestChain?: BaseChain;\n  llmChainInputs?: LLMChainInput;\n  headers?: Record<string, string>;\n  params?: Record<string, string>;\n  verbose?: boolean;\n};\n\n/**\n * Create a chain for querying an API from a OpenAPI spec.\n * @param spec OpenAPISpec or url/file/text string corresponding to one.\n * @param options Custom options passed into the chain\n * @returns OpenAPIChain\n */\nexport async function createOpenAPIChain(\n  spec: OpenAPIV3_1.Document | string,\n  options: OpenAPIChainOptions = {}\n) {\n  let convertedSpec;\n  if (typeof spec === \"string\") {\n    try {\n      convertedSpec = await OpenAPISpec.fromURL(spec);\n    } catch (e) {\n      try {\n        convertedSpec = OpenAPISpec.fromString(spec);\n      } catch (e) {\n        throw new Error(`Unable to parse spec from source ${spec}.`);\n      }\n    }\n  } else {\n    convertedSpec = OpenAPISpec.fromObject(spec);\n  }\n  const { openAIFunctions, defaultExecutionMethod } =\n    convertOpenAPISpecToOpenAIFunctions(convertedSpec);\n  if (defaultExecutionMethod === undefined) {\n    throw new Error(\n      `Could not parse any valid operations from the provided spec.`\n    );\n  }\n  const {\n    llm = new ChatOpenAI({ model: \"gpt-3.5-turbo-0613\" }),\n    prompt = ChatPromptTemplate.fromMessages([\n      HumanMessagePromptTemplate.fromTemplate(\n        \"Use the provided API's to respond to this user query:\\n\\n{query}\"\n      ),\n    ]),\n    requestChain = new SimpleRequestChain({\n      requestMethod: async (name, args) =>\n        defaultExecutionMethod(name, args, {\n          headers: options.headers,\n          params: options.params,\n        }),\n    }),\n    llmChainInputs = {},\n    verbose,\n    ...rest\n  } = options;\n  const formatChain = new LLMChain({\n    llm,\n    prompt,\n    outputParser: new JsonOutputFunctionsParser({ argsOnly: false }),\n    outputKey: \"function\",\n    llmKwargs: { functions: openAIFunctions },\n    ...llmChainInputs,\n  });\n  return new SequentialChain({\n    chains: [formatChain, requestChain],\n    outputVariables: [\"response\"],\n    inputVariables: formatChain.inputKeys,\n    verbose,\n    ...rest,\n  });\n}\n"],"mappings":";;;;;;;;;;;;;;;;AA4CA,SAAS,UAAUA,KAAaC,YAA4C;CAC1E,MAAM,yBAAyB,CAAC,GAAG,IAAI,SAAS,WAAW,AAAC,EAAC,IAC3D,CAAC,UAAU,MAAM,GAClB;CACD,MAAMC,YAAoC,CAAE;AAC5C,MAAK,MAAM,aAAa,wBAAwB;EAC9C,MAAM,iBAAiB,UAAU,QAAQ,QAAQ,GAAG,CAAC,QAAQ,OAAO,GAAG;EACvE,MAAM,QAAQ,WAAW;EACzB,IAAI;AACJ,MAAI,MAAM,QAAQ,MAAM,CACtB,KAAI,UAAU,WAAW,IAAI,EAAE;GAC7B,MAAM,YAAY,UAAU,SAAS,IAAI,GAAG,MAAM;GAClD,iBAAiB,CAAC,CAAC,EAAE,MAAM,KAAK,UAAU,EAAE;EAC7C,WAAU,UAAU,WAAW,IAAI,EAAE;GACpC,MAAM,YAAY,UAAU,SAAS,IAAI,GAAG,GAAG,eAAe,CAAC,CAAC,GAAG;GACnE,iBAAiB,GAAG,eAAe,CAAC,EAAE,MAAM,KAAK,UAAU,EAAE;EAC9D,OACC,iBAAiB,MAAM,KAAK,IAAI;WAEzB,OAAO,UAAU,UAAU;GACpC,MAAM,cAAc,UAAU,SAAS,IAAI,GAAG,MAAM;GACpD,MAAM,YAAY,OAAO,QAAQ,MAAM,CAAC,IACtC,CAAC,CAAC,GAAG,EAAE,KAAK,IAAI,cAAc,EAC/B;GACD,IAAI;AACJ,OAAI,UAAU,WAAW,IAAI,EAAE;IAC7B,iBAAiB;IACjB,iBAAiB;GAClB,WAAU,UAAU,WAAW,IAAI,EAAE;IACpC,iBAAiB;IACjB,iBAAiB;GAClB,OAAM;IACL,iBAAiB;IACjB,iBAAiB;GAClB;GACD,kBAAkB,UAAU,KAAK,eAAe;EACjD,WACK,UAAU,WAAW,IAAI,EAC3B,iBAAiB,CAAC,CAAC,EAAE,OAAO;WACnB,UAAU,WAAW,IAAI,EAClC,iBAAiB,CAAC,CAAC,EAAE,eAAe,CAAC,EAAE,OAAO;OAE9C,iBAAiB;EAGrB,UAAU,aAAa;CACxB;CACD,IAAI,eAAe;AACnB,MAAK,MAAM,CAAC,KAAK,SAAS,IAAI,OAAO,QAAQ,UAAU,EACrD,eAAe,aAAa,QAAQ,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,EAAE,SAAS;AAE3D,QAAO;AACR;;;;;;;AAQD,SAAS,iCACPC,QACAC,MACA;AACA,QAAO,OAAO,OACZ,CAACC,YAAmC,UAAU;EAC5C,IAAI;AACJ,MAAI,MAAM,QAAQ;GAChB,SAAS,KAAK,UAAU,MAAM,OAAO;GAErC,WAAW,WAAW,MAAM,QAAQ,iCAClC,QACA,KACD;EACF,WAAU,MAAM,SAAS;GACxB,MAAM,kBAAkB,OAAO,OAAO,MAAM,QAAQ,CAAC,GAAG;AACxD,OAAI,iBACF,SAAS,KAAK,UAAU,gBAAgB;AAE1C,OAAI,CAAC,OACH,QAAO;AAET,OAAI,OAAO,gBAAgB,QACzB,OAAO,cAAc,MAAM,eAAe;GAG5C,WAAW,WAAW,MAAM,QAAQ,iCAClC,QACA,KACD;EACF,MACC,QAAO;AAET,MAAI,MAAM,YAAY,MAAM,QAAQ,WAAW,SAAS,EACtD,WAAW,SAAS,KAAK,MAAM,KAAK;AAEtC,SAAO;CACR,GACD;EACE,MAAM;EACN,YAAY,CAAE;EACd,UAAU,CAAE;EACZ,sBAAsB,CAAE;CACzB,EACF;AACF;;;;;;;AASD,SAAgB,iCACdC,QACAF,MACiB;AACjB,KAAI,OAAO,SAAS,SAClB,QAAO,OAAO,KAAK,OAAO,cAAc,CAAE,EAAC,CAAC,OAC1C,CAACC,YAAmC,iBAAiB;AACnD,MAAI,CAAC,OAAO,WACV,QAAO;EAET,MAAM,kBAAkB,KAAK,UAAU,OAAO,WAAW,cAAc;AACvE,MAAI,gBAAgB,SAAS,OAC3B,QAAO;EAGT,WAAW,WAAW,gBAAgB,iCACpC,iBACA,KACD;AACD,OACG,gBAAgB,YACf,OAAO,UAAU,SAAS,aAAa,KACzC,WAAW,aAAa,QAExB,WAAW,SAAS,KAAK,aAAa;AAExC,SAAO;CACR,GACD;EACE,MAAM;EACN,YAAY,CAAE;EACd,UAAU,CAAE;EACZ,sBAAsB,CAAE;CACzB,EACF;AAEH,KAAI,OAAO,SAAS,QAClB,QAAO;EACL,MAAM;EACN,OAAO,iCAAiC,OAAO,SAAS,CAAE,GAAE,KAAK;EACjE,UAAU,OAAO;EACjB,UAAU,OAAO;CAClB;AAEH,QAAO,EACL,MAAM,OAAO,QAAQ,SACtB;AACF;;;;;;AAOD,SAAgB,oCAAoCD,MAGlD;AACA,KAAI,CAAC,KAAK,SAAS,MACjB,QAAO,EAAE,iBAAiB,CAAE,EAAE;CAEhC,MAAM,kBAAkB,CAAE;CAC1B,MAAMG,gBAAiE,CAAE;AACzE,MAAK,MAAM,QAAQ,OAAO,KAAK,KAAK,SAAS,MAAM,EAAE;EACnD,MAAM,iBAAiB,KAAK,qBAAqB,KAAK;AACtD,OAAK,MAAM,UAAU,KAAK,kBAAkB,KAAK,EAAE;GACjD,MAAM,YAAY,KAAK,aAAa,MAAM,OAAO;AACjD,OAAI,CAAC,UACH,QAAO,EAAE,iBAAiB,CAAE,EAAE;GAEhC,MAAM,gCAAgC,eACnC,OAAO,KAAK,0BAA0B,UAAU,CAAC,CACjD,OACC,CACEC,iBACA,UACG;AACH,QAAI,CAAC,gBAAgB,MAAM,KAEzB,gBAAgB,MAAM,MAAM,CAAE;IAEhC,gBAAgB,MAAM,IAAI,KAAK,MAAM;AACrC,WAAO;GACR,GACD,CAAE,EACH;GACH,MAAMC,mCAA2D;IAC/D,OAAO;IACP,QAAQ;IACR,QAAQ;IACR,MAAM;GACP;GACD,MAAMC,oBAMF,CAAE;AACN,QAAK,MAAM,iBAAiB,OAAO,KACjC,iCACD,CACC,KAAI,8BAA8B,gBAChC,kBAAkB,iCAAiC,kBACjD,iCACE,8BAA8B,gBAC9B,KACD;GAGP,MAAM,cAAc,KAAK,2BAA2B,UAAU;AAC9D,OAAI,aAAa,YAAY,QAAW;IACtC,MAAMC,qBAA4D,CAAE;AACpE,SAAK,MAAM,CAAC,WAAW,gBAAgB,IAAI,OAAO,QAChD,YAAY,QACb,CACC,KAAI,gBAAgB,WAAW,QAAW;KACxC,MAAM,SAAS,KAAK,UAAU,gBAAgB,OAAO;KACrD,mBAAmB,aAAa,iCAC9B,QACA,KACD;IACF;IAEH,MAAM,aAAa,OAAO,KAAK,mBAAmB;AAClD,QAAI,WAAW,WAAW,GACxB,kBAAkB,OAAO,mBAAmB,WAAW;aAC9C,WAAW,SAAS,GAC7B,kBAAkB,OAAO,EACvB,OAAO,OAAO,OAAO,mBAAmB,CACzC;GAEJ;GACD,MAAMC,iBACJ;IACE,MAAM,YAAY,sBAAsB,WAAW,MAAM,OAAO;IAChE,aAAa,UAAU,eAAe,UAAU,WAAW;IAC3D,YAAY;KACV,MAAM;KACN,YAAY;KAEZ,UAAU,OAAO,KAAK,kBAAkB;IACzC;GACF;GAEH,gBAAgB,KAAK,eAAe;GACpC,MAAM,WAAW,KAAK,WAAW,IAAI,SAAS,IAAI,IAC7C,KAAK,WAAW,IAAI,MAAM,GAAG,GAAG,GACjC,KAAK,WAAW;GACpB,cAAc,eAAe,QAAQ;IACnC;IACA,KAAK,UAAU;GAChB;EACF;CACF;AACD,QAAO;EACL;EACA,wBAAwB,OACtBC,MAEAC,aACAC,YAIG;GACH,MAAM,EACJ,SAAS,eACT,QAAQ,aACR,GAAG,MACJ,GAAG,WAAW,CAAE;GACjB,MAAM,EAAE,QAAQ,KAAK,GAAG,cAAc;GACtC,MAAM,gBAAgB,YAAY,UAAU,CAAE;GAC9C,MAAM,iBAAiB,OAAO,KAAK,cAAc,CAAC,OAEhD,CAACC,cAAmC,YAAY;AAC9C,QACE,cAAc,aAAa,MAC3B,cAAc,aAAa,QAC3B,cAAc,aAAa,QAG3B,aAAa,WAAW,cAAc;AAExC,WAAO;GACR,GACD,CAAE,EACH;GACD,MAAM,cAAc,IAAI,gBAAgB;IACtC,GAAG;IACH,GAAG;GACJ,GAAE,UAAU;GACb,MAAM,aAAa,YAAY;GAC/B,MAAM,eACJ,UAAU,KAAK,WAAW,IACzB,YAAY,SAAS,CAAC,CAAC,EAAE,aAAa,GAAG;GAC5C,MAAMC,UAAkC,CAAE;GAC1C,IAAI;AACJ,OAAI,YAAY,SAAS,QAAW;IAClC,IAAI,cAAc;AAClB,QAAI,OAAO,YAAY,SAAS,UAAU;AACxC,SAAI,OAAO,YAAY,SAAS,UAC9B,cAAc;KAEhB,OAAO,KAAK,UAAU,YAAY,KAAK;IACxC,OACC,OAAO,YAAY;IAErB,QAAQ,kBAAkB;GAC3B;GACD,MAAM,WAAW,MAAM,MAAM,cAAc;IACzC,GAAG;IACH;IACA,SAAS;KACP,GAAG;KACH,GAAG,YAAY;KACf,GAAG;IACJ;IACD;IACA,GAAG;GACJ,EAAC;GACF,IAAI;AACJ,OAAI,SAAS,SAAS,OAAO,SAAS,SAAS,KAC7C,SAAS,GAAG,SAAS,OAAO,EAAE,EAC5B,SAAS,WACV,KAAK,EAAE,KAAK,aAAa,EAAE,KAAK,UAAU,YAAY,EAAE;QAEzD,SAAS,MAAM,SAAS,MAAM;AAEhC,UAAO;EACR;CACF;AACF;;;;AAcD,IAAM,qBAAN,cAAiC,UAAU;CACzC,OAAO,UAAU;AACf,SAAO;CACR;CAED,AAAQ;CAER,WAAW;CAEX,YAAY;CAEZ,YAAYC,QAA8D;EACxE,OAAO;EACP,KAAK,gBAAgB,OAAO;CAC7B;CAED,IAAI,YAAY;AACd,SAAO,CAAC,KAAK,QAAS;CACvB;CAED,IAAI,aAAa;AACf,SAAO,CAAC,KAAK,SAAU;CACxB;CAED,aAAa;AACX,SAAO;CACR;;CAGD,MAAM,MACJC,QACAC,aACsB;EACtB,MAAM,gBAAgB,OAAO,KAAK;EAClC,MAAM,aAAa,cAAc;EACjC,MAAM,OAAO,cAAc;EAC3B,MAAM,WAAW,MAAM,KAAK,cAAc,YAAY,KAAK;AAC3D,SAAO,GAAG,KAAK,YAAY,SAAU;CACtC;AACF;;;;;;;AAqBD,eAAsB,mBACpBC,MACAC,UAA+B,CAAE,GACjC;CACA,IAAI;AACJ,KAAI,OAAO,SAAS,SAClB,KAAI;EACF,gBAAgB,MAAM,YAAY,QAAQ,KAAK;CAChD,SAAQ,GAAG;AACV,MAAI;GACF,gBAAgB,YAAY,WAAW,KAAK;EAC7C,SAAQC,KAAG;AACV,SAAM,IAAI,MAAM,CAAC,iCAAiC,EAAE,KAAK,CAAC,CAAC;EAC5D;CACF;MAED,gBAAgB,YAAY,WAAW,KAAK;CAE9C,MAAM,EAAE,iBAAiB,wBAAwB,GAC/C,oCAAoC,cAAc;AACpD,KAAI,2BAA2B,OAC7B,OAAM,IAAI,MACR,CAAC,4DAA4D,CAAC;CAGlE,MAAM,EACJ,MAAM,IAAI,WAAW,EAAE,OAAO,qBAAsB,IACpD,SAAS,mBAAmB,aAAa,CACvC,2BAA2B,aACzB,mEACD,AACF,EAAC,EACF,eAAe,IAAI,mBAAmB,EACpC,eAAe,OAAO,MAAM,SAC1B,uBAAuB,MAAM,MAAM;EACjC,SAAS,QAAQ;EACjB,QAAQ,QAAQ;CACjB,EAAC,CACL,IACD,iBAAiB,CAAE,GACnB,QACA,GAAG,MACJ,GAAG;CACJ,MAAM,cAAc,IAAI,SAAS;EAC/B;EACA;EACA,cAAc,IAAI,0BAA0B,EAAE,UAAU,MAAO;EAC/D,WAAW;EACX,WAAW,EAAE,WAAW,gBAAiB;EACzC,GAAG;CACJ;AACD,QAAO,IAAI,gBAAgB;EACzB,QAAQ,CAAC,aAAa,YAAa;EACnC,iBAAiB,CAAC,UAAW;EAC7B,gBAAgB,YAAY;EAC5B;EACA,GAAG;CACJ;AACF"}
+{"version":3,"file":"openapi.js","names":["url: string","pathParams: Record<string, string>","newParams: Record<string, string>","params: OpenAPIV3_1.ParameterObject[]","spec: OpenAPISpec","jsonSchema: JsonSchema7ObjectType","schema: OpenAPIV3_1.SchemaObject","nameToCallMap: Record<string, { method: string; url: string }>","operationParams: Record<string, OpenAPIV3_1.ParameterObject[]>","paramLocationToRequestArgNameMap: Record<string, string>","requestArgsSchema: Record<string, JsonSchema7ObjectType> & {\n        data?:\n          | JsonSchema7ObjectType\n          | {\n              anyOf?: JsonSchema7ObjectType[];\n            };\n      }","requestBodySchemas: Record<string, JsonSchema7ObjectType>","openAIFunction: OpenAIClient.Chat.ChatCompletionCreateParams.Function","name: string","requestArgs: Record<string, any>","options?: {\n        headers?: Record<string, string>;\n        params?: Record<string, string>;\n      }","filteredArgs: Record<string, any>","headers: Record<string, string>","config: { requestMethod: SimpleRequestChainExecutionMethod }","values: ChainValues","_runManager?: CallbackManagerForChainRun","spec: OpenAPIV3_1.Document | string","options: OpenAPIChainOptions"],"sources":["../../../src/chains/openai_functions/openapi.ts"],"sourcesContent":["import type { OpenAIClient } from \"@langchain/openai\";\nimport {\n  type JsonSchema7ObjectType,\n  type JsonSchema7ArrayType,\n  type JsonSchema7Type,\n} from \"@langchain/core/utils/json_schema\";\nimport type { OpenAPIV3_1 } from \"openapi-types\";\n\nimport { ChainValues } from \"@langchain/core/utils/types\";\nimport { BaseChatModel } from \"@langchain/core/language_models/chat_models\";\nimport { BaseFunctionCallOptions } from \"@langchain/core/language_models/base\";\nimport {\n  ChatPromptTemplate,\n  HumanMessagePromptTemplate,\n  BasePromptTemplate,\n} from \"@langchain/core/prompts\";\nimport { CallbackManagerForChainRun } from \"@langchain/core/callbacks/manager\";\nimport { OpenAPISpec } from \"../../util/openapi.js\";\nimport { BaseChain } from \"../base.js\";\nimport { LLMChain, LLMChainInput } from \"../llm_chain.js\";\nimport { SequentialChain } from \"../sequential_chain.js\";\nimport { JsonOutputFunctionsParser } from \"../../output_parsers/openai_functions.js\";\n\n/**\n * Type representing a function for executing OpenAPI requests.\n */\ntype OpenAPIExecutionMethod = (\n  name: string,\n  // eslint-disable-next-line @typescript-eslint/no-explicit-any\n  requestArgs: Record<string, any>,\n  options?: {\n    headers?: Record<string, string>;\n    params?: Record<string, string>;\n  }\n) => Promise<string>;\n\n/**\n * Formats a URL by replacing path parameters with their corresponding\n * values.\n * @param url The URL to format.\n * @param pathParams The path parameters to replace in the URL.\n * @returns The formatted URL.\n */\nfunction formatURL(url: string, pathParams: Record<string, string>): string {\n  const expectedPathParamNames = [...url.matchAll(/{(.*?)}/g)].map(\n    (match) => match[1]\n  );\n  const newParams: Record<string, string> = {};\n  for (const paramName of expectedPathParamNames) {\n    const cleanParamName = paramName.replace(/^\\.;/, \"\").replace(/\\*$/, \"\");\n    const value = pathParams[cleanParamName];\n    let formattedValue;\n    if (Array.isArray(value)) {\n      if (paramName.startsWith(\".\")) {\n        const separator = paramName.endsWith(\"*\") ? \".\" : \",\";\n        formattedValue = `.${value.join(separator)}`;\n      } else if (paramName.startsWith(\",\")) {\n        const separator = paramName.endsWith(\"*\") ? `${cleanParamName}=` : \",\";\n        formattedValue = `${cleanParamName}=${value.join(separator)}`;\n      } else {\n        formattedValue = value.join(\",\");\n      }\n    } else if (typeof value === \"object\") {\n      const kvSeparator = paramName.endsWith(\"*\") ? \"=\" : \",\";\n      const kvStrings = Object.entries(value).map(\n        ([k, v]) => k + kvSeparator + v\n      );\n      let entrySeparator;\n      if (paramName.startsWith(\".\")) {\n        entrySeparator = \".\";\n        formattedValue = \".\";\n      } else if (paramName.startsWith(\";\")) {\n        entrySeparator = \";\";\n        formattedValue = \";\";\n      } else {\n        entrySeparator = \",\";\n        formattedValue = \"\";\n      }\n      formattedValue += kvStrings.join(entrySeparator);\n    } else {\n      if (paramName.startsWith(\".\")) {\n        formattedValue = `.${value}`;\n      } else if (paramName.startsWith(\";\")) {\n        formattedValue = `;${cleanParamName}=${value}`;\n      } else {\n        formattedValue = value;\n      }\n    }\n    newParams[paramName] = formattedValue;\n  }\n  let formattedUrl = url;\n  for (const [key, newValue] of Object.entries(newParams)) {\n    formattedUrl = formattedUrl.replace(`{${key}}`, newValue);\n  }\n  return formattedUrl;\n}\n\n/**\n * Converts OpenAPI parameters to JSON schema format.\n * @param params The OpenAPI parameters to convert.\n * @param spec The OpenAPI specification that contains the parameters.\n * @returns The JSON schema representation of the OpenAPI parameters.\n */\nfunction convertOpenAPIParamsToJSONSchema(\n  params: OpenAPIV3_1.ParameterObject[],\n  spec: OpenAPISpec\n) {\n  return params.reduce(\n    (jsonSchema: JsonSchema7ObjectType, param) => {\n      let schema;\n      if (param.schema) {\n        schema = spec.getSchema(param.schema);\n        jsonSchema.properties[param.name] = convertOpenAPISchemaToJSONSchema(\n          schema,\n          spec\n        );\n      } else if (param.content) {\n        const mediaTypeSchema = Object.values(param.content)[0].schema;\n        if (mediaTypeSchema) {\n          schema = spec.getSchema(mediaTypeSchema);\n        }\n        if (!schema) {\n          return jsonSchema;\n        }\n        if (schema.description === undefined) {\n          schema.description = param.description ?? \"\";\n        }\n        jsonSchema.properties[param.name] = convertOpenAPISchemaToJSONSchema(\n          schema,\n          spec\n        );\n      } else {\n        return jsonSchema;\n      }\n      if (param.required && Array.isArray(jsonSchema.required)) {\n        jsonSchema.required.push(param.name);\n      }\n      return jsonSchema;\n    },\n    {\n      type: \"object\",\n      properties: {},\n      required: [],\n      additionalProperties: {},\n    }\n  );\n}\n\n// OpenAI throws errors on extraneous schema properties, e.g. if \"required\" is set on individual ones\n/**\n * Converts OpenAPI schemas to JSON schema format.\n * @param schema The OpenAPI schema to convert.\n * @param spec The OpenAPI specification that contains the schema.\n * @returns The JSON schema representation of the OpenAPI schema.\n */\nexport function convertOpenAPISchemaToJSONSchema(\n  schema: OpenAPIV3_1.SchemaObject,\n  spec: OpenAPISpec\n): JsonSchema7Type {\n  if (schema.type === \"object\") {\n    return Object.keys(schema.properties ?? {}).reduce(\n      (jsonSchema: JsonSchema7ObjectType, propertyName) => {\n        if (!schema.properties) {\n          return jsonSchema;\n        }\n        const openAPIProperty = spec.getSchema(schema.properties[propertyName]);\n        if (openAPIProperty.type === undefined) {\n          return jsonSchema;\n        }\n        jsonSchema.properties[propertyName] = convertOpenAPISchemaToJSONSchema(\n          openAPIProperty,\n          spec\n        );\n        if (\n          (openAPIProperty.required ||\n            schema.required?.includes(propertyName)) &&\n          jsonSchema.required !== undefined\n        ) {\n          jsonSchema.required.push(propertyName);\n        }\n        return jsonSchema;\n      },\n      {\n        type: \"object\",\n        properties: {},\n        required: [],\n        additionalProperties: {},\n      }\n    );\n  }\n  if (schema.type === \"array\") {\n    return {\n      type: \"array\",\n      items: convertOpenAPISchemaToJSONSchema(schema.items ?? {}, spec),\n      minItems: schema.minItems,\n      maxItems: schema.maxItems,\n    } as JsonSchema7ArrayType;\n  }\n  return {\n    type: schema.type ?? \"string\",\n  } as JsonSchema7Type;\n}\n\n/**\n * Converts an OpenAPI specification to OpenAI functions.\n * @param spec The OpenAPI specification to convert.\n * @returns An object containing the OpenAI functions derived from the OpenAPI specification and a default execution method.\n */\nexport function convertOpenAPISpecToOpenAIFunctions(spec: OpenAPISpec): {\n  openAIFunctions: OpenAIClient.Chat.ChatCompletionCreateParams.Function[];\n  defaultExecutionMethod?: OpenAPIExecutionMethod;\n} {\n  if (!spec.document.paths) {\n    return { openAIFunctions: [] };\n  }\n  const openAIFunctions = [];\n  const nameToCallMap: Record<string, { method: string; url: string }> = {};\n  for (const path of Object.keys(spec.document.paths)) {\n    const pathParameters = spec.getParametersForPath(path);\n    for (const method of spec.getMethodsForPath(path)) {\n      const operation = spec.getOperation(path, method);\n      if (!operation) {\n        return { openAIFunctions: [] };\n      }\n      const operationParametersByLocation = pathParameters\n        .concat(spec.getParametersForOperation(operation))\n        .reduce(\n          (\n            operationParams: Record<string, OpenAPIV3_1.ParameterObject[]>,\n            param\n          ) => {\n            if (!operationParams[param.in]) {\n              operationParams[param.in] = [];\n            }\n            operationParams[param.in].push(param);\n            return operationParams;\n          },\n          {}\n        );\n      const paramLocationToRequestArgNameMap: Record<string, string> = {\n        query: \"params\",\n        header: \"headers\",\n        cookie: \"cookies\",\n        path: \"path_params\",\n      };\n      const requestArgsSchema: Record<string, JsonSchema7ObjectType> & {\n        data?:\n          | JsonSchema7ObjectType\n          | {\n              anyOf?: JsonSchema7ObjectType[];\n            };\n      } = {};\n      for (const paramLocation of Object.keys(\n        paramLocationToRequestArgNameMap\n      )) {\n        if (operationParametersByLocation[paramLocation]) {\n          requestArgsSchema[paramLocationToRequestArgNameMap[paramLocation]] =\n            convertOpenAPIParamsToJSONSchema(\n              operationParametersByLocation[paramLocation],\n              spec\n            );\n        }\n      }\n      const requestBody = spec.getRequestBodyForOperation(operation);\n      if (requestBody?.content !== undefined) {\n        const requestBodySchemas: Record<string, JsonSchema7ObjectType> = {};\n        for (const [mediaType, mediaTypeObject] of Object.entries(\n          requestBody.content\n        )) {\n          if (mediaTypeObject.schema !== undefined) {\n            const schema = spec.getSchema(mediaTypeObject.schema);\n            requestBodySchemas[mediaType] = convertOpenAPISchemaToJSONSchema(\n              schema,\n              spec\n            ) as JsonSchema7ObjectType;\n          }\n        }\n        const mediaTypes = Object.keys(requestBodySchemas);\n        if (mediaTypes.length === 1) {\n          requestArgsSchema.data = requestBodySchemas[mediaTypes[0]];\n        } else if (mediaTypes.length > 1) {\n          requestArgsSchema.data = {\n            anyOf: Object.values(requestBodySchemas),\n          };\n        }\n      }\n      const openAIFunction: OpenAIClient.Chat.ChatCompletionCreateParams.Function =\n        {\n          name: OpenAPISpec.getCleanedOperationId(operation, path, method),\n          description: operation.description ?? operation.summary ?? \"\",\n          parameters: {\n            type: \"object\",\n            properties: requestArgsSchema,\n            // All remaining top-level parameters are required\n            required: Object.keys(requestArgsSchema),\n          },\n        };\n\n      openAIFunctions.push(openAIFunction);\n      const baseUrl = (spec.baseUrl ?? \"\").endsWith(\"/\")\n        ? (spec.baseUrl ?? \"\").slice(0, -1)\n        : spec.baseUrl ?? \"\";\n      nameToCallMap[openAIFunction.name] = {\n        method,\n        url: baseUrl + path,\n      };\n    }\n  }\n  return {\n    openAIFunctions,\n    defaultExecutionMethod: async (\n      name: string,\n      // eslint-disable-next-line @typescript-eslint/no-explicit-any\n      requestArgs: Record<string, any>,\n      options?: {\n        headers?: Record<string, string>;\n        params?: Record<string, string>;\n      }\n    ) => {\n      const {\n        headers: customHeaders,\n        params: customParams,\n        ...rest\n      } = options ?? {};\n      const { method, url } = nameToCallMap[name];\n      const requestParams = requestArgs.params ?? {};\n      const nonEmptyParams = Object.keys(requestParams).reduce(\n        // eslint-disable-next-line @typescript-eslint/no-explicit-any\n        (filteredArgs: Record<string, any>, argName) => {\n          if (\n            requestParams[argName] !== \"\" &&\n            requestParams[argName] !== null &&\n            requestParams[argName] !== undefined\n          ) {\n            filteredArgs[argName] = requestParams[argName];\n          }\n          return filteredArgs;\n        },\n        {}\n      );\n      const queryString = new URLSearchParams({\n        ...nonEmptyParams,\n        ...customParams,\n      }).toString();\n      const pathParams = requestArgs.path_params;\n      const formattedUrl =\n        formatURL(url, pathParams) +\n        (queryString.length ? `?${queryString}` : \"\");\n      const headers: Record<string, string> = {};\n      let body;\n      if (requestArgs.data !== undefined) {\n        let contentType = \"text/plain\";\n        if (typeof requestArgs.data !== \"string\") {\n          if (typeof requestArgs.data === \"object\") {\n            contentType = \"application/json\";\n          }\n          body = JSON.stringify(requestArgs.data);\n        } else {\n          body = requestArgs.data;\n        }\n        headers[\"content-type\"] = contentType;\n      }\n      const response = await fetch(formattedUrl, {\n        ...requestArgs,\n        method,\n        headers: {\n          ...headers,\n          ...requestArgs.headers,\n          ...customHeaders,\n        },\n        body,\n        ...rest,\n      });\n      let output;\n      if (response.status < 200 || response.status > 299) {\n        output = `${response.status}: ${\n          response.statusText\n        } for ${name} called with ${JSON.stringify(queryString)}`;\n      } else {\n        output = await response.text();\n      }\n      return output;\n    },\n  };\n}\n\n/**\n * Type representing a function for executing simple requests.\n */\ntype SimpleRequestChainExecutionMethod = (\n  name: string,\n  // eslint-disable-next-line @typescript-eslint/no-explicit-any\n  requestArgs: Record<string, any>\n) => Promise<string>;\n\n/**\n * A chain for making simple API requests.\n */\nclass SimpleRequestChain extends BaseChain {\n  static lc_name() {\n    return \"SimpleRequestChain\";\n  }\n\n  private requestMethod: SimpleRequestChainExecutionMethod;\n\n  inputKey = \"function\";\n\n  outputKey = \"response\";\n\n  constructor(config: { requestMethod: SimpleRequestChainExecutionMethod }) {\n    super();\n    this.requestMethod = config.requestMethod;\n  }\n\n  get inputKeys() {\n    return [this.inputKey];\n  }\n\n  get outputKeys() {\n    return [this.outputKey];\n  }\n\n  _chainType() {\n    return \"simple_request_chain\" as const;\n  }\n\n  /** @ignore */\n  async _call(\n    values: ChainValues,\n    _runManager?: CallbackManagerForChainRun\n  ): Promise<ChainValues> {\n    const inputKeyValue = values[this.inputKey];\n    const methodName = inputKeyValue.name;\n    const args = inputKeyValue.arguments;\n    const response = await this.requestMethod(methodName, args);\n    return { [this.outputKey]: response };\n  }\n}\n\n/**\n * Type representing the options for creating an OpenAPI chain.\n */\nexport type OpenAPIChainOptions = {\n  llm?: BaseChatModel<BaseFunctionCallOptions>;\n  prompt?: BasePromptTemplate;\n  requestChain?: BaseChain;\n  llmChainInputs?: LLMChainInput;\n  headers?: Record<string, string>;\n  params?: Record<string, string>;\n  verbose?: boolean;\n};\n\n/**\n * Create a chain for querying an API from a OpenAPI spec.\n * @param spec OpenAPISpec or url/file/text string corresponding to one.\n * @param options Custom options passed into the chain\n * @returns OpenAPIChain\n */\nexport async function createOpenAPIChain(\n  spec: OpenAPIV3_1.Document | string,\n  options: OpenAPIChainOptions = {}\n) {\n  let convertedSpec;\n  if (typeof spec === \"string\") {\n    try {\n      convertedSpec = await OpenAPISpec.fromURL(spec);\n    } catch {\n      try {\n        convertedSpec = OpenAPISpec.fromString(spec);\n      } catch {\n        throw new Error(`Unable to parse spec from source ${spec}.`);\n      }\n    }\n  } else {\n    convertedSpec = OpenAPISpec.fromObject(spec);\n  }\n  const { openAIFunctions, defaultExecutionMethod } =\n    convertOpenAPISpecToOpenAIFunctions(convertedSpec);\n  if (defaultExecutionMethod === undefined) {\n    throw new Error(\n      `Could not parse any valid operations from the provided spec.`\n    );\n  }\n\n  if (!options.llm) {\n    throw new Error(\"`llm` option is required\");\n  }\n\n  const {\n    llm = options.llm,\n    prompt = ChatPromptTemplate.fromMessages([\n      HumanMessagePromptTemplate.fromTemplate(\n        \"Use the provided API's to respond to this user query:\\n\\n{query}\"\n      ),\n    ]),\n    requestChain = new SimpleRequestChain({\n      requestMethod: async (name, args) =>\n        defaultExecutionMethod(name, args, {\n          headers: options.headers,\n          params: options.params,\n        }),\n    }),\n    llmChainInputs = {},\n    verbose,\n    ...rest\n  } = options;\n  const formatChain = new LLMChain({\n    llm,\n    prompt,\n    outputParser: new JsonOutputFunctionsParser({ argsOnly: false }),\n    outputKey: \"function\",\n    llmKwargs: { functions: openAIFunctions },\n    ...llmChainInputs,\n  });\n  return new SequentialChain({\n    chains: [formatChain, requestChain],\n    outputVariables: [\"response\"],\n    inputVariables: formatChain.inputKeys,\n    verbose,\n    ...rest,\n  });\n}\n"],"mappings":";;;;;;;;;;;;;;;AA2CA,SAAS,UAAUA,KAAaC,YAA4C;CAC1E,MAAM,yBAAyB,CAAC,GAAG,IAAI,SAAS,WAAW,AAAC,EAAC,IAC3D,CAAC,UAAU,MAAM,GAClB;CACD,MAAMC,YAAoC,CAAE;AAC5C,MAAK,MAAM,aAAa,wBAAwB;EAC9C,MAAM,iBAAiB,UAAU,QAAQ,QAAQ,GAAG,CAAC,QAAQ,OAAO,GAAG;EACvE,MAAM,QAAQ,WAAW;EACzB,IAAI;AACJ,MAAI,MAAM,QAAQ,MAAM,CACtB,KAAI,UAAU,WAAW,IAAI,EAAE;GAC7B,MAAM,YAAY,UAAU,SAAS,IAAI,GAAG,MAAM;GAClD,iBAAiB,CAAC,CAAC,EAAE,MAAM,KAAK,UAAU,EAAE;EAC7C,WAAU,UAAU,WAAW,IAAI,EAAE;GACpC,MAAM,YAAY,UAAU,SAAS,IAAI,GAAG,GAAG,eAAe,CAAC,CAAC,GAAG;GACnE,iBAAiB,GAAG,eAAe,CAAC,EAAE,MAAM,KAAK,UAAU,EAAE;EAC9D,OACC,iBAAiB,MAAM,KAAK,IAAI;WAEzB,OAAO,UAAU,UAAU;GACpC,MAAM,cAAc,UAAU,SAAS,IAAI,GAAG,MAAM;GACpD,MAAM,YAAY,OAAO,QAAQ,MAAM,CAAC,IACtC,CAAC,CAAC,GAAG,EAAE,KAAK,IAAI,cAAc,EAC/B;GACD,IAAI;AACJ,OAAI,UAAU,WAAW,IAAI,EAAE;IAC7B,iBAAiB;IACjB,iBAAiB;GAClB,WAAU,UAAU,WAAW,IAAI,EAAE;IACpC,iBAAiB;IACjB,iBAAiB;GAClB,OAAM;IACL,iBAAiB;IACjB,iBAAiB;GAClB;GACD,kBAAkB,UAAU,KAAK,eAAe;EACjD,WACK,UAAU,WAAW,IAAI,EAC3B,iBAAiB,CAAC,CAAC,EAAE,OAAO;WACnB,UAAU,WAAW,IAAI,EAClC,iBAAiB,CAAC,CAAC,EAAE,eAAe,CAAC,EAAE,OAAO;OAE9C,iBAAiB;EAGrB,UAAU,aAAa;CACxB;CACD,IAAI,eAAe;AACnB,MAAK,MAAM,CAAC,KAAK,SAAS,IAAI,OAAO,QAAQ,UAAU,EACrD,eAAe,aAAa,QAAQ,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,EAAE,SAAS;AAE3D,QAAO;AACR;;;;;;;AAQD,SAAS,iCACPC,QACAC,MACA;AACA,QAAO,OAAO,OACZ,CAACC,YAAmC,UAAU;EAC5C,IAAI;AACJ,MAAI,MAAM,QAAQ;GAChB,SAAS,KAAK,UAAU,MAAM,OAAO;GACrC,WAAW,WAAW,MAAM,QAAQ,iCAClC,QACA,KACD;EACF,WAAU,MAAM,SAAS;GACxB,MAAM,kBAAkB,OAAO,OAAO,MAAM,QAAQ,CAAC,GAAG;AACxD,OAAI,iBACF,SAAS,KAAK,UAAU,gBAAgB;AAE1C,OAAI,CAAC,OACH,QAAO;AAET,OAAI,OAAO,gBAAgB,QACzB,OAAO,cAAc,MAAM,eAAe;GAE5C,WAAW,WAAW,MAAM,QAAQ,iCAClC,QACA,KACD;EACF,MACC,QAAO;AAET,MAAI,MAAM,YAAY,MAAM,QAAQ,WAAW,SAAS,EACtD,WAAW,SAAS,KAAK,MAAM,KAAK;AAEtC,SAAO;CACR,GACD;EACE,MAAM;EACN,YAAY,CAAE;EACd,UAAU,CAAE;EACZ,sBAAsB,CAAE;CACzB,EACF;AACF;;;;;;;AASD,SAAgB,iCACdC,QACAF,MACiB;AACjB,KAAI,OAAO,SAAS,SAClB,QAAO,OAAO,KAAK,OAAO,cAAc,CAAE,EAAC,CAAC,OAC1C,CAACC,YAAmC,iBAAiB;AACnD,MAAI,CAAC,OAAO,WACV,QAAO;EAET,MAAM,kBAAkB,KAAK,UAAU,OAAO,WAAW,cAAc;AACvE,MAAI,gBAAgB,SAAS,OAC3B,QAAO;EAET,WAAW,WAAW,gBAAgB,iCACpC,iBACA,KACD;AACD,OACG,gBAAgB,YACf,OAAO,UAAU,SAAS,aAAa,KACzC,WAAW,aAAa,QAExB,WAAW,SAAS,KAAK,aAAa;AAExC,SAAO;CACR,GACD;EACE,MAAM;EACN,YAAY,CAAE;EACd,UAAU,CAAE;EACZ,sBAAsB,CAAE;CACzB,EACF;AAEH,KAAI,OAAO,SAAS,QAClB,QAAO;EACL,MAAM;EACN,OAAO,iCAAiC,OAAO,SAAS,CAAE,GAAE,KAAK;EACjE,UAAU,OAAO;EACjB,UAAU,OAAO;CAClB;AAEH,QAAO,EACL,MAAM,OAAO,QAAQ,SACtB;AACF;;;;;;AAOD,SAAgB,oCAAoCD,MAGlD;AACA,KAAI,CAAC,KAAK,SAAS,MACjB,QAAO,EAAE,iBAAiB,CAAE,EAAE;CAEhC,MAAM,kBAAkB,CAAE;CAC1B,MAAMG,gBAAiE,CAAE;AACzE,MAAK,MAAM,QAAQ,OAAO,KAAK,KAAK,SAAS,MAAM,EAAE;EACnD,MAAM,iBAAiB,KAAK,qBAAqB,KAAK;AACtD,OAAK,MAAM,UAAU,KAAK,kBAAkB,KAAK,EAAE;GACjD,MAAM,YAAY,KAAK,aAAa,MAAM,OAAO;AACjD,OAAI,CAAC,UACH,QAAO,EAAE,iBAAiB,CAAE,EAAE;GAEhC,MAAM,gCAAgC,eACnC,OAAO,KAAK,0BAA0B,UAAU,CAAC,CACjD,OACC,CACEC,iBACA,UACG;AACH,QAAI,CAAC,gBAAgB,MAAM,KACzB,gBAAgB,MAAM,MAAM,CAAE;IAEhC,gBAAgB,MAAM,IAAI,KAAK,MAAM;AACrC,WAAO;GACR,GACD,CAAE,EACH;GACH,MAAMC,mCAA2D;IAC/D,OAAO;IACP,QAAQ;IACR,QAAQ;IACR,MAAM;GACP;GACD,MAAMC,oBAMF,CAAE;AACN,QAAK,MAAM,iBAAiB,OAAO,KACjC,iCACD,CACC,KAAI,8BAA8B,gBAChC,kBAAkB,iCAAiC,kBACjD,iCACE,8BAA8B,gBAC9B,KACD;GAGP,MAAM,cAAc,KAAK,2BAA2B,UAAU;AAC9D,OAAI,aAAa,YAAY,QAAW;IACtC,MAAMC,qBAA4D,CAAE;AACpE,SAAK,MAAM,CAAC,WAAW,gBAAgB,IAAI,OAAO,QAChD,YAAY,QACb,CACC,KAAI,gBAAgB,WAAW,QAAW;KACxC,MAAM,SAAS,KAAK,UAAU,gBAAgB,OAAO;KACrD,mBAAmB,aAAa,iCAC9B,QACA,KACD;IACF;IAEH,MAAM,aAAa,OAAO,KAAK,mBAAmB;AAClD,QAAI,WAAW,WAAW,GACxB,kBAAkB,OAAO,mBAAmB,WAAW;aAC9C,WAAW,SAAS,GAC7B,kBAAkB,OAAO,EACvB,OAAO,OAAO,OAAO,mBAAmB,CACzC;GAEJ;GACD,MAAMC,iBACJ;IACE,MAAM,YAAY,sBAAsB,WAAW,MAAM,OAAO;IAChE,aAAa,UAAU,eAAe,UAAU,WAAW;IAC3D,YAAY;KACV,MAAM;KACN,YAAY;KAEZ,UAAU,OAAO,KAAK,kBAAkB;IACzC;GACF;GAEH,gBAAgB,KAAK,eAAe;GACpC,MAAM,WAAW,KAAK,WAAW,IAAI,SAAS,IAAI,IAC7C,KAAK,WAAW,IAAI,MAAM,GAAG,GAAG,GACjC,KAAK,WAAW;GACpB,cAAc,eAAe,QAAQ;IACnC;IACA,KAAK,UAAU;GAChB;EACF;CACF;AACD,QAAO;EACL;EACA,wBAAwB,OACtBC,MAEAC,aACAC,YAIG;GACH,MAAM,EACJ,SAAS,eACT,QAAQ,aACR,GAAG,MACJ,GAAG,WAAW,CAAE;GACjB,MAAM,EAAE,QAAQ,KAAK,GAAG,cAAc;GACtC,MAAM,gBAAgB,YAAY,UAAU,CAAE;GAC9C,MAAM,iBAAiB,OAAO,KAAK,cAAc,CAAC,OAEhD,CAACC,cAAmC,YAAY;AAC9C,QACE,cAAc,aAAa,MAC3B,cAAc,aAAa,QAC3B,cAAc,aAAa,QAE3B,aAAa,WAAW,cAAc;AAExC,WAAO;GACR,GACD,CAAE,EACH;GACD,MAAM,cAAc,IAAI,gBAAgB;IACtC,GAAG;IACH,GAAG;GACJ,GAAE,UAAU;GACb,MAAM,aAAa,YAAY;GAC/B,MAAM,eACJ,UAAU,KAAK,WAAW,IACzB,YAAY,SAAS,CAAC,CAAC,EAAE,aAAa,GAAG;GAC5C,MAAMC,UAAkC,CAAE;GAC1C,IAAI;AACJ,OAAI,YAAY,SAAS,QAAW;IAClC,IAAI,cAAc;AAClB,QAAI,OAAO,YAAY,SAAS,UAAU;AACxC,SAAI,OAAO,YAAY,SAAS,UAC9B,cAAc;KAEhB,OAAO,KAAK,UAAU,YAAY,KAAK;IACxC,OACC,OAAO,YAAY;IAErB,QAAQ,kBAAkB;GAC3B;GACD,MAAM,WAAW,MAAM,MAAM,cAAc;IACzC,GAAG;IACH;IACA,SAAS;KACP,GAAG;KACH,GAAG,YAAY;KACf,GAAG;IACJ;IACD;IACA,GAAG;GACJ,EAAC;GACF,IAAI;AACJ,OAAI,SAAS,SAAS,OAAO,SAAS,SAAS,KAC7C,SAAS,GAAG,SAAS,OAAO,EAAE,EAC5B,SAAS,WACV,KAAK,EAAE,KAAK,aAAa,EAAE,KAAK,UAAU,YAAY,EAAE;QAEzD,SAAS,MAAM,SAAS,MAAM;AAEhC,UAAO;EACR;CACF;AACF;;;;AAcD,IAAM,qBAAN,cAAiC,UAAU;CACzC,OAAO,UAAU;AACf,SAAO;CACR;CAED,AAAQ;CAER,WAAW;CAEX,YAAY;CAEZ,YAAYC,QAA8D;EACxE,OAAO;EACP,KAAK,gBAAgB,OAAO;CAC7B;CAED,IAAI,YAAY;AACd,SAAO,CAAC,KAAK,QAAS;CACvB;CAED,IAAI,aAAa;AACf,SAAO,CAAC,KAAK,SAAU;CACxB;CAED,aAAa;AACX,SAAO;CACR;;CAGD,MAAM,MACJC,QACAC,aACsB;EACtB,MAAM,gBAAgB,OAAO,KAAK;EAClC,MAAM,aAAa,cAAc;EACjC,MAAM,OAAO,cAAc;EAC3B,MAAM,WAAW,MAAM,KAAK,cAAc,YAAY,KAAK;AAC3D,SAAO,GAAG,KAAK,YAAY,SAAU;CACtC;AACF;;;;;;;AAqBD,eAAsB,mBACpBC,MACAC,UAA+B,CAAE,GACjC;CACA,IAAI;AACJ,KAAI,OAAO,SAAS,SAClB,KAAI;EACF,gBAAgB,MAAM,YAAY,QAAQ,KAAK;CAChD,QAAO;AACN,MAAI;GACF,gBAAgB,YAAY,WAAW,KAAK;EAC7C,QAAO;AACN,SAAM,IAAI,MAAM,CAAC,iCAAiC,EAAE,KAAK,CAAC,CAAC;EAC5D;CACF;MAED,gBAAgB,YAAY,WAAW,KAAK;CAE9C,MAAM,EAAE,iBAAiB,wBAAwB,GAC/C,oCAAoC,cAAc;AACpD,KAAI,2BAA2B,OAC7B,OAAM,IAAI,MACR,CAAC,4DAA4D,CAAC;AAIlE,KAAI,CAAC,QAAQ,IACX,OAAM,IAAI,MAAM;CAGlB,MAAM,EACJ,MAAM,QAAQ,KACd,SAAS,mBAAmB,aAAa,CACvC,2BAA2B,aACzB,mEACD,AACF,EAAC,EACF,eAAe,IAAI,mBAAmB,EACpC,eAAe,OAAO,MAAM,SAC1B,uBAAuB,MAAM,MAAM;EACjC,SAAS,QAAQ;EACjB,QAAQ,QAAQ;CACjB,EAAC,CACL,IACD,iBAAiB,CAAE,GACnB,QACA,GAAG,MACJ,GAAG;CACJ,MAAM,cAAc,IAAI,SAAS;EAC/B;EACA;EACA,cAAc,IAAI,0BAA0B,EAAE,UAAU,MAAO;EAC/D,WAAW;EACX,WAAW,EAAE,WAAW,gBAAiB;EACzC,GAAG;CACJ;AACD,QAAO,IAAI,gBAAgB;EACzB,QAAQ,CAAC,aAAa,YAAa;EACnC,iBAAiB,CAAC,UAAW;EAC7B,gBAAgB,YAAY;EAC5B;EACA,GAAG;CACJ;AACF"}

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

@@ -1 +1 @@
-{"version":3,"file":"tagging.cjs","names":["schema: FunctionParameters","llm: BaseChatModel<BaseFunctionCallOptions>","options: TaggingChainOptions","PromptTemplate","JsonOutputFunctionsParser","LLMChain","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  // eslint-disable-next-line @typescript-eslint/no-explicit-any\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,SAASC,wCAAe,aAAa,iBAAiB,CAAE,GAAG,MAAM,GACvE;CACF,MAAM,YAAY,oBAAoB,OAAO;CAC7C,MAAM,eAAe,IAAIC;AACzB,QAAO,IAAIC,2BAAS;EAClB;EACA;EACA,WAAW,EAAE,UAAW;EACxB;EACA,MAAM,CAAC,oBAAoB,SAAU;EACrC,GAAG;CACJ;AACF;;;;;;;;;;;;;;AAeD,SAAgB,0BAEdC,QACAL,KACAM,SAC0E;AAC1E,QAAO,wEACQ,OAAO,EACpB,KACA,QACD;AACF"}
+{"version":3,"file":"tagging.cjs","names":["schema: FunctionParameters","llm: BaseChatModel<BaseFunctionCallOptions>","options: TaggingChainOptions","PromptTemplate","JsonOutputFunctionsParser","LLMChain","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,SAASC,wCAAe,aAAa,iBAAiB,CAAE,GAAG,MAAM,GACvE;CACF,MAAM,YAAY,oBAAoB,OAAO;CAC7C,MAAM,eAAe,IAAIC;AACzB,QAAO,IAAIC,2BAAS;EAClB;EACA;EACA,WAAW,EAAE,UAAW;EACxB;EACA,MAAM,CAAC,oBAAoB,SAAU;EACrC,GAAG;CACJ;AACF;;;;;;;;;;;;;;AAeD,SAAgB,0BACdC,QACAL,KACAM,SAC0E;AAC1E,QAAO,wEACQ,OAAO,EACpB,KACA,QACD;AACF"}

package/dist/chains/openai_functions/tagging.d.cts

@@ -41,9 +41,7 @@ declare function createTaggingChain(schema: FunctionParameters, llm: BaseChatMod
  * Switch to expression language: https://js.langchain.com/docs/expression_language/
  * Will be removed in 0.2.0
  */
-declare function createTaggingChainFromZod(
-// eslint-disable-next-line @typescript-eslint/no-explicit-any
-schema: InteropZodObject, llm: BaseChatModel<BaseFunctionCallOptions>, options?: TaggingChainOptions): LLMChain<object, BaseChatModel<BaseFunctionCallOptions, AIMessageChunk>>;
+declare function createTaggingChainFromZod(schema: InteropZodObject, llm: BaseChatModel<BaseFunctionCallOptions>, options?: TaggingChainOptions): LLMChain<object, BaseChatModel<BaseFunctionCallOptions, AIMessageChunk>>;
 //#endregion
 export { TaggingChainOptions, createTaggingChain, createTaggingChainFromZod };
 //# sourceMappingURL=tagging.d.cts.map

package/dist/chains/openai_functions/tagging.d.cts.map

@@ -1 +1 @@
-{"version":3,"file":"tagging.d.cts","names":["BaseChatModel","BaseFunctionCallOptions","AIMessageChunk","PromptTemplate","InteropZodObject","FunctionParameters","LLMChain","LLMChainInput","TaggingChainOptions","Omit","createTaggingChain","createTaggingChainFromZod"],"sources":["../../../src/chains/openai_functions/tagging.d.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 { FunctionParameters } from \"../../output_parsers/openai_functions.js\";\nimport { LLMChain, LLMChainInput } from \"../llm_chain.js\";\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 * 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 declare function createTaggingChain(schema: FunctionParameters, llm: BaseChatModel<BaseFunctionCallOptions>, options?: TaggingChainOptions): LLMChain<object, BaseChatModel<BaseFunctionCallOptions, AIMessageChunk>>;\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 declare function createTaggingChainFromZod(\n// eslint-disable-next-line @typescript-eslint/no-explicit-any\nschema: InteropZodObject, llm: BaseChatModel<BaseFunctionCallOptions>, options?: TaggingChainOptions): LLMChain<object, BaseChatModel<BaseFunctionCallOptions, AIMessageChunk>>;\n"],"mappings":";;;;;;;;;;;;AAUA;AAA+B,KAAnBQ,mBAAAA,GAAmB;EAAA,MAClBL,CAAAA,EAAAA,cAAAA;CAAc,GACvBM,IAAKF,CAAAA,aAAAA,CAAAA,MAAAA,CAAAA,EAAAA,QAAAA,GAAAA,KAAAA,CAAAA;;AAAD;AAcR;;;;;;;;;;AAA4J;AAcpII,iBAdAD,kBAAAA,CAcyB,MAAA,EAdEL,kBAcF,EAAA,GAAA,EAd2BL,aAc3B,CAdyCC,uBAczC,CAAA,EAAA,OAAA,CAAA,EAd6EO,mBAc7E,CAAA,EAdmGF,QAcnG,CAAA,MAAA,EAdoHN,aAcpH,CAdkIC,uBAclI,EAd2JC,cAc3J,CAAA,CAAA;;;;;;;;;;AAE8D;;;;iBAFvFS,yBAAAA;;QAEhBP,uBAAuBJ,cAAcC,oCAAoCO,sBAAsBF,iBAAiBN,cAAcC,yBAAyBC"}
+{"version":3,"file":"tagging.d.cts","names":["BaseChatModel","BaseFunctionCallOptions","AIMessageChunk","PromptTemplate","InteropZodObject","FunctionParameters","LLMChain","LLMChainInput","TaggingChainOptions","Omit","createTaggingChain","createTaggingChainFromZod"],"sources":["../../../src/chains/openai_functions/tagging.d.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 { FunctionParameters } from \"../../output_parsers/openai_functions.js\";\nimport { LLMChain, LLMChainInput } from \"../llm_chain.js\";\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 * 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 declare function createTaggingChain(schema: FunctionParameters, llm: BaseChatModel<BaseFunctionCallOptions>, options?: TaggingChainOptions): LLMChain<object, BaseChatModel<BaseFunctionCallOptions, AIMessageChunk>>;\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 declare function createTaggingChainFromZod(schema: InteropZodObject, llm: BaseChatModel<BaseFunctionCallOptions>, options?: TaggingChainOptions): LLMChain<object, BaseChatModel<BaseFunctionCallOptions, AIMessageChunk>>;\n"],"mappings":";;;;;;;;;;;;AAUA;AAA+B,KAAnBQ,mBAAAA,GAAmB;EAAA,MAClBL,CAAAA,EAAAA,cAAAA;CAAc,GACvBM,IAAKF,CAAAA,aAAAA,CAAAA,MAAAA,CAAAA,EAAAA,QAAAA,GAAAA,KAAAA,CAAAA;;AAAD;AAcR;;;;;;;;;;AAA4J;AAcpII,iBAdAD,kBAAAA,CAcyB,MAAA,EAdEL,kBAcF,EAAA,GAAA,EAd2BL,aAc3B,CAdyCC,uBAczC,CAAA,EAAA,OAAA,CAAA,EAd6EO,mBAc7E,CAAA,EAdmGF,QAcnG,CAAA,MAAA,EAdoHN,aAcpH,CAdkIC,uBAclI,EAd2JC,cAc3J,CAAA,CAAA;;;;;;;;;;AAAgH;;;;iBAAzIS,yBAAAA,SAAkCP,uBAAuBJ,cAAcC,oCAAoCO,sBAAsBF,iBAAiBN,cAAcC,yBAAyBC"}

package/dist/chains/openai_functions/tagging.d.ts

@@ -41,9 +41,7 @@ declare function createTaggingChain(schema: FunctionParameters, llm: BaseChatMod
  * Switch to expression language: https://js.langchain.com/docs/expression_language/
  * Will be removed in 0.2.0
  */
-declare function createTaggingChainFromZod(
-// eslint-disable-next-line @typescript-eslint/no-explicit-any
-schema: InteropZodObject, llm: BaseChatModel<BaseFunctionCallOptions>, options?: TaggingChainOptions): LLMChain<object, BaseChatModel<BaseFunctionCallOptions, AIMessageChunk>>;
+declare function createTaggingChainFromZod(schema: InteropZodObject, llm: BaseChatModel<BaseFunctionCallOptions>, options?: TaggingChainOptions): LLMChain<object, BaseChatModel<BaseFunctionCallOptions, AIMessageChunk>>;
 //#endregion
 export { TaggingChainOptions, createTaggingChain, createTaggingChainFromZod };
 //# sourceMappingURL=tagging.d.ts.map

package/dist/chains/openai_functions/tagging.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"tagging.d.ts","names":["BaseChatModel","BaseFunctionCallOptions","AIMessageChunk","PromptTemplate","InteropZodObject","FunctionParameters","LLMChain","LLMChainInput","TaggingChainOptions","Omit","createTaggingChain","createTaggingChainFromZod"],"sources":["../../../src/chains/openai_functions/tagging.d.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 { FunctionParameters } from \"../../output_parsers/openai_functions.js\";\nimport { LLMChain, LLMChainInput } from \"../llm_chain.js\";\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 * 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 declare function createTaggingChain(schema: FunctionParameters, llm: BaseChatModel<BaseFunctionCallOptions>, options?: TaggingChainOptions): LLMChain<object, BaseChatModel<BaseFunctionCallOptions, AIMessageChunk>>;\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 declare function createTaggingChainFromZod(\n// eslint-disable-next-line @typescript-eslint/no-explicit-any\nschema: InteropZodObject, llm: BaseChatModel<BaseFunctionCallOptions>, options?: TaggingChainOptions): LLMChain<object, BaseChatModel<BaseFunctionCallOptions, AIMessageChunk>>;\n"],"mappings":";;;;;;;;;;;;AAUA;AAA+B,KAAnBQ,mBAAAA,GAAmB;EAAA,MAClBL,CAAAA,EAAAA,cAAAA;CAAc,GACvBM,IAAKF,CAAAA,aAAAA,CAAAA,MAAAA,CAAAA,EAAAA,QAAAA,GAAAA,KAAAA,CAAAA;;AAAD;AAcR;;;;;;;;;;AAA4J;AAcpII,iBAdAD,kBAAAA,CAcyB,MAAA,EAdEL,kBAcF,EAAA,GAAA,EAd2BL,aAc3B,CAdyCC,uBAczC,CAAA,EAAA,OAAA,CAAA,EAd6EO,mBAc7E,CAAA,EAdmGF,QAcnG,CAAA,MAAA,EAdoHN,aAcpH,CAdkIC,uBAclI,EAd2JC,cAc3J,CAAA,CAAA;;;;;;;;;;AAE8D;;;;iBAFvFS,yBAAAA;;QAEhBP,uBAAuBJ,cAAcC,oCAAoCO,sBAAsBF,iBAAiBN,cAAcC,yBAAyBC"}
+{"version":3,"file":"tagging.d.ts","names":["BaseChatModel","BaseFunctionCallOptions","AIMessageChunk","PromptTemplate","InteropZodObject","FunctionParameters","LLMChain","LLMChainInput","TaggingChainOptions","Omit","createTaggingChain","createTaggingChainFromZod"],"sources":["../../../src/chains/openai_functions/tagging.d.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 { FunctionParameters } from \"../../output_parsers/openai_functions.js\";\nimport { LLMChain, LLMChainInput } from \"../llm_chain.js\";\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 * 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 declare function createTaggingChain(schema: FunctionParameters, llm: BaseChatModel<BaseFunctionCallOptions>, options?: TaggingChainOptions): LLMChain<object, BaseChatModel<BaseFunctionCallOptions, AIMessageChunk>>;\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 declare function createTaggingChainFromZod(schema: InteropZodObject, llm: BaseChatModel<BaseFunctionCallOptions>, options?: TaggingChainOptions): LLMChain<object, BaseChatModel<BaseFunctionCallOptions, AIMessageChunk>>;\n"],"mappings":";;;;;;;;;;;;AAUA;AAA+B,KAAnBQ,mBAAAA,GAAmB;EAAA,MAClBL,CAAAA,EAAAA,cAAAA;CAAc,GACvBM,IAAKF,CAAAA,aAAAA,CAAAA,MAAAA,CAAAA,EAAAA,QAAAA,GAAAA,KAAAA,CAAAA;;AAAD;AAcR;;;;;;;;;;AAA4J;AAcpII,iBAdAD,kBAAAA,CAcyB,MAAA,EAdEL,kBAcF,EAAA,GAAA,EAd2BL,aAc3B,CAdyCC,uBAczC,CAAA,EAAA,OAAA,CAAA,EAd6EO,mBAc7E,CAAA,EAdmGF,QAcnG,CAAA,MAAA,EAdoHN,aAcpH,CAdkIC,uBAclI,EAd2JC,cAc3J,CAAA,CAAA;;;;;;;;;;AAAgH;;;;iBAAzIS,yBAAAA,SAAkCP,uBAAuBJ,cAAcC,oCAAoCO,sBAAsBF,iBAAiBN,cAAcC,yBAAyBC"}

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

@@ -1 +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  // eslint-disable-next-line @typescript-eslint/no-explicit-any\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,0BAEdC,QACAF,KACAG,SAC0E;AAC1E,QAAO,mBACL,aAAa,OAAO,EACpB,KACA,QACD;AACF"}
+{"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

@@ -2,7 +2,7 @@ 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 zod = require_rolldown_runtime.__toESM(require("zod"));
+const zod_v3 = require_rolldown_runtime.__toESM(require("zod/v3"));
 const __langchain_core_prompts = require_rolldown_runtime.__toESM(require("@langchain/core/prompts"));
 const __langchain_core_structured_query = require_rolldown_runtime.__toESM(require("@langchain/core/structured_query"));
 
@@ -32,9 +32,9 @@ var AttributeInfo = class {
 		this.description = description;
 	}
 };
-const queryInputSchema = /* @__PURE__ */ zod.z.object({
-	query: /* @__PURE__ */ zod.z.string().describe("text string to compare to document contents"),
-	filter: /* @__PURE__ */ zod.z.string().optional().describe("logical condition statement for filtering documents")
+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

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

@@ -1 +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\";\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): 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,MAAE,OAAO;CAChD,uBAAuBA,MACpB,QAAQ,CACR,SAAS,8CAA8C;CAC1D,wBAAwBA,MACrB,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,MACqE;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"}
+{"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

@@ -2,7 +2,7 @@ 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 { BaseLanguageModelInterface } from "@langchain/core/language_models/base";
-import { z } from "zod";
+import { z } from "zod/v3";
 import { InferInteropZodOutput, InputValues } from "@langchain/core/utils/types";
 import { Runnable, RunnableConfig } from "@langchain/core/runnables";
 import { Comparator, Operator, StructuredQuery } from "@langchain/core/structured_query";
@@ -74,7 +74,9 @@ type QueryConstructorRunnableOptions = {
 };
 /** @deprecated */
 type QueryConstructorChainOptions = QueryConstructorRunnableOptions;
-declare function loadQueryConstructorRunnable(opts: QueryConstructorRunnableOptions): Runnable<any, StructuredQuery, RunnableConfig<Record<string, any>>>;
+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

@@ -1 +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\";\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): 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;;;;;AAAiD;;6CAhBtDF,iCAAiCC,aAAakB;;iBAErEI,mBAAAA,OAA0BV;;;;KAItCW,+BAAAA;OACH3B;;iBAEUgB;aACJd;uBACUC;qBACFC;;;KAGXwB,4BAAAA,GAA+BD;iBACnBE,4BAAAA,OAAmCF,kCAAkCrB,cAAcD,iBAAiBE,eAAeuB"}
+{"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

@@ -3,7 +3,7 @@ import { DEFAULT_EXAMPLES, DEFAULT_PREFIX, DEFAULT_SCHEMA, DEFAULT_SUFFIX, EXAMP
 import { AsymmetricStructuredOutputParser } from "../../output_parsers/structured.js";
 import { Runnable, RunnableConfig } from "@langchain/core/runnables";
 import { InferInteropZodOutput, InputValues } from "@langchain/core/utils/types";
-import { z } from "zod";
+import { z } from "zod/v3";
 import { BaseLanguageModelInterface } from "@langchain/core/language_models/base";
 import { Comparator, Operator, StructuredQuery } from "@langchain/core/structured_query";
 
@@ -74,7 +74,9 @@ type QueryConstructorRunnableOptions = {
 };
 /** @deprecated */
 type QueryConstructorChainOptions = QueryConstructorRunnableOptions;
-declare function loadQueryConstructorRunnable(opts: QueryConstructorRunnableOptions): Runnable<any, StructuredQuery, RunnableConfig<Record<string, any>>>;
+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

@@ -1 +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\";\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): 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;;;;;AAAiD;;6CAhBtDF,iCAAiCC,aAAakB;;iBAErEI,mBAAAA,OAA0BV;;;;KAItCW,+BAAAA;OACH3B;;iBAEUgB;aACJd;uBACUC;qBACFC;;;KAGXwB,4BAAAA,GAA+BD;iBACnBE,4BAAAA,OAAmCF,kCAAkCrB,cAAcD,iBAAiBE,eAAeuB"}
+{"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

@@ -2,7 +2,7 @@ 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 { z } from "zod";
+import { z } from "zod/v3";
 import { FewShotPromptTemplate, interpolateFString } from "@langchain/core/prompts";
 import { Comparators, Operators, StructuredQuery } from "@langchain/core/structured_query";
 

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

@@ -1 +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\";\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): 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,MACqE;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"}
+{"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/question_answering/load.d.ts

@@ -1,6 +1,6 @@
 import { MapReduceDocumentsChain, MapReduceDocumentsChainInput, RefineDocumentsChain, StuffDocumentsChain } from "../combine_docs_chain.js";
 import { BasePromptTemplate } from "@langchain/core/prompts";
-import * as _langchain_core_language_models_base7 from "@langchain/core/language_models/base";
+import * as _langchain_core_language_models_base0 from "@langchain/core/language_models/base";
 import { BaseLanguageModelInterface } from "@langchain/core/language_models/base";
 
 //#region src/chains/question_answering/load.d.ts
@@ -16,7 +16,7 @@ type QAChainParams = ({
 } & MapReduceQAChainParams) | ({
   type?: "refine";
 } & RefineQAChainParams);
-declare const loadQAChain: (llm: BaseLanguageModelInterface<any, _langchain_core_language_models_base7.BaseLanguageModelCallOptions>, params?: QAChainParams) => MapReduceDocumentsChain | RefineDocumentsChain | StuffDocumentsChain;
+declare const loadQAChain: (llm: BaseLanguageModelInterface<any, _langchain_core_language_models_base0.BaseLanguageModelCallOptions>, params?: QAChainParams) => MapReduceDocumentsChain | RefineDocumentsChain | StuffDocumentsChain;
 /**
  * Represents the parameters for creating a StuffQAChain.
  */

package/dist/chains/question_answering/load.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"load.d.ts","names":["BaseLanguageModelInterface","BasePromptTemplate","StuffDocumentsChain","MapReduceDocumentsChain","RefineDocumentsChain","MapReduceDocumentsChainInput","QAChainParams","StuffQAChainParams","MapReduceQAChainParams","RefineQAChainParams","loadQAChain","_langchain_core_language_models_base7","BaseLanguageModelCallOptions","loadQAStuffChain","loadQAMapReduceChain","loadQARefineChain"],"sources":["../../../src/chains/question_answering/load.d.ts"],"sourcesContent":["import type { BaseLanguageModelInterface } from \"@langchain/core/language_models/base\";\nimport { BasePromptTemplate } from \"@langchain/core/prompts\";\nimport { StuffDocumentsChain, MapReduceDocumentsChain, RefineDocumentsChain, MapReduceDocumentsChainInput } from \"../combine_docs_chain.js\";\n/**\n * Represents the parameters for creating a QAChain. It can be of three\n * types: \"stuff\", \"map_reduce\", or \"refine\".\n */\nexport type QAChainParams = ({\n    type?: \"stuff\";\n} & StuffQAChainParams) | ({\n    type?: \"map_reduce\";\n} & MapReduceQAChainParams) | ({\n    type?: \"refine\";\n} & RefineQAChainParams);\nexport declare const loadQAChain: (llm: BaseLanguageModelInterface<any, import(\"@langchain/core/language_models/base\").BaseLanguageModelCallOptions>, params?: QAChainParams) => MapReduceDocumentsChain | RefineDocumentsChain | StuffDocumentsChain;\n/**\n * Represents the parameters for creating a StuffQAChain.\n */\nexport interface StuffQAChainParams {\n    prompt?: BasePromptTemplate;\n    verbose?: boolean;\n}\n/**\n * Loads a StuffQAChain based on the provided parameters. It takes an LLM\n * instance and StuffQAChainParams as parameters.\n * @param llm An instance of BaseLanguageModel.\n * @param params Parameters for creating a StuffQAChain.\n * @returns A StuffQAChain instance.\n */\nexport declare function loadQAStuffChain(llm: BaseLanguageModelInterface, params?: StuffQAChainParams): StuffDocumentsChain;\n/**\n * Represents the parameters for creating a MapReduceQAChain.\n */\nexport interface MapReduceQAChainParams {\n    returnIntermediateSteps?: MapReduceDocumentsChainInput[\"returnIntermediateSteps\"];\n    combineMapPrompt?: BasePromptTemplate;\n    combinePrompt?: BasePromptTemplate;\n    combineLLM?: BaseLanguageModelInterface;\n    verbose?: boolean;\n}\n/**\n * Loads a MapReduceQAChain based on the provided parameters. It takes an\n * LLM instance and MapReduceQAChainParams as parameters.\n * @param llm An instance of BaseLanguageModel.\n * @param params Parameters for creating a MapReduceQAChain.\n * @returns A MapReduceQAChain instance.\n */\nexport declare function loadQAMapReduceChain(llm: BaseLanguageModelInterface, params?: MapReduceQAChainParams): MapReduceDocumentsChain;\n/**\n * Represents the parameters for creating a RefineQAChain.\n */\nexport interface RefineQAChainParams {\n    questionPrompt?: BasePromptTemplate;\n    refinePrompt?: BasePromptTemplate;\n    refineLLM?: BaseLanguageModelInterface;\n    verbose?: boolean;\n}\n/**\n * Loads a RefineQAChain based on the provided parameters. It takes an LLM\n * instance and RefineQAChainParams as parameters.\n * @param llm An instance of BaseLanguageModel.\n * @param params Parameters for creating a RefineQAChain.\n * @returns A RefineQAChain instance.\n */\nexport declare function loadQARefineChain(llm: BaseLanguageModelInterface, params?: RefineQAChainParams): RefineDocumentsChain;\n"],"mappings":";;;;;;;;;;AAOA;AAAyB,KAAbM,aAAAA,GAAa,CAAA;EAAA,IAErBC,CAAAA,EAAAA,OAAAA;CAAkB,GAAlBA,kBAEAC,CAAAA,GAAAA,CAAAA;EAAsB,IAEtBC,CAAAA,EAAAA,YAAAA;AAAmB,CAAA,GAFnBD,sBAEmB,CAAA,GAAA,CAAA;EACFE,IAAAA,CAAAA,EAAAA,QAAgO;CAAA,GADjPD,mBACiP,CAAA;AAAAE,cAAhOD,WAAgOC,EAAAA,CAAAA,GAAAA,EAA7MX,0BAA+EY,CAAAA,GAAAA,EAA8HD,qCAAAA,CAA9HC,4BAAAA,CAAAA,EAAAA,MAAAA,CAAAA,EAAwCN,aAAxCM,EAAAA,GAA0DT,uBAA1DS,GAAoFR,oBAApFQ,GAA2GV,mBAA3GU;;;;AAAoFR,UAI1LG,kBAAAA,CAJ0LH;EAAoB,MAAGF,CAAAA,EAKrND,kBALqNC;EAAmB,OAAA,CAAA,EAAA,OAAA;AAIrP;AAWA;;;;;AAA2H;AAI3H;AAAuC,iBAJfW,gBAAAA,CAIe,GAAA,EAJOb,0BAIP,EAAA,MAAA,CAAA,EAJ4CO,kBAI5C,CAAA,EAJiEL,mBAIjE;;;;AAItBF,UAJAQ,sBAAAA,CAIAR;EAA0B,uBAAA,CAAA,EAHbK,4BAGa,CAAA,yBAAA,CAAA;EAUnBS,gBAAAA,CAAAA,EAZDb,kBAYqB;EAAA,aAAA,CAAA,EAXxBA,kBAWwB;EAAA,UAAMD,CAAAA,EAVjCA,0BAUiCA;EAA0B,OAAWQ,CAAAA,EAAAA,OAAAA;;AAAgD;AAIvI;;;;;AAG0C;AAUlBO,iBAjBAD,oBAAAA,CAiBiB,GAAA,EAjBSd,0BAiBT,EAAA,MAAA,CAAA,EAjB8CQ,sBAiB9C,CAAA,EAjBuEL,uBAiBvE;;;;AAAiEC,UAbzFK,mBAAAA,CAayFL;EAAoB,cAAA,CAAA,EAZzGH,kBAYyG;iBAX3GA;cACHD;;;;;;;;;;iBAUQe,iBAAAA,MAAuBf,qCAAqCS,sBAAsBL"}
+{"version":3,"file":"load.d.ts","names":["BaseLanguageModelInterface","BasePromptTemplate","StuffDocumentsChain","MapReduceDocumentsChain","RefineDocumentsChain","MapReduceDocumentsChainInput","QAChainParams","StuffQAChainParams","MapReduceQAChainParams","RefineQAChainParams","loadQAChain","_langchain_core_language_models_base0","BaseLanguageModelCallOptions","loadQAStuffChain","loadQAMapReduceChain","loadQARefineChain"],"sources":["../../../src/chains/question_answering/load.d.ts"],"sourcesContent":["import type { BaseLanguageModelInterface } from \"@langchain/core/language_models/base\";\nimport { BasePromptTemplate } from \"@langchain/core/prompts\";\nimport { StuffDocumentsChain, MapReduceDocumentsChain, RefineDocumentsChain, MapReduceDocumentsChainInput } from \"../combine_docs_chain.js\";\n/**\n * Represents the parameters for creating a QAChain. It can be of three\n * types: \"stuff\", \"map_reduce\", or \"refine\".\n */\nexport type QAChainParams = ({\n    type?: \"stuff\";\n} & StuffQAChainParams) | ({\n    type?: \"map_reduce\";\n} & MapReduceQAChainParams) | ({\n    type?: \"refine\";\n} & RefineQAChainParams);\nexport declare const loadQAChain: (llm: BaseLanguageModelInterface<any, import(\"@langchain/core/language_models/base\").BaseLanguageModelCallOptions>, params?: QAChainParams) => MapReduceDocumentsChain | RefineDocumentsChain | StuffDocumentsChain;\n/**\n * Represents the parameters for creating a StuffQAChain.\n */\nexport interface StuffQAChainParams {\n    prompt?: BasePromptTemplate;\n    verbose?: boolean;\n}\n/**\n * Loads a StuffQAChain based on the provided parameters. It takes an LLM\n * instance and StuffQAChainParams as parameters.\n * @param llm An instance of BaseLanguageModel.\n * @param params Parameters for creating a StuffQAChain.\n * @returns A StuffQAChain instance.\n */\nexport declare function loadQAStuffChain(llm: BaseLanguageModelInterface, params?: StuffQAChainParams): StuffDocumentsChain;\n/**\n * Represents the parameters for creating a MapReduceQAChain.\n */\nexport interface MapReduceQAChainParams {\n    returnIntermediateSteps?: MapReduceDocumentsChainInput[\"returnIntermediateSteps\"];\n    combineMapPrompt?: BasePromptTemplate;\n    combinePrompt?: BasePromptTemplate;\n    combineLLM?: BaseLanguageModelInterface;\n    verbose?: boolean;\n}\n/**\n * Loads a MapReduceQAChain based on the provided parameters. It takes an\n * LLM instance and MapReduceQAChainParams as parameters.\n * @param llm An instance of BaseLanguageModel.\n * @param params Parameters for creating a MapReduceQAChain.\n * @returns A MapReduceQAChain instance.\n */\nexport declare function loadQAMapReduceChain(llm: BaseLanguageModelInterface, params?: MapReduceQAChainParams): MapReduceDocumentsChain;\n/**\n * Represents the parameters for creating a RefineQAChain.\n */\nexport interface RefineQAChainParams {\n    questionPrompt?: BasePromptTemplate;\n    refinePrompt?: BasePromptTemplate;\n    refineLLM?: BaseLanguageModelInterface;\n    verbose?: boolean;\n}\n/**\n * Loads a RefineQAChain based on the provided parameters. It takes an LLM\n * instance and RefineQAChainParams as parameters.\n * @param llm An instance of BaseLanguageModel.\n * @param params Parameters for creating a RefineQAChain.\n * @returns A RefineQAChain instance.\n */\nexport declare function loadQARefineChain(llm: BaseLanguageModelInterface, params?: RefineQAChainParams): RefineDocumentsChain;\n"],"mappings":";;;;;;;;;;AAOA;AAAyB,KAAbM,aAAAA,GAAa,CAAA;EAAA,IAErBC,CAAAA,EAAAA,OAAAA;CAAkB,GAAlBA,kBAEAC,CAAAA,GAAAA,CAAAA;EAAsB,IAEtBC,CAAAA,EAAAA,YAAAA;AAAmB,CAAA,GAFnBD,sBAEmB,CAAA,GAAA,CAAA;EACFE,IAAAA,CAAAA,EAAAA,QAAgO;CAAA,GADjPD,mBACiP,CAAA;AAAAE,cAAhOD,WAAgOC,EAAAA,CAAAA,GAAAA,EAA7MX,0BAA+EY,CAAAA,GAAAA,EAA8HD,qCAAAA,CAA9HC,4BAAAA,CAAAA,EAAAA,MAAAA,CAAAA,EAAwCN,aAAxCM,EAAAA,GAA0DT,uBAA1DS,GAAoFR,oBAApFQ,GAA2GV,mBAA3GU;;;;AAAoFR,UAI1LG,kBAAAA,CAJ0LH;EAAoB,MAAGF,CAAAA,EAKrND,kBALqNC;EAAmB,OAAA,CAAA,EAAA,OAAA;AAIrP;AAWA;;;;;AAA2H;AAI3H;AAAuC,iBAJfW,gBAAAA,CAIe,GAAA,EAJOb,0BAIP,EAAA,MAAA,CAAA,EAJ4CO,kBAI5C,CAAA,EAJiEL,mBAIjE;;;;AAItBF,UAJAQ,sBAAAA,CAIAR;EAA0B,uBAAA,CAAA,EAHbK,4BAGa,CAAA,yBAAA,CAAA;EAUnBS,gBAAAA,CAAAA,EAZDb,kBAYqB;EAAA,aAAA,CAAA,EAXxBA,kBAWwB;EAAA,UAAMD,CAAAA,EAVjCA,0BAUiCA;EAA0B,OAAWQ,CAAAA,EAAAA,OAAAA;;AAAgD;AAIvI;;;;;AAG0C;AAUlBO,iBAjBAD,oBAAAA,CAiBiB,GAAA,EAjBSd,0BAiBT,EAAA,MAAA,CAAA,EAjB8CQ,sBAiB9C,CAAA,EAjBuEL,uBAiBvE;;;;AAAiEC,UAbzFK,mBAAAA,CAayFL;EAAoB,cAAA,CAAA,EAZzGH,kBAYyG;iBAX3GA;cACHD;;;;;;;;;;iBAUQe,iBAAAA,MAAuBf,qCAAqCS,sBAAsBL"}

package/dist/chains/question_answering/map_reduce_prompts.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"map_reduce_prompts.cjs","names":["PromptTemplate","SystemMessagePromptTemplate","HumanMessagePromptTemplate","ChatPromptTemplate","ConditionalPromptSelector","isChatModel"],"sources":["../../../src/chains/question_answering/map_reduce_prompts.ts"],"sourcesContent":["/* eslint-disable spaced-comment */\nimport {\n  ChatPromptTemplate,\n  SystemMessagePromptTemplate,\n  HumanMessagePromptTemplate,\n  PromptTemplate,\n} from \"@langchain/core/prompts\";\nimport {\n  ConditionalPromptSelector,\n  isChatModel,\n} from \"@langchain/core/example_selectors\";\n\nconst qa_template = `Use the following portion of a long document to see if any of the text is relevant to answer the question. \nReturn any relevant text verbatim.\n{context}\nQuestion: {question}\nRelevant text, if any:`;\nexport const DEFAULT_COMBINE_QA_PROMPT =\n  /*#__PURE__*/\n  PromptTemplate.fromTemplate(qa_template);\n\nconst system_template = `Use the following portion of a long document to see if any of the text is relevant to answer the question. \nReturn any relevant text verbatim.\n----------------\n{context}`;\nconst messages = [\n  /*#__PURE__*/ SystemMessagePromptTemplate.fromTemplate(system_template),\n  /*#__PURE__*/ HumanMessagePromptTemplate.fromTemplate(\"{question}\"),\n];\nconst CHAT_QA_PROMPT = /*#__PURE__*/ ChatPromptTemplate.fromMessages(messages);\n\nexport const COMBINE_QA_PROMPT_SELECTOR =\n  /*#__PURE__*/ new ConditionalPromptSelector(DEFAULT_COMBINE_QA_PROMPT, [\n    [isChatModel, CHAT_QA_PROMPT],\n  ]);\n\nconst combine_prompt = `Given the following extracted parts of a long document and a question, create a final answer. \nIf you don't know the answer, just say that you don't know. Don't try to make up an answer.\n\nQUESTION: Which state/country's law governs the interpretation of the contract?\n=========\nContent: This Agreement is governed by English law and the parties submit to the exclusive jurisdiction of the English courts in  relation to any dispute (contractual or non-contractual) concerning this Agreement save that either party may apply to any court for an  injunction or other relief to protect its Intellectual Property Rights.\n\nContent: No Waiver. Failure or delay in exercising any right or remedy under this Agreement shall not constitute a waiver of such (or any other)  right or remedy.\\n\\n11.7 Severability. The invalidity, illegality or unenforceability of any term (or part of a term) of this Agreement shall not affect the continuation  in force of the remainder of the term (if any) and this Agreement.\\n\\n11.8 No Agency. Except as expressly stated otherwise, nothing in this Agreement shall create an agency, partnership or joint venture of any  kind between the parties.\\n\\n11.9 No Third-Party Beneficiaries.\n\nContent: (b) if Google believes, in good faith, that the Distributor has violated or caused Google to violate any Anti-Bribery Laws (as  defined in Clause 8.5) or that such a violation is reasonably likely to occur,\n=========\nFINAL ANSWER: This Agreement is governed by English law.\n\nQUESTION: What did the president say about Michael Jackson?\n=========\nContent: Madam Speaker, Madam Vice President, our First Lady and Second Gentleman. Members of Congress and the Cabinet. Justices of the Supreme Court. My fellow Americans.  \\n\\nLast year COVID-19 kept us apart. This year we are finally together again. \\n\\nTonight, we meet as Democrats Republicans and Independents. But most importantly as Americans. \\n\\nWith a duty to one another to the American people to the Constitution. \\n\\nAnd with an unwavering resolve that freedom will always triumph over tyranny. \\n\\nSix days ago, Russia’s Vladimir Putin sought to shake the foundations of the free world thinking he could make it bend to his menacing ways. But he badly miscalculated. \\n\\nHe thought he could roll into Ukraine and the world would roll over. Instead he met a wall of strength he never imagined. \\n\\nHe met the Ukrainian people. \\n\\nFrom President Zelenskyy to every Ukrainian, their fearlessness, their courage, their determination, inspires the world. \\n\\nGroups of citizens blocking tanks with their bodies. Everyone from students to retirees teachers turned soldiers defending their homeland.\n\nContent: And we won’t stop. \\n\\nWe have lost so much to COVID-19. Time with one another. And worst of all, so much loss of life. \\n\\nLet’s use this moment to reset. Let’s stop looking at COVID-19 as a partisan dividing line and see it for what it is: A God-awful disease.  \\n\\nLet’s stop seeing each other as enemies, and start seeing each other for who we really are: Fellow Americans.  \\n\\nWe can’t change how divided we’ve been. But we can change how we move forward—on COVID-19 and other issues we must face together. \\n\\nI recently visited the New York City Police Department days after the funerals of Officer Wilbert Mora and his partner, Officer Jason Rivera. \\n\\nThey were responding to a 9-1-1 call when a man shot and killed them with a stolen gun. \\n\\nOfficer Mora was 27 years old. \\n\\nOfficer Rivera was 22. \\n\\nBoth Dominican Americans who’d grown up on the same streets they later chose to patrol as police officers. \\n\\nI spoke with their families and told them that we are forever in debt for their sacrifice, and we will carry on their mission to restore the trust and safety every community deserves.\n\nContent: And a proud Ukrainian people, who have known 30 years  of independence, have repeatedly shown that they will not tolerate anyone who tries to take their country backwards.  \\n\\nTo all Americans, I will be honest with you, as I’ve always promised. A Russian dictator, invading a foreign country, has costs around the world. \\n\\nAnd I’m taking robust action to make sure the pain of our sanctions  is targeted at Russia’s economy. And I will use every tool at our disposal to protect American businesses and consumers. \\n\\nTonight, I can announce that the United States has worked with 30 other countries to release 60 Million barrels of oil from reserves around the world.  \\n\\nAmerica will lead that effort, releasing 30 Million barrels from our own Strategic Petroleum Reserve. And we stand ready to do more if necessary, unified with our allies.  \\n\\nThese steps will help blunt gas prices here at home. And I know the news about what’s happening can seem alarming. \\n\\nBut I want you to know that we are going to be okay.\n\nContent: More support for patients and families. \\n\\nTo get there, I call on Congress to fund ARPA-H, the Advanced Research Projects Agency for Health. \\n\\nIt’s based on DARPA—the Defense Department project that led to the Internet, GPS, and so much more.  \\n\\nARPA-H will have a singular purpose—to drive breakthroughs in cancer, Alzheimer’s, diabetes, and more. \\n\\nA unity agenda for the nation. \\n\\nWe can do this. \\n\\nMy fellow Americans—tonight , we have gathered in a sacred space—the citadel of our democracy. \\n\\nIn this Capitol, generation after generation, Americans have debated great questions amid great strife, and have done great things. \\n\\nWe have fought for freedom, expanded liberty, defeated totalitarianism and terror. \\n\\nAnd built the strongest, freest, and most prosperous nation the world has ever known. \\n\\nNow is the hour. \\n\\nOur moment of responsibility. \\n\\nOur test of resolve and conscience, of history itself. \\n\\nIt is in this moment that our character is formed. Our purpose is found. Our future is forged. \\n\\nWell I know this nation.\n=========\nFINAL ANSWER: The president did not mention Michael Jackson.\n\nQUESTION: {question}\n=========\n{summaries}\n=========\nFINAL ANSWER:`;\nexport const COMBINE_PROMPT =\n  /*#__PURE__*/ PromptTemplate.fromTemplate(combine_prompt);\n\nconst system_combine_template = `Given the following extracted parts of a long document and a question, create a final answer. \nIf you don't know the answer, just say that you don't know. Don't try to make up an answer.\n----------------\n{summaries}`;\nconst combine_messages = [\n  /*#__PURE__*/ SystemMessagePromptTemplate.fromTemplate(\n    system_combine_template\n  ),\n  /*#__PURE__*/ HumanMessagePromptTemplate.fromTemplate(\"{question}\"),\n];\nconst CHAT_COMBINE_PROMPT =\n  /*#__PURE__*/ ChatPromptTemplate.fromMessages(combine_messages);\n\nexport const COMBINE_PROMPT_SELECTOR =\n  /*#__PURE__*/ new ConditionalPromptSelector(COMBINE_PROMPT, [\n    [isChatModel, CHAT_COMBINE_PROMPT],\n  ]);\n"],"mappings":";;;;;AAYA,MAAM,cAAc,CAAC;;;;sBAIC,CAAC;AACvB,MAAa,4CAEXA,wCAAe,aAAa,YAAY;AAE1C,MAAM,kBAAkB,CAAC;;;SAGhB,CAAC;AACV,MAAM,WAAW,iBACDC,qDAA4B,aAAa,gBAAgB,kBACzDC,oDAA2B,aAAa,aAAa,AACpE;AACD,MAAM,iCAA+BC,4CAAmB,aAAa,SAAS;AAE9E,MAAa,6CACG,IAAIC,6DAA0B,2BAA2B,CACrE,CAACC,gDAAa,cAAe,CAC9B;AAEH,MAAM,iBAAiB,CAAC;;;;;;;;;;;;;;;;;;;;;;;;;;;;;aA6BX,CAAC;AACd,MAAa,iCACGL,wCAAe,aAAa,eAAe;AAE3D,MAAM,0BAA0B,CAAC;;;WAGtB,CAAC;AACZ,MAAM,mBAAmB,iBACTC,qDAA4B,aACxC,wBACD,kBACaC,oDAA2B,aAAa,aAAa,AACpE;AACD,MAAM,sCACUC,4CAAmB,aAAa,iBAAiB;AAEjE,MAAa,0CACG,IAAIC,6DAA0B,gBAAgB,CAC1D,CAACC,gDAAa,mBAAoB,CACnC"}
+{"version":3,"file":"map_reduce_prompts.cjs","names":["PromptTemplate","SystemMessagePromptTemplate","HumanMessagePromptTemplate","ChatPromptTemplate","ConditionalPromptSelector","isChatModel"],"sources":["../../../src/chains/question_answering/map_reduce_prompts.ts"],"sourcesContent":["import {\n  ChatPromptTemplate,\n  SystemMessagePromptTemplate,\n  HumanMessagePromptTemplate,\n  PromptTemplate,\n} from \"@langchain/core/prompts\";\nimport {\n  ConditionalPromptSelector,\n  isChatModel,\n} from \"@langchain/core/example_selectors\";\n\nconst qa_template = `Use the following portion of a long document to see if any of the text is relevant to answer the question. \nReturn any relevant text verbatim.\n{context}\nQuestion: {question}\nRelevant text, if any:`;\nexport const DEFAULT_COMBINE_QA_PROMPT =\n  /*#__PURE__*/\n  PromptTemplate.fromTemplate(qa_template);\n\nconst system_template = `Use the following portion of a long document to see if any of the text is relevant to answer the question. \nReturn any relevant text verbatim.\n----------------\n{context}`;\nconst messages = [\n  /*#__PURE__*/ SystemMessagePromptTemplate.fromTemplate(system_template),\n  /*#__PURE__*/ HumanMessagePromptTemplate.fromTemplate(\"{question}\"),\n];\nconst CHAT_QA_PROMPT = /*#__PURE__*/ ChatPromptTemplate.fromMessages(messages);\n\nexport const COMBINE_QA_PROMPT_SELECTOR =\n  /*#__PURE__*/ new ConditionalPromptSelector(DEFAULT_COMBINE_QA_PROMPT, [\n    [isChatModel, CHAT_QA_PROMPT],\n  ]);\n\nconst combine_prompt = `Given the following extracted parts of a long document and a question, create a final answer. \nIf you don't know the answer, just say that you don't know. Don't try to make up an answer.\n\nQUESTION: Which state/country's law governs the interpretation of the contract?\n=========\nContent: This Agreement is governed by English law and the parties submit to the exclusive jurisdiction of the English courts in  relation to any dispute (contractual or non-contractual) concerning this Agreement save that either party may apply to any court for an  injunction or other relief to protect its Intellectual Property Rights.\n\nContent: No Waiver. Failure or delay in exercising any right or remedy under this Agreement shall not constitute a waiver of such (or any other)  right or remedy.\\n\\n11.7 Severability. The invalidity, illegality or unenforceability of any term (or part of a term) of this Agreement shall not affect the continuation  in force of the remainder of the term (if any) and this Agreement.\\n\\n11.8 No Agency. Except as expressly stated otherwise, nothing in this Agreement shall create an agency, partnership or joint venture of any  kind between the parties.\\n\\n11.9 No Third-Party Beneficiaries.\n\nContent: (b) if Google believes, in good faith, that the Distributor has violated or caused Google to violate any Anti-Bribery Laws (as  defined in Clause 8.5) or that such a violation is reasonably likely to occur,\n=========\nFINAL ANSWER: This Agreement is governed by English law.\n\nQUESTION: What did the president say about Michael Jackson?\n=========\nContent: Madam Speaker, Madam Vice President, our First Lady and Second Gentleman. Members of Congress and the Cabinet. Justices of the Supreme Court. My fellow Americans.  \\n\\nLast year COVID-19 kept us apart. This year we are finally together again. \\n\\nTonight, we meet as Democrats Republicans and Independents. But most importantly as Americans. \\n\\nWith a duty to one another to the American people to the Constitution. \\n\\nAnd with an unwavering resolve that freedom will always triumph over tyranny. \\n\\nSix days ago, Russia’s Vladimir Putin sought to shake the foundations of the free world thinking he could make it bend to his menacing ways. But he badly miscalculated. \\n\\nHe thought he could roll into Ukraine and the world would roll over. Instead he met a wall of strength he never imagined. \\n\\nHe met the Ukrainian people. \\n\\nFrom President Zelenskyy to every Ukrainian, their fearlessness, their courage, their determination, inspires the world. \\n\\nGroups of citizens blocking tanks with their bodies. Everyone from students to retirees teachers turned soldiers defending their homeland.\n\nContent: And we won’t stop. \\n\\nWe have lost so much to COVID-19. Time with one another. And worst of all, so much loss of life. \\n\\nLet’s use this moment to reset. Let’s stop looking at COVID-19 as a partisan dividing line and see it for what it is: A God-awful disease.  \\n\\nLet’s stop seeing each other as enemies, and start seeing each other for who we really are: Fellow Americans.  \\n\\nWe can’t change how divided we’ve been. But we can change how we move forward—on COVID-19 and other issues we must face together. \\n\\nI recently visited the New York City Police Department days after the funerals of Officer Wilbert Mora and his partner, Officer Jason Rivera. \\n\\nThey were responding to a 9-1-1 call when a man shot and killed them with a stolen gun. \\n\\nOfficer Mora was 27 years old. \\n\\nOfficer Rivera was 22. \\n\\nBoth Dominican Americans who’d grown up on the same streets they later chose to patrol as police officers. \\n\\nI spoke with their families and told them that we are forever in debt for their sacrifice, and we will carry on their mission to restore the trust and safety every community deserves.\n\nContent: And a proud Ukrainian people, who have known 30 years  of independence, have repeatedly shown that they will not tolerate anyone who tries to take their country backwards.  \\n\\nTo all Americans, I will be honest with you, as I’ve always promised. A Russian dictator, invading a foreign country, has costs around the world. \\n\\nAnd I’m taking robust action to make sure the pain of our sanctions  is targeted at Russia’s economy. And I will use every tool at our disposal to protect American businesses and consumers. \\n\\nTonight, I can announce that the United States has worked with 30 other countries to release 60 Million barrels of oil from reserves around the world.  \\n\\nAmerica will lead that effort, releasing 30 Million barrels from our own Strategic Petroleum Reserve. And we stand ready to do more if necessary, unified with our allies.  \\n\\nThese steps will help blunt gas prices here at home. And I know the news about what’s happening can seem alarming. \\n\\nBut I want you to know that we are going to be okay.\n\nContent: More support for patients and families. \\n\\nTo get there, I call on Congress to fund ARPA-H, the Advanced Research Projects Agency for Health. \\n\\nIt’s based on DARPA—the Defense Department project that led to the Internet, GPS, and so much more.  \\n\\nARPA-H will have a singular purpose—to drive breakthroughs in cancer, Alzheimer’s, diabetes, and more. \\n\\nA unity agenda for the nation. \\n\\nWe can do this. \\n\\nMy fellow Americans—tonight , we have gathered in a sacred space—the citadel of our democracy. \\n\\nIn this Capitol, generation after generation, Americans have debated great questions amid great strife, and have done great things. \\n\\nWe have fought for freedom, expanded liberty, defeated totalitarianism and terror. \\n\\nAnd built the strongest, freest, and most prosperous nation the world has ever known. \\n\\nNow is the hour. \\n\\nOur moment of responsibility. \\n\\nOur test of resolve and conscience, of history itself. \\n\\nIt is in this moment that our character is formed. Our purpose is found. Our future is forged. \\n\\nWell I know this nation.\n=========\nFINAL ANSWER: The president did not mention Michael Jackson.\n\nQUESTION: {question}\n=========\n{summaries}\n=========\nFINAL ANSWER:`;\nexport const COMBINE_PROMPT =\n  /*#__PURE__*/ PromptTemplate.fromTemplate(combine_prompt);\n\nconst system_combine_template = `Given the following extracted parts of a long document and a question, create a final answer. \nIf you don't know the answer, just say that you don't know. Don't try to make up an answer.\n----------------\n{summaries}`;\nconst combine_messages = [\n  /*#__PURE__*/ SystemMessagePromptTemplate.fromTemplate(\n    system_combine_template\n  ),\n  /*#__PURE__*/ HumanMessagePromptTemplate.fromTemplate(\"{question}\"),\n];\nconst CHAT_COMBINE_PROMPT =\n  /*#__PURE__*/ ChatPromptTemplate.fromMessages(combine_messages);\n\nexport const COMBINE_PROMPT_SELECTOR =\n  /*#__PURE__*/ new ConditionalPromptSelector(COMBINE_PROMPT, [\n    [isChatModel, CHAT_COMBINE_PROMPT],\n  ]);\n"],"mappings":";;;;;AAWA,MAAM,cAAc,CAAC;;;;sBAIC,CAAC;AACvB,MAAa,4CAEXA,wCAAe,aAAa,YAAY;AAE1C,MAAM,kBAAkB,CAAC;;;SAGhB,CAAC;AACV,MAAM,WAAW,iBACDC,qDAA4B,aAAa,gBAAgB,kBACzDC,oDAA2B,aAAa,aAAa,AACpE;AACD,MAAM,iCAA+BC,4CAAmB,aAAa,SAAS;AAE9E,MAAa,6CACG,IAAIC,6DAA0B,2BAA2B,CACrE,CAACC,gDAAa,cAAe,CAC9B;AAEH,MAAM,iBAAiB,CAAC;;;;;;;;;;;;;;;;;;;;;;;;;;;;;aA6BX,CAAC;AACd,MAAa,iCACGL,wCAAe,aAAa,eAAe;AAE3D,MAAM,0BAA0B,CAAC;;;WAGtB,CAAC;AACZ,MAAM,mBAAmB,iBACTC,qDAA4B,aACxC,wBACD,kBACaC,oDAA2B,aAAa,aAAa,AACpE;AACD,MAAM,sCACUC,4CAAmB,aAAa,iBAAiB;AAEjE,MAAa,0CACG,IAAIC,6DAA0B,gBAAgB,CAC1D,CAACC,gDAAa,mBAAoB,CACnC"}