bedrock-agentcore-starter-toolkit 0.1.19__py3-none-any.whl → 0.1.21__py3-none-any.whl

bedrock_agentcore_starter_toolkit/cli/runtime/commands.py

@@ -186,7 +186,7 @@ def configure(
     ),
     verbose: bool = typer.Option(False, "--verbose", "-v", help="Enable verbose output"),
     region: Optional[str] = typer.Option(None, "--region", "-r"),
-    protocol: Optional[str] = typer.Option(None, "--protocol", "-p", help="Server protocol (HTTP or MCP)"),
+    protocol: Optional[str] = typer.Option(None, "--protocol", "-p", help="Server protocol (HTTP or MCP or A2A)"),
     non_interactive: bool = typer.Option(
         False, "--non-interactive", "-ni", help="Skip prompts; use defaults unless overridden"
     ),
@@ -198,8 +198,8 @@ def configure(
     if not entrypoint:
         _handle_error("--entrypoint is required")
 
-    if protocol and protocol.upper() not in ["HTTP", "MCP"]:
-        _handle_error("Error: --protocol must be either HTTP or MCP")
+    if protocol and protocol.upper() not in ["HTTP", "MCP", "A2A"]:
+        _handle_error("Error: --protocol must be either HTTP or MCP or A2A")
 
     console.print("[cyan]Configuring Bedrock AgentCore...[/cyan]")
     try:

bedrock_agentcore_starter_toolkit/notebook/runtime/bedrock_agentcore.py

@@ -45,7 +45,7 @@ class Runtime:
         auto_create_execution_role: bool = False,
         authorizer_configuration: Optional[Dict[str, Any]] = None,
         region: Optional[str] = None,
-        protocol: Optional[Literal["HTTP", "MCP"]] = None,
+        protocol: Optional[Literal["HTTP", "MCP", "A2A"]] = None,
         disable_otel: bool = False,
         non_interactive: bool = True,
     ) -> ConfigureResult:
@@ -65,15 +65,15 @@ class Runtime:
             auto_create_execution_role: Whether to auto-create execution role (makes execution_role optional)
             authorizer_configuration: JWT authorizer configuration dictionary
             region: AWS region for deployment
