如何构建过滤器

先决条件

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

• 查询分析

我们可能需要进行查询分析以提取过滤器并将其传递到检索器中。我们要求 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" } }
  ]
}

下一步

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

接下来，查看本节中的其他查询分析指南，例如如何使用少量示例来提高性能。