@langchain/classic 1.0.0-alpha.1

package/dist/smith/runner_utils.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"runner_utils.js","names":["_chain: Serialized","_inputs: ChainValues","runId: string","run: Run","evaluator: RunEvaluatorLike","example?: Example","evaluator: any","config: RunTreeConfig","callbackManager: CallbackManager","fields: { func: AnyTraceableFunction }","input: RunInput","options?: Partial<RunnableConfig>","evaluator: LLMStringEvaluator","evaluationName: string","formatEvaluatorInputs: EvaluatorInputFormatter","config: EvalConfig | keyof EvaluatorType","evaluators: (RunEvaluator | DynamicRunEvaluator)[]","config: RunEvalConfig<keyof EvaluatorType>","modelOrFactory: ChainOrFactory","configs: RunnableConfig[]","examples: Example[]","chainOrFactory: ChainOrFactory","dataType?: DataType","datasetName: string","options?: RunOnDatasetParams","evaluationConfig: RunEvalConfig | undefined","wrappedRunnable: Runnable","runs: Run[]","evalResults: Record<\n    string,\n    { run_id: string; execution_time?: number; feedback: Feedback[] }\n  >","results: EvalResults","x: unknown"],"sources":["../../src/smith/runner_utils.ts"],"sourcesContent":["import { BaseLanguageModel } from \"@langchain/core/language_models/base\";\nimport { Serialized } from \"@langchain/core/load/serializable\";\nimport { mapStoredMessagesToChatMessages } from \"@langchain/core/messages\";\nimport {\n  Runnable,\n  RunnableConfig,\n  RunnableLambda,\n  getCallbackManagerForConfig,\n} from \"@langchain/core/runnables\";\nimport { LangChainTracer } from \"@langchain/core/tracers/tracer_langchain\";\nimport { BaseTracer } from \"@langchain/core/tracers/base\";\nimport { ChainValues } from \"@langchain/core/utils/types\";\nimport { AsyncCaller } from \"@langchain/core/utils/async_caller\";\nimport type {\n  CallbackManager,\n  CallbackManagerForChainRun,\n} from \"@langchain/core/callbacks/manager\";\nimport {\n  Client,\n  Example,\n  Feedback,\n  Run,\n  RunTree,\n  RunTreeConfig,\n} from \"langsmith\";\nimport { EvaluationResult, RunEvaluator } from \"langsmith/evaluation\";\nimport { DataType } from \"langsmith/schemas\";\nimport type { TraceableFunction } from \"langsmith/singletons/traceable\";\nimport { LLMStringEvaluator } from \"../evaluation/base.js\";\nimport { loadEvaluator } from \"../evaluation/loader.js\";\nimport { EvaluatorType } from \"../evaluation/types.js\";\nimport {\n  isOffTheShelfEvaluator,\n  type DynamicRunEvaluatorParams,\n  type EvalConfig,\n  type EvaluatorInputFormatter,\n  type RunEvalConfig,\n  type RunEvaluatorLike,\n  isCustomEvaluator,\n} from \"./config.js\";\nimport { randomName } from \"./name_generation.js\";\nimport { ProgressBar } from \"./progress.js\";\n\nexport type ChainOrFactory =\n  | Runnable\n  | (() => Runnable)\n  | AnyTraceableFunction\n  // eslint-disable-next-line @typescript-eslint/no-explicit-any\n  | ((obj: any) => any)\n  // eslint-disable-next-line @typescript-eslint/no-explicit-any\n  | ((obj: any) => Promise<any>)\n  | (() => (obj: unknown) => unknown)\n  | (() => (obj: unknown) => Promise<unknown>);\n\nclass SingleRunIdExtractor {\n  runIdPromiseResolver: (runId: string) => void;\n\n  runIdPromise: Promise<string>;\n\n  constructor() {\n    this.runIdPromise = new Promise<string>((extract) => {\n      this.runIdPromiseResolver = extract;\n    });\n  }\n\n  handleChainStart = (\n    _chain: Serialized,\n    _inputs: ChainValues,\n    runId: string\n  ) => {\n    this.runIdPromiseResolver(runId);\n  };\n\n  async extract(): Promise<string> {\n    return this.runIdPromise;\n  }\n}\n\nclass SingleRunExtractor extends BaseTracer {\n  runPromiseResolver: (run: Run) => void;\n\n  runPromise: Promise<Run>;\n\n  /** The name of the callback handler. */\n  name = \"single_run_extractor\";\n\n  constructor() {\n    super();\n    this.runPromise = new Promise<Run>((extract) => {\n      this.runPromiseResolver = extract;\n    });\n  }\n\n  async persistRun(run: Run) {\n    this.runPromiseResolver(run);\n  }\n\n  async extract(): Promise<Run> {\n    return this.runPromise;\n  }\n}\n\n/**\n * Wraps an evaluator function + implements the RunEvaluator interface.\n */\nclass DynamicRunEvaluator implements RunEvaluator {\n  evaluator: RunnableLambda<DynamicRunEvaluatorParams, EvaluationResult>;\n\n  constructor(evaluator: RunEvaluatorLike) {\n    this.evaluator = new RunnableLambda({ func: evaluator });\n  }\n\n  /**\n   * Evaluates a run with an optional example and returns the evaluation result.\n   * @param run The run to evaluate.\n   * @param example The optional example to use for evaluation.\n   * @returns A promise that extracts to the evaluation result.\n   */\n  async evaluateRun(run: Run, example?: Example): Promise<EvaluationResult> {\n    const extractor = new SingleRunIdExtractor();\n    const tracer = new LangChainTracer({ projectName: \"evaluators\" });\n    const result = await this.evaluator.invoke(\n      {\n        run,\n        example,\n        input: run.inputs,\n        prediction: run.outputs,\n        reference: example?.outputs,\n      },\n      {\n        callbacks: [extractor, tracer],\n      }\n    );\n    const runId = await extractor.extract();\n    return {\n      sourceRunId: runId,\n      ...result,\n    };\n  }\n}\n\n// eslint-disable-next-line @typescript-eslint/no-explicit-any\nfunction isLLMStringEvaluator(evaluator: any): evaluator is LLMStringEvaluator {\n  return evaluator && typeof evaluator.evaluateStrings === \"function\";\n}\n\n// eslint-disable-next-line @typescript-eslint/no-explicit-any\ntype AnyTraceableFunction = TraceableFunction<(...any: any[]) => any>;\n\n/**\n * Internal implementation of RunTree, which uses the\n * provided callback manager instead of the internal LangSmith client.\n *\n * The goal of this class is to ensure seamless interop when intergrated\n * with other Runnables.\n */\nclass CallbackManagerRunTree extends RunTree {\n  callbackManager: CallbackManager;\n\n  activeCallbackManager: CallbackManagerForChainRun | undefined = undefined;\n\n  constructor(config: RunTreeConfig, callbackManager: CallbackManager) {\n    super(config);\n\n    this.callbackManager = callbackManager;\n  }\n\n  createChild(config: RunTreeConfig): CallbackManagerRunTree {\n    const child = new CallbackManagerRunTree(\n      {\n        ...config,\n        parent_run: this,\n        project_name: this.project_name,\n        client: this.client,\n      },\n      this.activeCallbackManager?.getChild() ?? this.callbackManager\n    );\n    this.child_runs.push(child);\n    return child;\n  }\n\n  async postRun(): Promise<void> {\n    // how it is translated in comparison to basic RunTree?\n    this.activeCallbackManager = await this.callbackManager.handleChainStart(\n      typeof this.serialized !== \"object\" &&\n        this.serialized != null &&\n        \"lc\" in this.serialized\n        ? this.serialized\n        : {\n            id: [\"langchain\", \"smith\", \"CallbackManagerRunTree\"],\n            lc: 1,\n            type: \"not_implemented\",\n          },\n      this.inputs,\n      this.id,\n      this.run_type,\n      undefined,\n      undefined,\n      this.name\n    );\n  }\n\n  async patchRun(): Promise<void> {\n    if (this.error) {\n      await this.activeCallbackManager?.handleChainError(\n        this.error,\n        this.id,\n        this.parent_run?.id,\n        undefined,\n        undefined\n      );\n    } else {\n      await this.activeCallbackManager?.handleChainEnd(\n        this.outputs ?? {},\n        this.id,\n        this.parent_run?.id,\n        undefined,\n        undefined\n      );\n    }\n  }\n}\n\nclass RunnableTraceable<RunInput, RunOutput> extends Runnable<\n  RunInput,\n  RunOutput\n> {\n  lc_serializable = false;\n\n  lc_namespace = [\"langchain_core\", \"runnables\"];\n\n  protected func: AnyTraceableFunction;\n\n  constructor(fields: { func: AnyTraceableFunction }) {\n    super(fields);\n\n    if (!isLangsmithTraceableFunction(fields.func)) {\n      throw new Error(\n        \"RunnableTraceable requires a function that is wrapped in traceable higher-order function\"\n      );\n    }\n\n    this.func = fields.func;\n  }\n\n  async invoke(input: RunInput, options?: Partial<RunnableConfig>) {\n    const [config] = this._getOptionsList(options ?? {}, 1);\n    const callbackManager = await getCallbackManagerForConfig(config);\n\n    const partialConfig =\n      \"langsmith:traceable\" in this.func\n        ? (this.func[\"langsmith:traceable\"] as RunTreeConfig)\n        : { name: \"<lambda>\" };\n\n    if (!callbackManager) throw new Error(\"CallbackManager not found\");\n    const runTree = new CallbackManagerRunTree(\n      {\n        ...partialConfig,\n        parent_run: callbackManager?._parentRunId\n          ? new RunTree({ name: \"<parent>\", id: callbackManager?._parentRunId })\n          : undefined,\n      },\n      callbackManager\n    );\n\n    if (\n      typeof input === \"object\" &&\n      input != null &&\n      Object.keys(input).length === 1\n    ) {\n      if (\"args\" in input && Array.isArray(input)) {\n        return (await this.func(runTree, ...input)) as RunOutput;\n      }\n\n      if (\n        \"input\" in input &&\n        !(\n          typeof input === \"object\" &&\n          input != null &&\n          !Array.isArray(input) &&\n          // eslint-disable-next-line no-instanceof/no-instanceof\n          !(input instanceof Date)\n        )\n      ) {\n        try {\n          return (await this.func(runTree, input.input)) as RunOutput;\n        } catch {\n          return (await this.func(runTree, input)) as RunOutput;\n        }\n      }\n    }\n\n    return (await this.func(runTree, input)) as RunOutput;\n  }\n}\n\n/**\n * Wraps an off-the-shelf evaluator (loaded using loadEvaluator; of EvaluatorType[T])\n * and composes with a prepareData function so the user can prepare the trace and\n * dataset data for the evaluator.\n */\nclass PreparedRunEvaluator implements RunEvaluator {\n  evaluator: LLMStringEvaluator;\n\n  formatEvaluatorInputs: EvaluatorInputFormatter;\n\n  isStringEvaluator: boolean;\n\n  evaluationName: string;\n\n  constructor(\n    evaluator: LLMStringEvaluator,\n    evaluationName: string,\n    formatEvaluatorInputs: EvaluatorInputFormatter\n  ) {\n    this.evaluator = evaluator;\n    this.isStringEvaluator = typeof evaluator?.evaluateStrings === \"function\";\n    this.evaluationName = evaluationName;\n    this.formatEvaluatorInputs = formatEvaluatorInputs;\n  }\n\n  static async fromEvalConfig(\n    config: EvalConfig | keyof EvaluatorType\n  ): Promise<PreparedRunEvaluator> {\n    const evaluatorType =\n      typeof config === \"string\" ? config : config.evaluatorType;\n    const evalConfig = typeof config === \"string\" ? ({} as EvalConfig) : config;\n    const evaluator = await loadEvaluator(evaluatorType, evalConfig);\n    const feedbackKey = evalConfig?.feedbackKey ?? evaluator?.evaluationName;\n    if (!isLLMStringEvaluator(evaluator)) {\n      throw new Error(\n        `Evaluator of type ${evaluatorType} not yet supported. ` +\n          \"Please use a string evaluator, or implement your \" +\n          \"evaluation logic as a custom evaluator.\"\n      );\n    }\n    if (!feedbackKey) {\n      throw new Error(\n        `Evaluator of type ${evaluatorType} must have an evaluationName` +\n          ` or feedbackKey. Please manually provide a feedbackKey in the EvalConfig.`\n      );\n    }\n    return new PreparedRunEvaluator(\n      evaluator as LLMStringEvaluator,\n      feedbackKey,\n      evalConfig?.formatEvaluatorInputs\n    );\n  }\n\n  /**\n   * Evaluates a run with an optional example and returns the evaluation result.\n   * @param run The run to evaluate.\n   * @param example The optional example to use for evaluation.\n   * @returns A promise that extracts to the evaluation result.\n   */\n  async evaluateRun(run: Run, example?: Example): Promise<EvaluationResult> {\n    const { prediction, input, reference } = this.formatEvaluatorInputs({\n      rawInput: run.inputs,\n      rawPrediction: run.outputs,\n      rawReferenceOutput: example?.outputs,\n      run,\n    });\n    const extractor = new SingleRunIdExtractor();\n    const tracer = new LangChainTracer({ projectName: \"evaluators\" });\n    if (this.isStringEvaluator) {\n      const evalResult = await this.evaluator.evaluateStrings(\n        {\n          prediction: prediction as string,\n          reference: reference as string,\n          input: input as string,\n        },\n        {\n          callbacks: [extractor, tracer],\n        }\n      );\n      const runId = await extractor.extract();\n      return {\n        key: this.evaluationName,\n        comment: evalResult?.reasoning,\n        sourceRunId: runId,\n        ...evalResult,\n      };\n    }\n    throw new Error(\n      \"Evaluator not yet supported. \" +\n        \"Please use a string evaluator, or implement your \" +\n        \"evaluation logic as a custom evaluator.\"\n    );\n  }\n}\n\nclass LoadedEvalConfig {\n  constructor(public evaluators: (RunEvaluator | DynamicRunEvaluator)[]) {}\n\n  static async fromRunEvalConfig(\n    config: RunEvalConfig<keyof EvaluatorType>\n  ): Promise<LoadedEvalConfig> {\n    // Custom evaluators are applied \"as-is\"\n    const customEvaluators = (\n      config?.customEvaluators ?? config.evaluators?.filter(isCustomEvaluator)\n    )?.map((evaluator) => {\n      if (typeof evaluator === \"function\") {\n        return new DynamicRunEvaluator(evaluator);\n      } else {\n        return evaluator;\n      }\n    });\n\n    const offTheShelfEvaluators = await Promise.all(\n      config?.evaluators\n        ?.filter(isOffTheShelfEvaluator)\n        ?.map(\n          async (evaluator) =>\n            await PreparedRunEvaluator.fromEvalConfig(evaluator)\n        ) ?? []\n    );\n    return new LoadedEvalConfig(\n      (customEvaluators ?? []).concat(offTheShelfEvaluators ?? [])\n    );\n  }\n}\n\nexport interface RunOnDatasetParams\n  extends Omit<RunEvalConfig, \"customEvaluators\"> {\n  /**\n   * Name of the project for logging and tracking.\n   */\n  projectName?: string;\n\n  /**\n   * Additional metadata for the project.\n   */\n  projectMetadata?: Record<string, unknown>;\n\n  /**\n   * Client instance for LangSmith service interaction.\n   */\n  client?: Client;\n\n  /**\n   * Maximum concurrency level for dataset processing.\n   */\n  maxConcurrency?: number;\n\n  /**\n   * @deprecated Pass keys directly to the RunOnDatasetParams instead\n   */\n  evaluationConfig?: RunEvalConfig;\n}\n\n/**\n * Internals expect a constructor () -> Runnable. This function wraps/coerces\n * the provided LangChain object, custom function, or factory function into\n * a constructor of a runnable.\n * @param modelOrFactory The model or factory to create a wrapped model from.\n * @returns A function that returns the wrapped model.\n * @throws Error if the modelOrFactory is invalid.\n */\nconst createWrappedModel = async (modelOrFactory: ChainOrFactory) => {\n  if (Runnable.isRunnable(modelOrFactory)) {\n    return () => modelOrFactory;\n  }\n  if (typeof modelOrFactory === \"function\") {\n    if (isLangsmithTraceableFunction(modelOrFactory)) {\n      const wrappedModel = new RunnableTraceable({ func: modelOrFactory });\n      return () => wrappedModel;\n    }\n\n    try {\n      // If it works with no arguments, assume it's a factory\n      let res = (modelOrFactory as () => Runnable)();\n      if (\n        res &&\n        typeof (res as unknown as Promise<Runnable>).then === \"function\"\n      ) {\n        res = await res;\n      }\n      return modelOrFactory as () => Runnable;\n    } catch {\n      // Otherwise, it's a custom UDF, and we'll wrap\n      // the function in a lambda\n      const wrappedModel = new RunnableLambda({ func: modelOrFactory });\n      return () => wrappedModel;\n    }\n  }\n  throw new Error(\"Invalid modelOrFactory\");\n};\n\nconst loadExamples = async ({\n  datasetName,\n  client,\n  projectName,\n}: {\n  datasetName: string;\n  client: Client;\n  projectName: string;\n  maxConcurrency: number;\n}) => {\n  const exampleIterator = client.listExamples({ datasetName });\n  const configs: RunnableConfig[] = [];\n  const runExtractors = [];\n  const examples = [];\n  for await (const example of exampleIterator) {\n    const runExtractor = new SingleRunExtractor();\n    configs.push({\n      callbacks: [\n        new LangChainTracer({ exampleId: example.id, projectName }),\n        runExtractor,\n      ],\n    });\n    examples.push(example);\n    runExtractors.push(runExtractor);\n  }\n  return {\n    configs,\n    examples,\n    runExtractors,\n  };\n};\n\nconst applyEvaluators = async ({\n  evaluation,\n  runs,\n  examples,\n  client,\n  maxConcurrency,\n}: {\n  evaluation: LoadedEvalConfig;\n  runs: Run[];\n  examples: Example[];\n  client: Client;\n  maxConcurrency: number;\n}): Promise<{\n  [key: string]: {\n    execution_time?: number;\n    run_id: string;\n    feedback: Feedback[];\n  };\n}> => {\n  // TODO: Parallelize and/or put in callbacks to speed up evals.\n  const { evaluators } = evaluation;\n  const progress = new ProgressBar({\n    total: examples.length,\n    format: \"Running Evaluators: {bar} {percentage}% | {value}/{total}\\n\",\n  });\n  const caller = new AsyncCaller({\n    maxConcurrency,\n  });\n  const requests = runs.map(\n    async (\n      run,\n      i\n    ): Promise<{\n      run_id: string;\n      execution_time?: number;\n      feedback: Feedback[];\n    }> =>\n      caller.call(async () => {\n        const evaluatorResults = await Promise.allSettled(\n          evaluators.map((evaluator) =>\n            client.evaluateRun(run, evaluator, {\n              referenceExample: examples[i],\n              loadChildRuns: false,\n            })\n          )\n        );\n        progress.increment();\n        return {\n          execution_time:\n            run?.end_time && run.start_time\n              ? new Date(run.end_time).getTime() -\n                new Date(run.start_time).getTime()\n              : undefined,\n          feedback: evaluatorResults.map((evalResult) =>\n            evalResult.status === \"fulfilled\"\n              ? evalResult.value\n              : evalResult.reason\n          ),\n          run_id: run.id,\n        };\n      })\n  );\n  const results = await Promise.all(requests);\n\n  return results.reduce(\n    (acc, result, i) => ({\n      ...acc,\n      [examples[i].id]: result,\n    }),\n    {}\n  );\n};\n\nexport type EvalResults = {\n  projectName: string;\n  results: {\n    [key: string]: {\n      execution_time?: number;\n      run_id: string;\n      feedback: Feedback[];\n    };\n  };\n};\n\nconst getExamplesInputs = (\n  examples: Example[],\n  chainOrFactory: ChainOrFactory,\n  dataType?: DataType\n) => {\n  if (dataType === \"chat\") {\n    // For some batty reason, we store the chat dataset differently.\n    // { type: \"system\", data: { content: inputs.input } },\n    // But we need to create AIMesage, SystemMessage, etc.\n    return examples.map(({ inputs }) =>\n      mapStoredMessagesToChatMessages(inputs.input)\n    );\n  }\n  // If it's a language model and ALL example inputs have a single value,\n  // then we can be friendly and flatten the inputs to a list of strings.\n  const isLanguageModel =\n    typeof chainOrFactory === \"object\" &&\n    typeof (chainOrFactory as BaseLanguageModel)._llmType === \"function\";\n  if (\n    isLanguageModel &&\n    examples.every(({ inputs }) => Object.keys(inputs).length === 1)\n  ) {\n    return examples.map(({ inputs }) => Object.values(inputs)[0]);\n  }\n  return examples.map(({ inputs }) => inputs);\n};\n\n/**\n * Evaluates a given model or chain against a specified LangSmith dataset.\n *\n * This function fetches example records from the specified dataset,\n * runs the model or chain against each example, and returns the evaluation\n * results.\n *\n * @param chainOrFactory - A model or factory/constructor function to be evaluated. It can be a\n * Runnable instance, a factory function that returns a Runnable, or a user-defined\n * function or factory.\n *\n * @param datasetName - The name of the dataset against which the evaluation will be\n * performed. This dataset should already be defined and contain the relevant data\n * for evaluation.\n *\n * @param options - (Optional) Additional parameters for the evaluation process:\n *   - `evaluators` (RunEvalType[]): Evaluators to apply to a dataset run.\n *   - `formatEvaluatorInputs` (EvaluatorInputFormatter): Convert the evaluation data into formats that can be used by the evaluator.\n *   - `projectName` (string): Name of the project for logging and tracking.\n *   - `projectMetadata` (Record<string, unknown>): Additional metadata for the project.\n *   - `client` (Client): Client instance for LangSmith service interaction.\n *   - `maxConcurrency` (number): Maximum concurrency level for dataset processing.\n *\n * @returns A promise that resolves to an `EvalResults` object. This object includes\n * detailed results of the evaluation, such as execution time, run IDs, and feedback\n * for each entry in the dataset.\n *\n * @example\n * ```typescript\n * // Example usage for evaluating a model on a dataset\n * async function evaluateModel() {\n *   const chain = /* ...create your model or chain...*\\//\n *   const datasetName = 'example-dataset';\n *   const client = new Client(/* ...config... *\\//);\n *\n *   const results = await runOnDataset(chain, datasetName, {\n *     evaluators: [/* ...evaluators... *\\//],\n *     client,\n *   });\n *\n *   console.log('Evaluation Results:', results);\n * }\n *\n * evaluateModel();\n * ```\n * In this example, `runOnDataset` is used to evaluate a language model (or a chain of models) against\n * a dataset named 'example-dataset'. The evaluation process is configured using `RunOnDatasetParams[\"evaluators\"]`, which can\n * include both standard and custom evaluators. The `Client` instance is used to interact with LangChain services.\n * The function returns the evaluation results, which can be logged or further processed as needed.\n */\n\nexport async function runOnDataset(\n  chainOrFactory: ChainOrFactory,\n  datasetName: string,\n  options?: RunOnDatasetParams\n) {\n  const {\n    projectName,\n    projectMetadata,\n    client,\n    maxConcurrency,\n  }: RunOnDatasetParams = options ?? {};\n\n  const evaluationConfig: RunEvalConfig | undefined =\n    options?.evaluationConfig ??\n    (options?.evaluators != null\n      ? {\n          evaluators: options.evaluators,\n          formatEvaluatorInputs: options.formatEvaluatorInputs,\n        }\n      : undefined);\n\n  const wrappedModel = await createWrappedModel(chainOrFactory);\n  const testClient = client ?? new Client();\n  const testProjectName = projectName ?? randomName();\n  const dataset = await testClient.readDataset({ datasetName });\n  const datasetId = dataset.id;\n  const testConcurrency = maxConcurrency ?? 5;\n  const { configs, examples, runExtractors } = await loadExamples({\n    datasetName,\n    client: testClient,\n    projectName: testProjectName,\n    maxConcurrency: testConcurrency,\n  });\n\n  await testClient.createProject({\n    projectName: testProjectName,\n    referenceDatasetId: datasetId,\n    projectExtra: { metadata: { ...projectMetadata } },\n  });\n  const wrappedRunnable: Runnable = new RunnableLambda({\n    func: wrappedModel,\n  }).withConfig({ runName: \"evaluationRun\" });\n  const runInputs = getExamplesInputs(\n    examples,\n    chainOrFactory,\n    dataset.data_type\n  );\n  const progress = new ProgressBar({\n    total: runInputs.length,\n    format: \"Predicting: {bar} {percentage}% | {value}/{total}\",\n  });\n  // TODO: Collect the runs as well.\n  await wrappedRunnable\n    .withListeners({\n      onEnd: () => progress.increment(),\n    })\n    // TODO: Insert evaluation inline for immediate feedback.\n    .batch(runInputs, configs, {\n      maxConcurrency,\n      returnExceptions: true,\n    });\n\n  progress.complete();\n  const runs: Run[] = [];\n  for (let i = 0; i < examples.length; i += 1) {\n    runs.push(await runExtractors[i].extract());\n  }\n  let evalResults: Record<\n    string,\n    { run_id: string; execution_time?: number; feedback: Feedback[] }\n  > = {};\n  if (evaluationConfig) {\n    const loadedEvalConfig = await LoadedEvalConfig.fromRunEvalConfig(\n      evaluationConfig\n    );\n    evalResults = await applyEvaluators({\n      evaluation: loadedEvalConfig,\n      runs,\n      examples,\n      client: testClient,\n      maxConcurrency: testConcurrency,\n    });\n  }\n  const results: EvalResults = {\n    projectName: testProjectName,\n    results: evalResults ?? {},\n  };\n  return results;\n}\n\nfunction isLangsmithTraceableFunction(x: unknown): x is AnyTraceableFunction {\n  return typeof x === \"function\" && \"langsmith:traceable\" in x;\n}\n"],"mappings":";;;;;;;;;;;;;;AAsDA,IAAM,uBAAN,MAA2B;CACzB;CAEA;CAEA,cAAc;EACZ,KAAK,eAAe,IAAI,QAAgB,CAAC,YAAY;GACnD,KAAK,uBAAuB;EAC7B;CACF;CAED,mBAAmB,CACjBA,QACAC,SACAC,UACG;EACH,KAAK,qBAAqB,MAAM;CACjC;CAED,MAAM,UAA2B;AAC/B,SAAO,KAAK;CACb;AACF;AAED,IAAM,qBAAN,cAAiC,WAAW;CAC1C;CAEA;;CAGA,OAAO;CAEP,cAAc;EACZ,OAAO;EACP,KAAK,aAAa,IAAI,QAAa,CAAC,YAAY;GAC9C,KAAK,qBAAqB;EAC3B;CACF;CAED,MAAM,WAAWC,KAAU;EACzB,KAAK,mBAAmB,IAAI;CAC7B;CAED,MAAM,UAAwB;AAC5B,SAAO,KAAK;CACb;AACF;;;;AAKD,IAAM,sBAAN,MAAkD;CAChD;CAEA,YAAYC,WAA6B;EACvC,KAAK,YAAY,IAAI,eAAe,EAAE,MAAM,UAAW;CACxD;;;;;;;CAQD,MAAM,YAAYD,KAAUE,SAA8C;EACxE,MAAM,YAAY,IAAI;EACtB,MAAM,SAAS,IAAI,gBAAgB,EAAE,aAAa,aAAc;EAChE,MAAM,SAAS,MAAM,KAAK,UAAU,OAClC;GACE;GACA;GACA,OAAO,IAAI;GACX,YAAY,IAAI;GAChB,WAAW,SAAS;EACrB,GACD,EACE,WAAW,CAAC,WAAW,MAAO,EAC/B,EACF;EACD,MAAM,QAAQ,MAAM,UAAU,SAAS;AACvC,SAAO;GACL,aAAa;GACb,GAAG;EACJ;CACF;AACF;AAGD,SAAS,qBAAqBC,WAAiD;AAC7E,QAAO,aAAa,OAAO,UAAU,oBAAoB;AAC1D;;;;;;;;AAYD,IAAM,yBAAN,MAAM,+BAA+B,QAAQ;CAC3C;CAEA,wBAAgE;CAEhE,YAAYC,QAAuBC,iBAAkC;EACnE,MAAM,OAAO;EAEb,KAAK,kBAAkB;CACxB;CAED,YAAYD,QAA+C;EACzD,MAAM,QAAQ,IAAI,uBAChB;GACE,GAAG;GACH,YAAY;GACZ,cAAc,KAAK;GACnB,QAAQ,KAAK;EACd,GACD,KAAK,uBAAuB,UAAU,IAAI,KAAK;EAEjD,KAAK,WAAW,KAAK,MAAM;AAC3B,SAAO;CACR;CAED,MAAM,UAAyB;EAE7B,KAAK,wBAAwB,MAAM,KAAK,gBAAgB,iBACtD,OAAO,KAAK,eAAe,YACzB,KAAK,cAAc,QACnB,QAAQ,KAAK,aACX,KAAK,aACL;GACE,IAAI;IAAC;IAAa;IAAS;GAAyB;GACpD,IAAI;GACJ,MAAM;EACP,GACL,KAAK,QACL,KAAK,IACL,KAAK,UACL,QACA,QACA,KAAK,KACN;CACF;CAED,MAAM,WAA0B;AAC9B,MAAI,KAAK,OACP,MAAM,KAAK,uBAAuB,iBAChC,KAAK,OACL,KAAK,IACL,KAAK,YAAY,IACjB,QACA,OACD;OAED,MAAM,KAAK,uBAAuB,eAChC,KAAK,WAAW,CAAE,GAClB,KAAK,IACL,KAAK,YAAY,IACjB,QACA,OACD;CAEJ;AACF;AAED,IAAM,oBAAN,cAAqD,SAGnD;CACA,kBAAkB;CAElB,eAAe,CAAC,kBAAkB,WAAY;CAE9C,AAAU;CAEV,YAAYE,QAAwC;EAClD,MAAM,OAAO;AAEb,MAAI,CAAC,6BAA6B,OAAO,KAAK,CAC5C,OAAM,IAAI,MACR;EAIJ,KAAK,OAAO,OAAO;CACpB;CAED,MAAM,OAAOC,OAAiBC,SAAmC;EAC/D,MAAM,CAAC,OAAO,GAAG,KAAK,gBAAgB,WAAW,CAAE,GAAE,EAAE;EACvD,MAAM,kBAAkB,MAAM,4BAA4B,OAAO;EAEjE,MAAM,gBACJ,yBAAyB,KAAK,OACzB,KAAK,KAAK,yBACX,EAAE,MAAM,WAAY;AAE1B,MAAI,CAAC,gBAAiB,OAAM,IAAI,MAAM;EACtC,MAAM,UAAU,IAAI,uBAClB;GACE,GAAG;GACH,YAAY,iBAAiB,eACzB,IAAI,QAAQ;IAAE,MAAM;IAAY,IAAI,iBAAiB;GAAc,KACnE;EACL,GACD;AAGF,MACE,OAAO,UAAU,YACjB,SAAS,QACT,OAAO,KAAK,MAAM,CAAC,WAAW,GAC9B;AACA,OAAI,UAAU,SAAS,MAAM,QAAQ,MAAM,CACzC,QAAQ,MAAM,KAAK,KAAK,SAAS,GAAG,MAAM;AAG5C,OACE,WAAW,SACX,EACE,OAAO,UAAU,YACjB,SAAS,QACT,CAAC,MAAM,QAAQ,MAAM,IAErB,EAAE,iBAAiB,OAGrB,KAAI;AACF,WAAQ,MAAM,KAAK,KAAK,SAAS,MAAM,MAAM;GAC9C,QAAO;AACN,WAAQ,MAAM,KAAK,KAAK,SAAS,MAAM;GACxC;EAEJ;AAED,SAAQ,MAAM,KAAK,KAAK,SAAS,MAAM;CACxC;AACF;;;;;;AAOD,IAAM,uBAAN,MAAM,qBAA6C;CACjD;CAEA;CAEA;CAEA;CAEA,YACEC,WACAC,gBACAC,uBACA;EACA,KAAK,YAAY;EACjB,KAAK,oBAAoB,OAAO,WAAW,oBAAoB;EAC/D,KAAK,iBAAiB;EACtB,KAAK,wBAAwB;CAC9B;CAED,aAAa,eACXC,QAC+B;EAC/B,MAAM,gBACJ,OAAO,WAAW,WAAW,SAAS,OAAO;EAC/C,MAAM,aAAa,OAAO,WAAW,WAAY,CAAE,IAAkB;EACrE,MAAM,YAAY,MAAM,cAAc,eAAe,WAAW;EAChE,MAAM,cAAc,YAAY,eAAe,WAAW;AAC1D,MAAI,CAAC,qBAAqB,UAAU,CAClC,OAAM,IAAI,MACR,CAAC,kBAAkB,EAAE,cAAc,4GAAoB,CAEZ;AAG/C,MAAI,CAAC,YACH,OAAM,IAAI,MACR,CAAC,kBAAkB,EAAE,cAAc,qGAA4B,CACc;AAGjF,SAAO,IAAI,qBACT,WACA,aACA,YAAY;CAEf;;;;;;;CAQD,MAAM,YAAYZ,KAAUE,SAA8C;EACxE,MAAM,EAAE,YAAY,OAAO,WAAW,GAAG,KAAK,sBAAsB;GAClE,UAAU,IAAI;GACd,eAAe,IAAI;GACnB,oBAAoB,SAAS;GAC7B;EACD,EAAC;EACF,MAAM,YAAY,IAAI;EACtB,MAAM,SAAS,IAAI,gBAAgB,EAAE,aAAa,aAAc;AAChE,MAAI,KAAK,mBAAmB;GAC1B,MAAM,aAAa,MAAM,KAAK,UAAU,gBACtC;IACc;IACD;IACJ;GACR,GACD,EACE,WAAW,CAAC,WAAW,MAAO,EAC/B,EACF;GACD,MAAM,QAAQ,MAAM,UAAU,SAAS;AACvC,UAAO;IACL,KAAK,KAAK;IACV,SAAS,YAAY;IACrB,aAAa;IACb,GAAG;GACJ;EACF;AACD,QAAM,IAAI,MACR;CAIH;AACF;AAED,IAAM,mBAAN,MAAM,iBAAiB;CACrB,YAAmBW,YAAoD;EAApD;CAAsD;CAEzE,aAAa,kBACXC,QAC2B;EAE3B,MAAM,oBACJ,QAAQ,oBAAoB,OAAO,YAAY,OAAO,kBAAkB,GACvE,IAAI,CAAC,cAAc;AACpB,OAAI,OAAO,cAAc,WACvB,QAAO,IAAI,oBAAoB;OAE/B,QAAO;EAEV,EAAC;EAEF,MAAM,wBAAwB,MAAM,QAAQ,IAC1C,QAAQ,YACJ,OAAO,uBAAuB,EAC9B,IACA,OAAO,cACL,MAAM,qBAAqB,eAAe,UAAU,CACvD,IAAI,CAAE,EACV;AACD,SAAO,IAAI,kBACR,oBAAoB,CAAE,GAAE,OAAO,yBAAyB,CAAE,EAAC;CAE/D;AACF;;;;;;;;;AAsCD,MAAM,qBAAqB,OAAOC,mBAAmC;AACnE,KAAI,SAAS,WAAW,eAAe,CACrC,QAAO,MAAM;AAEf,KAAI,OAAO,mBAAmB,YAAY;AACxC,MAAI,6BAA6B,eAAe,EAAE;GAChD,MAAM,eAAe,IAAI,kBAAkB,EAAE,MAAM,eAAgB;AACnE,UAAO,MAAM;EACd;AAED,MAAI;GAEF,IAAI,MAAO,gBAAmC;AAC9C,OACE,OACA,OAAQ,IAAqC,SAAS,YAEtD,MAAM,MAAM;AAEd,UAAO;EACR,QAAO;GAGN,MAAM,eAAe,IAAI,eAAe,EAAE,MAAM,eAAgB;AAChE,UAAO,MAAM;EACd;CACF;AACD,OAAM,IAAI,MAAM;AACjB;AAED,MAAM,eAAe,OAAO,EAC1B,aACA,QACA,aAMD,KAAK;CACJ,MAAM,kBAAkB,OAAO,aAAa,EAAE,YAAa,EAAC;CAC5D,MAAMC,UAA4B,CAAE;CACpC,MAAM,gBAAgB,CAAE;CACxB,MAAM,WAAW,CAAE;AACnB,YAAW,MAAM,WAAW,iBAAiB;EAC3C,MAAM,eAAe,IAAI;EACzB,QAAQ,KAAK,EACX,WAAW,CACT,IAAI,gBAAgB;GAAE,WAAW,QAAQ;GAAI;EAAa,IAC1D,YACD,EACF,EAAC;EACF,SAAS,KAAK,QAAQ;EACtB,cAAc,KAAK,aAAa;CACjC;AACD,QAAO;EACL;EACA;EACA;CACD;AACF;AAED,MAAM,kBAAkB,OAAO,EAC7B,YACA,MACA,UACA,QACA,gBAOD,KAMK;CAEJ,MAAM,EAAE,YAAY,GAAG;CACvB,MAAM,WAAW,IAAI,YAAY;EAC/B,OAAO,SAAS;EAChB,QAAQ;CACT;CACD,MAAM,SAAS,IAAI,YAAY,EAC7B,eACD;CACD,MAAM,WAAW,KAAK,IACpB,OACE,KACA,MAMA,OAAO,KAAK,YAAY;EACtB,MAAM,mBAAmB,MAAM,QAAQ,WACrC,WAAW,IAAI,CAAC,cACd,OAAO,YAAY,KAAK,WAAW;GACjC,kBAAkB,SAAS;GAC3B,eAAe;EAChB,EAAC,CACH,CACF;EACD,SAAS,WAAW;AACpB,SAAO;GACL,gBACE,KAAK,YAAY,IAAI,aACjB,IAAI,KAAK,IAAI,UAAU,SAAS,GAChC,IAAI,KAAK,IAAI,YAAY,SAAS,GAClC;GACN,UAAU,iBAAiB,IAAI,CAAC,eAC9B,WAAW,WAAW,cAClB,WAAW,QACX,WAAW,OAChB;GACD,QAAQ,IAAI;EACb;CACF,EAAC,CACL;CACD,MAAM,UAAU,MAAM,QAAQ,IAAI,SAAS;AAE3C,QAAO,QAAQ,OACb,CAAC,KAAK,QAAQ,OAAO;EACnB,GAAG;GACF,SAAS,GAAG,KAAK;CACnB,IACD,CAAE,EACH;AACF;AAaD,MAAM,oBAAoB,CACxBC,UACAC,gBACAC,aACG;AACH,KAAI,aAAa,OAIf,QAAO,SAAS,IAAI,CAAC,EAAE,QAAQ,KAC7B,gCAAgC,OAAO,MAAM,CAC9C;CAIH,MAAM,kBACJ,OAAO,mBAAmB,YAC1B,OAAQ,eAAqC,aAAa;AAC5D,KACE,mBACA,SAAS,MAAM,CAAC,EAAE,QAAQ,KAAK,OAAO,KAAK,OAAO,CAAC,WAAW,EAAE,CAEhE,QAAO,SAAS,IAAI,CAAC,EAAE,QAAQ,KAAK,OAAO,OAAO,OAAO,CAAC,GAAG;AAE/D,QAAO,SAAS,IAAI,CAAC,EAAE,QAAQ,KAAK,OAAO;AAC5C;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAqDD,eAAsB,aACpBD,gBACAE,aACAC,SACA;CACA,MAAM,EACJ,aACA,iBACA,QACA,gBACmB,GAAG,WAAW,CAAE;CAErC,MAAMC,mBACJ,SAAS,qBACR,SAAS,cAAc,OACpB;EACE,YAAY,QAAQ;EACpB,uBAAuB,QAAQ;CAChC,IACD;CAEN,MAAM,eAAe,MAAM,mBAAmB,eAAe;CAC7D,MAAM,aAAa,UAAU,IAAI;CACjC,MAAM,kBAAkB,eAAe,YAAY;CACnD,MAAM,UAAU,MAAM,WAAW,YAAY,EAAE,YAAa,EAAC;CAC7D,MAAM,YAAY,QAAQ;CAC1B,MAAM,kBAAkB,kBAAkB;CAC1C,MAAM,EAAE,SAAS,UAAU,eAAe,GAAG,MAAM,aAAa;EAC9D;EACA,QAAQ;EACR,aAAa;EACb,gBAAgB;CACjB,EAAC;CAEF,MAAM,WAAW,cAAc;EAC7B,aAAa;EACb,oBAAoB;EACpB,cAAc,EAAE,UAAU,EAAE,GAAG,gBAAiB,EAAE;CACnD,EAAC;CACF,MAAMC,kBAA4B,IAAI,eAAe,EACnD,MAAM,aACP,GAAE,WAAW,EAAE,SAAS,gBAAiB,EAAC;CAC3C,MAAM,YAAY,kBAChB,UACA,gBACA,QAAQ,UACT;CACD,MAAM,WAAW,IAAI,YAAY;EAC/B,OAAO,UAAU;EACjB,QAAQ;CACT;CAED,MAAM,gBACH,cAAc,EACb,OAAO,MAAM,SAAS,WAAW,CAClC,EAAC,CAED,MAAM,WAAW,SAAS;EACzB;EACA,kBAAkB;CACnB,EAAC;CAEJ,SAAS,UAAU;CACnB,MAAMC,OAAc,CAAE;AACtB,MAAK,IAAI,IAAI,GAAG,IAAI,SAAS,QAAQ,KAAK,GACxC,KAAK,KAAK,MAAM,cAAc,GAAG,SAAS,CAAC;CAE7C,IAAIC,cAGA,CAAE;AACN,KAAI,kBAAkB;EACpB,MAAM,mBAAmB,MAAM,iBAAiB,kBAC9C,iBACD;EACD,cAAc,MAAM,gBAAgB;GAClC,YAAY;GACZ;GACA;GACA,QAAQ;GACR,gBAAgB;EACjB,EAAC;CACH;CACD,MAAMC,UAAuB;EAC3B,aAAa;EACb,SAAS,eAAe,CAAE;CAC3B;AACD,QAAO;AACR;AAED,SAAS,6BAA6BC,GAAuC;AAC3E,QAAO,OAAO,MAAM,cAAc,yBAAyB;AAC5D"}

