Skip to content

feat: add protocol mode support (HTTP, MCP, A2A)#550

Open
tejaskash wants to merge 8 commits intomainfrom
protocol-mode-support
Open

feat: add protocol mode support (HTTP, MCP, A2A)#550
tejaskash wants to merge 8 commits intomainfrom
protocol-mode-support

Conversation

@tejaskash
Copy link
Contributor

@tejaskash tejaskash commented Mar 17, 2026

Summary

Adds protocol mode as a first-class dimension to the agent creation flow. Users can now create HTTP (default), MCP, and A2A agents through both CLI flags and the TUI wizard.

  • Schema: New protocol field on AgentEnvSpec (optional, defaults to HTTP for backwards compat)
  • Templates: Protocol-first directory layout (python/{protocol}/{framework}/). New templates for MCP standalone and A2A (Strands, GoogleADK, LangGraph)
  • TUI wizard: Protocol selection step after build type. MCP skips framework/model/memory steps and jumps to confirm
  • Non-interactive: --protocol flag on create and add agent commands with full validation
  • Validation: Protocol/framework compatibility matrix enforced. MCP rejects --framework, --model-provider, --memory
  • CDK: Companion PR wires protocolConfiguration to CfnRuntime for non-HTTP protocols

AGUI support removed (CDK L1 doesn't support it yet). Can be re-added when L1 is ready.

Test plan

  • Unit tests: 645 targeted tests pass (schema, templates, validation, schema-mapper)
  • agentcore create --name McpTest --protocol MCP --language Python creates MCP standalone project
  • agentcore create --name A2aTest --protocol A2A --framework Strands --model-provider Bedrock --memory none --language Python creates A2A project
  • agentcore add agent --protocol MCP --name McpAgent --language Python in existing project
  • TUI: Create -> Add Agent -> New Agent -> select MCP -> skips framework/model -> confirm works
  • TUI: Protocol step shows HTTP, MCP, A2A options
  • Default (no --protocol) continues to work as HTTP

Add the protocol dimension to all agent creation flows: non-interactive
CLI, TUI wizard, and BYO agent. Each protocol mode routes to the correct
template directory under a new protocol-first layout
(python/{protocol}/{framework}/base/).

Key changes:
- Schema: ProtocolModeSchema, PROTOCOL_FRAMEWORK_MATRIX, helpers
- Templates: restructure under http/, add mcp/, a2a/, agui/ dirs
- Rendering: BaseRenderer uses protocol in template path, new McpRenderer
- CLI: --protocol flag on add agent and create commands with validation
- TUI: protocol step in wizard, conditional framework/model filtering
- Schema mapper: protocolMode passthrough, MCP skips model/memory/creds
- Tests: unit tests for schema, validation, rendering, and mapping
Add create-protocols.test.ts covering MCP, A2A, AGUI, and HTTP create
and add-agent flows. Fix MCP create defaulting undefined model provider
by falling back to Bedrock (no credential needed). Add protocol option
to createTestProject helper.
Previously HTTP agents omitted protocolMode for backwards compatibility.
Now all agents have an explicit protocolMode field set to HTTP, MCP,
A2A, or AGUI. This makes the config self-documenting and consistent.
Align templates with bedrock-agentcore-sdk-python PRs #349 and #350:

A2A templates now use serve_a2a() from bedrock_agentcore.runtime:
- Strands: StrandsA2AExecutor from strands.multiagent.a2a.executor
- GoogleADK: A2aAgentExecutor with Runner + AgentCard from a2a-sdk
- LangGraph: custom AgentExecutor subclass with AgentCard from a2a-sdk
- Dependencies: bedrock-agentcore[a2a], removed uvicorn/langgraph-a2a

AGUI templates now use serve_ag_ui() and AGUIApp from runtime:
- Strands: StrandsAgent from ag_ui_strands + serve_ag_ui()
- GoogleADK: AGUIApp decorator with Runner, yields AG-UI events
- LangGraph: AGUIApp decorator with graph.ainvoke, yields AG-UI events
- Dependencies: bedrock-agentcore[ag-ui], removed uvicorn/fake adapters
- Rename `protocolMode` field to `protocol` in AgentEnvSpec schema and
  all downstream types (GenerateConfig, AddAgentConfig, AgentRenderConfig)
- Remove AGUI protocol support (CDK L1 does not support it yet):
  delete all agui templates, remove from enums/matrix/tests/options
- Fix TUI crash in Add Agent -> New Agent flow: ADD_AGENT_STEP_LABELS
  was missing the 'protocol' step, causing StepIndicator to crash on
  undefined.length
- Pass protocol selection through AddAgentConfig so it isn't silently
  dropped when creating agents via the TUI add-agent flow
- Fix type error in create/validate.ts: capture validated ProtocolMode
  from safeParse instead of using untyped string
@tejaskash tejaskash requested a review from a team March 17, 2026 19:09
@github-actions github-actions bot added the size/xl PR size: XL label Mar 17, 2026
- Remove unnecessary type assertion on cliOptions.protocol
- Remove unused SDK_OPTIONS import from GenerateWizardUI
- Remove unused getSDKOptionsForProtocol import from useGenerateWizard
- Prefix unused destructured variable with underscore in test
- Run prettier on 4 files that failed format check
@github-actions github-actions bot added size/xl PR size: XL and removed size/xl PR size: XL labels Mar 17, 2026
The integration test still referenced the old `protocolMode` field name
instead of `protocol`, and included an AGUI test case that was removed
from the schema.
@github-actions github-actions bot added size/xl PR size: XL and removed size/xl PR size: XL labels Mar 17, 2026
@github-actions
Copy link
Contributor

github-actions bot commented Mar 17, 2026

Coverage Report

Status Category Percentage Covered / Total
🔵 Lines 42.94% 3873 / 9018
🔵 Statements 42.51% 4083 / 9603
🔵 Functions 44.27% 750 / 1694
🔵 Branches 44.69% 2582 / 5777
Generated in workflow #985 for commit 858644e by the Vitest Coverage Report Action

- Add model/__init__.py to all A2A template directories (strands,
  googleadk, langchain_langgraph) matching HTTP template structure
- Move build type validation before MCP early return in create/validate
  so invalid --build values are caught for MCP agents too
- Add clarifying comment for phantom framework/modelProvider defaults
  in MCP path of create/command.tsx
@github-actions github-actions bot added size/xl PR size: XL and removed size/xl PR size: XL labels Mar 17, 2026
@tejaskash tejaskash deployed to e2e-testing March 17, 2026 19:46 — with GitHub Actions Active
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

size/xl PR size: XL

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant