Replace PyPDF2 with pypdf in long_agent

pull/900/head
Pavan Kumar 1 month ago committed by GitHub
commit b307370e0a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

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

Loading…
Cancel
Save