From 44d06c2f2c3e851d5e904f5cbf2ab5cb9af9ee74 Mon Sep 17 00:00:00 2001
From: you-n-g <you-n-g@users.noreply.github.com>
Date: Sat, 31 Aug 2024 20:35:04 +0800
Subject: [PATCH] fix: package dependency. (#234)

* fix package

* fix lint
---
 rdagent/components/document_reader/document_reader.py | 6 +-----
 requirements.txt                                      | 3 ++-
 2 files changed, 3 insertions(+), 6 deletions(-)

diff --git a/rdagent/components/document_reader/document_reader.py b/rdagent/components/document_reader/document_reader.py
index cfb06105..46e57d6b 100644
--- a/rdagent/components/document_reader/document_reader.py
+++ b/rdagent/components/document_reader/document_reader.py
@@ -8,11 +8,7 @@ import fitz
 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
index 415d3fe3..e2eb6a08 100644
--- 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
\ No newline at end of file
+kaggle