跳至主要內容

Google Cloud SQL for PostgreSQL

Cloud SQL for PostgreSQL 是一項全代管資料庫服務，可協助您在 Google Cloud Platform 上設定、維護、管理和管理 PostgreSQL 關聯式資料庫。 透過 Cloud SQL for PostgreSQL 的 Langchain 整合，將資料庫應用程式擴充至建置 AI 驅動體驗。

此筆記本介紹如何使用 Cloud SQL for PostgreSQL 以 PostgresLoader 類別載入文件。

在 GitHub 上了解更多關於此套件的資訊。

Open In Colab

開始之前 (Before you begin)​

要執行此筆記本，您需要執行以下操作

🦜🔗 函式庫安裝 (Library Installation)​

安裝整合函式庫，langchain_google_cloud_sql_pg。

%pip install --upgrade --quiet  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)

🔐 驗證 (Authentication)​

以登入此筆記本的 IAM 使用者身分驗證 Google Cloud，以便存取您的 Google Cloud 專案。

  • 如果您使用 Colab 執行此筆記本，請使用以下儲存格並繼續。
  • 如果您使用 Vertex AI Workbench，請查看此處的設定說明。
from google.colab import auth

auth.authenticate_user()

☁ 設定您的 Google Cloud 專案 (Set Your Google Cloud Project)​

設定您的 Google Cloud 專案，以便您可以在此筆記本中利用 Google Cloud 資源。

如果您不知道您的專案 ID，請嘗試以下操作

# @title Project { display-mode: "form" }
PROJECT_ID = "gcp_project_id"  # @param {type:"string"}

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

基本用法 (Basic Usage)​

設定 Cloud SQL 資料庫值 (Set Cloud SQL database values)​

在 Cloud SQL 執行個體頁面 (Cloud SQL Instances page)中尋找您的資料庫變數。

# @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"}

Cloud SQL 引擎 (Cloud SQL Engine)​

將 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 資料庫驗證 (IAM database authentication) 將用作資料庫驗證的方法。 此函式庫使用屬於從環境來源的 應用程式預設憑證 (Application Default Credentials，ADC) 的 IAM 主體。

或者，也可以使用使用使用者名稱和密碼的 內建資料庫驗證 (built-in database authentication) 來存取 Cloud SQL 資料庫。 只需將選用的 user 和 password 引數提供給 PostgresEngine.from_instance()

  • 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 (Create PostgresLoader)​

from langchain_google_cloud_sql_pg import PostgresLoader

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

透過預設表格載入文件 (Load Documents via default table)​

載入器 (loader) 會從表格傳回文件列表，並將第一欄作為 page_content，將所有其他欄作為 metadata。 預設表格的第一欄將作為 page_content，第二欄將作為 metadata (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)

透過自訂表格/metadata 或自訂頁面內容欄載入文件 (Load documents via custom table/metadata or custom page content columns)​

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

設定頁面內容格式 (Set page content format)​

載入器會回傳一個 Documents 清單，每個資料列對應一個 document，頁面內容採用指定的字串格式，例如 text (以空格分隔的串聯字串)、JSON、YAML、CSV 等。 JSON 和 YAML 格式包含標頭，而 text 和 CSV 格式則不包含欄位標頭。

loader = await PostgresLoader.create(
    engine,
    table_name="products",
    content_columns=["product_name", "description"],
    format="YAML",
)
docs = await loader.aload()
print(docs)

此頁面是否對您有所幫助？