混合搜索

LangChain 中的标准搜索通过向量相似度完成。然而，许多向量存储实现（Astra DB、ElasticSearch、Neo4J、AzureSearch、Qdrant...）也支持更高级的搜索，结合向量相似度搜索与其他搜索技术（全文搜索、BM25等）。这通常被称为“混合”搜索。

步骤 1：确保您使用的向量存储支持混合搜索

目前，LangChain 中没有统一的方法执行混合搜索。每个向量存储库可能有其各自的方式。这通常作为关键字参数在 similarity_search 期间传入。

通过阅读文档或源代码，确定您使用的向量存储是否支持混合搜索，如果支持，请了解如何使用它。

步骤 2：将该参数作为链的可配置字段添加

这将使您能够轻松调用链，并在运行时配置任何相关标志。有关配置的更多信息，请参阅此文档。

步骤 3：使用该可配置字段调用链

现在，在运行时您可以使用可配置的字段调用此链。

代码示例

让我们通过一个具体的代码示例来了解其实际样子。在本例中，我们将使用 Astra DB 的 Cassandra/CQL 接口。

安装以下 Python 包：

!pip install "cassio>=0.1.7"

获取连接密钥。

初始化 cassio:

import cassio

cassio.init(
    database_id="Your database ID",
    token="Your application token",
    keyspace="Your key space",
)

使用标准的索引分析器创建 Cassandra VectorStore。需要索引分析器以启用术语匹配。

from cassio.table.cql import STANDARD_ANALYZER
from langchain_community.vectorstores import Cassandra
from langchain_openai import OpenAIEmbeddings

embeddings = OpenAIEmbeddings()
vectorstore = Cassandra(
    embedding=embeddings,
    table_name="test_hybrid",
    body_index_options=[STANDARD_ANALYZER],
    session=None,
    keyspace=None,
)

vectorstore.add_texts(
    [
        "In 2023, I visited Paris",
        "In 2022, I visited New York",
        "In 2021, I visited New Orleans",
    ]
)

API 参考:Cassandra |OpenAI 嵌入

如果我们执行标准的相似度搜索，我们会得到所有文档：

vectorstore.as_retriever().invoke("What city did I visit last?")

[Document(page_content='In 2022, I visited New York'),
Document(page_content='In 2023, I visited Paris'),
Document(page_content='In 2021, I visited New Orleans')]

Astra DB 向量存储的 body_search 参数可用于在术语 new 上过滤搜索结果。

vectorstore.as_retriever(search_kwargs={"body_search": "new"}).invoke(
    "What city did I visit last?"
)

[Document(page_content='In 2022, I visited New York'),
Document(page_content='In 2021, I visited New Orleans')]

我们现在可以创建用于问答的链了

from langchain_core.output_parsers import StrOutputParser
from langchain_core.prompts import ChatPromptTemplate
from langchain_core.runnables import (
    ConfigurableField,
    RunnablePassthrough,
)
from langchain_openai import ChatOpenAI

API 参考:StrOutputParser |聊天提示模板 |可配置字段 |可运行传递器 |ChatOpenAI

这是基本的问答链设置。

template = """Answer the question based only on the following context:
{context}
Question: {question}
"""
prompt = ChatPromptTemplate.from_template(template)

model = ChatOpenAI()

retriever = vectorstore.as_retriever()

在此我们标记检索器具有一个可配置字段。所有向量存储检索器都具有 search_kwargs 作为字段。这只是一个字典，包含与向量存储特定的字段

configurable_retriever = retriever.configurable_fields(
    search_kwargs=ConfigurableField(
        id="search_kwargs",
        name="Search Kwargs",
        description="The search kwargs to use",
    )
)

我们现在可以使用我们可配置的检索器创建链

chain = (
    {"context": configurable_retriever, "question": RunnablePassthrough()}
    | prompt
    | model
    | StrOutputParser()
)

chain.invoke("What city did I visit last?")

Paris

我们现在可以使用可配置选项调用链。search_kwargs是可配置字段的ID。该值是将用于Astra DB的搜索参数。

chain.invoke(
    "What city did I visit last?",
    config={"configurable": {"search_kwargs": {"body_search": "new"}}},
)

New York

代码示例​

代码示例