Python Agent에서 Gateway로

connection 생성기는 Python Agent를 AgentCore Gateway에 연결할 수 있습니다.

생성기는 에이전트가 배포될 때 IAM SigV4(httpx 요청 서명 사용)로 Gateway에 인증하고, 로컬에서 실행할 때는 Gateway 프로젝트에서 시작한 로컬 게이트웨이에 연결하도록 연결합니다.

전제 조건

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

Agent 컴포넌트(infra: agentcore)가 있는 Python 프로젝트
protocol: mcp 및 auth: iam이 설정된 agentcore-gateway 프로젝트

Gateway는 mcp 프로토콜을 제공해야 합니다 — 에이전트는 MCP 클라이언트로서 Gateway에 도달하므로 대상이 도구로 노출됩니다. Gateway는 또한 IAM 인증을 사용해야 합니다: 에이전트는 자체 실행 역할을 사용하여 SigV4로 요청에 서명합니다. 생성기는 Cognito 인증 게이트웨이를 거부합니다.

사용법

생성기 실행

이 제너레이터 실행@aws/nx-plugin:connection

CLI
VSCode

$ pnpm nx g @aws/nx-plugin:connection

$ yarn nx g @aws/nx-plugin:connection

$ npx nx g @aws/nx-plugin:connection

$ bunx nx g @aws/nx-plugin:connection

설치 Nx Console VSCode Plugin 아직 설치하지 않았다면
VSCode에서 Nx 콘솔 열기
클릭 Generate (UI) "Common Nx Commands" 섹션에서
검색 @aws/nx-plugin - connection
필수 매개변수 입력
클릭 Generate

명령 구성하기5

sourceProject필수

targetProject필수

sourceComponent

targetComponent

preferInstallDependencies

true

--dry-run어떤 파일이 변경될지 확인하기 위해 드라이 런을 수행할 수도 있습니다

에이전트 프로젝트를 소스로, Gateway 프로젝트를 대상으로 선택합니다.

옵션

제너레이터 옵션5 옵션

sourceProject필수string: 소스 프로젝트
targetProject필수string: 연결할 대상 프로젝트
sourceComponentstring: 연결할 소스 컴포넌트 (컴포넌트 이름, 소스 프로젝트 루트 기준 상대 경로, 또는 generator id). 프로젝트를 소스로 명시적으로 선택하려면 '.'을 사용하세요.
targetComponentstring: 연결할 대상 컴포넌트 (컴포넌트 이름, 대상 프로젝트 루트 기준 상대 경로, 또는 generator id). 프로젝트를 대상으로 명시적으로 선택하려면 '.'을 사용하세요.
preferInstallDependenciesboolean기본값: true: 생성기 실행 후 의존성 설치를 선호할지 여부입니다. 여러 생성기를 일괄 처리할 때 설치를 연기하려면 false로 설정하세요 (후속 생성기가 Nx 프로젝트 그래프를 계산할 수 있도록 필요한 경우 설치는 여전히 실행됩니다); 마지막에 한 번만 설치합니다.

생성기 출력

생성기는 공유 core-gateway 모듈을 agent_connection Python 프로젝트에 생성하고, Gateway별 래퍼를 추가하며, 에이전트를 수정합니다:

디렉터리packages/common/agent_connection
- 디렉터리<scope>_agent_connection
  - 디렉터리core/
    __init__.py Python package initialization
    agentcore_endpoints.py Framework-agnostic ARN/URL resolution
    agentcore_transport.py Shared AgentCore transport plumbing
    agentcore_gateway_mcp_transport.py Framework-agnostic Gateway MCP transport
    agentcore_gateway_mcp_client_<framework>.py Gateway MCP client for your agent’s framework
    디렉터리auth/ Framework-agnostic SigV4 / session-forwarding httpx.Auth
    …
  - 디렉터리app/
    __init__.py Python package initialization
    <gateway_snake>_client_<framework>.py Per-Gateway client wrapper
  - __init__.py Re-exports the Gateway client

클라이언트 접미사는 에이전트의 프레임워크(_strands 또는 _langchain)와 일치합니다.

또한 생성기는:

에이전트의 agent.py를 수정하여 Gateway 클라이언트를 가져오고 tools에 도구를 등록합니다
agent_connection을 에이전트의 워크스페이스 종속성으로 추가합니다
에이전트의 <agent>-dev 타겟이 Gateway의 dev 타겟에 의존하도록 연결합니다

연결된 Gateway 사용

생성기는 에이전트의 agent.py를 변환하여 Gateway 클라이언트를 사용합니다:

Strands
LangChain

from contextlib import contextmanager
from strands import Agent

