langchain/docs/extras/integrations/tools/google_drive.ipynb

Google Drive

This notebook walks through connecting a LangChain to the Google Drive API.

Prerequisites

1. Create a Google Cloud project or use an existing project
2. Enable the Google Drive API
3. Authorize credentials for desktop app
4. pip install --upgrade google-api-python-client google-auth-httplib2 google-auth-oauthlib

Instructions for retrieving your Google Docs data

By default, the GoogleDriveTools and GoogleDriveWrapper expects the credentials.json file to be ~/.credentials/credentials.json, but this is configurable using the GOOGLE_ACCOUNT_FILE environment variable. The location of token.json use the same directory (or use the parameter token_path). Note that token.json will be created automatically the first time you use the tool.

GoogleDriveSearchTool can retrieve a selection of files with some requests.

By default, If you use a folder_id, all the files inside this folder can be retrieved to Document, if the name match the query.

In [ ]:

#!pip install --upgrade google-api-python-client google-auth-httplib2 google-auth-oauthlib

You can obtain your folder and document id from the URL:

• Folder: https://drive.google.com/drive/u/0/folders/1yucgL9WGgWZdM1TOuKkeghlPizuzMYb5 -> folder id is "1yucgL9WGgWZdM1TOuKkeghlPizuzMYb5"
• Document: https://docs.google.com/document/d/1bfaMQ18_i56204VaQDVeAFpqEijJTgvurupdEDiaUQw/edit -> document id is "1bfaMQ18_i56204VaQDVeAFpqEijJTgvurupdEDiaUQw"

The special value root is for your personal home.

In [ ]:

folder_id="root"
#folder_id='1yucgL9WGgWZdM1TOuKkeghlPizuzMYb5'

By default, all files with these mime-type can be converted to 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)

It's possible to update or customize this. See the documentation of GoogleDriveAPIWrapper.

But, the corresponding packages must installed.

In [ ]:

#!pip install unstructured

In [ ]:

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

# 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
    )
)

In [ ]:

import logging
logging.basicConfig(level=logging.INFO)

In [ ]:

tool.run("machine learning")

In [ ]:

tool.description

In [ ]:

from langchain.agents import load_tools
tools = load_tools(["google-drive-search"],
                   folder_id=folder_id,
                   template="gdrive-query-in-folder",
                  )

Use within an Agent

In [ ]:

from langchain.llms import OpenAI
from langchain.agents import initialize_agent, AgentType
llm = OpenAI(temperature=0)
agent = initialize_agent(
    tools=tools,
    llm=llm,
    agent=AgentType.STRUCTURED_CHAT_ZERO_SHOT_REACT_DESCRIPTION,
)

In [ ]:

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