Google Firestore（原生模式）
Firestore 是一个无服务器的面向文档的数据库，能够扩展以满足任何需求。利用 Firestore 的 Langchain 集成，扩展您的数据库应用程序以构建由 AI 驱动的体验。
本笔记本介绍如何使用Firestore来存储向量，并使用FirestoreVectorStore类进行查询。
开始之前
要运行此笔记本，您需要执行以下操作：
在确认可以访问此笔记本运行时环境中的数据库后，填写以下值并在运行示例脚本之前运行单元格。
# @markdown Please specify a source for demo purpose.
COLLECTION_NAME = "test" # @param {type:"CollectionReference"|"string"}
🦜🔗 库安装
集成位于其自己的langchain-google-firestore包中，因此我们需要安装它。对于这个笔记本，我们还将安装langchain-google-genai以使用Google生成式AI嵌入。
%pip install -upgrade --quiet langchain-google-firestore langchain-google-vertexai
仅限Colab：取消注释以下单元格以重新启动内核，或使用按钮重新启动内核。对于Vertex AI Workbench，您可以使用顶部的按钮重新启动终端。
# # Automatically restart kernel after installs so that your environment can access the new packages
# import IPython
# app = IPython.Application.instance()
# app.kernel.do_shutdown(True)
☁ 设置您的Google Cloud项目
设置您的Google Cloud项目，以便您可以在此笔记本中利用Google Cloud资源。
如果您不知道您的项目ID，请尝试以下操作：
- 运行
gcloud config list。 - 运行
gcloud projects list。 - 查看支持页面：Locate the project ID。
# @markdown Please fill in the value below with your Google Cloud project ID and then run the cell.
PROJECT_ID = "extensions-testing" # @param {type:"string"}
# Set the project id
!gcloud config set project {PROJECT_ID}
🔐 认证
以登录此笔记本的IAM用户身份验证到Google Cloud，以便访问您的Google Cloud项目。
- 如果您正在使用Colab运行此笔记本，请使用下面的单元格并继续。
- 如果您正在使用Vertex AI Workbench，请查看这里的设置说明。
from google.colab import auth
auth.authenticate_user()
基本用法
初始化 FirestoreVectorStore
FirestoreVectorStore 允许您在 Firestore 数据库中存储新的向量。您可以使用它来存储来自任何模型的嵌入，包括来自 Google Generative AI 的嵌入。
from langchain_google_firestore import FirestoreVectorStore
from langchain_google_vertexai import VertexAIEmbeddings
embedding = VertexAIEmbeddings(
model_name="textembedding-gecko@latest",
project=PROJECT_ID,
)
# Sample data
ids = ["apple", "banana", "orange"]
fruits_texts = ['{"name": "apple"}', '{"name": "banana"}', '{"name": "orange"}']
# Create a vector store
vector_store = FirestoreVectorStore(
collection="fruits",
embedding=embedding,
)
# Add the fruits to the vector store
vector_store.add_texts(fruits_texts, ids=ids)
作为简写，您可以使用from_texts和from_documents方法在单个步骤中初始化和添加向量。
vector_store = FirestoreVectorStore.from_texts(
collection="fruits",
texts=fruits_texts,
embedding=embedding,
)
from langchain_core.documents import Document
fruits_docs = [Document(page_content=fruit) for fruit in fruits_texts]
vector_store = FirestoreVectorStore.from_documents(
collection="fruits",
documents=fruits_docs,
embedding=embedding,
)
删除向量
您可以使用delete方法从数据库中删除带有向量的文档。您需要提供要删除的向量的文档ID。这将从数据库中删除整个文档，包括它可能具有的任何其他字段。
vector_store.delete(ids)
更新向量
更新向量与添加向量类似。您可以使用add方法通过提供文档ID和新向量来更新文档的向量。
fruit_to_update = ['{"name": "apple","price": 12}']
apple_id = "apple"
vector_store.add_texts(fruit_to_update, ids=[apple_id])
相似性搜索
你可以使用FirestoreVectorStore对你存储的向量进行相似性搜索。这对于查找相似的文档或文本非常有用。
vector_store.similarity_search("I like fuji apples", k=3)
vector_store.max_marginal_relevance_search("fuji", 5)
你可以通过使用filters参数来为搜索添加一个预过滤器。这对于按特定字段或值进行过滤非常有用。
from google.cloud.firestore_v1.base_query import FieldFilter
vector_store.max_marginal_relevance_search(
"fuji", 5, filters=FieldFilter("content", "==", "apple")
)
自定义连接和认证
from google.api_core.client_options import ClientOptions
from google.cloud import firestore
from langchain_google_firestore import FirestoreVectorStore
client_options = ClientOptions()
client = firestore.Client(client_options=client_options)
# Create a vector store
vector_store = FirestoreVectorStore(
collection="fruits",
embedding=embedding,
client=client,
)