跳到主要内容
适用于 PostgreSQL 的 Cloud SQL 是一项完全托管的数据库服务，可帮助您在 Google Cloud Platform 上设置、维护、管理和管理 PostgreSQL 关系型数据库。利用适用于 PostgreSQL 的 Cloud SQL 的 LangChain 集成，扩展您的数据库应用程序以构建 AI 驱动的体验。
本 Notebook 介绍了如何使用 Cloud SQL for PostgreSQL 和 PostgresLoader 类加载文档。 在 GitHub 上了解有关该软件包的更多信息。 在 Colab 中打开

​
开始之前

要运行此 notebook，您需要执行以下操作

​
🦜🔗 库安装

安装集成库 langchain_google_cloud_sql_pg。 
pip install -qU  langchain_google_cloud_sql_pg
仅限 Colab： 取消注释以下单元格以重新启动内核，或使用按钮重新启动内核。对于 Vertex AI Workbench，您可以使用顶部的按钮重新启动终端。 
# # Automatically restart kernel after installs so that your environment can access the new packages
# import IPython

# app = IPython.Application.instance()
# app.kernel.do_shutdown(True)

​
🔐 身份验证

以登录到此 notebook 的 IAM 用户身份向 Google Cloud 进行身份验证，以访问您的 Google Cloud 项目。
  • 如果您正在使用 Colab 运行此 notebook，请使用下面的单元格并继续。
  • 如果您正在使用 Vertex AI Workbench，请查看此处的设置说明。
from google.colab import auth

auth.authenticate_user()

​
☁ 设置您的 Google Cloud 项目

设置您的 Google Cloud 项目，以便您可以在此 notebook 中利用 Google Cloud 资源。 如果您不知道您的项目 ID，请尝试以下操作：
  • 运行 gcloud config list。
  • 运行 gcloud projects list。
  • 查看支持页面：查找项目 ID。
# @title Project { display-mode: "form" }
PROJECT_ID = "gcp_project_id"  # @param {type:"string"}

# Set the project id
! gcloud config set project {PROJECT_ID}

​
基本用法

​
设置 Cloud SQL 数据库值

在云 SQL 实例页面中查找您的数据库变量。 
# @title Set Your Values Here { display-mode: "form" }
REGION = "us-central1"  # @param {type: "string"}
INSTANCE = "my-primary"  # @param {type: "string"}
DATABASE = "my-database"  # @param {type: "string"}
TABLE_NAME = "vector_store"  # @param {type: "string"}

​
云 SQL 引擎

将 PostgreSQL 建立为文档加载器的一个要求和参数是 PostgresEngine 对象。PostgresEngine 为您的 Cloud SQL for PostgreSQL 数据库配置了一个连接池，从而实现从您的应用程序成功连接并遵循行业最佳实践。 要使用 PostgresEngine.from_instance() 创建 PostgresEngine，您只需提供 4 个信息：
  1. project_id：Cloud SQL 实例所在的 Google Cloud 项目的项目 ID。
  2. region：Cloud SQL 实例所在的区域。
  3. instance：Cloud SQL 实例的名称。
  4. database：要连接到 Cloud SQL 实例上的数据库名称。
默认情况下，IAM 数据库身份验证将用作数据库身份验证方法。此库使用从环境中获取的应用程序默认凭据 (ADC) 所属的 IAM 主体。 或者，也可以使用内置数据库身份验证，即使用用户名和密码访问 Cloud SQL 数据库。只需向 PostgresEngine.from_instance() 提供可选的 user 和 password 参数：
  • user：用于内置数据库身份验证和登录的数据库用户
  • password：用于内置数据库身份验证和登录的数据库密码。
注意：本教程演示了异步接口。所有异步方法都有相应的同步方法。 
from langchain_google_cloud_sql_pg import PostgresEngine

engine = await PostgresEngine.afrom_instance(
    project_id=PROJECT_ID,
    region=REGION,
    instance=INSTANCE,
    database=DATABASE,
)

​
创建 PostgresLoader

from langchain_google_cloud_sql_pg import PostgresLoader

# Creating a basic PostgreSQL object
loader = await PostgresLoader.create(engine, table_name=TABLE_NAME)

​
通过默认表加载文档

加载器从表中返回文档列表，其中第一列作为 page_content，所有其他列作为元数据。默认表将第一列作为 page_content，第二列作为元数据 (JSON)。每行都成为一个文档。请注意，如果您希望文档具有 ID，则需要添加它们。 
from langchain_google_cloud_sql_pg import PostgresLoader

# Creating a basic PostgresLoader object
loader = await PostgresLoader.create(engine, table_name=TABLE_NAME)

docs = await loader.aload()
print(docs)

​
通过自定义表/元数据或自定义页面内容列加载文档

loader = await PostgresLoader.create(
    engine,
    table_name=TABLE_NAME,
    content_columns=["product_name"],  # Optional
    metadata_columns=["id"],  # Optional
)
docs = await loader.aload()
print(docs)

​
设置页面内容格式

加载器返回文档列表，每行一个文档，页面内容采用指定字符串格式，即文本（空格分隔的连接）、JSON、YAML、CSV 等。JSON 和 YAML 格式包含标题，而文本和 CSV 不包含字段标题。 
loader = await PostgresLoader.create(
    engine,
    table_name="products",
    content_columns=["product_name", "description"],
    format="YAML",
)
docs = await loader.aload()
print(docs)
在 GitHub 上编辑此页面源文件。
以编程方式连接这些文档到 Claude、VSCode 等，通过 MCP 获取实时答案。
© . This site is unofficial and not affiliated with LangChain, Inc.