package/dist/sql_db.cjs

@@ -0,0 +1,119 @@
+const require_rolldown_runtime = require('./_virtual/rolldown_runtime.cjs');
+const require_sql_utils = require('./util/sql_utils.cjs');
+const __langchain_core_load_serializable = require_rolldown_runtime.__toESM(require("@langchain/core/load/serializable"));
+
+//#region src/sql_db.ts
+var sql_db_exports = {};
+require_rolldown_runtime.__export(sql_db_exports, { SqlDatabase: () => SqlDatabase });
+/**
+* Class that represents a SQL database in the LangChain framework.
+*
+* @security **Security Notice**
+* This class generates SQL queries for the given database.
+* The SQLDatabase class provides a getTableInfo method that can be used
+* to get column information as well as sample data from the table.
+* To mitigate risk of leaking sensitive data, limit permissions
+* to read and scope to the tables that are needed.
+* Optionally, use the includesTables or ignoreTables class parameters
+* to limit which tables can/cannot be accessed.
+*
+* @link See https://js.langchain.com/docs/security for more information.
+*/
+var SqlDatabase = class SqlDatabase extends __langchain_core_load_serializable.Serializable {
+	lc_namespace = ["langchain", "sql_db"];
+	toJSON() {
+		return this.toJSONNotImplemented();
+	}
+	appDataSourceOptions;
+	appDataSource;
+	allTables = [];
+	includesTables = [];
+	ignoreTables = [];
+	sampleRowsInTableInfo = 3;
+	customDescription;
+	constructor(fields) {
+		super(...arguments);
+		this.appDataSource = fields.appDataSource;
+		this.appDataSourceOptions = fields.appDataSource.options;
+		if (fields?.includesTables && fields?.ignoreTables) throw new Error("Cannot specify both includeTables and ignoreTables");
+		this.includesTables = fields?.includesTables ?? [];
+		this.ignoreTables = fields?.ignoreTables ?? [];
+		this.sampleRowsInTableInfo = fields?.sampleRowsInTableInfo ?? this.sampleRowsInTableInfo;
+	}
+	static async fromDataSourceParams(fields) {
+		const sqlDatabase = new SqlDatabase(fields);
+		if (!sqlDatabase.appDataSource.isInitialized) await sqlDatabase.appDataSource.initialize();
+		sqlDatabase.allTables = await require_sql_utils.getTableAndColumnsName(sqlDatabase.appDataSource);
+		sqlDatabase.customDescription = Object.fromEntries(Object.entries(fields?.customDescription ?? {}).filter(([key, _]) => sqlDatabase.allTables.map((table) => table.tableName).includes(key)));
+		require_sql_utils.verifyIncludeTablesExistInDatabase(sqlDatabase.allTables, sqlDatabase.includesTables);
+		require_sql_utils.verifyIgnoreTablesExistInDatabase(sqlDatabase.allTables, sqlDatabase.ignoreTables);
+		return sqlDatabase;
+	}
+	static async fromOptionsParams(fields) {
+		const { DataSource } = await import("typeorm");
+		const dataSource = new DataSource(fields.appDataSourceOptions);
+		return SqlDatabase.fromDataSourceParams({
+			...fields,
+			appDataSource: dataSource
+		});
+	}
+	/**
+	* Get information about specified tables.
+	*
+	* Follows best practices as specified in: Rajkumar et al, 2022
+	* (https://arxiv.org/abs/2204.00498)
+	*
+	* If `sample_rows_in_table_info`, the specified number of sample rows will be
+	* appended to each table description. This can increase performance as
+	* demonstrated in the paper.
+	*/
+	async getTableInfo(targetTables) {
+		let selectedTables = this.includesTables.length > 0 ? this.allTables.filter((currentTable) => this.includesTables.includes(currentTable.tableName)) : this.allTables;
+		if (this.ignoreTables.length > 0) selectedTables = selectedTables.filter((currentTable) => !this.ignoreTables.includes(currentTable.tableName));
+		if (targetTables && targetTables.length > 0) {
+			require_sql_utils.verifyListTablesExistInDatabase(this.allTables, targetTables, "Wrong target table name:");
+			selectedTables = this.allTables.filter((currentTable) => targetTables.includes(currentTable.tableName));
+		}
+		return require_sql_utils.generateTableInfoFromTables(selectedTables, this.appDataSource, this.sampleRowsInTableInfo, this.customDescription);
+	}
+	/**
+	* Execute a SQL command and return a string representing the results.
+	* If the statement returns rows, a string of the results is returned.
+	* If the statement returns no rows, an empty string is returned.
+	*/
+	async run(command, fetch = "all") {
+		const res = await this.appDataSource.query(command);
+		if (fetch === "all") return JSON.stringify(res);
+		if (res?.length > 0) return JSON.stringify(res[0]);
+		return "";
+	}
+	serialize() {
+		return {
+			_type: "sql_database",
+			appDataSourceOptions: this.appDataSourceOptions,
+			includesTables: this.includesTables,
+			ignoreTables: this.ignoreTables,
+			sampleRowsInTableInfo: this.sampleRowsInTableInfo
+		};
+	}
+	/** @ignore */
+	static async imports() {
+		try {
+			const { DataSource } = await import("typeorm");
+			return { DataSource };
+		} catch (e) {
+			console.error(e);
+			throw new Error("Failed to load typeorm. Please install it with eg. `yarn add typeorm`.");
+		}
+	}
+};
+
+//#endregion
+exports.SqlDatabase = SqlDatabase;
+Object.defineProperty(exports, 'sql_db_exports', {
+  enumerable: true,
+  get: function () {
+    return sql_db_exports;
+  }
+});
+//# sourceMappingURL=sql_db.cjs.map

