探索Google Drive API：快速获取文档的指南

最新推荐文章于 2025-02-14 11:32:20 发布

aehrutktrjk

最新推荐文章于 2025-02-14 11:32:20 发布

阅读量512

点赞数 4

文章标签：服务器运维 python

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

版权

# 探索Google Drive API：快速获取文档的指南

## 引言
Google Drive是一个强大的云端存储服务，允许用户存储和访问各类文档。本篇文章旨在指导你如何使用Google Drive API从Google Drive中检索文档。

## 主要内容

### 准备工作
在动手编码之前，确保你满足以下前提条件：
1. 创建或使用现有的Google Cloud项目。
2. 启用Google Drive API。
3. 为桌面应用授权凭据。
4. 安装必要的Python库：

```bash
pip install --upgrade google-api-python-client google-auth-httplib2 google-auth-oauthlib

配置GoogleDriveRetriever

GoogleDriveRetriever是一个方便的工具，它默认会从~/.credentials/credentials.json读取凭据，但你可以通过环境变量GOOGLE_ACCOUNT_FILE配置不同路径。token.json文件会在首次使用时自动生成。

检索Google文档

你可以通过文件或文件夹ID检索文档。以下示例展示了如何使用langchain_googledrive库获取文档：

from langchain_googledrive.retrievers import GoogleDriveRetriever

folder_id = "root"  # 或者指定具体的文件夹ID

retriever = GoogleDriveRetriever(
    num_results=2,
)

for doc in retriever.invoke("machine learning"):
    print(doc.page_content.strip()[:60] + "...")

MIME类型支持

GoogleDriveRetriever支持多种MIME类型的文档，比如application/pdf和application/vnd.google-apps.document等。

使用自定义模板进行高级搜索

可以使用PromptTemplate进行复杂的自定义查询：

from langchain_core.prompts import PromptTemplate

retriever = GoogleDriveRetriever(
    template=PromptTemplate(
        input_variables=["query"],
        template="(fullText contains '{query}') "
        "and mimeType='application/vnd.google-apps.document' "
        "and modifiedTime > '2000-01-01T00:00:00' "
        "and trashed=false",
    ),
    num_results=2,
    includeItemsFromAllDrives=False,
    supportsAllDrives=False,
)
for doc in retriever.invoke("machine learning"):
    print(f"{doc.metadata['name']}:")
    print(doc.page_content.strip()[:60] + "...")