Use backwards-compatible langchain.chains import for RetrievalQA

medrax/rag/rag.py

@@ -9,7 +9,7 @@ from langchain_community.document_loaders import TextLoader, PyPDFLoader, Docx2t
 from langchain_text_splitters import RecursiveCharacterTextSplitter
 from langchain_pinecone import PineconeVectorStore
 from pinecone import Pinecone
-from langchain_community.chains import RetrievalQA
+from langchain.chains import RetrievalQA
 from langchain_community.chat_message_histories import ChatMessageHistory
 from langchain_core.runnables.history import RunnableWithMessageHistory
 from langchain_core.retrievers import BaseRetriever

medrax/tools/rag.py

@@ -1,6 +1,6 @@
 from langchain_core.tools import BaseTool
 from medrax.rag.rag import RAGConfig, CohereRAG
-from langchain_community.chains import RetrievalQA
+from langchain.chains import RetrievalQA
 from typing import Dict, Tuple, Any