package/dist/sql_db.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"sql_db.cjs","names":["Serializable","fields: SqlDatabaseDataSourceParams","getTableAndColumnsName","table: SqlTable","verifyIncludeTablesExistInDatabase","verifyIgnoreTablesExistInDatabase","fields: SqlDatabaseOptionsParams","targetTables?: Array<string>","verifyListTablesExistInDatabase","generateTableInfoFromTables","command: string","fetch: \"all\" | \"one\""],"sources":["../src/sql_db.ts"],"sourcesContent":["import type { DataSource as DataSourceT, DataSourceOptions } from \"typeorm\";\nimport { Serializable } from \"@langchain/core/load/serializable\";\nimport {\n  generateTableInfoFromTables,\n  getTableAndColumnsName,\n  SerializedSqlDatabase,\n  SqlDatabaseDataSourceParams,\n  SqlDatabaseOptionsParams,\n  SqlTable,\n  verifyIgnoreTablesExistInDatabase,\n  verifyIncludeTablesExistInDatabase,\n  verifyListTablesExistInDatabase,\n} from \"./util/sql_utils.js\";\n\nexport type { SqlDatabaseDataSourceParams, SqlDatabaseOptionsParams };\n\n/**\n * Class that represents a SQL database in the LangChain framework.\n *\n * @security **Security Notice**\n * This class generates SQL queries for the given database.\n * The SQLDatabase class provides a getTableInfo method that can be used\n * to get column information as well as sample data from the table.\n * To mitigate risk of leaking sensitive data, limit permissions\n * to read and scope to the tables that are needed.\n * Optionally, use the includesTables or ignoreTables class parameters\n * to limit which tables can/cannot be accessed.\n *\n * @link See https://js.langchain.com/docs/security for more information.\n */\nexport class SqlDatabase\n  extends Serializable\n  implements SqlDatabaseOptionsParams, SqlDatabaseDataSourceParams\n{\n  lc_namespace = [\"langchain\", \"sql_db\"];\n\n  toJSON() {\n    return this.toJSONNotImplemented();\n  }\n\n  appDataSourceOptions: DataSourceOptions;\n\n  appDataSource: DataSourceT;\n\n  allTables: Array<SqlTable> = [];\n\n  includesTables: Array<string> = [];\n\n  ignoreTables: Array<string> = [];\n\n  sampleRowsInTableInfo = 3;\n\n  customDescription?: Record<string, string>;\n\n  protected constructor(fields: SqlDatabaseDataSourceParams) {\n    super(...arguments);\n    this.appDataSource = fields.appDataSource;\n    this.appDataSourceOptions = fields.appDataSource.options;\n    if (fields?.includesTables && fields?.ignoreTables) {\n      throw new Error(\"Cannot specify both includeTables and ignoreTables\");\n    }\n    this.includesTables = fields?.includesTables ?? [];\n    this.ignoreTables = fields?.ignoreTables ?? [];\n    this.sampleRowsInTableInfo =\n      fields?.sampleRowsInTableInfo ?? this.sampleRowsInTableInfo;\n  }\n\n  static async fromDataSourceParams(\n    fields: SqlDatabaseDataSourceParams\n  ): Promise<SqlDatabase> {\n    const sqlDatabase = new SqlDatabase(fields);\n    if (!sqlDatabase.appDataSource.isInitialized) {\n      await sqlDatabase.appDataSource.initialize();\n    }\n    sqlDatabase.allTables = await getTableAndColumnsName(\n      sqlDatabase.appDataSource\n    );\n    sqlDatabase.customDescription = Object.fromEntries(\n      Object.entries(fields?.customDescription ?? {}).filter(([key, _]) =>\n        sqlDatabase.allTables\n          .map((table: SqlTable) => table.tableName)\n          .includes(key)\n      )\n    );\n    verifyIncludeTablesExistInDatabase(\n      sqlDatabase.allTables,\n      sqlDatabase.includesTables\n    );\n    verifyIgnoreTablesExistInDatabase(\n      sqlDatabase.allTables,\n      sqlDatabase.ignoreTables\n    );\n    return sqlDatabase;\n  }\n\n  static async fromOptionsParams(\n    fields: SqlDatabaseOptionsParams\n  ): Promise<SqlDatabase> {\n    const { DataSource } = await import(\"typeorm\");\n    const dataSource = new DataSource(fields.appDataSourceOptions);\n    return SqlDatabase.fromDataSourceParams({\n      ...fields,\n      appDataSource: dataSource,\n    });\n  }\n\n  /**\n   * Get information about specified tables.\n   *\n   * Follows best practices as specified in: Rajkumar et al, 2022\n   * (https://arxiv.org/abs/2204.00498)\n   *\n   * If `sample_rows_in_table_info`, the specified number of sample rows will be\n   * appended to each table description. This can increase performance as\n   * demonstrated in the paper.\n   */\n  async getTableInfo(targetTables?: Array<string>): Promise<string> {\n    let selectedTables =\n      this.includesTables.length > 0\n        ? this.allTables.filter((currentTable) =>\n            this.includesTables.includes(currentTable.tableName)\n          )\n        : this.allTables;\n\n    if (this.ignoreTables.length > 0) {\n      selectedTables = selectedTables.filter(\n        (currentTable) => !this.ignoreTables.includes(currentTable.tableName)\n      );\n    }\n\n    if (targetTables && targetTables.length > 0) {\n      verifyListTablesExistInDatabase(\n        this.allTables,\n        targetTables,\n        \"Wrong target table name:\"\n      );\n      selectedTables = this.allTables.filter((currentTable) =>\n        targetTables.includes(currentTable.tableName)\n      );\n    }\n\n    return generateTableInfoFromTables(\n      selectedTables,\n      this.appDataSource,\n      this.sampleRowsInTableInfo,\n      this.customDescription\n    );\n  }\n\n  /**\n   * Execute a SQL command and return a string representing the results.\n   * If the statement returns rows, a string of the results is returned.\n   * If the statement returns no rows, an empty string is returned.\n   */\n  async run(command: string, fetch: \"all\" | \"one\" = \"all\"): Promise<string> {\n    // TODO: Potential security issue here\n    const res = await this.appDataSource.query(command);\n\n    if (fetch === \"all\") {\n      return JSON.stringify(res);\n    }\n\n    if (res?.length > 0) {\n      return JSON.stringify(res[0]);\n    }\n\n    return \"\";\n  }\n\n  serialize(): SerializedSqlDatabase {\n    return {\n      _type: \"sql_database\",\n      appDataSourceOptions: this.appDataSourceOptions,\n      includesTables: this.includesTables,\n      ignoreTables: this.ignoreTables,\n      sampleRowsInTableInfo: this.sampleRowsInTableInfo,\n    };\n  }\n\n  /** @ignore */\n  static async imports() {\n    try {\n      const { DataSource } = await import(\"typeorm\");\n      return { DataSource };\n    } catch (e) {\n      console.error(e);\n      throw new Error(\n        \"Failed to load typeorm. Please install it with eg. `yarn add typeorm`.\"\n      );\n    }\n  }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;AA8BA,IAAa,cAAb,MAAa,oBACHA,gDAEV;CACE,eAAe,CAAC,aAAa,QAAS;CAEtC,SAAS;AACP,SAAO,KAAK,sBAAsB;CACnC;CAED;CAEA;CAEA,YAA6B,CAAE;CAE/B,iBAAgC,CAAE;CAElC,eAA8B,CAAE;CAEhC,wBAAwB;CAExB;CAEA,AAAU,YAAYC,QAAqC;EACzD,MAAM,GAAG,UAAU;EACnB,KAAK,gBAAgB,OAAO;EAC5B,KAAK,uBAAuB,OAAO,cAAc;AACjD,MAAI,QAAQ,kBAAkB,QAAQ,aACpC,OAAM,IAAI,MAAM;EAElB,KAAK,iBAAiB,QAAQ,kBAAkB,CAAE;EAClD,KAAK,eAAe,QAAQ,gBAAgB,CAAE;EAC9C,KAAK,wBACH,QAAQ,yBAAyB,KAAK;CACzC;CAED,aAAa,qBACXA,QACsB;EACtB,MAAM,cAAc,IAAI,YAAY;AACpC,MAAI,CAAC,YAAY,cAAc,eAC7B,MAAM,YAAY,cAAc,YAAY;EAE9C,YAAY,YAAY,MAAMC,yCAC5B,YAAY,cACb;EACD,YAAY,oBAAoB,OAAO,YACrC,OAAO,QAAQ,QAAQ,qBAAqB,CAAE,EAAC,CAAC,OAAO,CAAC,CAAC,KAAK,EAAE,KAC9D,YAAY,UACT,IAAI,CAACC,UAAoB,MAAM,UAAU,CACzC,SAAS,IAAI,CACjB,CACF;EACDC,qDACE,YAAY,WACZ,YAAY,eACb;EACDC,oDACE,YAAY,WACZ,YAAY,aACb;AACD,SAAO;CACR;CAED,aAAa,kBACXC,QACsB;EACtB,MAAM,EAAE,YAAY,GAAG,MAAM,OAAO;EACpC,MAAM,aAAa,IAAI,WAAW,OAAO;AACzC,SAAO,YAAY,qBAAqB;GACtC,GAAG;GACH,eAAe;EAChB,EAAC;CACH;;;;;;;;;;;CAYD,MAAM,aAAaC,cAA+C;EAChE,IAAI,iBACF,KAAK,eAAe,SAAS,IACzB,KAAK,UAAU,OAAO,CAAC,iBACrB,KAAK,eAAe,SAAS,aAAa,UAAU,CACrD,GACD,KAAK;AAEX,MAAI,KAAK,aAAa,SAAS,GAC7B,iBAAiB,eAAe,OAC9B,CAAC,iBAAiB,CAAC,KAAK,aAAa,SAAS,aAAa,UAAU,CACtE;AAGH,MAAI,gBAAgB,aAAa,SAAS,GAAG;GAC3CC,kDACE,KAAK,WACL,cACA,2BACD;GACD,iBAAiB,KAAK,UAAU,OAAO,CAAC,iBACtC,aAAa,SAAS,aAAa,UAAU,CAC9C;EACF;AAED,SAAOC,8CACL,gBACA,KAAK,eACL,KAAK,uBACL,KAAK,kBACN;CACF;;;;;;CAOD,MAAM,IAAIC,SAAiBC,QAAuB,OAAwB;EAExE,MAAM,MAAM,MAAM,KAAK,cAAc,MAAM,QAAQ;AAEnD,MAAI,UAAU,MACZ,QAAO,KAAK,UAAU,IAAI;AAG5B,MAAI,KAAK,SAAS,EAChB,QAAO,KAAK,UAAU,IAAI,GAAG;AAG/B,SAAO;CACR;CAED,YAAmC;AACjC,SAAO;GACL,OAAO;GACP,sBAAsB,KAAK;GAC3B,gBAAgB,KAAK;GACrB,cAAc,KAAK;GACnB,uBAAuB,KAAK;EAC7B;CACF;;CAGD,aAAa,UAAU;AACrB,MAAI;GACF,MAAM,EAAE,YAAY,GAAG,MAAM,OAAO;AACpC,UAAO,EAAE,WAAY;EACtB,SAAQ,GAAG;GACV,QAAQ,MAAM,EAAE;AAChB,SAAM,IAAI,MACR;EAEH;CACF;AACF"}

