引言
在构建基于RAG(检索增强生成)的系统时,Chroma数据库是一个非常流行的选择。然而,在使用SentenceTransformer模型进行文本嵌入时,许多开发者会遇到AttributeError: 'SentenceTransformer' object has no attribute 'embed_documents'的错误。本文将通过实例说明如何解决这一问题,并确保你能顺利使用Chroma数据库作为查询引擎。
问题描述
假设我们要构建一个RAG系统,其中我们需要将文本片段嵌入到Chroma数据库中以便后续查询。以下是我们尝试的代码:
# 加载文本片段并声明需要嵌入的列chunks=DataFrameLoader(final_df_for_chroma_injection,page_content_column='TEXT')