HyDE Retriever

此示例展示了如何使用 HyDE Retriever，它实现了如 本文 所述的假设文档嵌入 (HyDE)。

从高层次来看，HyDE 是一种嵌入技术，它接收查询，生成假设答案，然后嵌入生成的文档并将其用作最终示例。

为了使用 HyDE，我们因此需要提供一个基本嵌入模型，以及一个可用于生成这些文档的 LLM。默认情况下，HyDE 类附带一些默认提示供使用（有关详细信息，请参阅本文），但我们也可以创建自己的提示，这些提示应该有一个名为 {question} 的单一输入变量。

用法

提示

请参阅 此部分以获取有关安装集成软件包的通用说明。

npm

npm install @langchain/openai @langchain/core

Yarn

yarn add @langchain/openai @langchain/core

pnpm

pnpm add @langchain/openai @langchain/core

import { OpenAI, OpenAIEmbeddings } from "@langchain/openai";
import { MemoryVectorStore } from "langchain/vectorstores/memory";
import { HydeRetriever } from "langchain/retrievers/hyde";
import { Document } from "@langchain/core/documents";

const embeddings = new OpenAIEmbeddings();
const vectorStore = new MemoryVectorStore(embeddings);
const llm = new OpenAI();
const retriever = new HydeRetriever({
  vectorStore,
  llm,
  k: 1,
});

await vectorStore.addDocuments(
  [
    "My name is John.",
    "My name is Bob.",
    "My favourite food is pizza.",
    "My favourite food is pasta.",
  ].map((pageContent) => new Document({ pageContent }))
);

const results = await retriever.invoke("What is my favourite food?");

console.log(results);
/*
[
  Document { pageContent: 'My favourite food is pasta.', metadata: {} }
]
*/

API 参考

• OpenAI 从 @langchain/openai
• OpenAIEmbeddings 从 @langchain/openai
• MemoryVectorStore 从 langchain/vectorstores/memory
• HydeRetriever 从 langchain/retrievers/hyde
• Document 从 @langchain/core/documents

相关

• 检索器 概念指南
• 检索器 操作指南