package/dist/sql_db.d.cts

@@ -0,0 +1,60 @@
+import { SerializedSqlDatabase, SqlDatabaseDataSourceParams, SqlDatabaseOptionsParams, SqlTable } from "./util/sql_utils.cjs";
+import * as _langchain_core_load_serializable0 from "@langchain/core/load/serializable";
+import { Serializable } from "@langchain/core/load/serializable";
+import { DataSource, DataSourceOptions } from "typeorm";
+
+//#region src/sql_db.d.ts
+
+/**
+ * Class that represents a SQL database in the LangChain framework.
+ *
+ * @security **Security Notice**
+ * This class generates SQL queries for the given database.
+ * The SQLDatabase class provides a getTableInfo method that can be used
+ * to get column information as well as sample data from the table.
+ * To mitigate risk of leaking sensitive data, limit permissions
+ * to read and scope to the tables that are needed.
+ * Optionally, use the includesTables or ignoreTables class parameters
+ * to limit which tables can/cannot be accessed.
+ *
+ * @link See https://js.langchain.com/docs/security for more information.
+ */
+declare class SqlDatabase extends Serializable implements SqlDatabaseOptionsParams, SqlDatabaseDataSourceParams {
+  lc_namespace: string[];
+  toJSON(): _langchain_core_load_serializable0.SerializedNotImplemented;
+  appDataSourceOptions: DataSourceOptions;
+  appDataSource: DataSource;
+  allTables: Array<SqlTable>;
+  includesTables: Array<string>;
+  ignoreTables: Array<string>;
+  sampleRowsInTableInfo: number;
+  customDescription?: Record<string, string>;
+  protected constructor(fields: SqlDatabaseDataSourceParams);
+  static fromDataSourceParams(fields: SqlDatabaseDataSourceParams): Promise<SqlDatabase>;
+  static fromOptionsParams(fields: SqlDatabaseOptionsParams): Promise<SqlDatabase>;
+  /**
+   * Get information about specified tables.
+   *
+   * Follows best practices as specified in: Rajkumar et al, 2022
+   * (https://arxiv.org/abs/2204.00498)
+   *
+   * If `sample_rows_in_table_info`, the specified number of sample rows will be
+   * appended to each table description. This can increase performance as
+   * demonstrated in the paper.
+   */
+  getTableInfo(targetTables?: Array<string>): Promise<string>;
+  /**
+   * Execute a SQL command and return a string representing the results.
+   * If the statement returns rows, a string of the results is returned.
+   * If the statement returns no rows, an empty string is returned.
+   */
+  run(command: string, fetch?: "all" | "one"): Promise<string>;
+  serialize(): SerializedSqlDatabase;
+  /** @ignore */
+  static imports(): Promise<{
+    DataSource: typeof DataSource;
+  }>;
+}
+//#endregion
+export { SqlDatabase, type SqlDatabaseDataSourceParams, type SqlDatabaseOptionsParams };
+//# sourceMappingURL=sql_db.d.cts.map

