fix: package dependency. (#234)

* fix package * fix lint
microsoft · Aug 31, 2024 · 46be295 · 46be295
1 parent 97cf394
commit 46be295
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 6 deletions.
diff --git a/rdagent/components/document_reader/document_reader.py b/rdagent/components/document_reader/document_reader.py
@@ -8,11 +8,7 @@
 import requests
 from azure.ai.formrecognizer import DocumentAnalysisClient
 from azure.core.credentials import AzureKeyCredential
-from langchain.document_loaders import (
-    OnlinePDFLoader,
-    PyPDFDirectoryLoader,
-    PyPDFLoader,
-)
+from langchain_community.document_loaders import PyPDFDirectoryLoader, PyPDFLoader
 from PIL import Image
 
 if TYPE_CHECKING:

diff --git a/requirements.txt b/requirements.txt
@@ -23,6 +23,7 @@ pandarallel # parallelize pandas
 feedparser
 matplotlib
 langchain
+langchain-community
 tiktoken
 pymupdf  # Extract shotsreens from pdf
 
@@ -68,4 +69,4 @@ st-theme
 
 # kaggle crawler
 selenium
-kaggle
+kaggle