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,130 @@
|
|
|
1
|
+
import os
|
|
2
|
+
|
|
3
|
+
os.environ["LANGGRAPH_FAST_API"] = "true"
|
|
4
|
+
|
|
5
|
+
import uvicorn
|
|
6
|
+
from typing import Any, List
|
|
7
|
+
from fastapi import FastAPI
|
|
8
|
+
from fastapi.middleware.cors import CORSMiddleware
|
|
9
|
+
from langgraph.graph import StateGraph, START
|
|
10
|
+
from langgraph.graph.message import MessagesState
|
|
11
|
+
from langgraph.checkpoint.memory import MemorySaver
|
|
12
|
+
from langgraph.prebuilt import ToolNode, tools_condition
|
|
13
|
+
from langchain_core.tools import tool
|
|
14
|
+
from opentelemetry.instrumentation.langchain import LangchainInstrumentor
|
|
15
|
+
from ag_ui_langgraph import LangGraphAgent, add_langgraph_fastapi_endpoint
|
|
16
|
+
from model.load import load_model
|
|
17
|
+
|
|
18
|
+
LangchainInstrumentor().instrument()
|
|
19
|
+
|
|
20
|
+
|
|
21
|
+
@tool
|
|
22
|
+
def add_numbers(a: int, b: int) -> int:
|
|
23
|
+
"""Return the sum of two numbers."""
|
|
24
|
+
return a + b
|
|
25
|
+
|
|
26
|
+
|
|
27
|
+
backend_tools = [add_numbers]
|
|
28
|
+
|
|
29
|
+
{{#if needsOs}}
|
|
30
|
+
_MOUNT_PATHS = [
|
|
31
|
+
{{#if sessionStorageMountPath}}"{{sessionStorageMountPath}}",{{/if}}
|
|
32
|
+
{{#each efsMounts}}"{{mountPath}}",{{/each}}
|
|
33
|
+
{{#each s3Mounts}}"{{mountPath}}",{{/each}}
|
|
34
|
+
]
|
|
35
|
+
|
|
36
|
+
def _safe_resolve(path: str) -> str:
|
|
37
|
+
resolved = os.path.realpath(path)
|
|
38
|
+
if not any(resolved == os.path.realpath(m) or resolved.startswith(os.path.realpath(m) + os.sep) for m in _MOUNT_PATHS):
|
|
39
|
+
raise ValueError(f"Path '{path}' is not within any configured mount ({', '.join(_MOUNT_PATHS)})")
|
|
40
|
+
return resolved
|
|
41
|
+
|
|
42
|
+
@tool
|
|
43
|
+
def file_read(path: str) -> str:
|
|
44
|
+
"""Read a file from a mounted filesystem. Use the absolute path (e.g. /mnt/tools/data.txt)."""
|
|
45
|
+
try:
|
|
46
|
+
full_path = _safe_resolve(path)
|
|
47
|
+
with open(full_path) as f:
|
|
48
|
+
return f.read()
|
|
49
|
+
except ValueError as e:
|
|
50
|
+
return str(e)
|
|
51
|
+
except OSError as e:
|
|
52
|
+
return f"Error reading '{path}': {e.strerror}"
|
|
53
|
+
|
|
54
|
+
@tool
|
|
55
|
+
def file_write(path: str, content: str) -> str:
|
|
56
|
+
"""Write a file to a mounted filesystem. Use the absolute path (e.g. /mnt/tools/data.txt)."""
|
|
57
|
+
try:
|
|
58
|
+
full_path = _safe_resolve(path)
|
|
59
|
+
parent = os.path.dirname(full_path)
|
|
60
|
+
if parent:
|
|
61
|
+
os.makedirs(parent, exist_ok=True)
|
|
62
|
+
with open(full_path, "w") as f:
|
|
63
|
+
f.write(content)
|
|
64
|
+
return f"Written to {path}"
|
|
65
|
+
except ValueError as e:
|
|
66
|
+
return str(e)
|
|
67
|
+
except OSError as e:
|
|
68
|
+
return f"Error writing '{path}': {e.strerror}"
|
|
69
|
+
|
|
70
|
+
@tool
|
|
71
|
+
def list_files(path: str) -> str:
|
|
72
|
+
"""List files in a mounted filesystem directory. Use the absolute path (e.g. /mnt/tools)."""
|
|
73
|
+
try:
|
|
74
|
+
full_path = _safe_resolve(path)
|
|
75
|
+
entries = os.listdir(full_path)
|
|
76
|
+
return "\n".join(entries) if entries else "(empty directory)"
|
|
77
|
+
except ValueError as e:
|
|
78
|
+
return str(e)
|
|
79
|
+
except OSError as e:
|
|
80
|
+
return f"Error listing '{path}': {e.strerror}"
|
|
81
|
+
|
|
82
|
+
backend_tools.extend([file_read, file_write, list_files])
|
|
83
|
+
{{/if}}
|
|
84
|
+
model = load_model()
|
|
85
|
+
|
|
86
|
+
|
|
87
|
+
class AgentState(MessagesState):
|
|
88
|
+
tools: List[Any]
|
|
89
|
+
|
|
90
|
+
|
|
91
|
+
def chat_node(state: AgentState):
|
|
92
|
+
bound_model = model.bind_tools(
|
|
93
|
+
[*state.get("tools", []), *backend_tools],
|
|
94
|
+
)
|
|
95
|
+
response = bound_model.invoke(state["messages"])
|
|
96
|
+
return {"messages": [response]}
|
|
97
|
+
|
|
98
|
+
|
|
99
|
+
builder = StateGraph(AgentState)
|
|
100
|
+
builder.add_node("chat", chat_node)
|
|
101
|
+
builder.add_node("tools", ToolNode(tools=backend_tools))
|
|
102
|
+
builder.add_edge(START, "chat")
|
|
103
|
+
builder.add_conditional_edges("chat", tools_condition)
|
|
104
|
+
builder.add_edge("tools", "chat")
|
|
105
|
+
graph = builder.compile(checkpointer=MemorySaver())
|
|
106
|
+
|
|
107
|
+
agent = LangGraphAgent(
|
|
108
|
+
name="{{ name }}",
|
|
109
|
+
graph=graph,
|
|
110
|
+
description="A helpful assistant",
|
|
111
|
+
)
|
|
112
|
+
|
|
113
|
+
app = FastAPI()
|
|
114
|
+
app.add_middleware(
|
|
115
|
+
CORSMiddleware,
|
|
116
|
+
allow_origins=["*"],
|
|
117
|
+
allow_methods=["*"],
|
|
118
|
+
allow_headers=["*"],
|
|
119
|
+
)
|
|
120
|
+
|
|
121
|
+
add_langgraph_fastapi_endpoint(app=app, agent=agent, path="/invocations")
|
|
122
|
+
|
|
123
|
+
|
|
124
|
+
@app.get("/ping")
|
|
125
|
+
async def ping():
|
|
126
|
+
return {"status": "healthy"}
|
|
127
|
+
|
|
128
|
+
|
|
129
|
+
if __name__ == "__main__":
|
|
130
|
+
uvicorn.run(app, host="0.0.0.0", port=int(os.environ.get("PORT", "8080")))
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
# Package marker
|
|
@@ -0,0 +1,123 @@
|
|
|
1
|
+
{{#if (eq modelProvider "Bedrock")}}
|
|
2
|
+
from langchain_aws import ChatBedrock
|
|
3
|
+
|
|
4
|
+
# Uses global inference profile for Claude Sonnet 4.5
|
|
5
|
+
# https://docs.aws.amazon.com/bedrock/latest/userguide/inference-profiles-support.html
|
|
6
|
+
MODEL_ID = "global.anthropic.claude-sonnet-4-5-20250929-v1:0"
|
|
7
|
+
|
|
8
|
+
|
|
9
|
+
def load_model() -> ChatBedrock:
|
|
10
|
+
"""Get Bedrock model client using IAM credentials."""
|
|
11
|
+
return ChatBedrock(model_id=MODEL_ID)
|
|
12
|
+
{{/if}}
|
|
13
|
+
{{#if (eq modelProvider "Anthropic")}}
|
|
14
|
+
import os
|
|
15
|
+
from langchain_anthropic import ChatAnthropic
|
|
16
|
+
from bedrock_agentcore.identity.auth import requires_api_key
|
|
17
|
+
|
|
18
|
+
IDENTITY_PROVIDER_NAME = "{{identityProviders.[0].name}}"
|
|
19
|
+
IDENTITY_ENV_VAR = "{{identityProviders.[0].envVarName}}"
|
|
20
|
+
|
|
21
|
+
|
|
22
|
+
@requires_api_key(provider_name=IDENTITY_PROVIDER_NAME)
|
|
23
|
+
def _agentcore_identity_api_key_provider(api_key: str) -> str:
|
|
24
|
+
"""Fetch API key from AgentCore Identity."""
|
|
25
|
+
return api_key
|
|
26
|
+
|
|
27
|
+
|
|
28
|
+
def _get_api_key() -> str:
|
|
29
|
+
"""
|
|
30
|
+
Uses AgentCore Identity for API key management in deployed environments.
|
|
31
|
+
For local development, run via 'agentcore dev' which loads agentcore/.env.
|
|
32
|
+
"""
|
|
33
|
+
if os.getenv("LOCAL_DEV") == "1":
|
|
34
|
+
api_key = os.getenv(IDENTITY_ENV_VAR)
|
|
35
|
+
if not api_key:
|
|
36
|
+
raise RuntimeError(
|
|
37
|
+
f"{IDENTITY_ENV_VAR} not found. Add {IDENTITY_ENV_VAR}=your-key to .env.local"
|
|
38
|
+
)
|
|
39
|
+
return api_key
|
|
40
|
+
return _agentcore_identity_api_key_provider()
|
|
41
|
+
|
|
42
|
+
|
|
43
|
+
def load_model() -> ChatAnthropic:
|
|
44
|
+
"""Get authenticated Anthropic model client."""
|
|
45
|
+
return ChatAnthropic(
|
|
46
|
+
model="claude-sonnet-4-5-20250929",
|
|
47
|
+
api_key=_get_api_key()
|
|
48
|
+
)
|
|
49
|
+
{{/if}}
|
|
50
|
+
{{#if (eq modelProvider "OpenAI")}}
|
|
51
|
+
import os
|
|
52
|
+
from langchain_openai import ChatOpenAI
|
|
53
|
+
from bedrock_agentcore.identity.auth import requires_api_key
|
|
54
|
+
|
|
55
|
+
IDENTITY_PROVIDER_NAME = "{{identityProviders.[0].name}}"
|
|
56
|
+
IDENTITY_ENV_VAR = "{{identityProviders.[0].envVarName}}"
|
|
57
|
+
|
|
58
|
+
|
|
59
|
+
@requires_api_key(provider_name=IDENTITY_PROVIDER_NAME)
|
|
60
|
+
def _agentcore_identity_api_key_provider(api_key: str) -> str:
|
|
61
|
+
"""Fetch API key from AgentCore Identity."""
|
|
62
|
+
return api_key
|
|
63
|
+
|
|
64
|
+
|
|
65
|
+
def _get_api_key() -> str:
|
|
66
|
+
"""
|
|
67
|
+
Uses AgentCore Identity for API key management in deployed environments.
|
|
68
|
+
For local development, run via 'agentcore dev' which loads agentcore/.env.
|
|
69
|
+
"""
|
|
70
|
+
if os.getenv("LOCAL_DEV") == "1":
|
|
71
|
+
api_key = os.getenv(IDENTITY_ENV_VAR)
|
|
72
|
+
if not api_key:
|
|
73
|
+
raise RuntimeError(
|
|
74
|
+
f"{IDENTITY_ENV_VAR} not found. Add {IDENTITY_ENV_VAR}=your-key to .env.local"
|
|
75
|
+
)
|
|
76
|
+
return api_key
|
|
77
|
+
return _agentcore_identity_api_key_provider()
|
|
78
|
+
|
|
79
|
+
|
|
80
|
+
def load_model() -> ChatOpenAI:
|
|
81
|
+
"""Get authenticated OpenAI model client."""
|
|
82
|
+
return ChatOpenAI(
|
|
83
|
+
model="gpt-4.1",
|
|
84
|
+
api_key=_get_api_key()
|
|
85
|
+
)
|
|
86
|
+
{{/if}}
|
|
87
|
+
{{#if (eq modelProvider "Gemini")}}
|
|
88
|
+
import os
|
|
89
|
+
from langchain_google_genai import ChatGoogleGenerativeAI
|
|
90
|
+
from bedrock_agentcore.identity.auth import requires_api_key
|
|
91
|
+
|
|
92
|
+
IDENTITY_PROVIDER_NAME = "{{identityProviders.[0].name}}"
|
|
93
|
+
IDENTITY_ENV_VAR = "{{identityProviders.[0].envVarName}}"
|
|
94
|
+
|
|
95
|
+
|
|
96
|
+
@requires_api_key(provider_name=IDENTITY_PROVIDER_NAME)
|
|
97
|
+
def _agentcore_identity_api_key_provider(api_key: str) -> str:
|
|
98
|
+
"""Fetch API key from AgentCore Identity."""
|
|
99
|
+
return api_key
|
|
100
|
+
|
|
101
|
+
|
|
102
|
+
def _get_api_key() -> str:
|
|
103
|
+
"""
|
|
104
|
+
Uses AgentCore Identity for API key management in deployed environments.
|
|
105
|
+
For local development, run via 'agentcore dev' which loads agentcore/.env.
|
|
106
|
+
"""
|
|
107
|
+
if os.getenv("LOCAL_DEV") == "1":
|
|
108
|
+
api_key = os.getenv(IDENTITY_ENV_VAR)
|
|
109
|
+
if not api_key:
|
|
110
|
+
raise RuntimeError(
|
|
111
|
+
f"{IDENTITY_ENV_VAR} not found. Add {IDENTITY_ENV_VAR}=your-key to .env.local"
|
|
112
|
+
)
|
|
113
|
+
return api_key
|
|
114
|
+
return _agentcore_identity_api_key_provider()
|
|
115
|
+
|
|
116
|
+
|
|
117
|
+
def load_model() -> ChatGoogleGenerativeAI:
|
|
118
|
+
"""Get authenticated Gemini model client."""
|
|
119
|
+
return ChatGoogleGenerativeAI(
|
|
120
|
+
model="gemini-2.5-flash",
|
|
121
|
+
api_key=_get_api_key()
|
|
122
|
+
)
|
|
123
|
+
{{/if}}
|
|
@@ -0,0 +1,31 @@
|
|
|
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 AG-UI Agent using LangChain + LangGraph"
|
|
9
|
+
readme = "README.md"
|
|
10
|
+
requires-python = ">=3.10"
|
|
11
|
+
dependencies = [
|
|
12
|
+
"ag-ui-langgraph >= 0.0.31",
|
|
13
|
+
"ag-ui-protocol >= 0.1.10",
|
|
14
|
+
{{#if (eq modelProvider "Anthropic")}}"langchain-anthropic >= 0.3.0",
|
|
15
|
+
{{/if}}{{#if (eq modelProvider "Bedrock")}}"langchain-aws >= 0.2.0",
|
|
16
|
+
{{/if}}{{#if (eq modelProvider "Gemini")}}"langchain-google-genai >= 2.0.0",
|
|
17
|
+
"google-genai >= 1.0.0, < 2.0.0",
|
|
18
|
+
{{/if}}{{#if (eq modelProvider "OpenAI")}}"langchain-openai >= 0.2.0",
|
|
19
|
+
{{/if}}"aws-opentelemetry-distro ~= 0.18.0",
|
|
20
|
+
"opentelemetry-instrumentation-langchain >= 0.59.0",
|
|
21
|
+
"bedrock-agentcore >= 1.18.1",
|
|
22
|
+
"botocore[crt] >= 1.35.0",
|
|
23
|
+
"langgraph >= 0.3.25",
|
|
24
|
+
"langchain >= 0.3.0",
|
|
25
|
+
"langchain-core >= 0.3.0",
|
|
26
|
+
"fastapi >= 0.115.12",
|
|
27
|
+
"uvicorn >= 0.34.3",
|
|
28
|
+
]
|
|
29
|
+
|
|
30
|
+
[tool.hatch.build.targets.wheel]
|
|
31
|
+
packages = ["."]
|
|
@@ -0,0 +1,22 @@
|
|
|
1
|
+
# {{ name }}
|
|
2
|
+
|
|
3
|
+
An AG-UI agent deployed on Amazon Bedrock AgentCore using Strands SDK.
|
|
4
|
+
|
|
5
|
+
## Overview
|
|
6
|
+
|
|
7
|
+
This agent implements the AG-UI protocol, enabling streaming agent-to-UI communication. The agent exposes an `/invocations` endpoint that accepts AG-UI protocol requests and streams responses back to the client.
|
|
8
|
+
|
|
9
|
+
## Local Development
|
|
10
|
+
|
|
11
|
+
```bash
|
|
12
|
+
uv sync
|
|
13
|
+
uv run python main.py
|
|
14
|
+
```
|
|
15
|
+
|
|
16
|
+
The agent starts on port 8080.
|
|
17
|
+
|
|
18
|
+
## Deploy
|
|
19
|
+
|
|
20
|
+
```bash
|
|
21
|
+
agentcore deploy
|
|
22
|
+
```
|
|
@@ -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,105 @@
|
|
|
1
|
+
import os
|
|
2
|
+
|
|
3
|
+
# Suppress OpenTelemetry warnings during local development; remove for production
|
|
4
|
+
if os.getenv("LOCAL_DEV") == "1":
|
|
5
|
+
os.environ["OTEL_SDK_DISABLED"] = "true"
|
|
6
|
+
|
|
7
|
+
import uvicorn
|
|
8
|
+
from strands import Agent, tool
|
|
9
|
+
from ag_ui_strands import StrandsAgent, StrandsAgentConfig, create_strands_app
|
|
10
|
+
from model.load import load_model
|
|
11
|
+
{{#if hasMemory}}
|
|
12
|
+
from memory.session import get_memory_session_manager
|
|
13
|
+
{{/if}}
|
|
14
|
+
|
|
15
|
+
|
|
16
|
+
@tool
|
|
17
|
+
def add_numbers(a: int, b: int) -> int:
|
|
18
|
+
"""Return the sum of two numbers."""
|
|
19
|
+
return a + b
|
|
20
|
+
|
|
21
|
+
|
|
22
|
+
tools = [add_numbers]
|
|
23
|
+
|
|
24
|
+
{{#if needsOs}}
|
|
25
|
+
_MOUNT_PATHS = [
|
|
26
|
+
{{#if sessionStorageMountPath}}"{{sessionStorageMountPath}}",{{/if}}
|
|
27
|
+
{{#each efsMounts}}"{{mountPath}}",{{/each}}
|
|
28
|
+
{{#each s3Mounts}}"{{mountPath}}",{{/each}}
|
|
29
|
+
]
|
|
30
|
+
|
|
31
|
+
def _safe_resolve(path: str) -> str:
|
|
32
|
+
resolved = os.path.realpath(path)
|
|
33
|
+
if not any(resolved == os.path.realpath(m) or resolved.startswith(os.path.realpath(m) + os.sep) for m in _MOUNT_PATHS):
|
|
34
|
+
raise ValueError(f"Path '{path}' is not within any configured mount ({', '.join(_MOUNT_PATHS)})")
|
|
35
|
+
return resolved
|
|
36
|
+
|
|
37
|
+
@tool
|
|
38
|
+
def file_read(path: str) -> str:
|
|
39
|
+
"""Read a file from a mounted filesystem. Use the absolute path (e.g. /mnt/tools/data.txt)."""
|
|
40
|
+
try:
|
|
41
|
+
full_path = _safe_resolve(path)
|
|
42
|
+
with open(full_path) as f:
|
|
43
|
+
return f.read()
|
|
44
|
+
except ValueError as e:
|
|
45
|
+
return str(e)
|
|
46
|
+
except OSError as e:
|
|
47
|
+
return f"Error reading '{path}': {e.strerror}"
|
|
48
|
+
|
|
49
|
+
@tool
|
|
50
|
+
def file_write(path: str, content: str) -> str:
|
|
51
|
+
"""Write a file to a mounted filesystem. Use the absolute path (e.g. /mnt/tools/data.txt)."""
|
|
52
|
+
try:
|
|
53
|
+
full_path = _safe_resolve(path)
|
|
54
|
+
parent = os.path.dirname(full_path)
|
|
55
|
+
if parent:
|
|
56
|
+
os.makedirs(parent, exist_ok=True)
|
|
57
|
+
with open(full_path, "w") as f:
|
|
58
|
+
f.write(content)
|
|
59
|
+
return f"Written to {path}"
|
|
60
|
+
except ValueError as e:
|
|
61
|
+
return str(e)
|
|
62
|
+
except OSError as e:
|
|
63
|
+
return f"Error writing '{path}': {e.strerror}"
|
|
64
|
+
|
|
65
|
+
@tool
|
|
66
|
+
def list_files(path: str) -> str:
|
|
67
|
+
"""List files in a mounted filesystem directory. Use the absolute path (e.g. /mnt/tools)."""
|
|
68
|
+
try:
|
|
69
|
+
full_path = _safe_resolve(path)
|
|
70
|
+
entries = os.listdir(full_path)
|
|
71
|
+
return "\n".join(entries) if entries else "(empty directory)"
|
|
72
|
+
except ValueError as e:
|
|
73
|
+
return str(e)
|
|
74
|
+
except OSError as e:
|
|
75
|
+
return f"Error listing '{path}': {e.strerror}"
|
|
76
|
+
|
|
77
|
+
tools.extend([file_read, file_write, list_files])
|
|
78
|
+
{{/if}}
|
|
79
|
+
|
|
80
|
+
agent = Agent(
|
|
81
|
+
model=load_model(),
|
|
82
|
+
system_prompt="""You are a helpful assistant. Use tools when appropriate.
|
|
83
|
+
{{#if needsOs}}
|
|
84
|
+
You have access to the following mounted filesystems. Use file_read, file_write, and list_files with full absolute paths:
|
|
85
|
+
{{#if sessionStorageMountPath}}- {{sessionStorageMountPath}}: ephemeral session storage (lost when session ends)
|
|
86
|
+
{{/if}}{{#each efsMounts}}- {{mountPath}}: EFS persistent storage (persists across sessions and agent restarts)
|
|
87
|
+
{{/each}}{{#each s3Mounts}}- {{mountPath}}: S3 Files persistent storage (durable, backed by S3)
|
|
88
|
+
{{/each}}{{/if}}""",
|
|
89
|
+
tools=tools,
|
|
90
|
+
)
|
|
91
|
+
|
|
92
|
+
{{#if hasMemory}}
|
|
93
|
+
def session_manager_provider(input_data):
|
|
94
|
+
return get_memory_session_manager(input_data.thread_id, "default-user")
|
|
95
|
+
|
|
96
|
+
config = StrandsAgentConfig(session_manager_provider=session_manager_provider)
|
|
97
|
+
{{else}}
|
|
98
|
+
config = StrandsAgentConfig()
|
|
99
|
+
{{/if}}
|
|
100
|
+
|
|
101
|
+
agui_agent = StrandsAgent(agent=agent, name="{{ name }}", description="A helpful assistant", config=config)
|
|
102
|
+
app = create_strands_app(agui_agent, path="/invocations", ping_path="/ping")
|
|
103
|
+
|
|
104
|
+
if __name__ == "__main__":
|
|
105
|
+
uvicorn.run(app, host="0.0.0.0", port=int(os.environ.get("PORT", "8080")))
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
# Package marker
|
|
@@ -0,0 +1,123 @@
|
|
|
1
|
+
{{#if (eq modelProvider "Bedrock")}}
|
|
2
|
+
from strands.models.bedrock import BedrockModel
|
|
3
|
+
|
|
4
|
+
|
|
5
|
+
def load_model() -> BedrockModel:
|
|
6
|
+
"""Get Bedrock model client using IAM credentials."""
|
|
7
|
+
return BedrockModel(model_id="global.anthropic.claude-sonnet-4-5-20250929-v1:0")
|
|
8
|
+
{{/if}}
|
|
9
|
+
{{#if (eq modelProvider "Anthropic")}}
|
|
10
|
+
import os
|
|
11
|
+
|
|
12
|
+
from strands.models.anthropic import AnthropicModel
|
|
13
|
+
from bedrock_agentcore.identity.auth import requires_api_key
|
|
14
|
+
|
|
15
|
+
IDENTITY_PROVIDER_NAME = "{{identityProviders.[0].name}}"
|
|
16
|
+
IDENTITY_ENV_VAR = "{{identityProviders.[0].envVarName}}"
|
|
17
|
+
|
|
18
|
+
|
|
19
|
+
@requires_api_key(provider_name=IDENTITY_PROVIDER_NAME)
|
|
20
|
+
def _agentcore_identity_api_key_provider(api_key: str) -> str:
|
|
21
|
+
"""Fetch API key from AgentCore Identity."""
|
|
22
|
+
return api_key
|
|
23
|
+
|
|
24
|
+
|
|
25
|
+
def _get_api_key() -> str:
|
|
26
|
+
"""
|
|
27
|
+
Uses AgentCore Identity for API key management in deployed environments.
|
|
28
|
+
For local development, run via 'agentcore dev' which loads agentcore/.env.
|
|
29
|
+
"""
|
|
30
|
+
if os.getenv("LOCAL_DEV") == "1":
|
|
31
|
+
api_key = os.getenv(IDENTITY_ENV_VAR)
|
|
32
|
+
if not api_key:
|
|
33
|
+
raise RuntimeError(
|
|
34
|
+
f"{IDENTITY_ENV_VAR} not found. Add {IDENTITY_ENV_VAR}=your-key to .env.local"
|
|
35
|
+
)
|
|
36
|
+
return api_key
|
|
37
|
+
return _agentcore_identity_api_key_provider()
|
|
38
|
+
|
|
39
|
+
|
|
40
|
+
def load_model() -> AnthropicModel:
|
|
41
|
+
"""Get authenticated Anthropic model client."""
|
|
42
|
+
return AnthropicModel(
|
|
43
|
+
client_args={"api_key": _get_api_key()},
|
|
44
|
+
model_id="claude-sonnet-4-5-20250929",
|
|
45
|
+
max_tokens=5000,
|
|
46
|
+
)
|
|
47
|
+
{{/if}}
|
|
48
|
+
{{#if (eq modelProvider "OpenAI")}}
|
|
49
|
+
import os
|
|
50
|
+
|
|
51
|
+
from strands.models.openai import OpenAIModel
|
|
52
|
+
from bedrock_agentcore.identity.auth import requires_api_key
|
|
53
|
+
|
|
54
|
+
IDENTITY_PROVIDER_NAME = "{{identityProviders.[0].name}}"
|
|
55
|
+
IDENTITY_ENV_VAR = "{{identityProviders.[0].envVarName}}"
|
|
56
|
+
|
|
57
|
+
|
|
58
|
+
@requires_api_key(provider_name=IDENTITY_PROVIDER_NAME)
|
|
59
|
+
def _agentcore_identity_api_key_provider(api_key: str) -> str:
|
|
60
|
+
"""Fetch API key from AgentCore Identity."""
|
|
61
|
+
return api_key
|
|
62
|
+
|
|
63
|
+
|
|
64
|
+
def _get_api_key() -> str:
|
|
65
|
+
"""
|
|
66
|
+
Uses AgentCore Identity for API key management in deployed environments.
|
|
67
|
+
For local development, run via 'agentcore dev' which loads agentcore/.env.
|
|
68
|
+
"""
|
|
69
|
+
if os.getenv("LOCAL_DEV") == "1":
|
|
70
|
+
api_key = os.getenv(IDENTITY_ENV_VAR)
|
|
71
|
+
if not api_key:
|
|
72
|
+
raise RuntimeError(
|
|
73
|
+
f"{IDENTITY_ENV_VAR} not found. Add {IDENTITY_ENV_VAR}=your-key to .env.local"
|
|
74
|
+
)
|
|
75
|
+
return api_key
|
|
76
|
+
return _agentcore_identity_api_key_provider()
|
|
77
|
+
|
|
78
|
+
|
|
79
|
+
def load_model() -> OpenAIModel:
|
|
80
|
+
"""Get authenticated OpenAI model client."""
|
|
81
|
+
return OpenAIModel(
|
|
82
|
+
client_args={"api_key": _get_api_key()},
|
|
83
|
+
model_id="gpt-4.1",
|
|
84
|
+
)
|
|
85
|
+
{{/if}}
|
|
86
|
+
{{#if (eq modelProvider "Gemini")}}
|
|
87
|
+
import os
|
|
88
|
+
|
|
89
|
+
from strands.models.gemini import GeminiModel
|
|
90
|
+
from bedrock_agentcore.identity.auth import requires_api_key
|
|
91
|
+
|
|
92
|
+
IDENTITY_PROVIDER_NAME = "{{identityProviders.[0].name}}"
|
|
93
|
+
IDENTITY_ENV_VAR = "{{identityProviders.[0].envVarName}}"
|
|
94
|
+
|
|
95
|
+
|
|
96
|
+
@requires_api_key(provider_name=IDENTITY_PROVIDER_NAME)
|
|
97
|
+
def _agentcore_identity_api_key_provider(api_key: str) -> str:
|
|
98
|
+
"""Fetch API key from AgentCore Identity."""
|
|
99
|
+
return api_key
|
|
100
|
+
|
|
101
|
+
|
|
102
|
+
def _get_api_key() -> str:
|
|
103
|
+
"""
|
|
104
|
+
Uses AgentCore Identity for API key management in deployed environments.
|
|
105
|
+
For local development, run via 'agentcore dev' which loads agentcore/.env.
|
|
106
|
+
"""
|
|
107
|
+
if os.getenv("LOCAL_DEV") == "1":
|
|
108
|
+
api_key = os.getenv(IDENTITY_ENV_VAR)
|
|
109
|
+
if not api_key:
|
|
110
|
+
raise RuntimeError(
|
|
111
|
+
f"{IDENTITY_ENV_VAR} not found. Add {IDENTITY_ENV_VAR}=your-key to .env.local"
|
|
112
|
+
)
|
|
113
|
+
return api_key
|
|
114
|
+
return _agentcore_identity_api_key_provider()
|
|
115
|
+
|
|
116
|
+
|
|
117
|
+
def load_model() -> GeminiModel:
|
|
118
|
+
"""Get authenticated Gemini model client."""
|
|
119
|
+
return GeminiModel(
|
|
120
|
+
client_args={"api_key": _get_api_key()},
|
|
121
|
+
model_id="gemini-2.5-flash",
|
|
122
|
+
)
|
|
123
|
+
{{/if}}
|
|
@@ -0,0 +1,27 @@
|
|
|
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 AG-UI Agent using Strands SDK"
|
|
9
|
+
readme = "README.md"
|
|
10
|
+
requires-python = ">=3.12"
|
|
11
|
+
dependencies = [
|
|
12
|
+
{{#if (eq modelProvider "Anthropic")}}"anthropic >= 0.30.0",
|
|
13
|
+
{{/if}}"ag-ui-strands >= 0.1.7",
|
|
14
|
+
"ag-ui-protocol >= 0.1.10",
|
|
15
|
+
"aws-opentelemetry-distro ~= 0.18.0",
|
|
16
|
+
"bedrock-agentcore[strands-agents] >= 1.18.1",
|
|
17
|
+
"botocore[crt] >= 1.35.0",
|
|
18
|
+
"fastapi >= 0.115.12",
|
|
19
|
+
{{#if (eq modelProvider "Gemini")}}"google-genai >= 1.0.0",
|
|
20
|
+
{{/if}}{{#if (eq modelProvider "OpenAI")}}"openai >= 1.0.0",
|
|
21
|
+
{{/if}}"strands-agents >= 1.15.0",
|
|
22
|
+
"strands-agents-tools >= 0.2.14",
|
|
23
|
+
"uvicorn >= 0.34.3",
|
|
24
|
+
]
|
|
25
|
+
|
|
26
|
+
[tool.hatch.build.targets.wheel]
|
|
27
|
+
packages = ["."]
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
# Package marker
|
|
@@ -0,0 +1,46 @@
|
|
|
1
|
+
import os
|
|
2
|
+
import uuid
|
|
3
|
+
from typing import Optional
|
|
4
|
+
|
|
5
|
+
from bedrock_agentcore.memory.integrations.strands.config import AgentCoreMemoryConfig{{#if memoryProviders.[0].strategies.length}}, RetrievalConfig{{/if}}
|
|
6
|
+
from bedrock_agentcore.memory.integrations.strands.session_manager import AgentCoreMemorySessionManager
|
|
7
|
+
|
|
8
|
+
MEMORY_ID = os.getenv("{{memoryProviders.[0].envVarName}}")
|
|
9
|
+
REGION = os.getenv("AWS_REGION")
|
|
10
|
+
|
|
11
|
+
def get_memory_session_manager(session_id: Optional[str], actor_id: str) -> Optional[AgentCoreMemorySessionManager]:
|
|
12
|
+
if not MEMORY_ID:
|
|
13
|
+
return None
|
|
14
|
+
|
|
15
|
+
# AgentCoreMemoryConfig rejects None; OAuth/CUSTOM_JWT callers can reach us
|
|
16
|
+
# without a runtime session header, so synthesize one when absent.
|
|
17
|
+
session_id = session_id or uuid.uuid4().hex
|
|
18
|
+
|
|
19
|
+
{{#if memoryProviders.[0].strategies.length}}
|
|
20
|
+
retrieval_config = {
|
|
21
|
+
{{#if (includes memoryProviders.[0].strategies "SEMANTIC")}}
|
|
22
|
+
f"/users/{actor_id}/facts": RetrievalConfig(top_k=3, relevance_score=0.5),
|
|
23
|
+
{{/if}}
|
|
24
|
+
{{#if (includes memoryProviders.[0].strategies "USER_PREFERENCE")}}
|
|
25
|
+
f"/users/{actor_id}/preferences": RetrievalConfig(top_k=3, relevance_score=0.5),
|
|
26
|
+
{{/if}}
|
|
27
|
+
{{#if (includes memoryProviders.[0].strategies "EPISODIC")}}
|
|
28
|
+
f"/episodes/{actor_id}/{session_id}": RetrievalConfig(top_k=5, relevance_score=0.5),
|
|
29
|
+
{{/if}}
|
|
30
|
+
{{#if (includes memoryProviders.[0].strategies "SUMMARIZATION")}}
|
|
31
|
+
f"/summaries/{actor_id}": RetrievalConfig(top_k=3, relevance_score=0.5),
|
|
32
|
+
{{/if}}
|
|
33
|
+
}
|
|
34
|
+
{{/if}}
|
|
35
|
+
|
|
36
|
+
return AgentCoreMemorySessionManager(
|
|
37
|
+
AgentCoreMemoryConfig(
|
|
38
|
+
memory_id=MEMORY_ID,
|
|
39
|
+
session_id=session_id,
|
|
40
|
+
actor_id=actor_id,
|
|
41
|
+
{{#if memoryProviders.[0].strategies.length}}
|
|
42
|
+
retrieval_config=retrieval_config,
|
|
43
|
+
{{/if}}
|
|
44
|
+
),
|
|
45
|
+
REGION
|
|
46
|
+
)
|