Skip to main content
本笔记本介绍如何将 LangChain 连接到 Google Drive API。

​
前提条件

  1. 创建或使用已有的 Google Cloud 项目
  2. 启用 Google Drive API
  3. 授权桌面应用的凭据
  4. pip install -U google-api-python-client google-auth-httplib2 google-auth-oauthlib

​
获取 Google 文档数据的说明

默认情况下，GoogleDriveTools 和 GoogleDriveWrapper 期望 credentials.json 文件位于 ~/.credentials/credentials.json，但可以通过设置环境变量 GOOGLE_ACCOUNT_FILE 为 custom/path/to/credentials.json 来自定义路径。 token.json 的位置使用相同的目录（或使用参数 token_path）。注意 token.json 将在首次使用工具时自动创建。 GoogleDriveSearchTool 可以通过一些请求检索选定的文件。 默认情况下，如果使用 folder_id，该文件夹内所有名称与查询匹配的文件都可以被检索为 Document。 
pip install -qU  google-api-python-client google-auth-httplib2 google-auth-oauthlib langchain-community
您可以从 URL 中获取文件夹和文档 ID： 特殊值 root 代表您的个人主目录。 
folder_id = "root"
# folder_id='1yucgL9WGgWZdM1TOuKkeghlPizuzMYb5'
默认情况下，以下 MIME 类型的文件可以转换为 Document：
  • text/text
  • text/plain
  • text/html
  • text/csv
  • text/markdown
  • image/png
  • image/jpeg
  • application/epub+zip
  • application/pdf
  • application/rtf
  • application/vnd.google-apps.document（GDoc）
  • application/vnd.google-apps.presentation（GSlide）
  • application/vnd.google-apps.spreadsheet（GSheet）
  • application/vnd.google.colaboratory（Notebook colab）
  • application/vnd.openxmlformats-officedocument.presentationml.presentation（PPTX）
  • application/vnd.openxmlformats-officedocument.wordprocessingml.document（DOCX）
可以更新或自定义此列表。请参阅 GoogleDriveAPIWrapper 的文档。 但是，需要安装相应的软件包。 
pip install -qU  unstructured langchain-googledrive

import os

from langchain_googledrive.tools.google_drive.tool import GoogleDriveSearchTool
from langchain_googledrive.utilities.google_drive import GoogleDriveAPIWrapper

os.environ["GOOGLE_ACCOUNT_FILE"] = "custom/path/to/credentials.json"

# 默认情况下，仅在文件名中搜索
tool = GoogleDriveSearchTool(
    api_wrapper=GoogleDriveAPIWrapper(
        folder_id=folder_id,
        num_results=2,
        template="gdrive-query-in-folder",  # 在文档正文中搜索
    )
)

import logging

logging.basicConfig(level=logging.INFO)

tool.run("machine learning")

tool.description

"A wrapper around Google Drive Search. Useful for when you need to find a document in google drive. The input should be formatted as a list of entities separated with a space. As an example, a list of keywords is 'hello word'."

​
在 ReAct Agent 中使用工具

要创建使用 Google Drive 工具的 Agent，请安装 LangGraph： 
pip install -qU langgraph langchain-openai
并使用 create_agent 功能初始化 ReAct Agent。您还需要在 platform.openai.com 设置 OPEN_API_KEY 以访问 OpenAI 的聊天模型。 
import os

from langchain.chat_models import init_chat_model
from langchain.agents import create_agent


os.environ["OPENAI_API_KEY"] = "your-openai-api-key"


model = init_chat_model("gpt-4.1-mini", model_provider="openai", temperature=0)
agent = create_agent(model, tools=[tool])

events = agent.stream(
    {"messages": [("user", "Search in google drive, who is 'Yann LeCun' ?")]},
    stream_mode="values",
)
for event in events:
    event["messages"][-1].pretty_print()
Edit this page on GitHub or file an issue.
Connect these docs to Claude, VSCode, and more via MCP for real-time answers.