Fix all langchain.schema imports to langchain_core.documents

project/model/reranking.py

@@ -1,5 +1,5 @@
 from typing import List
-from langchain.schema import Document
+from langchain_core.documents import Document
 from flashrank.Ranker import Ranker, RerankRequest
 from project.utils.config_loader import load_config
 from project.logger.logging import get_logger

project/model/retriever.py

@@ -1,5 +1,5 @@
 from typing import List, Optional
-from langchain.schema import Document
+from langchain_core.documents import Document
 from langchain_community.vectorstores import FAISS
 from langchain.chains.query_constructor.base import AttributeInfo
 from langchain.retrievers.self_query.base import SelfQueryRetriever

project/source/data_preparation.py

@@ -3,7 +3,7 @@ from pathlib import Path
 from typing import List, Optional
 from langchain_community.document_loaders import PyPDFLoader, ArxivLoader
 from langchain_text_splitters import RecursiveCharacterTextSplitter
-from langchain.schema import Document
+from langchain_core.documents import Document
 from project.logger.logging import get_logger
 
 logger = get_logger(__name__)