LangChain中LLMChain弃用警告处理及替代实现代码请求
解决LangChain LLMChain弃用警告的替代实现
修改后的代码实现
首先移除旧的LLMChain导入,改用LangChain核心的管道语法(prompt | llm)实现链式调用:
# 移除 from langchain.chains import LLMChain from langchain_core.prompts import ChatPromptTemplate from langchain_core.runnables import Runnable from langchain_core.output_parsers import StrOutputParser # 可选,用于将模型输出转为字符串 def _create_chain(self, template: str) -> Runnable: prompt = ChatPromptTemplate.from_template(template) # 用管道语法替代LLMChain,若需要直接返回字符串结果,追加StrOutputParser return prompt | self.llm | StrOutputParser() # 若不需要解析,仅返回模型原始ChatMessage对象,可写为:return prompt | self.llm
关键说明
- 管道语法:
prompt | llm是LangChain 0.1+推荐的RunnableSequence实现方式,完全替代旧的LLMChain类 - 返回类型:将返回类型从
LLMChain改为Runnable,因为管道组合后得到的是RunnableSequence对象,属于Runnable子类 - 调用方式不变:修改后的chain仍可使用
invoke()、stream()等方法,比如:result = chain.invoke({"input": "你的查询内容"})
内容的提问来源于stack exchange,提问作者Ammar Ahmed Ansari
相关产品推荐
相关产品推荐

