本文档介绍了如何在Langchain中跨多个链维护状态，即如何在链之间传递变量。我们将通过一个实际示例，演示如何使用itemgetter在链之间共享变量，确保在后续链中正确引用先前链中的变量。本文将深入探讨Langchain中的状态管理机制，并提供详细的代码示例和解释，帮助开发者更好地理解和应用Langchain的链式调用功能。
在Langchain中构建复杂应用时，经常需要在多个链之间传递信息。一个常见的场景是，一个链的输出作为另一个链的输入，并且需要在后续链中引用先前链的输入变量。本文将介绍如何在Langchain中实现这种跨链状态维护，即如何在链之间传递变量。
使用 itemgetter 传递变量
itemgetter 是 Python 标准库 operator 模块中的一个函数，它可以根据键从字典中提取值。在 Langchain 中，我们可以利用 itemgetter 将先前链中的变量传递到后续链中。
以下是一个示例：
from langchain.prompts import PromptTemplate
from langchain.schema import StrOutputParser
from langchain.llms import OpenAI # 确保已安装 langchain-openai
from operator import itemgetter
import dotenv
dotenv.load_dotenv()
# 定义第一个 PromptTemplate
prompt1 = PromptTemplate(
input_variables=["name"],
template="I am {name}. Choose one profession for me in one word. Say AYYE when you do.",
)
# 定义第二个 PromptTemplate，需要 profession 和 name 两个输入变量
prompt2 = PromptTemplate(
input_variables=["profession", "name"],
template="I am a {profession}. Tell me king of pirates, what is my destiny? Refer to me by my given name.",
)
# 初始化 OpenAI 模型
llm = OpenAI()
# 创建第一个链
chain1 = prompt1 | llm | StrOutputParser()
# 创建第二个链，使用 itemgetter 传递 name 变量
chain2 = ({"profession": chain1, "name": itemgetter("name")} | prompt2 | llm | StrOutputParser())
# 调用第二个链，并传入 name 变量
response = chain2.invoke({"name": "Chopper"})
print(response)代码解释:
- itemgetter("name"): 这部分代码创建了一个函数，该函数接收一个字典作为输入，并返回字典中键为 "name" 的值。
- {"profession": chain1, "name": itemgetter("name")}: 这部分代码创建了一个字典，其中 "profession" 键对应的值是 chain1 的输出，而 "name" 键对应的值是使用 itemgetter("name") 从输入字典中提取的 "name" 变量。这样，chain2 就可以同时访问 chain1 的输出和原始的 "name" 变量。
- chain2.invoke({"name": "Chopper"}): 调用 chain2 时，我们传入一个包含 "name" 键的字典。itemgetter("name") 会从这个字典中提取 "Chopper" 作为 "name" 变量的值，并传递给 prompt2。
运行结果示例：
As a doctor, Chopper, your destiny, as proclaimed by the King of Pirates, is to heal and protect your crewmates, ensuring their well-being on their grand adventures.
注意事项:
- 确保 itemgetter 中指定的键存在于输入字典中。如果键不存在，将会抛出 KeyError 异常。
- itemgetter 只能用于提取字典中的值。如果需要进行更复杂的变量转换或处理，可以使用 Lambda 函数或其他自定义函数。
- 在复杂的链式调用中，可以使用多个 itemgetter 来传递多个变量。
总结:
通过使用 itemgetter，我们可以在 Langchain 中轻松地跨多个链传递变量，实现复杂的状态管理。这使得我们可以构建更加灵活和强大的 Langchain 应用。掌握这种技巧对于构建涉及多个步骤的复杂流程至关重要。
希望本教程能够帮助你理解如何在 Langchain 中维护状态并传递变量。通过灵活运用 itemgetter 和其他工具，你可以构建出更加复杂和强大的 Langchain 应用。
