使用Google Firestore构建AI应用：保存、加载和删除LangChain文档-CSDN博客

本文链接：https://blog.csdn.net/afTFODguAKBF/article/details/142364231

引言

Cloud Firestore是Google提供的无服务器文档数据库，能够根据需求进行扩展。借助Firestore的LangChain集成，您可以将数据库应用程序扩展到AI驱动的体验。在本文中，我们将探讨如何使用FirestoreLoader和FirestoreSaver在Firestore中保存、加载和删除LangChain文档。

主要内容

1. 环境准备

要开始使用Firestore与LangChain集成，您需要：

创建一个Google Cloud项目
启用Firestore API
创建Firestore数据库

确认您在运行环境中对数据库的访问权限，并配置相关参数。

2. 安装库

使用以下命令安装langchain-google-firestore库：

%pip install -upgrade --quiet langchain-google-firestore

3. 设置Google Cloud项目

请设置您的Google Cloud项目以便可以在笔记本中使用Google Cloud资源。

PROJECT_ID = "my-project-id"  # @param {type:"string"}
!gcloud config set project {PROJECT_ID}

4. 认证

在使用Colab的情况下，可以使用以下代码进行Google Cloud认证：

from google.colab import auth
auth.authenticate_user()

5. 基本使用

保存文档

通过FirestoreSaver将文档保存到Firestore中。

from langchain_core.documents import Document
from langchain_google_firestore import FirestoreSaver

saver = FirestoreSaver()
data = [Document(page_content="Hello, World!")]

saver.upsert_documents(data)

加载文档

使用FirestoreLoader从集合或子集合中加载LangChain文档。

from langchain_google_firestore import FirestoreLoader

loader_collection = FirestoreLoader("Collection")
data_collection = loader_collection.load()

删除文档

从Firestore集合中删除一组LangChain文档。

saver = FirestoreSaver()
saver.delete_documents(data)

代码示例

from langchain_core.documents import Document
from langchain_google_firestore import FirestoreSaver

# 保存文档示例
saver = FirestoreSaver("MyCollection")  # 指定集合，自动生成ID
data = [Document(page_content="Sample Content")]
saver.upsert_documents(data)

# 加载文档示例
from langchain_google_firestore import FirestoreLoader

loader = FirestoreLoader("MyCollection")
documents = loader.load()

for doc in documents:
    print(doc.page_content)

# 删除文档示例
saver.delete_documents(data)