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

Refactor: Email parser use with to handle buffer (#11496)

Browse source 
### What problem does this PR solve?
 Email parser use with to handle buffer

### Type of change

- [x] Refactoring
This commit is contained in:
Stephen Hu 2025-11-25 10:03:37 +08:00 • committed by GitHub
parent d1744aaaf3
commit 41665b0865
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 4 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
rag/app/email.py
View file

@ -51,9 +51,11 @@ def chunk(
attachment_res = []
if binary:
msg = BytesParser(policy=policy.default).parse(io.BytesIO(binary))
with io.BytesIO(binary) as buffer:
msg = BytesParser(policy=policy.default).parse(buffer)
else:
msg = BytesParser(policy=policy.default).parse(open(filename, "rb"))
with open(filename, "rb") as buffer:
msg = BytesParser(policy=policy.default).parse(buffer)
text_txt, html_txt = [], []
# get the email header info