如何构建过滤器

先决条件

本指南假设您熟悉以下内容

• 查询分析

我们可能希望进行查询分析以提取过滤器以传递到检索器。我们让 LLM 表示这些过滤器的其中一种方式是作为 Zod 架构。然后，问题就变成了如何将 Zod 架构转换为可以传递到检索器的过滤器。

这可以手动完成，但 LangChain 也提供了一些“翻译器”，它们能够将常见语法转换为特定于每个检索器的过滤器。在这里，我们将介绍如何使用这些翻译器。

设置

安装依赖项

npm

npm i @langchain/core zod

yarn

yarn add @langchain/core zod

pnpm

pnpm add @langchain/core zod

在本示例中，year 和 author 都是要过滤的属性。

import { z } from "zod";

const searchSchema = z.object({
  query: z.string(),
  startYear: z.number().optional(),
  author: z.string().optional(),
});

const searchQuery: z.infer<typeof searchSchema> = {
  query: "RAG",
  startYear: 2022,
  author: "LangChain",
};

import { Comparison, Comparator } from "langchain/chains/query_constructor/ir";

function constructComparisons(
  query: z.infer<typeof searchSchema>
): Comparison[] {
  const comparisons: Comparison[] = [];
  if (query.startYear !== undefined) {
    comparisons.push(
      new Comparison("gt" as Comparator, "start_year", query.startYear)
    );
  }
  if (query.author !== undefined) {
    comparisons.push(
      new Comparison("eq" as Comparator, "author", query.author)
    );
  }
  return comparisons;
}

const comparisons = constructComparisons(searchQuery);

import { Operation, Operator } from "langchain/chains/query_constructor/ir";

const _filter = new Operation("and" as Operator, comparisons);

import { ChromaTranslator } from "@langchain/community/structured_query/chroma";

new ChromaTranslator().visitOperation(_filter);

{
  "$and": [
    { start_year: { "$gt": 2022 } },
    { author: { "$eq": "LangChain" } }
  ]
}

后续步骤

您现在已经了解了如何从任意查询创建特定过滤器。

接下来，查看本节中其他一些查询分析指南，例如 如何使用少样本学习来提高性能。