本文档介绍了如何在Langchain中跨多个链维护状态，即如何在链之间传递变量。我们将通过一个实际示例，演示如何使用itemgetter在链之间共享变量，确保在后续链中正确引用先前链中的变量。本文将深入探讨Langchain中的状态管理机制，并提供详细的代码示例和解释，帮助开发者更好地理解和应用Langchain的链式调用功能。
在Langchain中构建复杂应用时，经常需要在多个链之间传递信息。一个常见的场景是，一个链的输出作为另一个链的输入，并且需要在后续链中引用先前链的输入变量。本文将介绍如何在Langchain中实现这种跨链状态维护，即如何在链之间传递变量。
使用 itemgetter 传递变量
itemgetter 是 Python 标准库 operator 模块中的一个函数，它可以根据键从字典中提取值。在 Langchain 中，我们可以利用 itemgetter 将先前链中的变量传递到后续链中。
以下是一个示例：
from langchain.prompts import PromptTemplatefrom langchain.schema import StrOutputParserfrom langchain.llms import OpenAI # 确保已安装 langchain-openaifrom operator import itemgetterimport dotenvdotenv.load_dotenv()# 定义第一个 PromptTemplateprompt1 = 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 应用。
以上就是在Langchain中跨链维护状态：变量传递与状态管理教程的详细内容，更多请关注创想鸟其它相关文章！
版权声明：本文内容由互联网用户自发贡献，该文观点仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。
如发现本站有涉嫌抄袭侵权/违法违规的内容， 请发送邮件至 chuangxiangniao@163.com 举报，一经查实，本站将立刻删除。
发布者：程序猿，转转请注明出处：https://www.chuangxiangniao.com/p/1368190.html
微信扫一扫 
支付宝扫一扫 