package/dist/sql_db.d.cts.map

@@ -0,0 +1 @@
+{"version":3,"file":"sql_db.d.cts","names":["DataSource","DataSourceT","DataSourceOptions","Serializable","SerializedSqlDatabase","SqlDatabaseDataSourceParams","SqlDatabaseOptionsParams","SqlTable","SqlDatabase","_langchain_core_load_serializable0","SerializedNotImplemented","Array","Record","Promise"],"sources":["../src/sql_db.d.ts"],"sourcesContent":["import type { DataSource as DataSourceT, DataSourceOptions } from \"typeorm\";\nimport { Serializable } from \"@langchain/core/load/serializable\";\nimport { SerializedSqlDatabase, SqlDatabaseDataSourceParams, SqlDatabaseOptionsParams, SqlTable } from \"./util/sql_utils.js\";\nexport type { SqlDatabaseDataSourceParams, SqlDatabaseOptionsParams };\n/**\n * Class that represents a SQL database in the LangChain framework.\n *\n * @security **Security Notice**\n * This class generates SQL queries for the given database.\n * The SQLDatabase class provides a getTableInfo method that can be used\n * to get column information as well as sample data from the table.\n * To mitigate risk of leaking sensitive data, limit permissions\n * to read and scope to the tables that are needed.\n * Optionally, use the includesTables or ignoreTables class parameters\n * to limit which tables can/cannot be accessed.\n *\n * @link See https://js.langchain.com/docs/security for more information.\n */\nexport declare class SqlDatabase extends Serializable implements SqlDatabaseOptionsParams, SqlDatabaseDataSourceParams {\n    lc_namespace: string[];\n    toJSON(): import(\"@langchain/core/load/serializable\").SerializedNotImplemented;\n    appDataSourceOptions: DataSourceOptions;\n    appDataSource: DataSourceT;\n    allTables: Array<SqlTable>;\n    includesTables: Array<string>;\n    ignoreTables: Array<string>;\n    sampleRowsInTableInfo: number;\n    customDescription?: Record<string, string>;\n    protected constructor(fields: SqlDatabaseDataSourceParams);\n    static fromDataSourceParams(fields: SqlDatabaseDataSourceParams): Promise<SqlDatabase>;\n    static fromOptionsParams(fields: SqlDatabaseOptionsParams): Promise<SqlDatabase>;\n    /**\n     * Get information about specified tables.\n     *\n     * Follows best practices as specified in: Rajkumar et al, 2022\n     * (https://arxiv.org/abs/2204.00498)\n     *\n     * If `sample_rows_in_table_info`, the specified number of sample rows will be\n     * appended to each table description. This can increase performance as\n     * demonstrated in the paper.\n     */\n    getTableInfo(targetTables?: Array<string>): Promise<string>;\n    /**\n     * Execute a SQL command and return a string representing the results.\n     * If the statement returns rows, a string of the results is returned.\n     * If the statement returns no rows, an empty string is returned.\n     */\n    run(command: string, fetch?: \"all\" | \"one\"): Promise<string>;\n    serialize(): SerializedSqlDatabase;\n    /** @ignore */\n    static imports(): Promise<{\n        DataSource: typeof DataSourceT;\n    }>;\n}\n"],"mappings":";;;;;;;;;AAkBA;;;;;;;;;;;;AAW8EQ,cAXzDA,WAAAA,SAAoBL,YAAAA,YAAwBG,wBAWaE,EAXaH,2BAWbG,CAAAA;EAAW,YAAnBK,EAAAA,MAAAA,EAAAA;EAAO,MACxCP,CAAAA,CAAAA,EAZLG,kCAAAA,CAE0BC,wBAUrBJ;EAAwB,oBAAWE,EAT9CN,iBAS8CM;EAAW,aAAnBK,EAR7CZ,UAQ6CY;EAAO,SAWvCF,EAlBjBA,KAkBiBA,CAlBXJ,QAkBWI,CAAAA;EAAK,cAAWE,EAjB5BF,KAiB4BE,CAAAA,MAAAA,CAAAA;EAAO,YAMNA,EAtB/BF,KAsB+BE,CAAAA,MAAAA,CAAAA;EAAO,qBACvCT,EAAAA,MAAAA;EAAqB,iBAGXH,CAAAA,EAxBHW,MAwBGX,CAAAA,MAAAA,EAAAA,MAAAA,CAAAA;EAAW,UADhBY,WAAAA,CAAAA,MAAAA,EAtBYR,2BAsBZQ;EAAO,OAhCYV,oBAAAA,CAAAA,MAAAA,EAWDE,2BAXCF,CAAAA,EAW6BU,OAX7BV,CAWqCK,WAXrCL,CAAAA;EAAY,OAAYG,iBAAAA,CAAAA,MAAAA,EAY5BA,wBAZ4BA,CAAAA,EAYDO,OAZCP,CAYOE,WAZPF,CAAAA;EAAwB;AAA6B;;;;;;;;;8BAuBtFK,gBAAgBE;;;;;;+CAMCA;eAChCT;;oBAEKS;uBACKZ"}

