콘텐츠로 이동

TypeScript Strands Agent를 관계형 데이터베이스에 연결

connection 생성기는 TypeScript Strands Agent관계형 데이터베이스 프로젝트에 연결하여 에이전트 팩토리 내에서 Prisma 클라이언트를 사용할 수 있도록 합니다.

이 생성기를 사용하기 전에 다음이 필요합니다:

  1. ts#strands-agent 프로젝트
  2. ts#rdb 프로젝트
  1. 설치 Nx Console VSCode Plugin 아직 설치하지 않았다면
  2. VSCode에서 Nx 콘솔 열기
  3. 클릭 Generate (UI) "Common Nx Commands" 섹션에서
  4. 검색 @aws/nx-plugin - connection
  5. 필수 매개변수 입력
    • 클릭 Generate

    Strands Agent 프로젝트를 소스로, 관계형 데이터베이스 프로젝트를 타겟으로 선택합니다. 프로젝트에 여러 에이전트 컴포넌트가 포함된 경우 sourceComponent를 지정하여 명확히 구분합니다.

    매개변수 타입 기본값 설명
    sourceProject 필수 string - 소스 프로젝트
    targetProject 필수 string - 연결할 대상 프로젝트
    sourceComponent string - 연결을 시작할 소스 컴포넌트 (컴포넌트 이름, 소스 프로젝트 루트 기준 상대 경로, 또는 generator id). 프로젝트를 소스로 명시적으로 선택하려면 '.'을 사용하세요.
    targetComponent string - 연결할 대상 컴포넌트 (컴포넌트 이름, 대상 프로젝트 루트 기준 상대 경로, 또는 generator id). 프로젝트를 대상으로 명시적으로 선택하려면 '.'을 사용하세요.

    생성기는 단일 파일을 수정합니다:

    • 디렉터리packages/my-service/src/my-agent
      • agent.ts getAgent 내부에서 가져온 Prisma 클라이언트가 도구에서 사용 가능

    또한 에이전트의 <agent-name>-serve-local 타겟이 데이터베이스의 serve-local 타겟에 의존하도록 업데이트됩니다.

    Prisma 클라이언트는 함수 매개변수를 통해 전달되는 대신 getAgent() 내부에서 직접 가져옵니다. 이는 strands-agent 런타임이 세션 컨텍스트를 전파하는 방식과 일치합니다 — withSessionIdgetAgent를 래핑하고 세션당 하나의 Agent 인스턴스를 캐시하므로 getPrisma()는 세션당 최대 한 번만 호출됩니다.

    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 */ });
    };

    생성된 에이전트 구성은 IGrantableIConnectable을 구현하므로 구성에서 직접 데이터베이스에 대한 네트워크 및 IAM 액세스 권한을 부여할 수 있습니다.

    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은 에이전트 런타임이 데이터베이스 포트에 도달할 수 있도록 보안 그룹 규칙을 엽니다. grantConnect는 에이전트의 실행 역할에 IAM rds-db:connect 권한을 부여합니다.

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

    이는 에이전트와 연결된 모든 데이터베이스를 시작합니다. SERVE_LOCAL=true 환경 변수는 각 Prisma 클라이언트가 Aurora 대신 로컬 Docker 데이터베이스에 연결하도록 합니다.