Skip to main content
本加载器介绍如何从 Gmail 加载数据。从 Gmail 加载数据有多种方式,本加载器目前对具体方式有较强的偏好。其工作方式是:首先查找您发送的所有消息,然后查找您回复了之前邮件的消息,接着获取该之前的邮件,并创建该邮件加上您的回复邮件的训练示例。 请注意,这里有明显的局限性。例如,所有创建的示例仅将之前的邮件作为上下文。 使用方式:
  • 设置 Google 开发者账户:前往 Google Developer Console,创建一个项目,并为该项目启用 Gmail API。这将为您生成一个 credentials.json 文件,稍后需要用到。
  • 安装 Google 客户端库:运行以下命令安装 Google 客户端库:
pip install -qU  google-auth google-auth-oauthlib google-auth-httplib2 google-api-python-client
import os.path

from google.auth.transport.requests import Request
from google.oauth2.credentials import Credentials
from google_auth_oauthlib.flow import InstalledAppFlow

SCOPES = ["https://www.googleapis.com/auth/gmail.readonly"]


creds = None
# The file token.json stores the user's access and refresh tokens, and is
# created automatically when the authorization flow completes for the first
# time.
if os.path.exists("email_token.json"):
    creds = Credentials.from_authorized_user_file("email_token.json", SCOPES)
# If there are no (valid) credentials available, let the user log in.
if not creds or not creds.valid:
    if creds and creds.expired and creds.refresh_token:
        creds.refresh(Request())
    else:
        flow = InstalledAppFlow.from_client_secrets_file(
            # your creds file here. Please create json file as here https://cloud.google.com/docs/authentication/getting-started
            "creds.json",
            SCOPES,
        )
        creds = flow.run_local_server(port=0)
    # Save the credentials for the next run
    with open("email_token.json", "w") as token:
        token.write(creds.to_json())
from langchain_community.chat_loaders.gmail import GMailLoader
loader = GMailLoader(creds=creds, n=3)
data = loader.load()
# Sometimes there can be errors which we silently ignore
len(data)
2
from langchain_community.chat_loaders.utils import (
    map_ai_messages,
)
# This makes messages sent by hchase@langchain.com the AI Messages
# This means you will train an LLM to predict as if it's responding as hchase
training_data = list(
    map_ai_messages(data, sender="Harrison Chase [hchase@langchain.com](mailto:hchase@langchain.com)")
)