agentcore-cli 0.31.1__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- agentcore/__init__.py +1 -0
- agentcore/agent_inspector/NOTICE +9 -0
- agentcore/agent_inspector/favicon.svg +60 -0
- agentcore/agent_inspector/index.css +1 -0
- agentcore/agent_inspector/index.html +14 -0
- agentcore/agent_inspector/index.js +289 -0
- agentcore/assets/AGENTS.md +60 -0
- agentcore/assets/README.md +108 -0
- agentcore/assets/agents/AGENTS.md +218 -0
- agentcore/assets/cdk/.prettierrc +8 -0
- agentcore/assets/cdk/README.md +26 -0
- agentcore/assets/cdk/bin/cdk.ts +216 -0
- agentcore/assets/cdk/cdk.json +88 -0
- agentcore/assets/cdk/gitignore.template +9 -0
- agentcore/assets/cdk/jest.config.js +9 -0
- agentcore/assets/cdk/lib/cdk-stack.ts +291 -0
- agentcore/assets/cdk/npmignore.template +6 -0
- agentcore/assets/cdk/package.json +30 -0
- agentcore/assets/cdk/test/cdk.test.ts +31 -0
- agentcore/assets/cdk/tsconfig.json +28 -0
- agentcore/assets/container/python/Dockerfile +47 -0
- agentcore/assets/container/python/dockerignore.template +27 -0
- agentcore/assets/container/typescript/Dockerfile +25 -0
- agentcore/assets/container/typescript/dockerignore.template +24 -0
- agentcore/assets/datasets/predefined-v1.jsonl +3 -0
- agentcore/assets/datasets/simulated-v1.jsonl +3 -0
- agentcore/assets/evaluators/autoevals-lambda/execution-role-policy.json +15 -0
- agentcore/assets/evaluators/autoevals-lambda/lambda_function.py +37 -0
- agentcore/assets/evaluators/autoevals-lambda/pyproject.toml +22 -0
- agentcore/assets/evaluators/deepeval-lambda/execution-role-policy.json +15 -0
- agentcore/assets/evaluators/deepeval-lambda/lambda_function.py +29 -0
- agentcore/assets/evaluators/deepeval-lambda/pyproject.toml +19 -0
- agentcore/assets/evaluators/python-lambda/execution-role-policy.json +10 -0
- agentcore/assets/evaluators/python-lambda/lambda_function.py +19 -0
- agentcore/assets/evaluators/python-lambda/pyproject.toml +15 -0
- agentcore/assets/harness/invoke.py.template +74 -0
- agentcore/assets/mcp/python/README.md +27 -0
- agentcore/assets/mcp/python/pyproject.toml +22 -0
- agentcore/assets/mcp/python/server.py +117 -0
- agentcore/assets/mcp/python-lambda/README.md +22 -0
- agentcore/assets/mcp/python-lambda/handler.py +144 -0
- agentcore/assets/mcp/python-lambda/pyproject.toml +15 -0
- agentcore/assets/python/a2a/googleadk/base/README.md +22 -0
- agentcore/assets/python/a2a/googleadk/base/gitignore.template +41 -0
- agentcore/assets/python/a2a/googleadk/base/main.py +124 -0
- agentcore/assets/python/a2a/googleadk/base/model/__init__.py +1 -0
- agentcore/assets/python/a2a/googleadk/base/model/load.py +41 -0
- agentcore/assets/python/a2a/googleadk/base/pyproject.toml +22 -0
- agentcore/assets/python/a2a/langchain_langgraph/base/README.md +22 -0
- agentcore/assets/python/a2a/langchain_langgraph/base/gitignore.template +41 -0
- agentcore/assets/python/a2a/langchain_langgraph/base/main.py +144 -0
- agentcore/assets/python/a2a/langchain_langgraph/base/model/__init__.py +1 -0
- agentcore/assets/python/a2a/langchain_langgraph/base/model/load.py +123 -0
- agentcore/assets/python/a2a/langchain_langgraph/base/pyproject.toml +26 -0
- agentcore/assets/python/a2a/strands/base/README.md +22 -0
- agentcore/assets/python/a2a/strands/base/gitignore.template +41 -0
- agentcore/assets/python/a2a/strands/base/main.py +113 -0
- agentcore/assets/python/a2a/strands/base/model/__init__.py +1 -0
- agentcore/assets/python/a2a/strands/base/model/load.py +123 -0
- agentcore/assets/python/a2a/strands/base/pyproject.toml +23 -0
- agentcore/assets/python/a2a/strands/capabilities/memory/__init__.py +1 -0
- agentcore/assets/python/a2a/strands/capabilities/memory/session.py +46 -0
- agentcore/assets/python/agui/googleadk/base/README.md +30 -0
- agentcore/assets/python/agui/googleadk/base/gitignore.template +41 -0
- agentcore/assets/python/agui/googleadk/base/main.py +91 -0
- agentcore/assets/python/agui/googleadk/base/model/__init__.py +1 -0
- agentcore/assets/python/agui/googleadk/base/model/load.py +41 -0
- agentcore/assets/python/agui/googleadk/base/pyproject.toml +25 -0
- agentcore/assets/python/agui/langchain_langgraph/base/README.md +22 -0
- agentcore/assets/python/agui/langchain_langgraph/base/gitignore.template +41 -0
- agentcore/assets/python/agui/langchain_langgraph/base/main.py +130 -0
- agentcore/assets/python/agui/langchain_langgraph/base/model/__init__.py +1 -0
- agentcore/assets/python/agui/langchain_langgraph/base/model/load.py +123 -0
- agentcore/assets/python/agui/langchain_langgraph/base/pyproject.toml +31 -0
- agentcore/assets/python/agui/strands/base/README.md +22 -0
- agentcore/assets/python/agui/strands/base/gitignore.template +41 -0
- agentcore/assets/python/agui/strands/base/main.py +105 -0
- agentcore/assets/python/agui/strands/base/model/__init__.py +1 -0
- agentcore/assets/python/agui/strands/base/model/load.py +123 -0
- agentcore/assets/python/agui/strands/base/pyproject.toml +27 -0
- agentcore/assets/python/agui/strands/capabilities/memory/__init__.py +1 -0
- agentcore/assets/python/agui/strands/capabilities/memory/session.py +46 -0
- agentcore/assets/python/http/autogen/base/README.md +45 -0
- agentcore/assets/python/http/autogen/base/gitignore.template +41 -0
- agentcore/assets/python/http/autogen/base/main.py +145 -0
- agentcore/assets/python/http/autogen/base/mcp_client/__init__.py +1 -0
- agentcore/assets/python/http/autogen/base/mcp_client/client.py +28 -0
- agentcore/assets/python/http/autogen/base/model/__init__.py +1 -0
- agentcore/assets/python/http/autogen/base/model/load.py +136 -0
- agentcore/assets/python/http/autogen/base/pyproject.toml +34 -0
- agentcore/assets/python/http/bma/base/Dockerfile +41 -0
- agentcore/assets/python/http/bma/base/README.md +178 -0
- agentcore/assets/python/http/bma/base/client.py +200 -0
- agentcore/assets/python/http/bma/base/lifecycle/server.py +641 -0
- agentcore/assets/python/http/bma/base/otel/collector.yaml +69 -0
- agentcore/assets/python/http/bma/base/plugins/acr-report/.codex-plugin/plugin.json +6 -0
- agentcore/assets/python/http/bma/base/plugins/acr-report/skills/acr-report/SKILL.md +16 -0
- agentcore/assets/python/http/bma/base/policies/bma-acr-policy.json +11 -0
- agentcore/assets/python/http/bma/base/policies/bma-session-policy.json +35 -0
- agentcore/assets/python/http/bma/base/policies/bma-session-trust.json +20 -0
- agentcore/assets/python/http/bma/base/pyproject.toml +21 -0
- agentcore/assets/python/http/googleadk/base/README.md +45 -0
- agentcore/assets/python/http/googleadk/base/gitignore.template +41 -0
- agentcore/assets/python/http/googleadk/base/main.py +207 -0
- agentcore/assets/python/http/googleadk/base/mcp_client/__init__.py +1 -0
- agentcore/assets/python/http/googleadk/base/mcp_client/client.py +75 -0
- agentcore/assets/python/http/googleadk/base/model/__init__.py +1 -0
- agentcore/assets/python/http/googleadk/base/model/load.py +41 -0
- agentcore/assets/python/http/googleadk/base/pyproject.toml +24 -0
- agentcore/assets/python/http/langchain_langgraph/base/README.md +45 -0
- agentcore/assets/python/http/langchain_langgraph/base/gitignore.template +41 -0
- agentcore/assets/python/http/langchain_langgraph/base/main.py +227 -0
- agentcore/assets/python/http/langchain_langgraph/base/mcp_client/__init__.py +1 -0
- agentcore/assets/python/http/langchain_langgraph/base/mcp_client/client.py +79 -0
- agentcore/assets/python/http/langchain_langgraph/base/model/__init__.py +1 -0
- agentcore/assets/python/http/langchain_langgraph/base/model/load.py +123 -0
- agentcore/assets/python/http/langchain_langgraph/base/pyproject.toml +38 -0
- agentcore/assets/python/http/openaiagents/base/README.md +45 -0
- agentcore/assets/python/http/openaiagents/base/gitignore.template +41 -0
- agentcore/assets/python/http/openaiagents/base/main.py +200 -0
- agentcore/assets/python/http/openaiagents/base/mcp_client/__init__.py +1 -0
- agentcore/assets/python/http/openaiagents/base/mcp_client/client.py +74 -0
- agentcore/assets/python/http/openaiagents/base/model/__init__.py +1 -0
- agentcore/assets/python/http/openaiagents/base/model/load.py +37 -0
- agentcore/assets/python/http/openaiagents/base/pyproject.toml +21 -0
- agentcore/assets/python/http/strands/base/README.md +46 -0
- agentcore/assets/python/http/strands/base/gitignore.template +41 -0
- agentcore/assets/python/http/strands/base/main.py +714 -0
- agentcore/assets/python/http/strands/base/mcp_client/__init__.py +1 -0
- agentcore/assets/python/http/strands/base/mcp_client/client.py +116 -0
- agentcore/assets/python/http/strands/base/model/__init__.py +1 -0
- agentcore/assets/python/http/strands/base/model/load.py +239 -0
- agentcore/assets/python/http/strands/base/model/mantle_compat.py +21 -0
- agentcore/assets/python/http/strands/base/pyproject.toml +31 -0
- agentcore/assets/python/http/strands/base/skills/fetcher.py +279 -0
- agentcore/assets/python/http/strands/capabilities/execution-limits/hooks/execution_limits.py +54 -0
- agentcore/assets/python/http/strands/capabilities/memory/__init__.py +1 -0
- agentcore/assets/python/http/strands/capabilities/memory/session.py +47 -0
- agentcore/assets/python/http/strands/capabilities/payments/__init__.py +4 -0
- agentcore/assets/python/http/strands/capabilities/payments/payments.py +142 -0
- agentcore/assets/python/mcp/standalone/base/README.md +36 -0
- agentcore/assets/python/mcp/standalone/base/gitignore.template +41 -0
- agentcore/assets/python/mcp/standalone/base/main.py +83 -0
- agentcore/assets/python/mcp/standalone/base/pyproject.toml +16 -0
- agentcore/assets/typescript/.gitkeep +0 -0
- agentcore/assets/typescript/http/strands/base/README.md +42 -0
- agentcore/assets/typescript/http/strands/base/gitignore.template +22 -0
- agentcore/assets/typescript/http/strands/base/main.ts +149 -0
- agentcore/assets/typescript/http/strands/base/mcp_client/client.ts +11 -0
- agentcore/assets/typescript/http/strands/base/model/load.ts +102 -0
- agentcore/assets/typescript/http/strands/base/package.json +38 -0
- agentcore/assets/typescript/http/strands/base/tsconfig.json +19 -0
- agentcore/assets/typescript/http/strands/capabilities/memory/memory.ts +52 -0
- agentcore/assets/typescript/http/vercelai/base/README.md +42 -0
- agentcore/assets/typescript/http/vercelai/base/gitignore.template +22 -0
- agentcore/assets/typescript/http/vercelai/base/main.ts +70 -0
- agentcore/assets/typescript/http/vercelai/base/model/load.ts +121 -0
- agentcore/assets/typescript/http/vercelai/base/package.json +35 -0
- agentcore/assets/typescript/http/vercelai/base/tsconfig.json +19 -0
- agentcore/aws/__init__.py +263 -0
- agentcore/aws/_aws_errors.py +72 -0
- agentcore/aws/_http.py +176 -0
- agentcore/aws/_util.py +37 -0
- agentcore/aws/account.py +165 -0
- agentcore/aws/agentcore.py +971 -0
- agentcore/aws/agentcore_ab_tests.py +230 -0
- agentcore/aws/agentcore_batch_evaluation.py +227 -0
- agentcore/aws/agentcore_config_bundles.py +220 -0
- agentcore/aws/agentcore_control.py +994 -0
- agentcore/aws/agentcore_datasets.py +222 -0
- agentcore/aws/agentcore_harness.py +330 -0
- agentcore/aws/agentcore_payments.py +430 -0
- agentcore/aws/agentcore_recommendation.py +140 -0
- agentcore/aws/agui_parser.py +131 -0
- agentcore/aws/agui_types.py +342 -0
- agentcore/aws/api_client.py +87 -0
- agentcore/aws/arn.py +18 -0
- agentcore/aws/aws_context.py +23 -0
- agentcore/aws/bedrock.py +44 -0
- agentcore/aws/bedrock_agent.py +129 -0
- agentcore/aws/bedrock_import.py +252 -0
- agentcore/aws/bedrock_import_types.py +111 -0
- agentcore/aws/cloudwatch.py +136 -0
- agentcore/aws/connect_shell.py +266 -0
- agentcore/aws/iam.py +41 -0
- agentcore/aws/partition.py +58 -0
- agentcore/aws/policy_generation.py +116 -0
- agentcore/aws/region.py +54 -0
- agentcore/aws/retry.py +27 -0
- agentcore/aws/shell_framer.py +136 -0
- agentcore/aws/stage_endpoint.py +39 -0
- agentcore/aws/target_region.py +54 -0
- agentcore/aws/transaction_search.py +97 -0
- agentcore/cdk/__init__.py +10 -0
- agentcore/cdk/constants.py +13 -0
- agentcore/cdk/local_cdk_project.py +48 -0
- agentcore/cdk/toolkit.py +706 -0
- agentcore/cli/__init__.py +0 -0
- agentcore/cli/commands/__init__.py +0 -0
- agentcore/cli/commands/_jobs_common.py +39 -0
- agentcore/cli/commands/_run_wizards.py +639 -0
- agentcore/cli/commands/add/__init__.py +37 -0
- agentcore/cli/commands/add/_commander.py +52 -0
- agentcore/cli/commands/add/_common.py +47 -0
- agentcore/cli/commands/add/agent_wizard.py +528 -0
- agentcore/cli/commands/add/auth_options.py +154 -0
- agentcore/cli/commands/add/skill_action.py +105 -0
- agentcore/cli/commands/add/skill_command.py +87 -0
- agentcore/cli/commands/add/tool_action.py +160 -0
- agentcore/cli/commands/add/tool_command.py +117 -0
- agentcore/cli/commands/add/validate.py +333 -0
- agentcore/cli/commands/capacity_provider/__init__.py +86 -0
- agentcore/cli/commands/capacity_provider/action.py +114 -0
- agentcore/cli/commands/capacity_provider/constants.py +22 -0
- agentcore/cli/commands/config.py +91 -0
- agentcore/cli/commands/config_bundle.py +251 -0
- agentcore/cli/commands/create/__init__.py +643 -0
- agentcore/cli/commands/create/action.py +406 -0
- agentcore/cli/commands/create/dependencies.py +91 -0
- agentcore/cli/commands/create/harness_validate.py +82 -0
- agentcore/cli/commands/create/validate.py +195 -0
- agentcore/cli/commands/create/wizard.py +127 -0
- agentcore/cli/commands/dataset.py +119 -0
- agentcore/cli/commands/deploy/__init__.py +31 -0
- agentcore/cli/commands/deploy/_guards.py +25 -0
- agentcore/cli/commands/deploy/actions.py +840 -0
- agentcore/cli/commands/deploy/command.py +349 -0
- agentcore/cli/commands/deploy/progress.py +115 -0
- agentcore/cli/commands/deploy/types.py +68 -0
- agentcore/cli/commands/deploy/utils.py +43 -0
- agentcore/cli/commands/deploy/validate.py +21 -0
- agentcore/cli/commands/dev/__init__.py +5 -0
- agentcore/cli/commands/dev/_deps.py +50 -0
- agentcore/cli/commands/dev/browser_mode.py +270 -0
- agentcore/cli/commands/dev/command.py +450 -0
- agentcore/cli/commands/dev/terminal_mode.py +379 -0
- agentcore/cli/commands/exec/__init__.py +29 -0
- agentcore/cli/commands/exec/action.py +442 -0
- agentcore/cli/commands/exec/command.py +206 -0
- agentcore/cli/commands/exec/types.py +35 -0
- agentcore/cli/commands/export/__init__.py +114 -0
- agentcore/cli/commands/export/constants.py +21 -0
- agentcore/cli/commands/export/fetch_harness_spec.py +234 -0
- agentcore/cli/commands/export/harness_action.py +385 -0
- agentcore/cli/commands/export/harness_mapper.py +868 -0
- agentcore/cli/commands/export/harness_resolver.py +95 -0
- agentcore/cli/commands/export/types.py +90 -0
- agentcore/cli/commands/export/wizard.py +97 -0
- agentcore/cli/commands/feedback/__init__.py +69 -0
- agentcore/cli/commands/feedback/action.py +30 -0
- agentcore/cli/commands/feedback/consent_prompt.py +35 -0
- agentcore/cli/commands/feedback/wizard.py +49 -0
- agentcore/cli/commands/fetch/__init__.py +69 -0
- agentcore/cli/commands/fetch/action.py +62 -0
- agentcore/cli/commands/help.py +65 -0
- agentcore/cli/commands/import_/__init__.py +167 -0
- agentcore/cli/commands/import_/_js.py +72 -0
- agentcore/cli/commands/import_/actions.py +457 -0
- agentcore/cli/commands/import_/constants.py +45 -0
- agentcore/cli/commands/import_/import_evaluator.py +140 -0
- agentcore/cli/commands/import_/import_gateway.py +520 -0
- agentcore/cli/commands/import_/import_memory.py +128 -0
- agentcore/cli/commands/import_/import_online_eval.py +165 -0
- agentcore/cli/commands/import_/import_pipeline.py +145 -0
- agentcore/cli/commands/import_/import_runtime.py +214 -0
- agentcore/cli/commands/import_/import_utils.py +410 -0
- agentcore/cli/commands/import_/phase1_update.py +110 -0
- agentcore/cli/commands/import_/phase2_import.py +206 -0
- agentcore/cli/commands/import_/resource_import.py +228 -0
- agentcore/cli/commands/import_/template_utils.py +148 -0
- agentcore/cli/commands/import_/types.py +150 -0
- agentcore/cli/commands/import_/yaml_parser.py +265 -0
- agentcore/cli/commands/init.py +196 -0
- agentcore/cli/commands/invoke/__init__.py +32 -0
- agentcore/cli/commands/invoke/_deps.py +73 -0
- agentcore/cli/commands/invoke/action.py +840 -0
- agentcore/cli/commands/invoke/command.py +490 -0
- agentcore/cli/commands/invoke/constants.py +40 -0
- agentcore/cli/commands/invoke/guards.py +23 -0
- agentcore/cli/commands/invoke/interactive.py +604 -0
- agentcore/cli/commands/invoke/resolve.py +149 -0
- agentcore/cli/commands/invoke/resolve_prompt.py +60 -0
- agentcore/cli/commands/invoke/types.py +61 -0
- agentcore/cli/commands/invoke/validate.py +33 -0
- agentcore/cli/commands/job_actions.py +239 -0
- agentcore/cli/commands/job_history.py +380 -0
- agentcore/cli/commands/logs/__init__.py +53 -0
- agentcore/cli/commands/logs/action.py +159 -0
- agentcore/cli/commands/logs/filter_pattern.py +17 -0
- agentcore/cli/commands/logs/guards.py +51 -0
- agentcore/cli/commands/logs_evals.py +42 -0
- agentcore/cli/commands/package.py +118 -0
- agentcore/cli/commands/remove/__init__.py +172 -0
- agentcore/cli/commands/remove/skill_command.py +131 -0
- agentcore/cli/commands/remove/tool_command.py +59 -0
- agentcore/cli/commands/run.py +808 -0
- agentcore/cli/commands/run_ingest.py +97 -0
- agentcore/cli/commands/shared/__init__.py +0 -0
- agentcore/cli/commands/shared/arn_utils.py +9 -0
- agentcore/cli/commands/shared/filesystem_utils.py +421 -0
- agentcore/cli/commands/shared/header_utils.py +87 -0
- agentcore/cli/commands/shared/indexed_key_parser.py +76 -0
- agentcore/cli/commands/shared/lifecycle_utils.py +72 -0
- agentcore/cli/commands/shared/region_utils.py +20 -0
- agentcore/cli/commands/shared/vpc_utils.py +99 -0
- agentcore/cli/commands/status/__init__.py +30 -0
- agentcore/cli/commands/status/_guards.py +25 -0
- agentcore/cli/commands/status/action.py +741 -0
- agentcore/cli/commands/status/command.py +321 -0
- agentcore/cli/commands/status/constants.py +36 -0
- agentcore/cli/commands/status/format_knowledge_base.py +111 -0
- agentcore/cli/commands/telemetry.py +79 -0
- agentcore/cli/commands/traces/__init__.py +155 -0
- agentcore/cli/commands/traces/action.py +123 -0
- agentcore/cli/commands/update/__init__.py +63 -0
- agentcore/cli/commands/update/action.py +164 -0
- agentcore/cli/commands/validate.py +165 -0
- agentcore/cli/constants.py +96 -0
- agentcore/cli/errors.py +118 -0
- agentcore/cli/external_requirements/__init__.py +67 -0
- agentcore/cli/external_requirements/checks.py +280 -0
- agentcore/cli/external_requirements/detect.py +55 -0
- agentcore/cli/external_requirements/versions.py +57 -0
- agentcore/cli/groups.py +23 -0
- agentcore/cli/interactive.py +85 -0
- agentcore/cli/main.py +137 -0
- agentcore/cli/naming.py +11 -0
- agentcore/cli/notices.py +42 -0
- agentcore/cli/project.py +34 -0
- agentcore/cli/run_command.py +27 -0
- agentcore/cli/tree.py +135 -0
- agentcore/cli/update_notifier.py +123 -0
- agentcore/cli/variadic.py +50 -0
- agentcore/cloudformation/__init__.py +84 -0
- agentcore/cloudformation/bootstrap.py +64 -0
- agentcore/cloudformation/logical_ids.py +37 -0
- agentcore/cloudformation/outputs.py +566 -0
- agentcore/cloudformation/stack_discovery.py +75 -0
- agentcore/cloudformation/stack_events.py +107 -0
- agentcore/cloudformation/stack_status.py +98 -0
- agentcore/cloudformation/types.py +44 -0
- agentcore/lib/__init__.py +68 -0
- agentcore/lib/constants.py +77 -0
- agentcore/lib/dependency_management/__init__.py +23 -0
- agentcore/lib/dependency_management/messages.py +92 -0
- agentcore/lib/dependency_management/plan.py +98 -0
- agentcore/lib/dependency_management/semver.py +137 -0
- agentcore/lib/dependency_management/sync.py +139 -0
- agentcore/lib/dependency_management/types.py +56 -0
- agentcore/lib/errors/__init__.py +5 -0
- agentcore/lib/errors/types.py +352 -0
- agentcore/lib/errors/validation.py +121 -0
- agentcore/lib/packaging/__init__.py +81 -0
- agentcore/lib/packaging/build_args.py +23 -0
- agentcore/lib/packaging/build_context.py +26 -0
- agentcore/lib/packaging/build_context_dockerignore.py +45 -0
- agentcore/lib/packaging/container.py +95 -0
- agentcore/lib/packaging/helpers.py +289 -0
- agentcore/lib/packaging/node.py +153 -0
- agentcore/lib/packaging/python.py +135 -0
- agentcore/lib/packaging/types.py +69 -0
- agentcore/lib/packaging/uv.py +45 -0
- agentcore/lib/result.py +104 -0
- agentcore/lib/schemas/__init__.py +1 -0
- agentcore/lib/schemas/io/__init__.py +27 -0
- agentcore/lib/schemas/io/config_io.py +458 -0
- agentcore/lib/schemas/io/global_config.py +239 -0
- agentcore/lib/schemas/io/path_resolver.py +132 -0
- agentcore/lib/secrets/__init__.py +18 -0
- agentcore/lib/secrets/cipher.py +70 -0
- agentcore/lib/secrets/key_provider.py +163 -0
- agentcore/lib/secrets/sensitive_keys.py +55 -0
- agentcore/lib/time_constants.py +4 -0
- agentcore/lib/utils/__init__.py +47 -0
- agentcore/lib/utils/aws_account.py +15 -0
- agentcore/lib/utils/credentials.py +68 -0
- agentcore/lib/utils/dotenv.py +32 -0
- agentcore/lib/utils/env.py +105 -0
- agentcore/lib/utils/json_format.py +29 -0
- agentcore/lib/utils/json_rpc.py +22 -0
- agentcore/lib/utils/platform.py +33 -0
- agentcore/lib/utils/polling.py +86 -0
- agentcore/lib/utils/subprocess.py +135 -0
- agentcore/lib/utils/time_parser.py +48 -0
- agentcore/lib/utils/validation.py +106 -0
- agentcore/logging/__init__.py +10 -0
- agentcore/logging/_common.py +77 -0
- agentcore/logging/cdk_io_host.py +36 -0
- agentcore/logging/create_logger.py +107 -0
- agentcore/logging/dev_logger.py +63 -0
- agentcore/logging/exec_logger.py +159 -0
- agentcore/logging/invoke_logger.py +154 -0
- agentcore/logging/remove_logger.py +95 -0
- agentcore/operations/__init__.py +0 -0
- agentcore/operations/agent/__init__.py +0 -0
- agentcore/operations/agent/config_bundle_defaults.py +31 -0
- agentcore/operations/agent/generate/__init__.py +28 -0
- agentcore/operations/agent/generate/schema_mapper.py +319 -0
- agentcore/operations/agent/generate/types.py +172 -0
- agentcore/operations/agent/generate/write_agent_to_project.py +67 -0
- agentcore/operations/agent/import_/__init__.py +31 -0
- agentcore/operations/agent/import_/base_translator.py +449 -0
- agentcore/operations/agent/import_/constants.py +23 -0
- agentcore/operations/agent/import_/execute.py +127 -0
- agentcore/operations/agent/import_/langgraph_translator.py +294 -0
- agentcore/operations/agent/import_/pyproject_generator.py +73 -0
- agentcore/operations/agent/import_/strands_translator.py +289 -0
- agentcore/operations/config_bundle/__init__.py +1 -0
- agentcore/operations/config_bundle/bundle_name_variants.py +9 -0
- agentcore/operations/config_bundle/diff_versions.py +52 -0
- agentcore/operations/config_bundle/resolve_bundle.py +65 -0
- agentcore/operations/connectors/__init__.py +3 -0
- agentcore/operations/connectors/translators.py +63 -0
- agentcore/operations/dataset/__init__.py +21 -0
- agentcore/operations/dataset/delete.py +7 -0
- agentcore/operations/dataset/publish.py +16 -0
- agentcore/operations/dataset/pull.py +21 -0
- agentcore/operations/dataset/push.py +201 -0
- agentcore/operations/dataset/resolve_dataset.py +65 -0
- agentcore/operations/dataset/status.py +18 -0
- agentcore/operations/dataset/wait.py +25 -0
- agentcore/operations/deploy/__init__.py +152 -0
- agentcore/operations/deploy/backfill_vpc_id.py +90 -0
- agentcore/operations/deploy/change_detection.py +84 -0
- agentcore/operations/deploy/constants.py +58 -0
- agentcore/operations/deploy/dependency_sync.py +105 -0
- agentcore/operations/deploy/ensure_target.py +30 -0
- agentcore/operations/deploy/gateway_status.py +44 -0
- agentcore/operations/deploy/managed_memory_notice.py +36 -0
- agentcore/operations/deploy/payment_connector_authorization.py +106 -0
- agentcore/operations/deploy/post_deploy_datasets.py +96 -0
- agentcore/operations/deploy/post_deploy_knowledge_bases.py +104 -0
- agentcore/operations/deploy/post_deploy_observability.py +28 -0
- agentcore/operations/deploy/post_deploy_online_evals.py +60 -0
- agentcore/operations/deploy/pre_deploy_identity.py +505 -0
- agentcore/operations/deploy/preflight.py +385 -0
- agentcore/operations/deploy/region_override.py +45 -0
- agentcore/operations/deploy/teardown.py +153 -0
- agentcore/operations/dev/__init__.py +61 -0
- agentcore/operations/dev/_http.py +79 -0
- agentcore/operations/dev/_server.py +165 -0
- agentcore/operations/dev/codezip_dev_server.py +151 -0
- agentcore/operations/dev/config.py +128 -0
- agentcore/operations/dev/constants.py +9 -0
- agentcore/operations/dev/container_dev_server.py +237 -0
- agentcore/operations/dev/container_runtime.py +37 -0
- agentcore/operations/dev/dev_server.py +244 -0
- agentcore/operations/dev/gateway_env.py +25 -0
- agentcore/operations/dev/invoke.py +214 -0
- agentcore/operations/dev/invoke_a2a.py +204 -0
- agentcore/operations/dev/invoke_agui.py +88 -0
- agentcore/operations/dev/invoke_mcp.py +145 -0
- agentcore/operations/dev/invoke_types.py +30 -0
- agentcore/operations/dev/load_dev_env.py +26 -0
- agentcore/operations/dev/memory_env.py +19 -0
- agentcore/operations/dev/otel/__init__.py +5 -0
- agentcore/operations/dev/otel/collector.py +234 -0
- agentcore/operations/dev/otel/transforms.py +331 -0
- agentcore/operations/dev/payment_env.py +50 -0
- agentcore/operations/dev/server.py +28 -0
- agentcore/operations/dev/sse_transform.py +30 -0
- agentcore/operations/dev/utils.py +152 -0
- agentcore/operations/dev/web_ui/__init__.py +31 -0
- agentcore/operations/dev/web_ui/constants.py +25 -0
- agentcore/operations/dev/web_ui/handlers/__init__.py +36 -0
- agentcore/operations/dev/web_ui/handlers/_common.py +36 -0
- agentcore/operations/dev/web_ui/handlers/_deployed.py +37 -0
- agentcore/operations/dev/web_ui/handlers/a2a_proxy.py +37 -0
- agentcore/operations/dev/web_ui/handlers/cloudwatch_traces.py +71 -0
- agentcore/operations/dev/web_ui/handlers/harness_invocation.py +89 -0
- agentcore/operations/dev/web_ui/handlers/harness_utils.py +95 -0
- agentcore/operations/dev/web_ui/handlers/invocations.py +327 -0
- agentcore/operations/dev/web_ui/handlers/mcp_proxy.py +119 -0
- agentcore/operations/dev/web_ui/handlers/memory.py +84 -0
- agentcore/operations/dev/web_ui/handlers/resources.py +351 -0
- agentcore/operations/dev/web_ui/handlers/start.py +182 -0
- agentcore/operations/dev/web_ui/handlers/status.py +25 -0
- agentcore/operations/dev/web_ui/handlers/traces.py +50 -0
- agentcore/operations/dev/web_ui/run_web_ui.py +60 -0
- agentcore/operations/dev/web_ui/web_server.py +297 -0
- agentcore/operations/eval/__init__.py +8 -0
- agentcore/operations/eval/list_eval_runs.py +18 -0
- agentcore/operations/eval/logs_eval.py +119 -0
- agentcore/operations/eval/pause_resume.py +90 -0
- agentcore/operations/eval/run_eval.py +354 -0
- agentcore/operations/eval/shared/__init__.py +0 -0
- agentcore/operations/eval/shared/dataset_loader.py +68 -0
- agentcore/operations/eval/shared/dataset_session_provider.py +122 -0
- agentcore/operations/eval/shared/evaluator_runner.py +89 -0
- agentcore/operations/eval/shared/scenario_executor.py +71 -0
- agentcore/operations/eval/shared/span_collector.py +255 -0
- agentcore/operations/eval/storage.py +67 -0
- agentcore/operations/feedback/__init__.py +21 -0
- agentcore/operations/feedback/aperture_client.py +98 -0
- agentcore/operations/feedback/build_payload.py +97 -0
- agentcore/operations/feedback/constants.py +26 -0
- agentcore/operations/feedback/submit_feedback.py +150 -0
- agentcore/operations/fetch_access/__init__.py +36 -0
- agentcore/operations/fetch_access/_target.py +17 -0
- agentcore/operations/fetch_access/fetch_gateway_token.py +75 -0
- agentcore/operations/fetch_access/fetch_harness_token.py +72 -0
- agentcore/operations/fetch_access/fetch_runtime_token.py +76 -0
- agentcore/operations/fetch_access/list_agents.py +18 -0
- agentcore/operations/fetch_access/list_gateways.py +26 -0
- agentcore/operations/fetch_access/list_harnesses.py +23 -0
- agentcore/operations/fetch_access/oauth_token.py +162 -0
- agentcore/operations/fetch_access/types.py +34 -0
- agentcore/operations/harness/__init__.py +1 -0
- agentcore/operations/harness/orphan.py +84 -0
- agentcore/operations/harness/skill_utils.py +56 -0
- agentcore/operations/identity/__init__.py +33 -0
- agentcore/operations/identity/api_key_credential_provider.py +68 -0
- agentcore/operations/identity/oauth2_credential_provider.py +95 -0
- agentcore/operations/ingest/__init__.py +184 -0
- agentcore/operations/init/__init__.py +3 -0
- agentcore/operations/init/files.py +59 -0
- agentcore/operations/insights/__init__.py +21 -0
- agentcore/operations/insights/insights_storage.py +67 -0
- agentcore/operations/insights/run_insights.py +148 -0
- agentcore/operations/invoke/__init__.py +5 -0
- agentcore/operations/invoke/resolve_agent_context.py +85 -0
- agentcore/operations/jobs/__init__.py +38 -0
- agentcore/operations/jobs/ab_test/__init__.py +0 -0
- agentcore/operations/jobs/ab_test/build_options.py +126 -0
- agentcore/operations/jobs/ab_test/format.py +121 -0
- agentcore/operations/jobs/ab_test/handler.py +412 -0
- agentcore/operations/jobs/ab_test/promote.py +142 -0
- agentcore/operations/jobs/ab_test/resolve.py +221 -0
- agentcore/operations/jobs/batch_evaluation/__init__.py +0 -0
- agentcore/operations/jobs/batch_evaluation/build_source.py +68 -0
- agentcore/operations/jobs/batch_evaluation/dataset_phase1.py +95 -0
- agentcore/operations/jobs/batch_evaluation/format.py +58 -0
- agentcore/operations/jobs/batch_evaluation/handler.py +203 -0
- agentcore/operations/jobs/insights/__init__.py +0 -0
- agentcore/operations/jobs/insights/format.py +65 -0
- agentcore/operations/jobs/insights/handler.py +178 -0
- agentcore/operations/jobs/recommendation/__init__.py +0 -0
- agentcore/operations/jobs/recommendation/apply_to_bundle.py +76 -0
- agentcore/operations/jobs/recommendation/build_config.py +271 -0
- agentcore/operations/jobs/recommendation/fetch_session_spans.py +93 -0
- agentcore/operations/jobs/recommendation/format.py +77 -0
- agentcore/operations/jobs/recommendation/handler.py +331 -0
- agentcore/operations/jobs/shared/__init__.py +0 -0
- agentcore/operations/jobs/shared/constants.py +85 -0
- agentcore/operations/jobs/shared/engine.py +187 -0
- agentcore/operations/jobs/shared/format.py +104 -0
- agentcore/operations/jobs/shared/region.py +34 -0
- agentcore/operations/jobs/shared/resolve_agent_state.py +30 -0
- agentcore/operations/jobs/shared/storage.py +117 -0
- agentcore/operations/jobs/shared/wait.py +30 -0
- agentcore/operations/knowledge_base/__init__.py +33 -0
- agentcore/operations/knowledge_base/connector_config.py +105 -0
- agentcore/operations/knowledge_base/hydrate_data_sources.py +59 -0
- agentcore/operations/mcp/__init__.py +4 -0
- agentcore/operations/memory/__init__.py +13 -0
- agentcore/operations/memory/_records.py +72 -0
- agentcore/operations/memory/list_memory_records.py +45 -0
- agentcore/operations/memory/retrieve_memory_records.py +52 -0
- agentcore/operations/node/__init__.py +15 -0
- agentcore/operations/node/setup.py +36 -0
- agentcore/operations/python/__init__.py +17 -0
- agentcore/operations/python/setup.py +47 -0
- agentcore/operations/resolve_agent.py +159 -0
- agentcore/operations/session/__init__.py +86 -0
- agentcore/operations/traces/__init__.py +29 -0
- agentcore/operations/traces/get_trace.py +202 -0
- agentcore/operations/traces/insights_query.py +74 -0
- agentcore/operations/traces/list_traces.py +46 -0
- agentcore/operations/traces/trace_url.py +25 -0
- agentcore/operations/traces/types.py +55 -0
- agentcore/primitives/__init__.py +0 -0
- agentcore/primitives/_gateway_target_wizard.py +293 -0
- agentcore/primitives/_gateway_validation.py +442 -0
- agentcore/primitives/_harness_cli.py +557 -0
- agentcore/primitives/_harness_validation.py +163 -0
- agentcore/primitives/_harness_wizard.py +726 -0
- agentcore/primitives/_knowledge_base_wizard.py +259 -0
- agentcore/primitives/_payment_wizard.py +331 -0
- agentcore/primitives/_policy_form.py +160 -0
- agentcore/primitives/agent.py +817 -0
- agentcore/primitives/auth_utils.py +77 -0
- agentcore/primitives/base.py +178 -0
- agentcore/primitives/capacity_provider.py +501 -0
- agentcore/primitives/config_bundle.py +329 -0
- agentcore/primitives/credential.py +387 -0
- agentcore/primitives/credential_utils.py +45 -0
- agentcore/primitives/dataset.py +211 -0
- agentcore/primitives/deployed_gateways.py +34 -0
- agentcore/primitives/evaluator.py +909 -0
- agentcore/primitives/gateway.py +496 -0
- agentcore/primitives/gateway_target.py +988 -0
- agentcore/primitives/harness.py +706 -0
- agentcore/primitives/knowledge_base.py +477 -0
- agentcore/primitives/memory.py +387 -0
- agentcore/primitives/online_eval_config.py +365 -0
- agentcore/primitives/online_insights.py +275 -0
- agentcore/primitives/payment_connector.py +427 -0
- agentcore/primitives/payment_eligible.py +20 -0
- agentcore/primitives/payment_manager.py +605 -0
- agentcore/primitives/payment_validation.py +68 -0
- agentcore/primitives/policy.py +622 -0
- agentcore/primitives/policy_engine.py +358 -0
- agentcore/primitives/registry.py +56 -0
- agentcore/primitives/runtime_endpoint.py +361 -0
- agentcore/primitives/secret_flag_warning.py +17 -0
- agentcore/schema/__init__.py +14 -0
- agentcore/schema/_zod.py +395 -0
- agentcore/schema/constants.py +306 -0
- agentcore/schema/schemas/__init__.py +40 -0
- agentcore/schema/schemas/agent_env.py +545 -0
- agentcore/schema/schemas/agentcore_project.py +530 -0
- agentcore/schema/schemas/auth.py +286 -0
- agentcore/schema/schemas/aws_targets.py +72 -0
- agentcore/schema/schemas/connections.py +187 -0
- agentcore/schema/schemas/deployed_state.py +322 -0
- agentcore/schema/schemas/mcp.py +650 -0
- agentcore/schema/schemas/mcp_defs.py +54 -0
- agentcore/schema/schemas/primitives/__init__.py +52 -0
- agentcore/schema/schemas/primitives/ab_test.py +162 -0
- agentcore/schema/schemas/primitives/capacity_provider.py +214 -0
- agentcore/schema/schemas/primitives/config_bundle.py +48 -0
- agentcore/schema/schemas/primitives/dataset.py +55 -0
- agentcore/schema/schemas/primitives/evaluator.py +171 -0
- agentcore/schema/schemas/primitives/harness.py +710 -0
- agentcore/schema/schemas/primitives/knowledge_base.py +90 -0
- agentcore/schema/schemas/primitives/memory.py +113 -0
- agentcore/schema/schemas/primitives/online_eval_config.py +77 -0
- agentcore/schema/schemas/primitives/payment.py +114 -0
- agentcore/schema/schemas/primitives/policy.py +83 -0
- agentcore/schema/schemas/primitives/tags.py +50 -0
- agentcore/schema/schemas/zod_util.py +5 -0
- agentcore/schema/types/__init__.py +3 -0
- agentcore/schema/types/path.py +16 -0
- agentcore/templates/__init__.py +115 -0
- agentcore/templates/_handlebars.py +782 -0
- agentcore/templates/_js.py +164 -0
- agentcore/templates/base_renderer.py +107 -0
- agentcore/templates/cdk_renderer.py +129 -0
- agentcore/templates/evaluator_renderer.py +39 -0
- agentcore/templates/gateway_target_renderer.py +69 -0
- agentcore/templates/llm_context/README.md +16 -0
- agentcore/templates/llm_context/agentcore.ts +685 -0
- agentcore/templates/llm_context/aws-targets.ts +51 -0
- agentcore/templates/profiles.py +151 -0
- agentcore/templates/render.py +92 -0
- agentcore/templates/renderers.py +83 -0
- agentcore/templates/schema_assets.py +23 -0
- agentcore/templates/template_root.py +28 -0
- agentcore/templates/types.py +192 -0
- agentcore_cli-0.31.1.dist-info/METADATA +84 -0
- agentcore_cli-0.31.1.dist-info/RECORD +655 -0
- agentcore_cli-0.31.1.dist-info/WHEEL +4 -0
- agentcore_cli-0.31.1.dist-info/entry_points.txt +2 -0
- agentcore_cli-0.31.1.dist-info/licenses/LICENSE +175 -0
- agentcore_cli-0.31.1.dist-info/licenses/NOTICE +88 -0
|
@@ -0,0 +1,69 @@
|
|
|
1
|
+
# The CloudWatch agent runs this OpenTelemetry Collector configuration. It receives OTLP from
|
|
2
|
+
# the exec-server, puts the runtime session ID on each span and log, and signs each request
|
|
3
|
+
# with the ACR role. The exec-server sends the log group headers that the Runtime sets in
|
|
4
|
+
# OTEL_EXPORTER_OTLP_TRACES_HEADERS and OTEL_EXPORTER_OTLP_LOGS_HEADERS, and the collector
|
|
5
|
+
# sends them on to X-Ray and CloudWatch Logs.
|
|
6
|
+
receivers:
|
|
7
|
+
otlp:
|
|
8
|
+
protocols:
|
|
9
|
+
http:
|
|
10
|
+
endpoint: 127.0.0.1:4318
|
|
11
|
+
include_metadata: true
|
|
12
|
+
processors:
|
|
13
|
+
attributes:
|
|
14
|
+
actions:
|
|
15
|
+
- key: session.id
|
|
16
|
+
value: ${env:AGENTCORE_RUNTIME_SID}
|
|
17
|
+
action: upsert
|
|
18
|
+
exporters:
|
|
19
|
+
otlphttp/traces:
|
|
20
|
+
traces_endpoint: https://xray.${env:AWS_REGION}.amazonaws.com/v1/traces
|
|
21
|
+
auth:
|
|
22
|
+
authenticator: headers_setter/traces
|
|
23
|
+
otlphttp/logs:
|
|
24
|
+
logs_endpoint: https://logs.${env:AWS_REGION}.amazonaws.com/v1/logs
|
|
25
|
+
auth:
|
|
26
|
+
authenticator: headers_setter/logs
|
|
27
|
+
extensions:
|
|
28
|
+
sigv4auth/xray:
|
|
29
|
+
region: ${env:AWS_REGION}
|
|
30
|
+
service: xray
|
|
31
|
+
sigv4auth/logs:
|
|
32
|
+
region: ${env:AWS_REGION}
|
|
33
|
+
service: logs
|
|
34
|
+
headers_setter/traces:
|
|
35
|
+
additional_auth: sigv4auth/xray
|
|
36
|
+
headers:
|
|
37
|
+
- key: x-aws-log-group
|
|
38
|
+
from_context: x-aws-log-group
|
|
39
|
+
action: upsert
|
|
40
|
+
- key: x-aws-log-stream
|
|
41
|
+
from_context: x-aws-log-stream
|
|
42
|
+
action: upsert
|
|
43
|
+
headers_setter/logs:
|
|
44
|
+
additional_auth: sigv4auth/logs
|
|
45
|
+
headers:
|
|
46
|
+
- key: x-aws-log-group
|
|
47
|
+
from_context: x-aws-log-group
|
|
48
|
+
action: upsert
|
|
49
|
+
- key: x-aws-log-stream
|
|
50
|
+
from_context: x-aws-log-stream
|
|
51
|
+
action: upsert
|
|
52
|
+
- key: x-aws-metric-namespace
|
|
53
|
+
from_context: x-aws-metric-namespace
|
|
54
|
+
action: upsert
|
|
55
|
+
service:
|
|
56
|
+
extensions:
|
|
57
|
+
- sigv4auth/xray
|
|
58
|
+
- sigv4auth/logs
|
|
59
|
+
- headers_setter/traces
|
|
60
|
+
- headers_setter/logs
|
|
61
|
+
pipelines:
|
|
62
|
+
traces:
|
|
63
|
+
receivers: [otlp]
|
|
64
|
+
processors: [attributes]
|
|
65
|
+
exporters: [otlphttp/traces]
|
|
66
|
+
logs:
|
|
67
|
+
receivers: [otlp]
|
|
68
|
+
processors: [attributes]
|
|
69
|
+
exporters: [otlphttp/logs]
|
|
@@ -0,0 +1,16 @@
|
|
|
1
|
+
---
|
|
2
|
+
name: acr-report
|
|
3
|
+
description: Save a report of the Python version, user ID, and working directory of the ACR in the workspace.
|
|
4
|
+
---
|
|
5
|
+
|
|
6
|
+
Run these commands in the workspace directory:
|
|
7
|
+
|
|
8
|
+
- `python3 --version`
|
|
9
|
+
- `id -u`
|
|
10
|
+
- `pwd`
|
|
11
|
+
|
|
12
|
+
Save the output as `acr-report.txt` in the workspace. The first line of the
|
|
13
|
+
file is `# ACR report`, and each command and its output follow it. Read the
|
|
14
|
+
file back and report its path.
|
|
15
|
+
|
|
16
|
+
If a command fails, put its error in the report. Do not change the commands.
|
|
@@ -0,0 +1,11 @@
|
|
|
1
|
+
{
|
|
2
|
+
"Version": "2012-10-17",
|
|
3
|
+
"Statement": [
|
|
4
|
+
{
|
|
5
|
+
"Sid": "AttachToBmaEnvironment",
|
|
6
|
+
"Effect": "Allow",
|
|
7
|
+
"Action": ["bedrock-mantle:RegisterEnvironment", "bedrock-mantle:ConnectEnvironment"],
|
|
8
|
+
"Resource": "arn:*:bedrock-mantle:*:*:project/*"
|
|
9
|
+
}
|
|
10
|
+
]
|
|
11
|
+
}
|
|
@@ -0,0 +1,35 @@
|
|
|
1
|
+
{
|
|
2
|
+
"Version": "2012-10-17",
|
|
3
|
+
"Statement": [
|
|
4
|
+
{
|
|
5
|
+
"Sid": "CreateInference",
|
|
6
|
+
"Effect": "Allow",
|
|
7
|
+
"Action": "bedrock-mantle:CreateInference",
|
|
8
|
+
"Resource": "*"
|
|
9
|
+
},
|
|
10
|
+
{
|
|
11
|
+
"Sid": "InvokeAgentCoreRuntime",
|
|
12
|
+
"Effect": "Allow",
|
|
13
|
+
"Action": "bedrock-agentcore:InvokeAgentRuntime",
|
|
14
|
+
"Resource": [
|
|
15
|
+
"arn:${AWS::Partition}:bedrock-agentcore:${AWS::Region}:${AWS::AccountId}:runtime/*",
|
|
16
|
+
"arn:${AWS::Partition}:bedrock-agentcore:${AWS::Region}:${AWS::AccountId}:runtime/*/runtime-endpoint/*"
|
|
17
|
+
]
|
|
18
|
+
},
|
|
19
|
+
{
|
|
20
|
+
"Sid": "StopAgentCoreRuntimeSession",
|
|
21
|
+
"Effect": "Allow",
|
|
22
|
+
"Action": "bedrock-agentcore:StopRuntimeSession",
|
|
23
|
+
"Resource": [
|
|
24
|
+
"arn:${AWS::Partition}:bedrock-agentcore:${AWS::Region}:${AWS::AccountId}:runtime/*",
|
|
25
|
+
"arn:${AWS::Partition}:bedrock-agentcore:${AWS::Region}:${AWS::AccountId}:runtime/*/runtime-endpoint/*"
|
|
26
|
+
]
|
|
27
|
+
},
|
|
28
|
+
{
|
|
29
|
+
"Sid": "InvokeAgentCoreGateway",
|
|
30
|
+
"Effect": "Allow",
|
|
31
|
+
"Action": "bedrock-agentcore:InvokeGateway",
|
|
32
|
+
"Resource": "arn:${AWS::Partition}:bedrock-agentcore:${AWS::Region}:${AWS::AccountId}:gateway/*"
|
|
33
|
+
}
|
|
34
|
+
]
|
|
35
|
+
}
|
|
@@ -0,0 +1,20 @@
|
|
|
1
|
+
{
|
|
2
|
+
"Version": "2012-10-17",
|
|
3
|
+
"Statement": [
|
|
4
|
+
{
|
|
5
|
+
"Sid": "AssumeFromBmaSession",
|
|
6
|
+
"Effect": "Allow",
|
|
7
|
+
"Principal": { "Service": "bedrock-mantle.amazonaws.com" },
|
|
8
|
+
"Action": "sts:AssumeRole",
|
|
9
|
+
"Condition": {
|
|
10
|
+
"StringEquals": { "aws:SourceAccount": "${AWS::AccountId}" },
|
|
11
|
+
"ArnLike": {
|
|
12
|
+
"aws:SourceArn": [
|
|
13
|
+
"arn:${AWS::Partition}:bedrock-mantle:${AWS::Region}:${AWS::AccountId}:session/*",
|
|
14
|
+
"arn:${AWS::Partition}:bedrock-mantle:${AWS::Region}:${AWS::AccountId}:project/*"
|
|
15
|
+
]
|
|
16
|
+
}
|
|
17
|
+
}
|
|
18
|
+
}
|
|
19
|
+
]
|
|
20
|
+
}
|
|
@@ -0,0 +1,21 @@
|
|
|
1
|
+
[project]
|
|
2
|
+
name = "{{ name }}"
|
|
3
|
+
version = "0.1.0"
|
|
4
|
+
description = "AgentCore Runtime environment for Bedrock Managed Agents"
|
|
5
|
+
readme = "README.md"
|
|
6
|
+
requires-python = ">=3.12"
|
|
7
|
+
dependencies = [
|
|
8
|
+
"aws-opentelemetry-distro",
|
|
9
|
+
"bedrock-agentcore",
|
|
10
|
+
]
|
|
11
|
+
|
|
12
|
+
# The dependencies of client.py. The image build runs `uv sync --no-dev`.
|
|
13
|
+
[dependency-groups]
|
|
14
|
+
dev = [
|
|
15
|
+
"aws-bedrock-token-generator>=1.1.0",
|
|
16
|
+
"boto3>=1.43.0",
|
|
17
|
+
"openai>=3.16.2",
|
|
18
|
+
]
|
|
19
|
+
|
|
20
|
+
[tool.uv]
|
|
21
|
+
package = false
|
|
@@ -0,0 +1,45 @@
|
|
|
1
|
+
This is a project generated by the agentcore create CLI tool!
|
|
2
|
+
|
|
3
|
+
# Layout
|
|
4
|
+
|
|
5
|
+
The generated application code lives at the agent root directory. At the root, there is a `.gitignore` file, an
|
|
6
|
+
`agentcore/` folder which represents the configurations and state associated with this project. Other `agentcore`
|
|
7
|
+
commands like `deploy`, `dev`, and `invoke` rely on the configuration stored here.
|
|
8
|
+
|
|
9
|
+
## Agent Root
|
|
10
|
+
|
|
11
|
+
The main entrypoint to your app is defined in `main.py`. Using the AgentCore SDK `@app.entrypoint` decorator, this
|
|
12
|
+
file defines a Starlette ASGI app with the Google ADK framework running within.
|
|
13
|
+
|
|
14
|
+
`model/load.py` instantiates your chosen model provider (Gemini).
|
|
15
|
+
|
|
16
|
+
## Input Validation
|
|
17
|
+
|
|
18
|
+
Validate invocation input before forwarding it to the agent framework. Keep plain prompts typed as strings and pass
|
|
19
|
+
only prompt text to the agent.
|
|
20
|
+
|
|
21
|
+
## Environment Variables
|
|
22
|
+
|
|
23
|
+
| Variable | Required | Description |
|
|
24
|
+
| --- | --- | --- |
|
|
25
|
+
{{#if hasIdentity}}| `{{identityProviders.[0].envVarName}}` | Yes | {{modelProvider}} API key (local) or Identity provider name (deployed) |
|
|
26
|
+
{{/if}}| `LOCAL_DEV` | No | Set to `1` to use `.env.local` instead of AgentCore Identity |
|
|
27
|
+
|
|
28
|
+
# Developing locally
|
|
29
|
+
|
|
30
|
+
If installation was successful, a virtual environment is already created with dependencies installed.
|
|
31
|
+
|
|
32
|
+
Activate the environment with `source .venv/bin/activate` on macOS/Linux, `.venv\Scripts\activate.bat` in Windows
|
|
33
|
+
Command Prompt, or `.\.venv\Scripts\activate.ps1` in Windows PowerShell.
|
|
34
|
+
|
|
35
|
+
`agentcore dev` will start a local server on 0.0.0.0:8080.
|
|
36
|
+
|
|
37
|
+
In a new terminal, you can invoke that server with:
|
|
38
|
+
|
|
39
|
+
`agentcore invoke --dev "What can you do"`
|
|
40
|
+
|
|
41
|
+
# Deployment
|
|
42
|
+
|
|
43
|
+
After providing credentials, `agentcore deploy` will deploy your project into Amazon Bedrock AgentCore.
|
|
44
|
+
|
|
45
|
+
Use `agentcore invoke` to invoke your deployed agent.
|
|
@@ -0,0 +1,41 @@
|
|
|
1
|
+
# Environment variables
|
|
2
|
+
.env
|
|
3
|
+
|
|
4
|
+
# Python
|
|
5
|
+
__pycache__/
|
|
6
|
+
*.py[cod]
|
|
7
|
+
*$py.class
|
|
8
|
+
*.so
|
|
9
|
+
.Python
|
|
10
|
+
build/
|
|
11
|
+
develop-eggs/
|
|
12
|
+
dist/
|
|
13
|
+
downloads/
|
|
14
|
+
eggs/
|
|
15
|
+
.eggs/
|
|
16
|
+
lib/
|
|
17
|
+
lib64/
|
|
18
|
+
parts/
|
|
19
|
+
sdist/
|
|
20
|
+
var/
|
|
21
|
+
wheels/
|
|
22
|
+
*.egg-info/
|
|
23
|
+
.installed.cfg
|
|
24
|
+
*.egg
|
|
25
|
+
|
|
26
|
+
# Virtual environments
|
|
27
|
+
.venv/
|
|
28
|
+
venv/
|
|
29
|
+
ENV/
|
|
30
|
+
env/
|
|
31
|
+
|
|
32
|
+
# IDE
|
|
33
|
+
.vscode/
|
|
34
|
+
.idea/
|
|
35
|
+
*.swp
|
|
36
|
+
*.swo
|
|
37
|
+
*~
|
|
38
|
+
|
|
39
|
+
# OS
|
|
40
|
+
.DS_Store
|
|
41
|
+
Thumbs.db
|
|
@@ -0,0 +1,207 @@
|
|
|
1
|
+
{{#if needsOs}}
|
|
2
|
+
import os
|
|
3
|
+
{{/if}}
|
|
4
|
+
from collections import OrderedDict
|
|
5
|
+
from google.adk.agents import Agent
|
|
6
|
+
from google.adk.runners import Runner
|
|
7
|
+
from google.adk.sessions import InMemorySessionService
|
|
8
|
+
from google.genai import types
|
|
9
|
+
from bedrock_agentcore.runtime import BedrockAgentCoreApp
|
|
10
|
+
from model.load import load_model
|
|
11
|
+
{{#if hasGateway}}
|
|
12
|
+
from mcp_client.client import get_all_gateway_mcp_toolsets
|
|
13
|
+
{{else}}
|
|
14
|
+
from mcp_client.client import get_streamable_http_mcp_client
|
|
15
|
+
{{/if}}
|
|
16
|
+
|
|
17
|
+
app = BedrockAgentCoreApp()
|
|
18
|
+
log = app.logger
|
|
19
|
+
|
|
20
|
+
APP_NAME = "{{ name }}"
|
|
21
|
+
|
|
22
|
+
# https://google.github.io/adk-docs/agents/models/
|
|
23
|
+
MODEL_ID = "gemini-2.5-flash"
|
|
24
|
+
|
|
25
|
+
|
|
26
|
+
# Define a simple function tool
|
|
27
|
+
def add_numbers(a: int, b: int) -> int:
|
|
28
|
+
"""Return the sum of two numbers"""
|
|
29
|
+
return a + b
|
|
30
|
+
|
|
31
|
+
|
|
32
|
+
# Define a collection of tools used by the model
|
|
33
|
+
tools = [add_numbers]
|
|
34
|
+
|
|
35
|
+
{{#if needsOs}}
|
|
36
|
+
_MOUNT_PATHS = [
|
|
37
|
+
{{#if sessionStorageMountPath}}"{{sessionStorageMountPath}}",{{/if}}
|
|
38
|
+
{{#each efsMounts}}"{{mountPath}}",{{/each}}
|
|
39
|
+
{{#each s3Mounts}}"{{mountPath}}",{{/each}}
|
|
40
|
+
]
|
|
41
|
+
|
|
42
|
+
def _safe_resolve(path: str) -> str:
|
|
43
|
+
resolved = os.path.realpath(path)
|
|
44
|
+
if not any(resolved == os.path.realpath(m) or resolved.startswith(os.path.realpath(m) + os.sep) for m in _MOUNT_PATHS):
|
|
45
|
+
raise ValueError(f"Path '{path}' is not within any configured mount ({', '.join(_MOUNT_PATHS)})")
|
|
46
|
+
return resolved
|
|
47
|
+
|
|
48
|
+
def file_read(path: str) -> str:
|
|
49
|
+
"""Read a file from a mounted filesystem. Use the absolute path (e.g. /mnt/tools/data.txt)."""
|
|
50
|
+
try:
|
|
51
|
+
full_path = _safe_resolve(path)
|
|
52
|
+
with open(full_path) as f:
|
|
53
|
+
return f.read()
|
|
54
|
+
except ValueError as e:
|
|
55
|
+
return str(e)
|
|
56
|
+
except OSError as e:
|
|
57
|
+
return f"Error reading '{path}': {e.strerror}"
|
|
58
|
+
|
|
59
|
+
def file_write(path: str, content: str) -> str:
|
|
60
|
+
"""Write a file to a mounted filesystem. Use the absolute path (e.g. /mnt/tools/data.txt)."""
|
|
61
|
+
try:
|
|
62
|
+
full_path = _safe_resolve(path)
|
|
63
|
+
parent = os.path.dirname(full_path)
|
|
64
|
+
if parent:
|
|
65
|
+
os.makedirs(parent, exist_ok=True)
|
|
66
|
+
with open(full_path, "w") as f:
|
|
67
|
+
f.write(content)
|
|
68
|
+
return f"Written to {path}"
|
|
69
|
+
except ValueError as e:
|
|
70
|
+
return str(e)
|
|
71
|
+
except OSError as e:
|
|
72
|
+
return f"Error writing '{path}': {e.strerror}"
|
|
73
|
+
|
|
74
|
+
def list_files(path: str) -> str:
|
|
75
|
+
"""List files in a mounted filesystem directory. Use the absolute path (e.g. /mnt/tools)."""
|
|
76
|
+
try:
|
|
77
|
+
full_path = _safe_resolve(path)
|
|
78
|
+
entries = os.listdir(full_path)
|
|
79
|
+
return "\n".join(entries) if entries else "(empty directory)"
|
|
80
|
+
except ValueError as e:
|
|
81
|
+
return str(e)
|
|
82
|
+
except OSError as e:
|
|
83
|
+
return f"Error listing '{path}': {e.strerror}"
|
|
84
|
+
|
|
85
|
+
tools.extend([file_read, file_write, list_files])
|
|
86
|
+
{{/if}}
|
|
87
|
+
|
|
88
|
+
AGENT_INSTRUCTION = """
|
|
89
|
+
I can answer your questions using the knowledge I have!
|
|
90
|
+
{{#if needsOs}}
|
|
91
|
+
You have access to the following mounted filesystems. Use file_read, file_write, and list_files with full absolute paths:
|
|
92
|
+
{{#if sessionStorageMountPath}}- {{sessionStorageMountPath}}: ephemeral session storage (lost when session ends)
|
|
93
|
+
{{/if}}{{#each efsMounts}}- {{mountPath}}: EFS persistent storage (persists across sessions and agent restarts)
|
|
94
|
+
{{/each}}{{#each s3Mounts}}- {{mountPath}}: S3 Files persistent storage (durable, backed by S3)
|
|
95
|
+
{{/each}}{{/if}}
|
|
96
|
+
"""
|
|
97
|
+
|
|
98
|
+
# Get MCP Toolset
|
|
99
|
+
{{#if hasGateway}}
|
|
100
|
+
mcp_toolset = get_all_gateway_mcp_toolsets()
|
|
101
|
+
{{else}}
|
|
102
|
+
mcp_client = get_streamable_http_mcp_client()
|
|
103
|
+
mcp_toolset = [mcp_client] if mcp_client else []
|
|
104
|
+
{{/if}}
|
|
105
|
+
|
|
106
|
+
_credentials_loaded = False
|
|
107
|
+
|
|
108
|
+
def ensure_credentials_loaded():
|
|
109
|
+
global _credentials_loaded
|
|
110
|
+
if not _credentials_loaded:
|
|
111
|
+
load_model()
|
|
112
|
+
_credentials_loaded = True
|
|
113
|
+
|
|
114
|
+
|
|
115
|
+
# Agent Definition
|
|
116
|
+
agent = Agent(
|
|
117
|
+
model=MODEL_ID,
|
|
118
|
+
name="{{ name }}",
|
|
119
|
+
description="Agent to answer questions",
|
|
120
|
+
instruction=AGENT_INSTRUCTION,
|
|
121
|
+
tools=mcp_toolset + tools,
|
|
122
|
+
)
|
|
123
|
+
|
|
124
|
+
|
|
125
|
+
# Module-level session service and runner preserve history across invocations.
|
|
126
|
+
# InMemorySessionService retains every (app_name, user_id, session_id) triple
|
|
127
|
+
# forever, so we bound it to 128 active sessions with LRU eviction (the
|
|
128
|
+
# least-recently-used session is deleted and its history reset) to keep a
|
|
129
|
+
# long-running process from growing without limit. For durable history, swap in
|
|
130
|
+
# a persistent session service (e.g. DatabaseSessionService).
|
|
131
|
+
_SESSION_LIMIT = 128
|
|
132
|
+
_session_service = InMemorySessionService()
|
|
133
|
+
_session_keys = OrderedDict()
|
|
134
|
+
_runner = None
|
|
135
|
+
|
|
136
|
+
|
|
137
|
+
def get_or_create_runner():
|
|
138
|
+
global _runner
|
|
139
|
+
if _runner is None:
|
|
140
|
+
ensure_credentials_loaded()
|
|
141
|
+
_runner = Runner(
|
|
142
|
+
agent=agent,
|
|
143
|
+
app_name=APP_NAME,
|
|
144
|
+
session_service=_session_service,
|
|
145
|
+
)
|
|
146
|
+
return _runner
|
|
147
|
+
|
|
148
|
+
|
|
149
|
+
async def get_or_create_session(user_id, session_id):
|
|
150
|
+
key = (user_id, session_id)
|
|
151
|
+
if key in _session_keys:
|
|
152
|
+
_session_keys.move_to_end(key)
|
|
153
|
+
else:
|
|
154
|
+
while len(_session_keys) >= _SESSION_LIMIT:
|
|
155
|
+
(old_user_id, old_session_id), _ = _session_keys.popitem(last=False)
|
|
156
|
+
await _session_service.delete_session(
|
|
157
|
+
app_name=APP_NAME, user_id=old_user_id, session_id=old_session_id
|
|
158
|
+
)
|
|
159
|
+
_session_keys[key] = True
|
|
160
|
+
|
|
161
|
+
session = await _session_service.get_session(
|
|
162
|
+
app_name=APP_NAME, user_id=user_id, session_id=session_id
|
|
163
|
+
)
|
|
164
|
+
if session is None:
|
|
165
|
+
session = await _session_service.create_session(
|
|
166
|
+
app_name=APP_NAME, user_id=user_id, session_id=session_id
|
|
167
|
+
)
|
|
168
|
+
return session
|
|
169
|
+
|
|
170
|
+
|
|
171
|
+
# Agent Interaction
|
|
172
|
+
async def call_agent_async(query, user_id, session_id):
|
|
173
|
+
content = types.Content(role="user", parts=[types.Part(text=query)])
|
|
174
|
+
runner = get_or_create_runner()
|
|
175
|
+
session = await get_or_create_session(user_id, session_id)
|
|
176
|
+
events = runner.run_async(
|
|
177
|
+
user_id=user_id, session_id=session.id, new_message=content
|
|
178
|
+
)
|
|
179
|
+
|
|
180
|
+
final_response = None
|
|
181
|
+
async for event in events:
|
|
182
|
+
if event.is_final_response():
|
|
183
|
+
final_response = event.content.parts[0].text
|
|
184
|
+
|
|
185
|
+
return final_response
|
|
186
|
+
|
|
187
|
+
|
|
188
|
+
@app.entrypoint
|
|
189
|
+
async def invoke(payload, context):
|
|
190
|
+
log.info("Invoking Agent.....")
|
|
191
|
+
|
|
192
|
+
# Process the user prompt
|
|
193
|
+
prompt = payload.get("prompt", "What can you help me with?")
|
|
194
|
+
if not isinstance(prompt, str):
|
|
195
|
+
raise ValueError("prompt must be a string")
|
|
196
|
+
session_id = getattr(context, "session_id", "default_session")
|
|
197
|
+
user_id = payload.get("user_id", "default_user")
|
|
198
|
+
|
|
199
|
+
# Run the agent
|
|
200
|
+
result = await call_agent_async(prompt, user_id, session_id)
|
|
201
|
+
|
|
202
|
+
# Return result
|
|
203
|
+
return {"result": result}
|
|
204
|
+
|
|
205
|
+
|
|
206
|
+
if __name__ == "__main__":
|
|
207
|
+
app.run()
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
# Package marker
|
|
@@ -0,0 +1,75 @@
|
|
|
1
|
+
import os
|
|
2
|
+
import logging
|
|
3
|
+
from google.adk.tools.mcp_tool.mcp_toolset import MCPToolset
|
|
4
|
+
from google.adk.tools.mcp_tool.mcp_session_manager import StreamableHTTPConnectionParams
|
|
5
|
+
|
|
6
|
+
logger = logging.getLogger(__name__)
|
|
7
|
+
|
|
8
|
+
{{#if hasGateway}}
|
|
9
|
+
{{#if (includes gatewayAuthTypes "AWS_IAM")}}
|
|
10
|
+
import httpx
|
|
11
|
+
from mcp_proxy_for_aws.sigv4_helper import SigV4HTTPXAuth, create_aws_session
|
|
12
|
+
{{/if}}
|
|
13
|
+
{{#if (includes gatewayAuthTypes "CUSTOM_JWT")}}
|
|
14
|
+
from bedrock_agentcore.identity import requires_access_token
|
|
15
|
+
{{/if}}
|
|
16
|
+
|
|
17
|
+
{{#each gatewayProviders}}
|
|
18
|
+
{{#if (eq authType "CUSTOM_JWT")}}
|
|
19
|
+
@requires_access_token(
|
|
20
|
+
provider_name="{{credentialProviderName}}",
|
|
21
|
+
scopes=[{{#if scopes}}"{{scopes}}"{{/if}}],
|
|
22
|
+
auth_flow="M2M",
|
|
23
|
+
)
|
|
24
|
+
def _get_bearer_token_{{snakeCase name}}(*, access_token: str):
|
|
25
|
+
"""Obtain OAuth access token via AgentCore Identity for {{name}}."""
|
|
26
|
+
return access_token
|
|
27
|
+
|
|
28
|
+
{{/if}}
|
|
29
|
+
{{/each}}
|
|
30
|
+
|
|
31
|
+
def get_all_gateway_mcp_toolsets() -> list[MCPToolset]:
|
|
32
|
+
"""Returns MCP Toolsets for all configured gateways."""
|
|
33
|
+
toolsets = []
|
|
34
|
+
{{#each gatewayProviders}}
|
|
35
|
+
url = os.environ.get("{{envVarName}}")
|
|
36
|
+
if url:
|
|
37
|
+
{{#if (eq authType "AWS_IAM")}}
|
|
38
|
+
session = create_aws_session()
|
|
39
|
+
auth = SigV4HTTPXAuth(session.get_credentials(), "bedrock-agentcore", session.region_name)
|
|
40
|
+
toolsets.append(MCPToolset(tool_name_prefix="{{snakeCase name}}", connection_params=StreamableHTTPConnectionParams(
|
|
41
|
+
url=url,
|
|
42
|
+
httpx_client_factory=lambda **kwargs: httpx.AsyncClient(auth=auth, **kwargs)
|
|
43
|
+
)))
|
|
44
|
+
{{else if (eq authType "CUSTOM_JWT")}}
|
|
45
|
+
token = _get_bearer_token_{{snakeCase name}}()
|
|
46
|
+
headers = {"Authorization": f"Bearer {token}"} if token else None
|
|
47
|
+
toolsets.append(MCPToolset(tool_name_prefix="{{snakeCase name}}", connection_params=StreamableHTTPConnectionParams(url=url, headers=headers)))
|
|
48
|
+
{{else}}
|
|
49
|
+
toolsets.append(MCPToolset(tool_name_prefix="{{snakeCase name}}", connection_params=StreamableHTTPConnectionParams(url=url)))
|
|
50
|
+
{{/if}}
|
|
51
|
+
else:
|
|
52
|
+
logger.warning("{{envVarName}} not set — {{name}} gateway tools unavailable")
|
|
53
|
+
{{/each}}
|
|
54
|
+
return toolsets
|
|
55
|
+
{{else}}
|
|
56
|
+
{{#if isVpc}}
|
|
57
|
+
# VPC mode: external MCP endpoints are not reachable without a NAT gateway.
|
|
58
|
+
# Add an AgentCore Gateway with `agentcore add gateway`, or configure your own endpoint below.
|
|
59
|
+
|
|
60
|
+
def get_streamable_http_mcp_client() -> MCPToolset | None:
|
|
61
|
+
"""No MCP server configured. Add a gateway with `agentcore add gateway`."""
|
|
62
|
+
return None
|
|
63
|
+
{{else}}
|
|
64
|
+
# ExaAI provides information about code through web searches, crawling and code context searches through their platform. Requires no authentication
|
|
65
|
+
EXAMPLE_MCP_ENDPOINT = "https://mcp.exa.ai/mcp"
|
|
66
|
+
|
|
67
|
+
|
|
68
|
+
def get_streamable_http_mcp_client() -> MCPToolset:
|
|
69
|
+
"""Returns an MCP Toolset compatible with Google ADK."""
|
|
70
|
+
# to use an MCP server that supports bearer authentication, add headers={"Authorization": f"Bearer {access_token}"}
|
|
71
|
+
return MCPToolset(
|
|
72
|
+
connection_params=StreamableHTTPConnectionParams(url=EXAMPLE_MCP_ENDPOINT)
|
|
73
|
+
)
|
|
74
|
+
{{/if}}
|
|
75
|
+
{{/if}}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
# Package marker
|
|
@@ -0,0 +1,41 @@
|
|
|
1
|
+
import os
|
|
2
|
+
from bedrock_agentcore.identity.auth import requires_api_key
|
|
3
|
+
|
|
4
|
+
IDENTITY_PROVIDER_NAME = "{{identityProviders.[0].name}}"
|
|
5
|
+
IDENTITY_ENV_VAR = "{{identityProviders.[0].envVarName}}"
|
|
6
|
+
|
|
7
|
+
|
|
8
|
+
@requires_api_key(provider_name=IDENTITY_PROVIDER_NAME)
|
|
9
|
+
def _agentcore_identity_api_key_provider(api_key: str) -> str:
|
|
10
|
+
"""Fetch API key from AgentCore Identity."""
|
|
11
|
+
return api_key
|
|
12
|
+
|
|
13
|
+
|
|
14
|
+
def _get_api_key() -> str:
|
|
15
|
+
"""
|
|
16
|
+
Uses AgentCore Identity for API key management in deployed environments.
|
|
17
|
+
For local development, run via 'agentcore dev' which loads agentcore/.env.
|
|
18
|
+
"""
|
|
19
|
+
if os.getenv("LOCAL_DEV") == "1":
|
|
20
|
+
api_key = os.getenv(IDENTITY_ENV_VAR)
|
|
21
|
+
if not api_key:
|
|
22
|
+
raise RuntimeError(
|
|
23
|
+
f"{IDENTITY_ENV_VAR} not found. Add {IDENTITY_ENV_VAR}=your-key to .env.local"
|
|
24
|
+
)
|
|
25
|
+
return api_key
|
|
26
|
+
return _agentcore_identity_api_key_provider()
|
|
27
|
+
|
|
28
|
+
|
|
29
|
+
def load_model() -> None:
|
|
30
|
+
"""
|
|
31
|
+
Set up Gemini API key authentication.
|
|
32
|
+
Uses AgentCore Identity for API key management in deployed environments,
|
|
33
|
+
and falls back to .env file for local development.
|
|
34
|
+
Sets the GOOGLE_API_KEY environment variable for the Google ADK.
|
|
35
|
+
"""
|
|
36
|
+
api_key = _get_api_key()
|
|
37
|
+
# Use Google AI Studios API Key Authentication.
|
|
38
|
+
# https://google.github.io/adk-docs/agents/models/#google-ai-studio
|
|
39
|
+
os.environ["GOOGLE_API_KEY"] = api_key
|
|
40
|
+
# Set to TRUE is using Google Vertex AI, Set to FALSE for Google AI Studio
|
|
41
|
+
os.environ["GOOGLE_GENAI_USE_VERTEXAI"] = "FALSE"
|
|
@@ -0,0 +1,24 @@
|
|
|
1
|
+
[build-system]
|
|
2
|
+
requires = ["hatchling"]
|
|
3
|
+
build-backend = "hatchling.build"
|
|
4
|
+
|
|
5
|
+
[project]
|
|
6
|
+
name = "{{ name }}"
|
|
7
|
+
version = "0.1.0"
|
|
8
|
+
description = "AgentCore Runtime Application using Google ADK"
|
|
9
|
+
readme = "README.md"
|
|
10
|
+
requires-python = ">=3.10"
|
|
11
|
+
dependencies = [
|
|
12
|
+
"aws-opentelemetry-distro >= 0.17.0",
|
|
13
|
+
"google-adk >= 1.35.0, < 2.0.0",
|
|
14
|
+
"google-genai >= 1.0.0, < 2.0.0",
|
|
15
|
+
# 1.13.0 was yanked for broken imports.
|
|
16
|
+
"opentelemetry-resourcedetector-gcp >= 1.9.0a0, < 2.0.0, != 1.13.0",
|
|
17
|
+
"bedrock-agentcore >= 1.18.1",
|
|
18
|
+
"botocore[crt] >= 1.35.0",
|
|
19
|
+
{{#if hasGateway}}{{#if (includes gatewayAuthTypes "AWS_IAM")}}"mcp-proxy-for-aws >= 1.1.0",
|
|
20
|
+
{{/if}}{{/if}}
|
|
21
|
+
]
|
|
22
|
+
|
|
23
|
+
[tool.hatch.build.targets.wheel]
|
|
24
|
+
packages = ["."]
|
|
@@ -0,0 +1,45 @@
|
|
|
1
|
+
This is a project generated by the agentcore create CLI tool!
|
|
2
|
+
|
|
3
|
+
# Layout
|
|
4
|
+
|
|
5
|
+
The generated application code lives at the agent root directory. At the root, there is a `.gitignore` file, an
|
|
6
|
+
`agentcore/` folder which represents the configurations and state associated with this project. Other `agentcore`
|
|
7
|
+
commands like `deploy`, `dev`, and `invoke` rely on the configuration stored here.
|
|
8
|
+
|
|
9
|
+
## Agent Root
|
|
10
|
+
|
|
11
|
+
The main entrypoint to your app is defined in `main.py`. Using the AgentCore SDK `@app.entrypoint` decorator, this
|
|
12
|
+
file defines a Starlette ASGI app with the LangChain/LangGraph framework running within.
|
|
13
|
+
|
|
14
|
+
`model/load.py` instantiates your chosen model provider.
|
|
15
|
+
|
|
16
|
+
## Input Validation
|
|
17
|
+
|
|
18
|
+
Validate invocation input before forwarding it to the agent framework. Keep plain prompts typed as strings and pass
|
|
19
|
+
only prompt text to the agent.
|
|
20
|
+
|
|
21
|
+
## Environment Variables
|
|
22
|
+
|
|
23
|
+
| Variable | Required | Description |
|
|
24
|
+
| --- | --- | --- |
|
|
25
|
+
{{#if hasIdentity}}| `{{identityProviders.[0].envVarName}}` | Yes | {{modelProvider}} API key (local) or Identity provider name (deployed) |
|
|
26
|
+
{{/if}}| `LOCAL_DEV` | No | Set to `1` to use `.env.local` instead of AgentCore Identity |
|
|
27
|
+
|
|
28
|
+
# Developing locally
|
|
29
|
+
|
|
30
|
+
If installation was successful, a virtual environment is already created with dependencies installed.
|
|
31
|
+
|
|
32
|
+
Activate the environment with `source .venv/bin/activate` on macOS/Linux, `.venv\Scripts\activate.bat` in Windows
|
|
33
|
+
Command Prompt, or `.\.venv\Scripts\activate.ps1` in Windows PowerShell.
|
|
34
|
+
|
|
35
|
+
`agentcore dev` will start a local server on 0.0.0.0:8080.
|
|
36
|
+
|
|
37
|
+
In a new terminal, you can invoke that server with:
|
|
38
|
+
|
|
39
|
+
`agentcore invoke --dev "What can you do"`
|
|
40
|
+
|
|
41
|
+
# Deployment
|
|
42
|
+
|
|
43
|
+
After providing credentials, `agentcore deploy` will deploy your project into Amazon Bedrock AgentCore.
|
|
44
|
+
|
|
45
|
+
Use `agentcore invoke` to invoke your deployed agent.
|