LangChain导入HuggingFacePipeline时报TypeError错误求助

问题详情 #

运行以下Python代码时：

from transformers import pipeline
import langchain
from langchain.llms import HuggingFacePipeline

model_name = "bert-base-uncased"
task = "question-answering"

执行到from langchain.llms import HuggingFacePipeline行时出现错误：

TypeError: issubclass() arg 1 must be a class

运行环境为Databricks集群，配置如下：

• Runtime：13.2 ML（包含Apache Spark 3.4.0、GPU、Scala 2.12）
• Worker & Driver类型：Standard_NC21s_v3，224GB内存、2块GPU
• 已安装对应依赖库

解决方案 #

• 对齐langchain与transformers版本：该错误多因版本不兼容导致，Databricks默认预装的库版本可能存在冲突。执行以下命令安装经过验证的兼容版本：

  pip install langchain==0.0.350 transformers==4.35.2
  

• 修正导入路径：langchain 0.1.x版本后，HuggingFacePipeline组件迁移至langchain_community库，若使用新版本langchain，需修改导入语句：

  from langchain_community.llms import HuggingFacePipeline
  

• 重启集群：安装完新库版本后，必须重启Databricks集群，确保环境依赖完全更新，避免旧版本残留干扰。

内容的提问来源于stack exchange，提问作者Dolev Mitz