Skip to main content
Open In ColabOpen on GitHub

Google Cloud端硬盘

这个笔记本将介绍如何连接一个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

Instructions for retrieving your Google Docs data​

默认情况下，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 --upgrade --quiet  google-api-python-client google-auth-httplib2 google-auth-oauthlib langchain-community

您可以通过URL获取您的文件夹和文档ID：

The special value root is for your personal home.

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
  • 应用程序/rtf
  • application/vnd.google-apps.document (GDoc)
  • application/vnd.google-apps.presentation (谷歌幻灯片)
  • 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 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"

# 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
"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代理中​

要在创建使用Google Jobs工具的代理之前，请安装Langchain

%pip install --upgrade --quiet langgraph langchain-openai

并使用create_react_agent功能初始化一个ReAct代理。您还需要设置您的OPEN_API_KEY（访问https://platform.openai.com）以访问OpenAI的聊天模型。

import os

from langchain.chat_models import init_chat_model
from langgraph.prebuilt import create_react_agent

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


llm = init_chat_model("gpt-4o-mini", model_provider="openai", temperature=0)
agent = create_react_agent(llm, 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()