-            protocol: agent server protocol, must be either HTTP or MCP
+            protocol: agent server protocol, must be either HTTP or MCP or A2A
             disable_otel: Whether to disable OpenTelemetry observability (default: False)
             non_interactive: Skip interactive prompts and use defaults (default: True)
 
         Returns:
             ConfigureResult with configuration details
         """
-        if protocol and protocol.upper() not in ["HTTP", "MCP"]:
-            raise ValueError("protocol must be either HTTP or MCP")
+        if protocol and protocol.upper() not in ["HTTP", "MCP", "A2A"]:
+            raise ValueError("protocol must be either HTTP or MCP or A2A")
 
         # Parse entrypoint to get agent name
         file_path, file_name = parse_entrypoint(entrypoint)

bedrock_agentcore_starter_toolkit/operations/runtime/configure.py

@@ -59,7 +59,7 @@ def configure_bedrock_agentcore(
         request_header_configuration: Request header configuration dictionary
         verbose: Whether to provide verbose output during configuration
         region: AWS region for deployment
-        protocol: agent server protocol, must be either HTTP or MCP
+        protocol: agent server protocol, must be either HTTP or MCP or A2A
         non_interactive: Skip interactive prompts and use defaults
 
     Returns:
@@ -205,6 +205,7 @@ def configure_bedrock_agentcore(
         requirements_file,
         memory_id,
         memory_name,
+        protocol,
     )
 
     # Check if .dockerignore was created

bedrock_agentcore_starter_toolkit/services/runtime.py

@@ -98,6 +98,8 @@ class BedrockAgentCoreClient:
         control_plane_url = get_control_plane_endpoint(region)
         data_plane_url = get_data_plane_endpoint(region)
 
+        self.dp_endpoint = data_plane_url
+
         self.logger.debug("Initializing Bedrock AgentCore client for region: %s", region)
         self.logger.debug("Control plane: %s", control_plane_url)
         self.logger.debug("Data plane: %s", data_plane_url)

bedrock_agentcore_starter_toolkit/utils/runtime/container.py

@@ -110,6 +110,7 @@ class ContainerRuntime:
         requirements_file: Optional[str] = None,
         memory_id: Optional[str] = None,
         memory_name: Optional[str] = None,
+        protocol: Optional[str] = None,
     ) -> Path:
         """Generate Dockerfile from template."""
         current_platform = self._get_current_platform()
@@ -168,6 +169,7 @@ class ContainerRuntime:
             "observability_enabled": enable_observability,
             "memory_id": memory_id,
             "memory_name": memory_name,
+            "protocol": protocol or "HTTP",
         }
 
         dockerfile_path = output_dir / "Dockerfile"

bedrock_agentcore_starter_toolkit/utils/runtime/schema.py

@@ -43,7 +43,18 @@ class NetworkConfiguration(BaseModel):
 class ProtocolConfiguration(BaseModel):
     """Protocol configuration for BedrockAgentCore deployment."""
 
-    server_protocol: str = Field(default="HTTP", description="Server protocol for deployment, either HTTP or MCP")
+    server_protocol: str = Field(
+        default="HTTP", description="Server protocol for deployment, either HTTP or MCP or A2A"
+    )
+
+    @field_validator("server_protocol")
+    @classmethod
+    def validate_protocol(cls, v: str) -> str:
+        """Validate protocol is one of the supported types."""
+        allowed = ["HTTP", "MCP", "A2A"]
+        if v.upper() not in allowed:
+            raise ValueError(f"Protocol must be one of {allowed}, got: {v}")
+        return v.upper()
 
     def to_aws_dict(self) -> dict:
         """Convert to AWS API format with camelCase keys."""

bedrock_agentcore_starter_toolkit/utils/runtime/templates/Dockerfile.j2

@@ -40,8 +40,9 @@ ENV DOCKER_CONTAINER=1
 RUN useradd -m -u 1000 bedrock_agentcore
 USER bedrock_agentcore
 
-EXPOSE 8080
+EXPOSE 9000
 EXPOSE 8000
+EXPOSE 8080
 
 # Copy entire project (respecting .dockerignore)
 COPY . .

{bedrock_agentcore_starter_toolkit-0.1.19.dist-info → bedrock_agentcore_starter_toolkit-0.1.21.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: bedrock-agentcore-starter-toolkit
-Version: 0.1.19
+Version: 0.1.21
 Summary: A starter toolkit for using Bedrock AgentCore
 Project-URL: Homepage, https://github.com/aws/bedrock-agentcore-starter-toolkit
 Project-URL: Bug Tracker, https://github.com/aws/bedrock-agentcore-starter-toolkit/issues

{bedrock_agentcore_starter_toolkit-0.1.19.dist-info → bedrock_agentcore_starter_toolkit-0.1.21.dist-info}/RECORD

@@ -9,11 +9,11 @@ bedrock_agentcore_starter_toolkit/cli/import_agent/__init__.py,sha256=tyM3dXM3Tc
 bedrock_agentcore_starter_toolkit/cli/import_agent/agent_info.py,sha256=CcPXbKNnI9fc7NyFmZBjB3PkB0wAz2Q_kBGoUVcnngA,9736
 bedrock_agentcore_starter_toolkit/cli/import_agent/commands.py,sha256=vpA66fGwKf6cxBelT3Yb_Af5Ow3AMquGLIDe-tBLjmc,22419
 bedrock_agentcore_starter_toolkit/cli/runtime/__init__.py,sha256=0zKPPoYThoDIr3DZhIQJavq5nVTKRsgcE1s9--wx118,60
-bedrock_agentcore_starter_toolkit/cli/runtime/commands.py,sha256=0k86wtUZmSPZA4w6jpmMRYjKpSU0XosWljkVwWpt4yc,51234
+bedrock_agentcore_starter_toolkit/cli/runtime/commands.py,sha256=oFviB4Fx2Ff43IHxwWg1q8QLYU2IK3IUS96_MO0IH5k,51255
 bedrock_agentcore_starter_toolkit/cli/runtime/configuration_manager.py,sha256=72UVvnzD3aqaOBNhIMMHjNN5-mgR2oUnnWyA33oFoLM,14602
 bedrock_agentcore_starter_toolkit/notebook/__init__.py,sha256=wH1ZzIVKsKT_P0qX2kIIoyVxeHj8K40odfR1YI3McHw,129
 bedrock_agentcore_starter_toolkit/notebook/runtime/__init__.py,sha256=DoGfB_uGFLANJVE9rSZtTH3ymw76WBWmD9vORvBIdXs,66
-bedrock_agentcore_starter_toolkit/notebook/runtime/bedrock_agentcore.py,sha256=jUpCv7MEnnUvsih37MVdQqYLJ_z2fIKNQuxgO1w7zV4,15714
+bedrock_agentcore_starter_toolkit/notebook/runtime/bedrock_agentcore.py,sha256=pGIO6lOdOR71BX6uTW6xhOKAuYSx6jxtvIggy8NmHHI,15742
 bedrock_agentcore_starter_toolkit/operations/__init__.py,sha256=L7sCNjfZviiVVoh2f3NEs2sbjNqFkmIRI3ZPYMMWMz0,51
 bedrock_agentcore_starter_toolkit/operations/gateway/__init__.py,sha256=5Qo1GeN-DghNp9g0coFUs7WpUJDboJoidOVs-5Co7fo,233
 bedrock_agentcore_starter_toolkit/operations/gateway/client.py,sha256=9KQzRAEGWbBDVVacdB1GqN_HJ06vwTBs6tLwfxYROoY,26498
@@ -38,7 +38,7 @@ bedrock_agentcore_starter_toolkit/operations/memory/models/strategies/semantic.p
 bedrock_agentcore_starter_toolkit/operations/memory/models/strategies/summary.py,sha256=X0sVh-6Hw9PPnpPQucG_M7gVyXUiElZInFXBN8WDFUQ,1015
 bedrock_agentcore_starter_toolkit/operations/memory/models/strategies/user_preference.py,sha256=JJPAzt_Shix5a-RJEf7xOEzN5kxtX3dpP8xoMvry61U,996
 bedrock_agentcore_starter_toolkit/operations/runtime/__init__.py,sha256=0jRUuwSoqh4R_WqzfP4XAXngrgyFK5uH8JGXUVars6Y,793
-bedrock_agentcore_starter_toolkit/operations/runtime/configure.py,sha256=wQAYfKfYOp_YXh90bfwzMWB8vhW76fndZ30q-nbfE4M,12823
+bedrock_agentcore_starter_toolkit/operations/runtime/configure.py,sha256=qNioReDz0xW0TTc1ihpcd6Cm1t3cM0u4kYrxMttivYI,12848
 bedrock_agentcore_starter_toolkit/operations/runtime/create_role.py,sha256=1-b_wBvkOXNh-HJsxATwHfXQqj0dpdETds0FNSTY0BE,16116
 bedrock_agentcore_starter_toolkit/operations/runtime/destroy.py,sha256=w80NppiAFEQefrQpBoF4YBL2pCreKM_TFNYG68irdYs,25147
 bedrock_agentcore_starter_toolkit/operations/runtime/invoke.py,sha256=49bfVSGICisXlOiqtFc6vUfd24z3ibQxPPeZy1QMoWw,6764
@@ -48,7 +48,7 @@ bedrock_agentcore_starter_toolkit/operations/runtime/status.py,sha256=4lK-zp29rZ
 bedrock_agentcore_starter_toolkit/services/__init__.py,sha256=s7QtYYFCCX2ff0gZHUdDxCDI3zhUq0fPsjevZbF9xdA,66
 bedrock_agentcore_starter_toolkit/services/codebuild.py,sha256=-ys84zixz2Y4g3kATnITrQ5cjREU_aZjjgMc3D_giFU,14311
 bedrock_agentcore_starter_toolkit/services/ecr.py,sha256=nW9wIZcXI6amZeLVSmM9F6awVBQP1-zrFXTozSNEDjo,2805
-bedrock_agentcore_starter_toolkit/services/runtime.py,sha256=NmjsGfOfdUr-ySFB6hDd9YjIFJsdRied-yVCQhbcOdI,22365
+bedrock_agentcore_starter_toolkit/services/runtime.py,sha256=lqBB-4uJuxNKbnPHpRskQ7bajiZ2MpjnbqPnKnvTinY,22408
 bedrock_agentcore_starter_toolkit/services/xray.py,sha256=CRlcfVXpghVy7PvZqLUC4rp49Sw5DQ98rUU61HAluRM,6363
 bedrock_agentcore_starter_toolkit/services/import_agent/__init__.py,sha256=ig-xanZqA3oJdRTucYz9xF9_2yi31ADRVIKFsnIw_l4,68
 bedrock_agentcore_starter_toolkit/services/import_agent/utils.py,sha256=yCnzqv3S8sbQi6ArTz3MeR4ggRcojbAgMbBb7KAf0sA,16077
@@ -63,18 +63,18 @@ bedrock_agentcore_starter_toolkit/services/import_agent/scripts/bedrock_to_stran
 bedrock_agentcore_starter_toolkit/utils/endpoints.py,sha256=1gIDRd1oO1fymYpiedVit7m6zl5k6N8Ns9N-2ix7ZaE,1153
 bedrock_agentcore_starter_toolkit/utils/logging_config.py,sha256=NtZDyndNKCAbz7jZ0leb13bb3UmjjRUTSVwI8MMlOfw,2191
 bedrock_agentcore_starter_toolkit/utils/runtime/config.py,sha256=qRQid59rD3zJ0d0hcBY4-8R52PNIWEIUt9iR3biuz_c,4495
-bedrock_agentcore_starter_toolkit/utils/runtime/container.py,sha256=vrwIIoDCJ_rLLGbJLbLRhjlBzjdVBnR3I0HjuByJQX4,15839
+bedrock_agentcore_starter_toolkit/utils/runtime/container.py,sha256=h6IxLFORkyVaZIhx2flrLGf2qFxVXXD26C0F2FtsAjo,15923
 bedrock_agentcore_starter_toolkit/utils/runtime/entrypoint.py,sha256=d-XjEwvQOdpRHvBGLdIBCs1fgUwNwB0EL_WkcdQXwXQ,5893
 bedrock_agentcore_starter_toolkit/utils/runtime/logs.py,sha256=ZZ9PD4QO0BSms5KphhUb3-6-IPTkmwkY-Zn2AWM9Aew,1601
 bedrock_agentcore_starter_toolkit/utils/runtime/policy_template.py,sha256=CgER7YXPh0BpR6JcTcumDL_k8bhmfLSEok1sf09-31I,2054
-bedrock_agentcore_starter_toolkit/utils/runtime/schema.py,sha256=19YP2yd7g5-JdTz-S93iXsrubKo9c6VYpDDtgrh8luU,7590
-bedrock_agentcore_starter_toolkit/utils/runtime/templates/Dockerfile.j2,sha256=fnnw8DsRcda2ZQDvLxAz1RYTml_IMG5q0d0iKn3DB1I,1595
+bedrock_agentcore_starter_toolkit/utils/runtime/schema.py,sha256=019Lt1RkDka7ZiM5uCnucY69-vC5LYIhY-Os0AxXsS4,7959
+bedrock_agentcore_starter_toolkit/utils/runtime/templates/Dockerfile.j2,sha256=3CK9AG26w3tMmLA6XDI_rpjZZNDhneJj3ghtTM9YAf8,1607
 bedrock_agentcore_starter_toolkit/utils/runtime/templates/dockerignore.template,sha256=b_70sBi0MwkTuA9TqxPqFcWK7TcmpaXBJ6M2Ipox65Q,691
 bedrock_agentcore_starter_toolkit/utils/runtime/templates/execution_role_policy.json.j2,sha256=hsYLCgaWMfXuwva90yb1DVEO_HUkIjHHNcPKb60E18k,6180
 bedrock_agentcore_starter_toolkit/utils/runtime/templates/execution_role_trust_policy.json.j2,sha256=PPJF6Ofq70W5DUE0NlbmnZjw5RkgepPgkskxEgEG28o,473
-bedrock_agentcore_starter_toolkit-0.1.19.dist-info/METADATA,sha256=GG02BGPitQkKkWVqf2DI9VmjRq8l_hpo7qLR1ETsWaE,9999
-bedrock_agentcore_starter_toolkit-0.1.19.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-bedrock_agentcore_starter_toolkit-0.1.19.dist-info/entry_points.txt,sha256=Tf94DkUf2Tp8P7p8MEXLxre7A7Pp_XNukteiz0wHnk8,77
-bedrock_agentcore_starter_toolkit-0.1.19.dist-info/licenses/LICENSE.txt,sha256=nNPOMinitYdtfbhdQgsPgz1UowBznU6QVN3Xs0pSTKU,10768
-bedrock_agentcore_starter_toolkit-0.1.19.dist-info/licenses/NOTICE.txt,sha256=rkBsg8DbKqfIoQbveqX9foR4uJPUVAokbkr02pRPilE,8674
-bedrock_agentcore_starter_toolkit-0.1.19.dist-info/RECORD,,
+bedrock_agentcore_starter_toolkit-0.1.21.dist-info/METADATA,sha256=Bn1VGJrPP0p5HvynOBD1-fktM-915TBCw_brI8AKa0c,9999
+bedrock_agentcore_starter_toolkit-0.1.21.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+bedrock_agentcore_starter_toolkit-0.1.21.dist-info/entry_points.txt,sha256=Tf94DkUf2Tp8P7p8MEXLxre7A7Pp_XNukteiz0wHnk8,77
+bedrock_agentcore_starter_toolkit-0.1.21.dist-info/licenses/LICENSE.txt,sha256=nNPOMinitYdtfbhdQgsPgz1UowBznU6QVN3Xs0pSTKU,10768
+bedrock_agentcore_starter_toolkit-0.1.21.dist-info/licenses/NOTICE.txt,sha256=rkBsg8DbKqfIoQbveqX9foR4uJPUVAokbkr02pRPilE,8674
+bedrock_agentcore_starter_toolkit-0.1.21.dist-info/RECORD,,