Langchain 使用LCEL实现时，如何保留中间数据

问题

目前在 LangChain 0.3 版本中，LCEL (LangChain Expression Language) 是一个新功能，它通过声明性方式定义链式调用、模型推理和数据流。对于很多常见的任务，它可以极大地简化链条配置。
例如，我们可以利用LCEL语法轻松的将不同组件连接起来(类似于管道），形成一个完整的链条，让程序顺着链条一步一步执行下去。

 chain = step1_chain | step2_chain | step3_chain
 chain.invoke(x)  #执行链条

执行这样一个简单的链条可以直接获得最终的结果，但是如果想保留中间每一步计算的结果，应该如何修改呢？

解决方法

答案就是使用RunnablePassthrough.assign。
查看官方文档，可以找到这两段描述

In some cases, it may be useful to pass the input through while adding some keys to the output. In this case, you can use the assign method

If you’d like to mimic the dict packaging of input and output in LLMChain, you can use a RunnablePassthrough.assign

看一个具体的例子，这个例子对用户输入的问题进行实体识别，然后对识别的结果进行实体链接，同时完成schema的获取； 最后，利用这些中间结果，将用户问题改写为针对neo4j数据库的查询语句。

cypher_response = (
    RunnablePassthrough.assign(names=entity_chain)
    | RunnablePassthrough.assign(
        entities_list=lambda x: map_to_database(x['names']), 
        schema=lambda _: graph.get_schema
    )
    | RunnablePassthrough.assign(
        cypher = (cypher_prompt | llm | StrOutputParser()) ) # 将嵌套的cypher处理逻辑直接放在一起
)

这样调用模型的输出包含了所有的中间结果：

{'question': '发现某化工园区多种化学品泄露，园区内主要存储的是乙醇、硫酸、氯气和过氧化氢，需要考虑哪些风险因素',
 'names': Entities(names=['乙醇', '硫酸', '氯气', '过氧化氢', '化工园区', '化学品泄露', '风险因素']),
 'entities_list': '乙醇 maps to 乙醇(化学品)/溶于乙醇(理化特性)/无水乙醇(化学品) in database\n硫酸 maps to 硫酸(化学品)/亚硫酸(化学品)/发烟硫酸(化学品) in database\n氯气 maps to 强氯气味(理化特性) in database\n过氧化氢 maps to 过氧化氢(化学品) in database\n',
 'schema': 'Node properties:\n安全隐患 {id: STRING, name: STRING, wcc: INTEGER}\n组织机构 {wcc: INTEGER, id: STRING, name: STRING}\n职务 {wcc: INTEGER, id: STRING, name: STRING}\n安全措施 {wcc: INTEGER, id: STRING, name: STRING}\n\n急救措施 {name: STRING, id: STRING, wcc: INTEGER}\n泄漏应急处理 {name: STRING, id: STRING, wcc: INTEGER}\n化学品英文名 {name: STRING, id: STRING, wcc: INTEGER}\n温度 {name: STRING, wcc: INTEGER}\n相对湿度 {name: STRING, wcc: INTEGER}\n慢性影响 {wcc: INTEGER, name: STRING, id: STRING}\n活性反应 {name: STRING, id: STRING, wcc: INTEGER}\n灭火剂 {id: STRING, name: STRING, wcc: INTEGER}\nRelationship properties:\n\nThe relationships:\n(:安全隐患)-[:MENTIONS]->(:职务)\n(:安全隐患)-[:MENTIONS]->(:组织机构)\n(:安全隐患)-[:MENTIONS]->(:安全措施)\n(:安全隐患)-[:MENTIONS]->(:危险源)\n(:安全隐患)-[:MENTIONS]->(:区域)\n(:安全隐患)-[:MENTIONS]->(:活动操作)\n(:安全隐患)-[:MENTIONS]->(:设备设施)\n(:安全隐患)-[:MENTIONS]->(:用途)\n(:安全隐患)-[:MENTIONS]->(:化学品类别)\n(:安全隐患)-[:MENTIONS]->(:化学品)\n(:安全隐患)-[:MENTIONS]->(:灭火剂)\n(:安全隐患)-[:MENTIONS]->(:禁忌物)\n(:安全隐患)-[:MENTIONS]->(:中毒表现)\n(:化学品类别)-[:包含]->(:化学品)\n(:化学品类别)-[:包含]->(:禁忌物)\n(:化学品类别)-[:禁忌关系_可以混存]->(:危险性类别)\n(:化学品类别)-[:禁忌关系_可以混存]->(:危险性说明)\n(:化学品类别)-[:禁忌关系_应隔离存储]->(:化学品类别)\n(:化学品类别)-[:别名]->(:化学品)\n(:化学品类别)-[:危险性类别]->(:危险性类别)\n(:化学品类别)-[:危险性类别]->(:燃烧与爆炸危险性)\n(:化学品类别)-[:危险性类别]->(:化学品)\n(:化学品类别)-[:外观与性状]->(:理化特性)\n(:化学品类别)-[:熔点]->(:理化特性)\n(:化学品类别)-[:溶解性]->(:理化特性)\n(:化学品类别)-[:禁忌物]->(:化学品类别)\n(:化学品类别)-[:禁忌物]->(:禁忌物)\n(:化学品类别)-[:禁忌物]->(:化学品)',
 'cypher': '```cypher\nMATCH (c1:化学品 {name: "乙醇"})-[:禁忌物|:灭火方法(适用)|:灭火方法(禁用)|:燃烧与爆炸危险性|:应急措施|:泄漏应急处理|:急救措施|:危险性说明|:危险性类别|:环境影响|:中毒表现|:理化特性|:毒性|:存储条件]->(riskFactor)\nMATCH (c2:化学品 {name: "硫酸"})-[:禁忌物|:灭火方法(适用)|:灭火方法(禁用)|:燃烧与爆炸危险性|:应急措施|:泄漏应急处理|:急救措施|:危险性说明|:危险性类别|:环境影响|:中毒表现|:理化特性|:毒性|:存储条件]->(riskFactor)\nMATCH (c3:化学品 {name: "氯气"})-[:禁忌物|:灭火方法(适用)|:灭火方法(禁用)|:燃烧与爆炸危险性|:应急措施|:泄漏应急处理|:急救措施|:危险性说明|:危险性类别|:环境影响|:中毒表现|:理化特性|:毒性|:存储条件]->(riskFactor)\nMATCH (c4:化学品 {name: "过氧化氢"})-[:禁忌物|:灭火方法(适用)|:灭火方法(禁用)|:燃烧与爆炸危险性|:应急措施|:泄漏应急处理|:急救措施|:危险性说明|:危险性类别|:环境影响|:中毒表现|:理化特性|:毒性|:存储条件]->(riskFactor)\nRETURN DISTINCT riskFactor\n```'
 }

Reference

RunnablePassthrough
Migrating from LLMChain