from my_scope_agent_connection import MyGatewayClientStrands

@contextmanager
def get_agent():
    my_gateway = MyGatewayClientStrands.create()
    with (
        my_gateway,
    ):
        yield Agent(
            system_prompt="...",
            tools=[*my_gateway.list_tools_sync()],
        )

MyGatewayClientStrands.create()는 list_tools_sync()가 Gateway를 통해 사용 가능한 모든 도구를 제공하는 단일 컨텍스트 관리 가능한 MCPClient를 반환합니다.

from langchain.agents import create_agent
from langchain_aws import ChatBedrockConverse

from my_scope_agent_connection import MyGatewayClientLangChain

def get_agent():
    my_gateway = MyGatewayClientLangChain.create()
    return create_agent(
        model=ChatBedrockConverse(model=MODEL_ID, region_name=REGION),
        system_prompt="...",
        tools=[*my_gateway],
    )

MyGatewayClientLangChain.create()는 langchain-mcp-adapters를 통해 로드된 도구 목록을 반환합니다. 각 도구는 호출마다 새로운 세션을 열기 때문에 with 블록이 필요하지 않습니다.

두 경우 모두 클라이언트는 모드에 따라 동일하게 동작합니다:

배포 모드 (LOCAL_DEV 미설정): Gateway의 MCP 엔드포인트를 가리키는 도구, SigV4 서명됨.
로컬 모드 (LOCAL_DEV=true): Gateway 프로젝트의 dev 타겟에서 시작한 로컬 게이트웨이를 가리키는 plain-HTTP 도구.

세션 ID는 X-Amzn-Bedrock-AgentCore-Runtime-Session-Id 헤더를 통해 다운스트림 MCP 서버로 자동으로 전파됩니다.

인프라

생성기를 실행한 후 에이전트에 Gateway를 호출할 권한을 부여해야 합니다.

CDK
Terraform

const gateway = new MyGateway(this, 'MyGateway');
const myAgent = new MyAgent(this, 'MyAgent');

// Grant the agent permissions to invoke the Gateway
gateway.grantInvokeAccess(myAgent);

Gateway URL은 생성된 CDK 구성에 의해 런타임 구성의 agentcore.gateways.<ClassName> 네임스페이스에 자동으로 등록되므로 에이전트가 런타임에 이를 검색할 수 있습니다.

module "my_gateway" {
  source = "../../common/terraform/src/app/gateways/my-gateway"
}

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

  # Grant the agent permission to invoke the Gateway
  additional_iam_policy_statements = [{
    Effect   = "Allow"
    Action   = ["bedrock-agentcore:InvokeGateway"]
    Resource = [module.my_gateway.gateway_arn]
  }]
}

Gateway URL은 생성된 Terraform 모듈에 의해 런타임 구성의 agentcore.gateways.<ClassName> 네임스페이스에 자동으로 등록되므로 에이전트가 런타임에 이를 검색할 수 있습니다.

로컬 개발

생성기는 에이전트의 dev 타겟을 다음과 같이 구성합니다:

연결된 Gateway의 로컬 게이트웨이와 연결된 모든 MCP 서버를 시작합니다
LOCAL_DEV=true를 설정하여 생성된 클라이언트가 배포된 Gateway 대신 로컬 게이트웨이를 가리키도록 합니다

다음 명령으로 에이전트를 로컬에서 실행합니다:

pnpm nx <agent-name>-dev <project-name>

yarn nx <agent-name>-dev <project-name>

npx nx <agent-name>-dev <project-name>

bunx nx <agent-name>-dev <project-name>

에이전트를 로컬에서 배포된 Gateway에 대해 실행하려면(예: Cedar 정책을 테스트하기 위해) 에이전트의 serve 타겟을 사용합니다. LOCAL_DEV가 설정되지 않으면 클라이언트는 런타임 구성에서 배포된 Gateway URL을 확인하고 로컬 AWS 자격 증명으로 요청에 SigV4 서명합니다:

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

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

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

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

로컬 충실도

로컬 게이트웨이는 배포된 Gateway를 대신하므로:

Cedar 정책 평가 없음. 정책에 관계없이 모든 도구가 에이전트에 표시됩니다. 배포된 Gateway에 대해 정책을 테스트하려면 serve 타겟을 사용하세요.
도구 이름 접두사는 유지됩니다. 각 로컬 MCP 서버의 도구는 <target-name>___<tool-name>으로 노출되며, 이는 배포된 Gateway가 생성하는 것과 일치합니다. 이를 통해 에이전트의 시스템 프롬프트와 참조하는 Cedar 작업 이름이 로컬 및 배포 실행 간에 일관성을 유지합니다.