[使用Google Cloud SQL for PostgreSQL存储聊天消息历史的完整指南]

用户026700199494
103 阅读3分钟 
# 使用Google Cloud SQL for PostgreSQL存储聊天消息历史的完整指南

## 引言

在现代应用开发中，管理和存储聊天消息历史是一个常见的需求。在本篇文章中，我们将探讨如何使用Google Cloud SQL for PostgreSQL来高效管理聊天记录。我们将介绍PostgresChatMessageHistory类的使用，并结合代码示例，帮助你快速掌握这一技术。

## 主要内容

### 1. 设置Google Cloud环境

在开始之前，确保你已创建一个Google Cloud项目，并启用了Cloud SQL Admin API。以下是基本的设置步骤：
- 创建Google Cloud项目
- 启用Cloud SQL Admin API
- 创建Cloud SQL for PostgreSQL实例
- 创建数据库并添加IAM数据库用户（可选）

### 2. 安装必要库

我们需要安装`langchain-google-cloud-sql-pg`和`langchain-google-vertexai`库来进行集成：
```bash
%pip install --upgrade --quiet langchain-google-cloud-sql-pg langchain-google-vertexai

3. 认证和项目设置

确保使用合适的Google Cloud账户进行认证，并设置项目ID：

from google.colab import auth
auth.authenticate_user()

# 设置项目ID
PROJECT_ID = "my-project-id"
!gcloud config set project {PROJECT_ID}

4. 启用API

启用必要的API服务，如Cloud SQL Admin API和Vertex AI API：

!gcloud services enable sqladmin.googleapis.com
!gcloud services enable aiplatform.googleapis.com

5. 连接PostgresEngine

创建PostgresEngine实例用于连接数据库：

from langchain_google_cloud_sql_pg import PostgresEngine

engine = PostgresEngine.from_instance(
    project_id=PROJECT_ID, 
    region="us-central1", 
    instance="my-postgresql-instance", 
    database="my-database"
)
# 使用API代理服务提高访问稳定性

6. 初始化聊天记录表

使用PostgresEngine创建存储聊天消息的表：

engine.init_chat_history_table(table_name="message_store")

7. 使用PostgresChatMessageHistory

创建PostgresChatMessageHistory实例并添加消息：

from langchain_google_cloud_sql_pg import PostgresChatMessageHistory

history = PostgresChatMessageHistory.create_sync(
    engine, 
    session_id="test_session", 
    table_name="message_store"
)

history.add_user_message("hi!")
history.add_ai_message("whats up?")

# 输出消息历史
history.messages

代码示例

# 示例代码展示了如何使用上面的步骤来实现完整的聊天消息存储和检索

# 初始化 PostgresEngine
engine = PostgresEngine.from_instance(
    project_id=PROJECT_ID, 
    region="us-central1", 
    instance="my-postgresql-instance", 
    database="my-database"
)

# 初始化聊天记录表
engine.init_chat_history_table(table_name="message_store")

# 创建聊天历史实例
history = PostgresChatMessageHistory.create_sync(
    engine, 
    session_id="test_session", 
    table_name="message_store"
)

# 添加聊天消息
history.add_user_message("hi!")
history.add_ai_message("whats up?")

# 获取聊天消息历史
print(history.messages)

常见问题和解决方案

  • 连接失败问题：如果无法连接数据库，确保你的网络环境支持访问Google Cloud SQL。可以考虑使用API代理服务来提高访问稳定性。
  • 不正确的凭据：确认你使用的账户拥有访问对应Google Cloud项目的权限，并在环境中配置了正确的Application Default Credentials。

总结和进一步学习资源

在本文中，我们学习了如何通过Google Cloud SQL for PostgreSQL存储和管理聊天消息历史。从环境配置到代码实现，提供了完整的参考和实践路径。对于想深入了解的开发者，建议参考以下资源：

参考资料

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

---END---
avatar
文章
阅读
粉丝