package/dist/sql_db.d.ts

@@ -0,0 +1,60 @@
+import { SerializedSqlDatabase, SqlDatabaseDataSourceParams, SqlDatabaseOptionsParams, SqlTable } from "./util/sql_utils.js";
+import * as _langchain_core_load_serializable0 from "@langchain/core/load/serializable";
+import { Serializable } from "@langchain/core/load/serializable";
+import { DataSource, DataSourceOptions } from "typeorm";
+
+//#region src/sql_db.d.ts
+
+/**
+ * Class that represents a SQL database in the LangChain framework.
+ *
+ * @security **Security Notice**
+ * This class generates SQL queries for the given database.
+ * The SQLDatabase class provides a getTableInfo method that can be used
+ * to get column information as well as sample data from the table.
+ * To mitigate risk of leaking sensitive data, limit permissions
+ * to read and scope to the tables that are needed.
+ * Optionally, use the includesTables or ignoreTables class parameters
+ * to limit which tables can/cannot be accessed.
+ *
+ * @link See https://js.langchain.com/docs/security for more information.
+ */
+declare class SqlDatabase extends Serializable implements SqlDatabaseOptionsParams, SqlDatabaseDataSourceParams {
+  lc_namespace: string[];
+  toJSON(): _langchain_core_load_serializable0.SerializedNotImplemented;
+  appDataSourceOptions: DataSourceOptions;
+  appDataSource: DataSource;
+  allTables: Array<SqlTable>;
+  includesTables: Array<string>;
+  ignoreTables: Array<string>;
+  sampleRowsInTableInfo: number;
+  customDescription?: Record<string, string>;
+  protected constructor(fields: SqlDatabaseDataSourceParams);
+  static fromDataSourceParams(fields: SqlDatabaseDataSourceParams): Promise<SqlDatabase>;
+  static fromOptionsParams(fields: SqlDatabaseOptionsParams): Promise<SqlDatabase>;
+  /**
+   * Get information about specified tables.
+   *
+   * Follows best practices as specified in: Rajkumar et al, 2022
+   * (https://arxiv.org/abs/2204.00498)
+   *
+   * If `sample_rows_in_table_info`, the specified number of sample rows will be
+   * appended to each table description. This can increase performance as
+   * demonstrated in the paper.
+   */
+  getTableInfo(targetTables?: Array<string>): Promise<string>;
+  /**
+   * Execute a SQL command and return a string representing the results.
+   * If the statement returns rows, a string of the results is returned.
+   * If the statement returns no rows, an empty string is returned.
+   */
+  run(command: string, fetch?: "all" | "one"): Promise<string>;
+  serialize(): SerializedSqlDatabase;
+  /** @ignore */
+  static imports(): Promise<{
+    DataSource: typeof DataSource;
+  }>;
+}
+//#endregion
+export { SqlDatabase, type SqlDatabaseDataSourceParams, type SqlDatabaseOptionsParams };
+//# sourceMappingURL=sql_db.d.ts.map

package/dist/sql_db.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"sql_db.d.ts","names":["DataSource","DataSourceT","DataSourceOptions","Serializable","SerializedSqlDatabase","SqlDatabaseDataSourceParams","SqlDatabaseOptionsParams","SqlTable","SqlDatabase","_langchain_core_load_serializable0","SerializedNotImplemented","Array","Record","Promise"],"sources":["../src/sql_db.d.ts"],"sourcesContent":["import type { DataSource as DataSourceT, DataSourceOptions } from \"typeorm\";\nimport { Serializable } from \"@langchain/core/load/serializable\";\nimport { SerializedSqlDatabase, SqlDatabaseDataSourceParams, SqlDatabaseOptionsParams, SqlTable } from \"./util/sql_utils.js\";\nexport type { SqlDatabaseDataSourceParams, SqlDatabaseOptionsParams };\n/**\n * Class that represents a SQL database in the LangChain framework.\n *\n * @security **Security Notice**\n * This class generates SQL queries for the given database.\n * The SQLDatabase class provides a getTableInfo method that can be used\n * to get column information as well as sample data from the table.\n * To mitigate risk of leaking sensitive data, limit permissions\n * to read and scope to the tables that are needed.\n * Optionally, use the includesTables or ignoreTables class parameters\n * to limit which tables can/cannot be accessed.\n *\n * @link See https://js.langchain.com/docs/security for more information.\n */\nexport declare class SqlDatabase extends Serializable implements SqlDatabaseOptionsParams, SqlDatabaseDataSourceParams {\n    lc_namespace: string[];\n    toJSON(): import(\"@langchain/core/load/serializable\").SerializedNotImplemented;\n    appDataSourceOptions: DataSourceOptions;\n    appDataSource: DataSourceT;\n    allTables: Array<SqlTable>;\n    includesTables: Array<string>;\n    ignoreTables: Array<string>;\n    sampleRowsInTableInfo: number;\n    customDescription?: Record<string, string>;\n    protected constructor(fields: SqlDatabaseDataSourceParams);\n    static fromDataSourceParams(fields: SqlDatabaseDataSourceParams): Promise<SqlDatabase>;\n    static fromOptionsParams(fields: SqlDatabaseOptionsParams): Promise<SqlDatabase>;\n    /**\n     * Get information about specified tables.\n     *\n     * Follows best practices as specified in: Rajkumar et al, 2022\n     * (https://arxiv.org/abs/2204.00498)\n     *\n     * If `sample_rows_in_table_info`, the specified number of sample rows will be\n     * appended to each table description. This can increase performance as\n     * demonstrated in the paper.\n     */\n    getTableInfo(targetTables?: Array<string>): Promise<string>;\n    /**\n     * Execute a SQL command and return a string representing the results.\n     * If the statement returns rows, a string of the results is returned.\n     * If the statement returns no rows, an empty string is returned.\n     */\n    run(command: string, fetch?: \"all\" | \"one\"): Promise<string>;\n    serialize(): SerializedSqlDatabase;\n    /** @ignore */\n    static imports(): Promise<{\n        DataSource: typeof DataSourceT;\n    }>;\n}\n"],"mappings":";;;;;;;;;AAkBA;;;;;;;;;;;;AAW8EQ,cAXzDA,WAAAA,SAAoBL,YAAAA,YAAwBG,wBAWaE,EAXaH,2BAWbG,CAAAA;EAAW,YAAnBK,EAAAA,MAAAA,EAAAA;EAAO,MACxCP,CAAAA,CAAAA,EAZLG,kCAAAA,CAE0BC,wBAUrBJ;EAAwB,oBAAWE,EAT9CN,iBAS8CM;EAAW,aAAnBK,EAR7CZ,UAQ6CY;EAAO,SAWvCF,EAlBjBA,KAkBiBA,CAlBXJ,QAkBWI,CAAAA;EAAK,cAAWE,EAjB5BF,KAiB4BE,CAAAA,MAAAA,CAAAA;EAAO,YAMNA,EAtB/BF,KAsB+BE,CAAAA,MAAAA,CAAAA;EAAO,qBACvCT,EAAAA,MAAAA;EAAqB,iBAGXH,CAAAA,EAxBHW,MAwBGX,CAAAA,MAAAA,EAAAA,MAAAA,CAAAA;EAAW,UADhBY,WAAAA,CAAAA,MAAAA,EAtBYR,2BAsBZQ;EAAO,OAhCYV,oBAAAA,CAAAA,MAAAA,EAWDE,2BAXCF,CAAAA,EAW6BU,OAX7BV,CAWqCK,WAXrCL,CAAAA;EAAY,OAAYG,iBAAAA,CAAAA,MAAAA,EAY5BA,wBAZ4BA,CAAAA,EAYDO,OAZCP,CAYOE,WAZPF,CAAAA;EAAwB;AAA6B;;;;;;;;;8BAuBtFK,gBAAgBE;;;;;;+CAMCA;eAChCT;;oBAEKS;uBACKZ"}

package/dist/sql_db.js

