klas/t6_mem0
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[Bug fix] Fix missing dependency issue with gmail (#862)

Browse Source
This commit is contained in:
Deshraj Yadav
2023-10-27 20:02:03 -07:00
committed by GitHub
parent f6c4f86986
commit 29bd038579
3 changed files with 13 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
embedchain/loaders/gmail.py
View File

@@ -7,9 +7,9 @@ from textwrap import dedent
from bs4 import BeautifulSoup
try:
from llama_index import download_loader
from llama_hub.gmail.base import GmailReader
except ImportError:
raise ImportError("Notion requires extra dependencies. Install with `pip install embedchain[community]`") from None
raise ImportError("Gmail requires extra dependencies. Install with `pip install embedchain[gmail]`") from None
from embedchain.loaders.base_loader import BaseLoader
from embedchain.utils import clean_string
@@ -32,7 +32,6 @@ class GmailLoader(BaseLoader):
dev account. Refer this `https://cloud.google.com/docs/authentication/api-keys`"
)
GmailReader = download_loader("GmailReader")
loader = GmailReader(query=query, service=None, results_per_page=20)
documents = loader.load_data()
logging.info(f"Gmail Loader: {len(documents)} mails found for query- {query}")