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 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