@@ -0,0 +1,113 @@
+import { __export } from "./_virtual/rolldown_runtime.js";
+import { generateTableInfoFromTables, getTableAndColumnsName, verifyIgnoreTablesExistInDatabase, verifyIncludeTablesExistInDatabase, verifyListTablesExistInDatabase } from "./util/sql_utils.js";
+import { Serializable } from "@langchain/core/load/serializable";
+
+//#region src/sql_db.ts
+var sql_db_exports = {};
+__export(sql_db_exports, { SqlDatabase: () => SqlDatabase });
+/**
+* Class that represents a SQL database in the LangChain framework.
+*
+* @security **Security Notice**
+* This class generates SQL queries for the given database.
+* The SQLDatabase class provides a getTableInfo method that can be used
+* to get column information as well as sample data from the table.
+* To mitigate risk of leaking sensitive data, limit permissions
+* to read and scope to the tables that are needed.
+* Optionally, use the includesTables or ignoreTables class parameters
+* to limit which tables can/cannot be accessed.
+*
+* @link See https://js.langchain.com/docs/security for more information.
+*/
+var SqlDatabase = class SqlDatabase extends Serializable {
+	lc_namespace = ["langchain", "sql_db"];
+	toJSON() {
+		return this.toJSONNotImplemented();
+	}
+	appDataSourceOptions;
+	appDataSource;
+	allTables = [];
+	includesTables = [];
+	ignoreTables = [];
+	sampleRowsInTableInfo = 3;
+	customDescription;
+	constructor(fields) {
+		super(...arguments);
+		this.appDataSource = fields.appDataSource;
+		this.appDataSourceOptions = fields.appDataSource.options;
+		if (fields?.includesTables && fields?.ignoreTables) throw new Error("Cannot specify both includeTables and ignoreTables");
+		this.includesTables = fields?.includesTables ?? [];
+		this.ignoreTables = fields?.ignoreTables ?? [];
+		this.sampleRowsInTableInfo = fields?.sampleRowsInTableInfo ?? this.sampleRowsInTableInfo;
+	}
+	static async fromDataSourceParams(fields) {
+		const sqlDatabase = new SqlDatabase(fields);
+		if (!sqlDatabase.appDataSource.isInitialized) await sqlDatabase.appDataSource.initialize();
+		sqlDatabase.allTables = await getTableAndColumnsName(sqlDatabase.appDataSource);
+		sqlDatabase.customDescription = Object.fromEntries(Object.entries(fields?.customDescription ?? {}).filter(([key, _]) => sqlDatabase.allTables.map((table) => table.tableName).includes(key)));
+		verifyIncludeTablesExistInDatabase(sqlDatabase.allTables, sqlDatabase.includesTables);
+		verifyIgnoreTablesExistInDatabase(sqlDatabase.allTables, sqlDatabase.ignoreTables);
+		return sqlDatabase;
+	}
+	static async fromOptionsParams(fields) {
+		const { DataSource } = await import("typeorm");
+		const dataSource = new DataSource(fields.appDataSourceOptions);
+		return SqlDatabase.fromDataSourceParams({
+			...fields,
+			appDataSource: dataSource
+		});
+	}
+	/**
+	* Get information about specified tables.
+	*
+	* Follows best practices as specified in: Rajkumar et al, 2022
+	* (https://arxiv.org/abs/2204.00498)
+	*
+	* If `sample_rows_in_table_info`, the specified number of sample rows will be
+	* appended to each table description. This can increase performance as
+	* demonstrated in the paper.
+	*/
+	async getTableInfo(targetTables) {
+		let selectedTables = this.includesTables.length > 0 ? this.allTables.filter((currentTable) => this.includesTables.includes(currentTable.tableName)) : this.allTables;
+		if (this.ignoreTables.length > 0) selectedTables = selectedTables.filter((currentTable) => !this.ignoreTables.includes(currentTable.tableName));
+		if (targetTables && targetTables.length > 0) {
+			verifyListTablesExistInDatabase(this.allTables, targetTables, "Wrong target table name:");
+			selectedTables = this.allTables.filter((currentTable) => targetTables.includes(currentTable.tableName));
+		}
+		return generateTableInfoFromTables(selectedTables, this.appDataSource, this.sampleRowsInTableInfo, this.customDescription);
+	}
+	/**
+	* Execute a SQL command and return a string representing the results.
+	* If the statement returns rows, a string of the results is returned.
+	* If the statement returns no rows, an empty string is returned.
+	*/
+	async run(command, fetch = "all") {
+		const res = await this.appDataSource.query(command);
+		if (fetch === "all") return JSON.stringify(res);
+		if (res?.length > 0) return JSON.stringify(res[0]);
+		return "";
+	}
+	serialize() {
+		return {
+			_type: "sql_database",
+			appDataSourceOptions: this.appDataSourceOptions,
+			includesTables: this.includesTables,
+			ignoreTables: this.ignoreTables,
+			sampleRowsInTableInfo: this.sampleRowsInTableInfo
+		};
+	}
+	/** @ignore */
+	static async imports() {
+		try {
+			const { DataSource } = await import("typeorm");
+			return { DataSource };
+		} catch (e) {
+			console.error(e);
+			throw new Error("Failed to load typeorm. Please install it with eg. `yarn add typeorm`.");
+		}
+	}
+};
+
+//#endregion
+export { SqlDatabase, sql_db_exports };
+//# sourceMappingURL=sql_db.js.map

package/dist/sql_db.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"sql_db.js","names":["fields: SqlDatabaseDataSourceParams","table: SqlTable","fields: SqlDatabaseOptionsParams","targetTables?: Array<string>","command: string","fetch: \"all\" | \"one\""],"sources":["../src/sql_db.ts"],"sourcesContent":["import type { DataSource as DataSourceT, DataSourceOptions } from \"typeorm\";\nimport { Serializable } from \"@langchain/core/load/serializable\";\nimport {\n  generateTableInfoFromTables,\n  getTableAndColumnsName,\n  SerializedSqlDatabase,\n  SqlDatabaseDataSourceParams,\n  SqlDatabaseOptionsParams,\n  SqlTable,\n  verifyIgnoreTablesExistInDatabase,\n  verifyIncludeTablesExistInDatabase,\n  verifyListTablesExistInDatabase,\n} from \"./util/sql_utils.js\";\n\nexport type { SqlDatabaseDataSourceParams, SqlDatabaseOptionsParams };\n\n/**\n * Class that represents a SQL database in the LangChain framework.\n *\n * @security **Security Notice**\n * This class generates SQL queries for the given database.\n * The SQLDatabase class provides a getTableInfo method that can be used\n * to get column information as well as sample data from the table.\n * To mitigate risk of leaking sensitive data, limit permissions\n * to read and scope to the tables that are needed.\n * Optionally, use the includesTables or ignoreTables class parameters\n * to limit which tables can/cannot be accessed.\n *\n * @link See https://js.langchain.com/docs/security for more information.\n */\nexport class SqlDatabase\n  extends Serializable\n  implements SqlDatabaseOptionsParams, SqlDatabaseDataSourceParams\n{\n  lc_namespace = [\"langchain\", \"sql_db\"];\n\n  toJSON() {\n    return this.toJSONNotImplemented();\n  }\n\n  appDataSourceOptions: DataSourceOptions;\n\n  appDataSource: DataSourceT;\n\n  allTables: Array<SqlTable> = [];\n\n  includesTables: Array<string> = [];\n\n  ignoreTables: Array<string> = [];\n\n  sampleRowsInTableInfo = 3;\n\n  customDescription?: Record<string, string>;\n\n  protected constructor(fields: SqlDatabaseDataSourceParams) {\n    super(...arguments);\n    this.appDataSource = fields.appDataSource;\n    this.appDataSourceOptions = fields.appDataSource.options;\n    if (fields?.includesTables && fields?.ignoreTables) {\n      throw new Error(\"Cannot specify both includeTables and ignoreTables\");\n    }\n    this.includesTables = fields?.includesTables ?? [];\n    this.ignoreTables = fields?.ignoreTables ?? [];\n    this.sampleRowsInTableInfo =\n      fields?.sampleRowsInTableInfo ?? this.sampleRowsInTableInfo;\n  }\n\n  static async fromDataSourceParams(\n    fields: SqlDatabaseDataSourceParams\n  ): Promise<SqlDatabase> {\n    const sqlDatabase = new SqlDatabase(fields);\n    if (!sqlDatabase.appDataSource.isInitialized) {\n      await sqlDatabase.appDataSource.initialize();\n    }\n    sqlDatabase.allTables = await getTableAndColumnsName(\n      sqlDatabase.appDataSource\n    );\n    sqlDatabase.customDescription = Object.fromEntries(\n      Object.entries(fields?.customDescription ?? {}).filter(([key, _]) =>\n        sqlDatabase.allTables\n          .map((table: SqlTable) => table.tableName)\n          .includes(key)\n      )\n    );\n    verifyIncludeTablesExistInDatabase(\n      sqlDatabase.allTables,\n      sqlDatabase.includesTables\n    );\n    verifyIgnoreTablesExistInDatabase(\n      sqlDatabase.allTables,\n      sqlDatabase.ignoreTables\n    );\n    return sqlDatabase;\n  }\n\n  static async fromOptionsParams(\n    fields: SqlDatabaseOptionsParams\n  ): Promise<SqlDatabase> {\n    const { DataSource } = await import(\"typeorm\");\n    const dataSource = new DataSource(fields.appDataSourceOptions);\n    return SqlDatabase.fromDataSourceParams({\n      ...fields,\n      appDataSource: dataSource,\n    });\n  }\n\n  /**\n   * Get information about specified tables.\n   *\n   * Follows best practices as specified in: Rajkumar et al, 2022\n   * (https://arxiv.org/abs/2204.00498)\n   *\n   * If `sample_rows_in_table_info`, the specified number of sample rows will be\n   * appended to each table description. This can increase performance as\n   * demonstrated in the paper.\n   */\n  async getTableInfo(targetTables?: Array<string>): Promise<string> {\n    let selectedTables =\n      this.includesTables.length > 0\n        ? this.allTables.filter((currentTable) =>\n            this.includesTables.includes(currentTable.tableName)\n          )\n        : this.allTables;\n\n    if (this.ignoreTables.length > 0) {\n      selectedTables = selectedTables.filter(\n        (currentTable) => !this.ignoreTables.includes(currentTable.tableName)\n      );\n    }\n\n    if (targetTables && targetTables.length > 0) {\n      verifyListTablesExistInDatabase(\n        this.allTables,\n        targetTables,\n        \"Wrong target table name:\"\n      );\n      selectedTables = this.allTables.filter((currentTable) =>\n        targetTables.includes(currentTable.tableName)\n      );\n    }\n\n    return generateTableInfoFromTables(\n      selectedTables,\n      this.appDataSource,\n      this.sampleRowsInTableInfo,\n      this.customDescription\n    );\n  }\n\n  /**\n   * Execute a SQL command and return a string representing the results.\n   * If the statement returns rows, a string of the results is returned.\n   * If the statement returns no rows, an empty string is returned.\n   */\n  async run(command: string, fetch: \"all\" | \"one\" = \"all\"): Promise<string> {\n    // TODO: Potential security issue here\n    const res = await this.appDataSource.query(command);\n\n    if (fetch === \"all\") {\n      return JSON.stringify(res);\n    }\n\n    if (res?.length > 0) {\n      return JSON.stringify(res[0]);\n    }\n\n    return \"\";\n  }\n\n  serialize(): SerializedSqlDatabase {\n    return {\n      _type: \"sql_database\",\n      appDataSourceOptions: this.appDataSourceOptions,\n      includesTables: this.includesTables,\n      ignoreTables: this.ignoreTables,\n      sampleRowsInTableInfo: this.sampleRowsInTableInfo,\n    };\n  }\n\n  /** @ignore */\n  static async imports() {\n    try {\n      const { DataSource } = await import(\"typeorm\");\n      return { DataSource };\n    } catch (e) {\n      console.error(e);\n      throw new Error(\n        \"Failed to load typeorm. Please install it with eg. `yarn add typeorm`.\"\n      );\n    }\n  }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;AA8BA,IAAa,cAAb,MAAa,oBACH,aAEV;CACE,eAAe,CAAC,aAAa,QAAS;CAEtC,SAAS;AACP,SAAO,KAAK,sBAAsB;CACnC;CAED;CAEA;CAEA,YAA6B,CAAE;CAE/B,iBAAgC,CAAE;CAElC,eAA8B,CAAE;CAEhC,wBAAwB;CAExB;CAEA,AAAU,YAAYA,QAAqC;EACzD,MAAM,GAAG,UAAU;EACnB,KAAK,gBAAgB,OAAO;EAC5B,KAAK,uBAAuB,OAAO,cAAc;AACjD,MAAI,QAAQ,kBAAkB,QAAQ,aACpC,OAAM,IAAI,MAAM;EAElB,KAAK,iBAAiB,QAAQ,kBAAkB,CAAE;EAClD,KAAK,eAAe,QAAQ,gBAAgB,CAAE;EAC9C,KAAK,wBACH,QAAQ,yBAAyB,KAAK;CACzC;CAED,aAAa,qBACXA,QACsB;EACtB,MAAM,cAAc,IAAI,YAAY;AACpC,MAAI,CAAC,YAAY,cAAc,eAC7B,MAAM,YAAY,cAAc,YAAY;EAE9C,YAAY,YAAY,MAAM,uBAC5B,YAAY,cACb;EACD,YAAY,oBAAoB,OAAO,YACrC,OAAO,QAAQ,QAAQ,qBAAqB,CAAE,EAAC,CAAC,OAAO,CAAC,CAAC,KAAK,EAAE,KAC9D,YAAY,UACT,IAAI,CAACC,UAAoB,MAAM,UAAU,CACzC,SAAS,IAAI,CACjB,CACF;EACD,mCACE,YAAY,WACZ,YAAY,eACb;EACD,kCACE,YAAY,WACZ,YAAY,aACb;AACD,SAAO;CACR;CAED,aAAa,kBACXC,QACsB;EACtB,MAAM,EAAE,YAAY,GAAG,MAAM,OAAO;EACpC,MAAM,aAAa,IAAI,WAAW,OAAO;AACzC,SAAO,YAAY,qBAAqB;GACtC,GAAG;GACH,eAAe;EAChB,EAAC;CACH;;;;;;;;;;;CAYD,MAAM,aAAaC,cAA+C;EAChE,IAAI,iBACF,KAAK,eAAe,SAAS,IACzB,KAAK,UAAU,OAAO,CAAC,iBACrB,KAAK,eAAe,SAAS,aAAa,UAAU,CACrD,GACD,KAAK;AAEX,MAAI,KAAK,aAAa,SAAS,GAC7B,iBAAiB,eAAe,OAC9B,CAAC,iBAAiB,CAAC,KAAK,aAAa,SAAS,aAAa,UAAU,CACtE;AAGH,MAAI,gBAAgB,aAAa,SAAS,GAAG;GAC3C,gCACE,KAAK,WACL,cACA,2BACD;GACD,iBAAiB,KAAK,UAAU,OAAO,CAAC,iBACtC,aAAa,SAAS,aAAa,UAAU,CAC9C;EACF;AAED,SAAO,4BACL,gBACA,KAAK,eACL,KAAK,uBACL,KAAK,kBACN;CACF;;;;;;CAOD,MAAM,IAAIC,SAAiBC,QAAuB,OAAwB;EAExE,MAAM,MAAM,MAAM,KAAK,cAAc,MAAM,QAAQ;AAEnD,MAAI,UAAU,MACZ,QAAO,KAAK,UAAU,IAAI;AAG5B,MAAI,KAAK,SAAS,EAChB,QAAO,KAAK,UAAU,IAAI,GAAG;AAG/B,SAAO;CACR;CAED,YAAmC;AACjC,SAAO;GACL,OAAO;GACP,sBAAsB,KAAK;GAC3B,gBAAgB,KAAK;GACrB,cAAc,KAAK;GACnB,uBAAuB,KAAK;EAC7B;CACF;;CAGD,aAAa,UAAU;AACrB,MAAI;GACF,MAAM,EAAE,YAAY,GAAG,MAAM,OAAO;AACpC,UAAO,EAAE,WAAY;EACtB,SAAQ,GAAG;GACV,QAAQ,MAAM,EAAE;AAChB,SAAM,IAAI,MACR;EAEH;CACF;AACF"}

