SuperAPI AI 工具集

databricks-lakebase-provisioned

数据与分析

查看项目

更新于 10-15 20:32

技能介绍

用途:该技能提供在 Databricks 上使用 Lakebase Provisioned(托管 PostgreSQL)承载 OLTP 事务型工作负载的模式与最佳实践,适用于需要 PostgreSQL 数据库的应用、为 Databricks Apps 增加持久化状态、从 Delta Lake 做反向 ETL,以及为 LangChain 应用存储聊天/智能体记忆等场景。

主要能力:提供全托管 PostgreSQL 实例、基于 OAuth 令牌的认证(通过 Databricks SDK,约 1 小时过期)、Unity Catalog 治理注册、从 Delta 表向 PostgreSQL 的反向 ETL 同步,以及与 Databricks Apps 的集成。可用区域(AWS):us-east-1、us-east-2、us-west-2、eu-central-1、eu-west-1、ap-south-1、ap-southeast-1、ap-southeast-2。容量按计算单元 CU_1、CU_2、CU_4、CU_8 划分。

使用方式:可用 Python SDK 或 CLI。SDK 中通过 WorkspaceClient 调用 w.database.create_database_instance 创建实例,用 generate_database_credential 生成令牌作为连接密码;连接串须使用 sslmode=require。长运行应用必须实现令牌刷新(建议约 50 分钟刷新一次,令牌 1 小时过期)。CLI 示例:databricks database create-database-instance --name my-lakebase-instance --capacity CU_1;databricks database generate-database-credential --request-id $(uuidgen) --json '{"instance_names": ["my-lakebase-instance"]}'。还提供 MCP 工具 manage_lakebase_database(create_or_update/get/list/delete,type="provisioned")、manage_lakebase_sync(反向 ETL,调度策略 TRIGGERED/SNAPSHOT/CONTINUOUS)和 generate_lakebase_credential。

依赖与限制:需 Databricks SDK for Python >= 0.61.0(推荐 0.81.0+)、psycopg 3.x、SQLAlchemy 2.x(postgresql+psycopg)。常见问题包括长查询中令牌过期、macOS 上 DNS 解析失败、实例被停止导致连接拒绝、权限未授予、未使用 sslmode=require 报 SSL 错误。Lakebase Autoscaling 为更新的自动扩缩产品但区域覆盖有限,本技能聚焦更广泛可用的 Provisioned 版本。资料中未提供具体收费与价格信息。