Replace PyPDF2 with pypdf in long_agent

pull/900/head^2
Pavan Kumar 1 month ago
parent 74715f191b
commit 6f02e4e470

@ -1,7 +1,7 @@
import concurrent.futures import concurrent.futures
import os import os
from typing import Union, List from typing import Union, List
import PyPDF2 import pypdf
import markdown import markdown
from pathlib import Path from pathlib import Path
from swarms.utils.litellm_tokenizer import count_tokens from swarms.utils.litellm_tokenizer import count_tokens
@ -57,7 +57,7 @@ class LongAgent:
text = "" text = ""
with open(file_path, "rb") as file: with open(file_path, "rb") as file:
pdf_reader = PyPDF2.PdfReader(file) pdf_reader = pypdf.PdfReader(file)
for page in pdf_reader.pages: for page in pdf_reader.pages:
text += page.extract_text() text += page.extract_text()

Loading…
Cancel
Save