package/dist/storage/encoder_backed.cjs

@@ -0,0 +1,94 @@
+const require_rolldown_runtime = require('../_virtual/rolldown_runtime.cjs');
+const __langchain_core_documents = require_rolldown_runtime.__toESM(require("@langchain/core/documents"));
+const __langchain_core_stores = require_rolldown_runtime.__toESM(require("@langchain/core/stores"));
+
+//#region src/storage/encoder_backed.ts
+var encoder_backed_exports = {};
+require_rolldown_runtime.__export(encoder_backed_exports, {
+	EncoderBackedStore: () => EncoderBackedStore,
+	createDocumentStoreFromByteStore: () => createDocumentStoreFromByteStore
+});
+/**
+* Class that provides a layer of abstraction over the base storage,
+* allowing for the encoding and decoding of keys and values. It extends
+* the BaseStore class.
+*/
+var EncoderBackedStore = class extends __langchain_core_stores.BaseStore {
+	lc_namespace = ["langchain", "storage"];
+	store;
+	keyEncoder;
+	valueSerializer;
+	valueDeserializer;
+	constructor(fields) {
+		super(fields);
+		this.store = fields.store;
+		this.keyEncoder = fields.keyEncoder;
+		this.valueSerializer = fields.valueSerializer;
+		this.valueDeserializer = fields.valueDeserializer;
+	}
+	/**
+	* Method to get multiple keys at once. It works with the encoded keys and
+	* serialized values.
+	* @param keys Array of keys to get
+	* @returns Promise that resolves with an array of values or undefined for each key
+	*/
+	async mget(keys) {
+		const encodedKeys = keys.map(this.keyEncoder);
+		const values = await this.store.mget(encodedKeys);
+		return values.map((value) => {
+			if (value === void 0) return void 0;
+			return this.valueDeserializer(value);
+		});
+	}
+	/**
+	* Method to set multiple keys at once. It works with the encoded keys and
+	* serialized values.
+	* @param keyValuePairs Array of key-value pairs to set
+	* @returns Promise that resolves when the operation is complete
+	*/
+	async mset(keyValuePairs) {
+		const encodedPairs = keyValuePairs.map(([key, value]) => [this.keyEncoder(key), this.valueSerializer(value)]);
+		return this.store.mset(encodedPairs);
+	}
+	/**
+	* Method to delete multiple keys at once. It works with the encoded keys.
+	* @param keys Array of keys to delete
+	* @returns Promise that resolves when the operation is complete
+	*/
+	async mdelete(keys) {
+		const encodedKeys = keys.map(this.keyEncoder);
+		return this.store.mdelete(encodedKeys);
+	}
+	/**
+	* Method to yield keys. It works with the encoded keys.
+	* @param prefix Optional prefix to filter keys
+	* @returns AsyncGenerator that yields keys
+	*/
+	async *yieldKeys(prefix) {
+		yield* this.store.yieldKeys(prefix);
+	}
+};
+function createDocumentStoreFromByteStore(store) {
+	const encoder = new TextEncoder();
+	const decoder = new TextDecoder();
+	return new EncoderBackedStore({
+		store,
+		keyEncoder: (key) => key,
+		valueSerializer: (doc) => encoder.encode(JSON.stringify({
+			pageContent: doc.pageContent,
+			metadata: doc.metadata
+		})),
+		valueDeserializer: (bytes) => new __langchain_core_documents.Document(JSON.parse(decoder.decode(bytes)))
+	});
+}
+
+//#endregion
+exports.EncoderBackedStore = EncoderBackedStore;
+exports.createDocumentStoreFromByteStore = createDocumentStoreFromByteStore;
+Object.defineProperty(exports, 'encoder_backed_exports', {
+  enumerable: true,
+  get: function () {
+    return encoder_backed_exports;
+  }
+});
+//# sourceMappingURL=encoder_backed.cjs.map

package/dist/storage/encoder_backed.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"encoder_backed.cjs","names":["BaseStore","fields: {\n    store: BaseStore<string, SerializedType>;\n    keyEncoder: (key: K) => string;\n    valueSerializer: (value: V) => SerializedType;\n    valueDeserializer: (value: SerializedType) => V;\n  }","keys: K[]","keyValuePairs: [K, V][]","encodedPairs: [string, SerializedType][]","prefix?: string | undefined","store: BaseStore<string, Uint8Array>","key: string","doc: Document","bytes: Uint8Array","Document"],"sources":["../../src/storage/encoder_backed.ts"],"sourcesContent":["import { Document } from \"@langchain/core/documents\";\nimport { BaseStore } from \"@langchain/core/stores\";\n\n/**\n * Class that provides a layer of abstraction over the base storage,\n * allowing for the encoding and decoding of keys and values. It extends\n * the BaseStore class.\n */\n// eslint-disable-next-line @typescript-eslint/no-explicit-any\nexport class EncoderBackedStore<K, V, SerializedType = any> extends BaseStore<\n  K,\n  V\n> {\n  lc_namespace = [\"langchain\", \"storage\"];\n\n  store: BaseStore<string, SerializedType>;\n\n  keyEncoder: (key: K) => string;\n\n  valueSerializer: (value: V) => SerializedType;\n\n  valueDeserializer: (value: SerializedType) => V;\n\n  constructor(fields: {\n    store: BaseStore<string, SerializedType>;\n    keyEncoder: (key: K) => string;\n    valueSerializer: (value: V) => SerializedType;\n    valueDeserializer: (value: SerializedType) => V;\n  }) {\n    super(fields);\n    this.store = fields.store;\n    this.keyEncoder = fields.keyEncoder;\n    this.valueSerializer = fields.valueSerializer;\n    this.valueDeserializer = fields.valueDeserializer;\n  }\n\n  /**\n   * Method to get multiple keys at once. It works with the encoded keys and\n   * serialized values.\n   * @param keys Array of keys to get\n   * @returns Promise that resolves with an array of values or undefined for each key\n   */\n  async mget(keys: K[]): Promise<(V | undefined)[]> {\n    const encodedKeys = keys.map(this.keyEncoder);\n    const values = await this.store.mget(encodedKeys);\n    return values.map((value) => {\n      if (value === undefined) {\n        return undefined;\n      }\n      return this.valueDeserializer(value);\n    });\n  }\n\n  /**\n   * Method to set multiple keys at once. It works with the encoded keys and\n   * serialized values.\n   * @param keyValuePairs Array of key-value pairs to set\n   * @returns Promise that resolves when the operation is complete\n   */\n  async mset(keyValuePairs: [K, V][]): Promise<void> {\n    const encodedPairs: [string, SerializedType][] = keyValuePairs.map(\n      ([key, value]) => [this.keyEncoder(key), this.valueSerializer(value)]\n    );\n    return this.store.mset(encodedPairs);\n  }\n\n  /**\n   * Method to delete multiple keys at once. It works with the encoded keys.\n   * @param keys Array of keys to delete\n   * @returns Promise that resolves when the operation is complete\n   */\n  async mdelete(keys: K[]): Promise<void> {\n    const encodedKeys = keys.map(this.keyEncoder);\n    return this.store.mdelete(encodedKeys);\n  }\n\n  /**\n   * Method to yield keys. It works with the encoded keys.\n   * @param prefix Optional prefix to filter keys\n   * @returns AsyncGenerator that yields keys\n   */\n  async *yieldKeys(prefix?: string | undefined): AsyncGenerator<string | K> {\n    yield* this.store.yieldKeys(prefix);\n  }\n}\n\nexport function createDocumentStoreFromByteStore(\n  store: BaseStore<string, Uint8Array>\n) {\n  const encoder = new TextEncoder();\n  const decoder = new TextDecoder();\n  return new EncoderBackedStore({\n    store,\n    keyEncoder: (key: string) => key,\n    valueSerializer: (doc: Document) =>\n      encoder.encode(\n        JSON.stringify({ pageContent: doc.pageContent, metadata: doc.metadata })\n      ),\n    valueDeserializer: (bytes: Uint8Array) =>\n      new Document(JSON.parse(decoder.decode(bytes))),\n  });\n}\n"],"mappings":";;;;;;;;;;;;;;;AASA,IAAa,qBAAb,cAAoEA,kCAGlE;CACA,eAAe,CAAC,aAAa,SAAU;CAEvC;CAEA;CAEA;CAEA;CAEA,YAAYC,QAKT;EACD,MAAM,OAAO;EACb,KAAK,QAAQ,OAAO;EACpB,KAAK,aAAa,OAAO;EACzB,KAAK,kBAAkB,OAAO;EAC9B,KAAK,oBAAoB,OAAO;CACjC;;;;;;;CAQD,MAAM,KAAKC,MAAuC;EAChD,MAAM,cAAc,KAAK,IAAI,KAAK,WAAW;EAC7C,MAAM,SAAS,MAAM,KAAK,MAAM,KAAK,YAAY;AACjD,SAAO,OAAO,IAAI,CAAC,UAAU;AAC3B,OAAI,UAAU,OACZ,QAAO;AAET,UAAO,KAAK,kBAAkB,MAAM;EACrC,EAAC;CACH;;;;;;;CAQD,MAAM,KAAKC,eAAwC;EACjD,MAAMC,eAA2C,cAAc,IAC7D,CAAC,CAAC,KAAK,MAAM,KAAK,CAAC,KAAK,WAAW,IAAI,EAAE,KAAK,gBAAgB,MAAM,AAAC,EACtE;AACD,SAAO,KAAK,MAAM,KAAK,aAAa;CACrC;;;;;;CAOD,MAAM,QAAQF,MAA0B;EACtC,MAAM,cAAc,KAAK,IAAI,KAAK,WAAW;AAC7C,SAAO,KAAK,MAAM,QAAQ,YAAY;CACvC;;;;;;CAOD,OAAO,UAAUG,QAAyD;EACxE,OAAO,KAAK,MAAM,UAAU,OAAO;CACpC;AACF;AAED,SAAgB,iCACdC,OACA;CACA,MAAM,UAAU,IAAI;CACpB,MAAM,UAAU,IAAI;AACpB,QAAO,IAAI,mBAAmB;EAC5B;EACA,YAAY,CAACC,QAAgB;EAC7B,iBAAiB,CAACC,QAChB,QAAQ,OACN,KAAK,UAAU;GAAE,aAAa,IAAI;GAAa,UAAU,IAAI;EAAU,EAAC,CACzE;EACH,mBAAmB,CAACC,UAClB,IAAIC,oCAAS,KAAK,MAAM,QAAQ,OAAO,MAAM,CAAC;CACjD;AACF"}