diff --git a/populate_database.py b/populate_database.py index 3d2a1ab8a..41ab9ecac 100644 --- a/populate_database.py +++ b/populate_database.py @@ -1,11 +1,11 @@ import argparse import os import shutil -from langchain.document_loaders.pdf import PyPDFDirectoryLoader +from langchain_community.document_loaders import PyPDFDirectoryLoader from langchain_text_splitters import RecursiveCharacterTextSplitter from langchain.schema.document import Document from get_embedding_function import get_embedding_function -from langchain.vectorstores.chroma import Chroma +from langchain_community.vectorstores import Chroma CHROMA_PATH = "chroma" diff --git a/requirements.txt b/requirements.txt index b290b554c..db0bd5b20 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,5 +1,6 @@ pypdf langchain +langchain_community chromadb # Vector storage pytest boto3