gmakstutis/cognee
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add type hint in extract_text_from_file

Browse source
This commit is contained in:
Leon Luithlen 2024-11-04 17:24:46 +01:00
parent cb205069bc
commit 5b07a0a3d2
1 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
cognee/infrastructure/files/utils/extract_text_from_file.py
View file

@ -1,7 +1,8 @@
from typing import BinaryIO from typing import BinaryIO
from pypdf import PdfReader from pypdf import PdfReader
import filetype
def extract_text_from_file(file: BinaryIO, file_type) -> str: def extract_text_from_file(file: BinaryIO, file_type: filetype.Type) -> str:
"""Extract text from a file""" """Extract text from a file"""
if file_type.extension == "pdf": if file_type.extension == "pdf":
reader = PdfReader(stream = file) reader = PdfReader(stream = file)