github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-06-26 00:23:25 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix multipart email body retrieval (#9790)

Browse Source 
Description: 
Gmail message retrieval in GmailGetMessage and GmailSearch returned an
empty string when encountering multipart emails. This change correctly
extracts the email body for multipart emails.

Dependencies: None

@hwchase17 @vowelparrot
This commit is contained in:
Juhee Kim 2023-09-04 08:04:36 +09:00 committed by GitHub
parent 7d8bb78e5c
commit 50ca44c79f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 20 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
libs/langchain/langchain/tools/gmail/get_message.py
View File

@ -46,7 +46,16 @@ class GmailGetMessage(GmailBaseTool):
subject = email_msg["Subject"]
sender = email_msg["From"]
message_body = email_msg.get_payload()
message_body = ""
if email_msg.is_multipart():
for part in email_msg.walk():
ctype = part.get_content_type()
cdispo = str(part.get("Content-Disposition"))
if ctype == "text/plain" and "attachment" not in cdispo:
message_body = part.get_payload(decode=True).decode("utf-8")
break
else:
message_body = email_msg.get_payload(decode=True).decode("utf-8")
body = clean_email_body(message_body)

11
libs/langchain/langchain/tools/gmail/search.py
View File

@ -91,7 +91,16 @@ class GmailSearch(GmailBaseTool):
subject = email_msg["Subject"]
sender = email_msg["From"]
message_body = email_msg.get_payload()
message_body = ""
if email_msg.is_multipart():
for part in email_msg.walk():
ctype = part.get_content_type()
cdispo = str(part.get("Content-Disposition"))
if ctype == "text/plain" and "attachment" not in cdispo:
message_body = part.get_payload(decode=True).decode("utf-8")
break
else:
message_body = email_msg.get_payload(decode=True).decode("utf-8")
body = clean_email_body(message_body)