TypeScript Strands Agent 连接到关系数据库

connection 生成器将 TypeScript Strands Agent 连接到 关系数据库 项目，使 Prisma 客户端在您的 agent 工厂内可用。

前置条件

在使用此生成器之前，请确保您拥有：

1. 一个 ts#strands-agent 项目
2. 一个 ts#rdb 项目

用法

运行生成器

VSCode

1. 安装 Nx Console VSCode Plugin 如果您尚未安装
2. 在VSCode中打开Nx控制台
3. 点击 Generate (UI) 在"Common Nx Commands"部分
4. 搜索 @aws/nx-plugin - connection
5. 填写必需参数
6. 点击 Generate

CLI

pnpm

pnpm nx g @aws/nx-plugin:connection

yarn

yarn nx g @aws/nx-plugin:connection

npm

npx nx g @aws/nx-plugin:connection

bun

bunx nx g @aws/nx-plugin:connection

您还可以执行试运行以查看哪些文件会被更改

pnpm

pnpm nx g @aws/nx-plugin:connection --dry-run

yarn

yarn nx g @aws/nx-plugin:connection --dry-run

npm

npx nx g @aws/nx-plugin:connection --dry-run

bun

bunx nx g @aws/nx-plugin:connection --dry-run

选择您的 Strands Agent 项目作为源，选择您的关系数据库项目作为目标。如果项目包含多个 agent 组件，请指定 sourceComponent 以消除歧义。

选项

参数	类型	默认值	描述
sourceProject 必需	string	-	源项目
targetProject 必需	string	-	要连接到的目标项目
sourceComponent	string	-	要从其连接的源组件（组件名称、相对于源项目根目录的路径或生成器 ID）。使用 '.' 显式选择项目作为源。
targetComponent	string	-	要连接到的目标组件（组件名称、相对于目标项目根目录的路径或生成器 ID）。使用 '.' 显式选择项目作为目标。

生成器输出

生成器修改单个文件：

• 文件夹packages/my-service/src/my-agent

  • agent.ts 在 getAgent 内获取 Prisma 客户端并可用于工具

此外，agent 的 <agent-name>-serve-local 目标会更新为依赖于数据库的 serve-local 目标。

工作原理

Prisma 客户端直接在 getAgent() 内获取，而不是通过函数参数传递。这与 strands-agent 运行时传播会话上下文的方式一致 —— withSessionId 包装 getAgent 并为每个会话缓存一个 Agent 实例，因此 getPrisma() 每个会话最多调用一次。

Agent 定义

getAgent 被更新为在其主体顶部导入并调用 Prisma getter：

packages/my-service/src/my-agent/agent.ts

import { getPrisma as getMyDb } from ':my-scope/my-db';

export const getAgent = async () => {
  const myDb = await getMyDb();
  return new Agent({ /* use myDb in tools */ });
};

多个数据库

使用不同的目标再次运行生成器会将第二个数据库添加到第一个数据库旁边：

packages/my-service/src/my-agent/agent.ts

import { getPrisma as getMyDb } from ':my-scope/my-db';
import { getPrisma as getOtherDb } from ':my-scope/other-db';

export const getAgent = async () => {
  const myDb = await getMyDb();
  const otherDb = await getOtherDb();
  return new Agent({ /* use both clients in tools */ });
};

基础设施

生成的 agent 构造实现了 IGrantable 和 IConnectable，因此您可以直接在构造上授予对数据库的网络和 IAM 访问权限。

CDK

packages/infra/src/stacks/application-stack.ts

import { MyDatabase } from ':my-scope/common-constructs';

const db = new MyDatabase(this, 'Db', { vpc, ... });
const myAgent = new MyAgent(this, 'MyAgent', { vpc, ... });

db.allowDefaultPortFrom(myAgent);
db.grantConnect(myAgent);

allowDefaultPortFrom 打开安全组规则，以便 agent 运行时可以访问数据库端口。grantConnect 向 agent 的执行角色授予 IAM rds-db:connect 权限。

Terraform

将数据库模块输出传递到您的 agent 模块中，以便它可以访问数据库并读取其运行时配置：

packages/infra/src/main.tf

module "my_database" {
  source              = "../../common/terraform/src/app/dbs/my-database"
  vpc_id              = module.vpc.vpc_id
  database_subnet_ids = module.vpc.private_isolated_subnet_ids
}

module "my_agent" {
  source = "../../common/terraform/src/app/agents/my-agent"

  appconfig_application_id     = module.my_database.appconfig_application_id
  database_cluster_resource_id = module.my_database.cluster_resource_id
  database_runtime_user        = module.my_database.database_runtime_user
  database_security_group_id   = module.my_database.security_group_id
  database_port                = module.my_database.cluster_port
}

确保 agent 的执行角色具有 rds-db:connect 权限，并且其安全组可以在数据库端口上访问数据库安全组。

本地开发

pnpm

pnpm nx <agent-name>-serve-local <project-name>

yarn

yarn nx <agent-name>-serve-local <project-name>

npm

npx nx <agent-name>-serve-local <project-name>

bun

bunx nx <agent-name>-serve-local <project-name>

这将启动 agent 和所有连接的数据库。SERVE_LOCAL=true 环境变量使每个 Prisma 客户端连接到其本地 Docker 数据库而不是 Aurora。