跳到主要內容

Google 雲端硬碟

本筆記本逐步說明如何將 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 文件資料的說明​

預設情況下，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

特殊值 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

在 Agent 中使用​

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' ?")

此頁面是否有幫助？