谷歌 Drive

此笔记本介绍如何将 LangChain 连接到 Google Drive API。

先决条件

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

检索 Google Docs 数据的说明

默认情况下，GoogleDriveTools 和 GoogleDriveWrapper 预计 credentials.json 文件位于 ~/.credentials/credentials.json，但可以通过 GOOGLE_ACCOUNT_FILE 环境变量进行配置。token.json 的位置使用相同的目录（或使用参数 token_path）。请注意，token.json 将在您第一次使用该工具时自动创建。

GoogleDriveSearchTool 可以通过一些请求检索选定的文件。

默认情况下，如果您使用 folder_id，则该文件夹中的所有文件都将被检索到 Document 中，前提是文件名与查询匹配。

%pip install --upgrade --quiet  google-api-python-client google-auth-httplib2 google-auth-oauthlib langchain-community

您可以从 URL 获取您的文件夹和文档 ID

• 文件夹：https://drive.google.com/drive/u/0/folders/1yucgL9WGgWZdM1TOuKkeghlPizuzMYb5 -> 文件夹 ID 是 "1yucgL9WGgWZdM1TOuKkeghlPizuzMYb5"
• 文档：https://docs.google.com/document/d/1bfaMQ18_i56204VaQDVeAFpqEijJTgvurupdEDiaUQw/edit -> 文档 ID 是 "1bfaMQ18_i56204VaQDVeAFpqEijJTgvurupdEDiaUQw"

特殊值 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 --upgrade --quiet  unstructured

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

# By default, search only in the filename.
tool = GoogleDriveSearchTool(
    api_wrapper=GoogleDriveAPIWrapper(
        folder_id=folder_id,
        num_results=2,
        template="gdrive-query-in-folder",  # Search in the body of documents
    )
)

import logging

logging.basicConfig(level=logging.INFO)

tool.run("machine learning")

tool.description

from langchain.agents import load_tools

tools = load_tools(
    ["google-drive-search"],
    folder_id=folder_id,
    template="gdrive-query-in-folder",
)

API 参考：load_tools

在代理中使用

from langchain.agents import AgentType, initialize_agent
from langchain_openai import OpenAI

llm = OpenAI(temperature=0)
agent = initialize_agent(
    tools=tools,
    llm=llm,
    agent=AgentType.STRUCTURED_CHAT_ZERO_SHOT_REACT_DESCRIPTION,
)

API 参考：AgentType | initialize_agent | OpenAI

agent.run("Search in google drive, who is 'Yann LeCun' ?")

相关

• 工具 概念指南
• 工具 操作指南