Chroma 报错

Question

from langchain.chains import LLMChain,StuffDocumentsChain
from langchain.document_transformers import (
    LongContextReorder
)
from langchain.embeddings import HuggingFaceBgeEmbeddings
from langchain.vectorstores import  Chroma

#使用huggingface托管的开源LLM来做嵌入，MiniLM-L6-v2是一个较小的LLM
model_path = '/media/data/github/dataroot/models/huggingface/all-MiniLM-L6-v2'
embedings = HuggingFaceBgeEmbeddings(model_name=model_path)

text = [
    "篮球是一项伟大的运动。",
    "带我飞往月球是我最喜欢的歌曲之一。",
    "凯尔特人队是我最喜欢的球队。",
    "这是一篇关于波士顿凯尔特人的文件。",
    "我非常喜欢去看电影。",
    "波士顿凯尔特人队以20分的优势赢得了比赛。",
    "这只是一段随机的文字。",
    "《艾尔登之环》是过去15年最好的游戏之一。",
    "L.科内特是凯尔特人队最好的球员之一。",
    "拉里.伯德是一位标志性的NBA球员。"
]


retrieval = Chroma.from_texts(text,embedings).as_retriever(
    search_kwargs={"k": 10}
)
query = "关于我的喜好都知道什么?"

#根据相关性返回文本块
docs = retrieval.get_relevant_documents(query)
docs

代码报错

Answer 1

tomiezhang 2024-04-18 03:46:57

langchain==0.0.279 openai==0.27.8 

chromadb==0.4.10

检查下这几个包的版本吧，大概率环境依赖问题。

或者直接http://149.28.145.211:8008/notebooks/README.ipynb 线上比较运行学习