Python Agent đến MCP

Trình tạo connection có thể kết nối Python Agent của bạn với máy chủ MCP (có thể là TypeScript hoặc Python).

Trình tạo thiết lập tất cả các kết nối cần thiết để agent của bạn có thể khám phá và gọi các công cụ của máy chủ MCP, cả khi triển khai lên AWS (thông qua Bedrock AgentCore) và khi chạy cục bộ.

Điều kiện tiên quyết

Trước khi sử dụng trình tạo này, hãy đảm bảo bạn có:

1. Một dự án Python với thành phần Python Agent (Strands hoặc LangChain)
2. Một dự án với thành phần máy chủ MCP (có thể là ts#mcp-server hoặc py#mcp-server)
3. Cả hai thành phần được tạo với infra: agentcore

Cách sử dụng

Chạy Trình tạo

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

Bạn cũng có thể thực hiện chạy thử để xem những tệp nào sẽ bị thay đổi

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

VSCode

1. Cài đặt Nx Console VSCode Plugin nếu bạn chưa cài đặt
2. Mở Nx Console trong VSCode
3. Nhấp Generate (UI) trong phần "Common Nx Commands"
4. Tìm kiếm @aws/nx-plugin - connection
5. Điền các tham số bắt buộc
6. Nhấp Generate

Chọn dự án agent của bạn làm nguồn và dự án máy chủ MCP của bạn làm đích. Nếu các dự án của bạn chứa nhiều thành phần, hãy chỉ định các tùy chọn sourceComponent và targetComponent để phân biệt.

Tùy chọn

Tham số	Kiểu	Mặc định	Mô tả
sourceProject Bắt buộc	string	-	Dự án nguồn
targetProject Bắt buộc	string	-	Dự án đích để kết nối tới
sourceComponent	string	-	Component nguồn để kết nối từ đó (tên component, đường dẫn tương đối so với thư mục gốc của dự án nguồn, hoặc generator id). Sử dụng '.' để chọn rõ ràng dự án làm nguồn.
targetComponent	string	-	Component đích để kết nối tới (tên component, đường dẫn tương đối so với thư mục gốc của dự án đích, hoặc generator id). Sử dụng '.' để chọn rõ ràng dự án làm đích.
preferInstallDependencies	boolean	true	Có nên cài đặt các dependencies sau khi generator chạy hay không. Đặt thành false để trì hoãn việc cài đặt khi chạy nhiều generator liên tiếp (việc cài đặt vẫn sẽ chạy nếu cần thiết để các generator tiếp theo có thể tính toán Nx project graph); cài đặt một lần vào cuối.

Kết quả của Trình tạo

Trình tạo tạo một dự án Python agent_connection được chia sẻ tại packages/common/agent_connection/ (nếu nó chưa tồn tại). Các mô-đun client theo từng kết nối được tạo vào dự án được chia sẻ này:

• Thư mụcpackages/common/agent_connection

  • Thư mục<scope>_agent_connection

    • __init__.py Re-exports per-connection clients
    • Thư mụccore

      • agentcore_endpoints.py Framework-agnostic ARN/URL resolution
      • agentcore_mcp_transport.py Framework-agnostic MCP transport
      • agentcore_mcp_client_<framework>.py MCP client wrapping the transport for your agent’s framework
      • Thư mụcauth/ Framework-agnostic SigV4 / session-forwarding httpx.Auth

        • …
    • Thư mụcapp

      • <mcp_server_name>_client_<framework>.py Per-connection client for each MCP server

Hậu tố client khớp với framework của agent của bạn (_strands hoặc _langchain).

Ngoài ra, trình tạo:

• Chuyển đổi agent.py của agent của bạn để import và sử dụng các công cụ của máy chủ MCP thông qua client dựa trên class
• Thêm dự án agent_connection làm phụ thuộc workspace của dự án agent của bạn
• Cập nhật target dev của agent để phụ thuộc vào target serve của máy chủ MCP

Sử dụng Máy chủ MCP Đã kết nối

Trình tạo chuyển đổi agent.py của agent của bạn để sử dụng các công cụ của máy chủ MCP:

Strands

packages/my-project/my_module/agent/agent.py

from contextlib import contextmanager
from strands import Agent

from my_scope_agent_connection import MyMcpServerClientStrands

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

Client Strands là một context manager, được nhập vào trong khối with xung quanh agent.

LangChain

packages/my-project/my_module/agent/agent.py

from langchain.agents import create_agent
from langchain_aws import ChatBedrockConverse

from my_scope_agent_connection import MyMcpServerClientLangChain

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

Client LangChain trả về một danh sách các công cụ được tải thông qua langchain-mcp-adapters. Mỗi công cụ mở một phiên MCP mới cho mỗi lần gọi, vì vậy các công cụ vẫn có thể sử dụng được trong suốt vòng đời của agent — không cần khối with.

ID phiên AgentCore được truyền tự động đến máy chủ MCP thông qua header X-Amzn-Bedrock-AgentCore-Runtime-Session-Id cho cả hai framework, đảm bảo tính nhất quán cho Bedrock AgentCore Observability.

Cơ sở hạ tầng

CDK

Sau khi chạy trình tạo kết nối, bạn cần cấp quyền cho agent để gọi máy chủ MCP:

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

const mcpServer = new MyMcpServer(this, 'MyMcpServer');
const myAgent = new MyAgent(this, 'MyAgent');

// Grant the agent permissions to invoke the MCP server
mcpServer.grantInvokeAccess(myAgent);

ARN runtime AgentCore của máy chủ MCP được tự động đăng ký trong namespace agentcore của Runtime Configuration bởi construct CDK được tạo, vì vậy agent có thể khám phá nó tại runtime.

Terraform

Sau khi chạy trình tạo kết nối, bạn cần cấp quyền cho agent để gọi máy chủ MCP trong cấu hình Terraform của bạn:

packages/infra/src/main.tf

module "inventory_mcp_server" {
  source = "../../common/terraform/src/app/mcp-servers/inventory-mcp"
}

module "story_agent" {
  source = "../../common/terraform/src/app/agents/story-agent"
}

# Grant the agent permissions to invoke the MCP server
resource "aws_iam_policy" "agent_invoke_mcp" {
  name = "AgentInvokeMcpPolicy"
  policy = jsonencode({
    Version = "2012-10-17"
    Statement = [{
      Effect   = "Allow"
      Action   = "bedrock-agentcore:InvokeAgent"
      Resource = module.inventory_mcp_server.agent_core_runtime_arn
    }]
  })
}

resource "aws_iam_role_policy_attachment" "agent_invoke_mcp" {
  role       = module.story_agent.agent_core_runtime_role_arn
  policy_arn = aws_iam_policy.agent_invoke_mcp.arn
}

ARN runtime AgentCore của máy chủ MCP được tự động đăng ký trong namespace agentcore của Runtime Configuration bởi mô-đun Terraform được tạo, vì vậy agent có thể khám phá nó tại runtime.

Phát triển Cục bộ

Trình tạo cấu hình target dev của agent để:

1. Tự động khởi động (các) máy chủ MCP đã kết nối
2. Đặt LOCAL_DEV=true để client được tạo sử dụng truyền tải HTTP trực tiếp thay vì AgentCore

Chạy agent cục bộ với:

pnpm

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

yarn

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

npm

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

bun

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

Điều này sẽ khởi động cả agent và tất cả các máy chủ MCP đã kết nối, với agent kết nối trực tiếp đến các máy chủ MCP thông qua HTTP trên các cổng cục bộ được chỉ định của chúng.